Issue #1470548: Add test for fragment producing with XMLGenerator.

serhiy-storchaka · serhiy-storchaka · commit bd9b56162ba6 · 2013-02-25T13:47:20.000+02:00
diff --git a/Lib/test/test_sax.py b/Lib/test/test_sax.py
@@ -493,6 +493,21 @@ def func(out):
         func(result)
         self.assertFalse(result.closed)
 
+    def test_xmlgen_fragment(self):
+        result = self.ioclass()
+        gen = XMLGenerator(result)
+
+        # Don't call gen.startDocument()
+        gen.startElement("foo", {"a": "1.0"})
+        gen.characters("Hello")
+        gen.endElement("foo")
+        gen.startElement("bar", {"b": "2.0"})
+        gen.endElement("bar")
+        # Don't call gen.endDocument()
+
+        self.assertEqual(result.getvalue(),
+            self.xml('<foo a="1.0">Hello</foo><bar b="2.0"></bar>')[len(self.xml('')):])
+
 class StringXmlgenTest(XmlgenTest, unittest.TestCase):
     ioclass = StringIO
 
