Mercurial > piecrust2
view tests/test_processing_tree.py @ 217:1f4c3dae1fe8
bake: Better error handling for site baking.
The site baker now keeps track of whether any worker saw any error. It returns
the current record after a bake.
The `bake` command uses this return value to figure out what kind of exit code
to return.
author | Ludovic Chabant <ludovic@chabant.com> |
---|---|
date | Sat, 31 Jan 2015 17:35:36 -0800 |
parents | f485ba500df3 |
children | 72f17534d58e |
line wrap: on
line source
from piecrust.processing.base import CopyFileProcessor, SimpleFileProcessor from piecrust.processing.tree import ProcessingTreeBuilder, ProcessingTreeNode class MockProcessor(SimpleFileProcessor): def __init__(self): super(MockProcessor, self).__init__({'mock': 'out'}) self.processed = [] def _doProcess(self, in_path, out_path): self.processed.append((in_path, out_path)) mock_processors = [MockProcessor(), CopyFileProcessor()] IDX_MOCK = 0 IDX_COPY = 1 def test_mock_node(): node = ProcessingTreeNode('/foo.mock', list(mock_processors)) assert node.getProcessor() == mock_processors[IDX_MOCK] def test_copy_node(): node = ProcessingTreeNode('/foo.other', list(mock_processors)) assert node.getProcessor() == mock_processors[IDX_COPY] def test_build_simple_tree(): builder = ProcessingTreeBuilder(mock_processors) root = builder.build('/foo.mock') assert root is not None assert root.getProcessor() == mock_processors[IDX_MOCK] assert not root.is_leaf assert len(root.outputs) == 1 out = root.outputs[0] assert out.getProcessor() == mock_processors[IDX_COPY]