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

diff --git a/Lib/test/test_sax.py b/Lib/test/test_sax.py
index 218d5199fbbd887c6660ec2a5f90d1ecbe8b024c..bfa7d72d1644e94f63b8c1227ca21b83225e7815 100644 (file)
--- a/Lib/test/test_sax.py
+++ b/Lib/test/test_sax.py
@@ -493,6 +493,21 @@ class XmlgenTest:
         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