view tests/test_processing_tree.py @ 298:b7ab1b503510

data: Fix incorrect next/previous page URLs in pagination data. Consolidate splitting an URL between its first URL and its sub page number. Be careful about the index page's URL not losing its slash.
author Ludovic Chabant <ludovic@chabant.com>
date Wed, 11 Mar 2015 23:46:42 -0700
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]