Ticket #584: recurse.py

import genshi.template.markup

template_src = """<module
    xmlns:py="http://genshi.edgewall.org/"
    xmlns:xi="http://www.w3.org/2001/XInclude" py:strip="True">
    <div py:content="module['name']" />
    <py:for each="module in module.get('children', [])">
        <!--! Here's where the recursion occurs -->
        <xi:include href="module.xml" />
    </py:for>
</module>
"""

module = dict(
    type='module',
    name='module 1',
    children=[
        dict(type='module', name='module 2'),
    ],
)


class SimpleLoader(object):

    def __init__(self, auto_reload):
        self.auto_reload = auto_reload
        self.tmpl = genshi.template.markup.MarkupTemplate(
            template_src, filepath=None, filename='module.xml', loader=self)
        self.loaded = 0

    def load(self, name, **kwargs):
        self.loaded += 1
        return self.tmpl


def test_simple_loader(auto_reload):
    loader = SimpleLoader(auto_reload)
    tmpl = loader.load('module.xml')
    print(tmpl.generate(module=module))
    print "Loaded", loader.loaded, "templates."


if __name__ == '__main__':
    print "Testing with auto_reload = True, expecting:"
    print "  - 1 load on prepare"
    print "  - 1 load during rendering"
    test_simple_loader(auto_reload=True)

    print "Testing with auto_reload = False, expecting:"
    print "  - 1 load on prepare"
    print "  - 1 load to check filepath during prepare"
    print "  - 1 load to during rendering"
    test_simple_loader(auto_reload=False)