Refactor _compile() into a base class, and add a new test that checks the line-wrapping of attributes by TALInterpreter.do_startTag().

ce3191fa · Fred Drake · 6ae5b596 · ce3191fa
Commit ce3191fa authored Jun 26, 2001 by Fred Drake
Show whitespace changes
Inline Side-by-side

Showing with 29 additions and 1 deletion

lib/python/TAL/tests/test_talinterpreter.py lib/python/TAL/tests/test_talinterpreter.py +29 -1

No files found.
--- a/lib/python/TAL/tests/test_talinterpreter.py
+++ b/lib/python/TAL/tests/test_talinterpreter.py
@@ -6,13 +6,15 @@ import sys
 import utils
 import unittest
+from StringIO import StringIO
 from TAL.TALDefs import METALError
 from TAL.HTMLTALParser import HTMLTALParser
 from TAL.TALInterpreter import TALInterpreter
 from TAL.DummyEngine import DummyEngine
-class MacroErrorsTestCase(unittest.TestCase):
+class TestCaseBase(unittest.TestCase):
    def _compile(self, source):
        parser = HTMLTALParser()
@@ -20,6 +22,9 @@ class MacroErrorsTestCase(unittest.TestCase):
        program, macros = parser.getCode()
        return program, macros
+class MacroErrorsTestCase(TestCaseBase):
    def setUp(self):
        dummy, macros = self._compile('<p metal:define-macro="M">Booh</p>')
        self.macro = macros['M']
@@ -42,9 +47,32 @@ class MacroErrorsTestCase(unittest.TestCase):
        self.macro[0] = ("version", "duh")
+class OutputPresentationTestCase(TestCaseBase):
+    def check_attribute_wrapping(self):
+        # To make sure the attribute-wrapping code is invoked, we have to
+        # include at least one TAL/METAL attribute to avoid having the start
+        # tag optimized into a rawtext instruction.
+        INPUT = r"""
+        <html this='element' has='a' lot='of' attributes=', so' the='output'
+              needs='to' be='line' wrapped='.' tal:define='foo nothing'>
+        </html>"""
+        EXPECTED = r'''
+        <html this="element" has="a" lot="of"
+              attributes=", so" the="output" needs="to"
+              be="line" wrapped=".">
+        </html>''' "\n"
+        program, macros = self._compile(INPUT)
+        sio = StringIO()
+        interp = TALInterpreter(program, {}, DummyEngine(), sio, wrap=60)
+        interp()
+        self.assertEqual(sio.getvalue(), EXPECTED)
 def test_suite():
    suite = unittest.TestSuite()
    suite.addTest(unittest.makeSuite(MacroErrorsTestCase, "check_"))
+    suite.addTest(unittest.makeSuite(OutputPresentationTestCase, "check_"))
    return suite