Ticket #192: join_unit_tests.patch

genshi/tests/builder.py

@@ -14 +14 @@
 import doctest
 import unittest
 
-from genshi.builder import Element, tag
+from genshi.builder import Element, tag, Fragment
 from genshi.core import Attrs, Markup, Stream
 from genshi.input import XML
 
@@ -64 +64 @@
                                    'genshi</a>'))
 
 
+class FragmentTestCase(unittest.TestCase):
+
+    def test_join_with_fragment(self):
+        tags = [tag.span('test'),tag(','),tag.p(tag.span('A'),tag.span('B')),
+                tag(','),tag('text')]
+        self.assertEqual(
+            unicode(tag(',').join([tags[0],tags[2],tags[4]]).generate()),
+            unicode(tag(tags).generate()) )
+
+    def test_join_with_element(self):
+        tags = [tag.span('test'),tag.b(','),tag.p(tag.span('A'),tag.span('B')),
+                tag.b(','),tag('text')]
+        self.assertEqual(
+            unicode(tag.b(',').join([tags[0],tags[2],tags[4]]).generate()),
+            unicode(tag(tags).generate()) )
+
+    def test_join_empty_list(self):
+        self.assertEqual(
+            unicode(tag(',').join([]).generate()),
+            unicode(Fragment().generate()))
+        self.assertEqual(
+            unicode(tag.b(',').join([]).generate()),
+            unicode(Fragment().generate()))
+
+    def test_join_list_one_element(self):
+        self.assertEqual(
+            unicode(tag(',').join([tag.div('test')]).generate()),
+            unicode(tag.div('test').generate()))
+        self.assertEqual(
+            unicode(tag.b(',').join([tag.div('test')]).generate()),
+            unicode(tag.div('test').generate()))
+
+
 def suite():
     suite = unittest.TestSuite()
     suite.addTest(doctest.DocTestSuite(Element.__module__))
     suite.addTest(unittest.makeSuite(ElementFactoryTestCase, 'test'))
+    suite.addTest(unittest.makeSuite(FragmentTestCase, 'test'))
     return suite