Ticket #269: test_pyif_with_pystrip.py

#!/usr/bin/env python

import unittest

from genshi.template import MarkupTemplate

tmpl_data = """<div xmlns="http://www.w3.org/1999/xhtml" xmlns:py="http://genshi.edgewall.org/">
   REPLACEME
</div>"""

simple_replace = """<py:if test="True" py:strip="">
    var x = 'bar';
</py:if>"""

replace_and_variable = """<py:if test="True" py:strip="">
    var x = '$d';
</py:if>"""
    


class TestPyStripWithPyIf(unittest.TestCase):
    def test_pystrip_and_pyif(self):
        def generate(replacement_string):
            data = tmpl_data.replace('REPLACEME', replacement_string)
            tmpl = MarkupTemplate(data)
            return tmpl.generate(d="bar").render('xhtml', doctype='xhtml')
        
        simple_output = generate(simple_replace)
        wrong_output = generate(replace_and_variable)
        self.assertEquals(simple_output, wrong_output)