75 lines
3.4 KiB
Diff
75 lines
3.4 KiB
Diff
Index: lxml-4.9.1/src/lxml/tests/test_elementtree.py
|
|
===================================================================
|
|
--- lxml-4.9.1.orig/src/lxml/tests/test_elementtree.py
|
|
+++ lxml-4.9.1/src/lxml/tests/test_elementtree.py
|
|
@@ -4373,12 +4373,14 @@ class _XMLPullParserTest(unittest.TestCa
|
|
root = parser.close()
|
|
return root
|
|
|
|
- def _feed(self, parser, data, chunk_size=None):
|
|
+ def _feed(self, parser, data, chunk_size=None, flush=False):
|
|
if chunk_size is None:
|
|
parser.feed(data)
|
|
else:
|
|
for i in range(0, len(data), chunk_size):
|
|
parser.feed(data[i:i+chunk_size])
|
|
+ if flush:
|
|
+ parser.flush()
|
|
|
|
def assert_events(self, parser, expected, max_events=None):
|
|
self.assertEqual(
|
|
@@ -4396,29 +4398,30 @@ class _XMLPullParserTest(unittest.TestCa
|
|
self.assertEqual([(action, elem.tag) for action, elem in events],
|
|
expected)
|
|
|
|
- def test_simple_xml(self):
|
|
- for chunk_size in (None, 1, 5):
|
|
- #with self.subTest(chunk_size=chunk_size):
|
|
- parser = self.etree.XMLPullParser()
|
|
- self.assert_event_tags(parser, [])
|
|
- self._feed(parser, "<!-- comment -->\n", chunk_size)
|
|
- self.assert_event_tags(parser, [])
|
|
- self._feed(parser,
|
|
- "<root>\n <element key='value'>text</element",
|
|
- chunk_size)
|
|
- self.assert_event_tags(parser, [])
|
|
- self._feed(parser, ">\n", chunk_size)
|
|
- self.assert_event_tags(parser, [('end', 'element')])
|
|
- self._feed(parser, "<element>text</element>tail\n", chunk_size)
|
|
- self._feed(parser, "<empty-element/>\n", chunk_size)
|
|
- self.assert_event_tags(parser, [
|
|
- ('end', 'element'),
|
|
- ('end', 'empty-element'),
|
|
- ])
|
|
- self._feed(parser, "</root>\n", chunk_size)
|
|
- self.assert_event_tags(parser, [('end', 'root')])
|
|
- root = self._close_and_return_root(parser)
|
|
- self.assertEqual(root.tag, 'root')
|
|
+ def test_simple_xml(self, chunk_size=None, flush=False):
|
|
+ parser = self.etree.XMLPullParser()
|
|
+ self.assert_event_tags(parser, [])
|
|
+ self._feed(parser, "<!-- comment -->\n", chunk_size, flush)
|
|
+ self.assert_event_tags(parser, [])
|
|
+ self._feed(parser,
|
|
+ "<root>\n <element key='value'>text</element",
|
|
+ chunk_size, flush)
|
|
+ self.assert_event_tags(parser, [])
|
|
+ self._feed(parser, ">\n", chunk_size, flush)
|
|
+ self.assert_event_tags(parser, [('end', 'element')])
|
|
+ self._feed(parser, "<element>text</element>tail\n", chunk_size, flush)
|
|
+ self._feed(parser, "<empty-element/>\n", chunk_size, flush)
|
|
+ self.assert_event_tags(parser, [
|
|
+ ('end', 'element'),
|
|
+ ('end', 'empty-element'),
|
|
+ ])
|
|
+ self._feed(parser, "</root>\n", chunk_size, flush)
|
|
+ self.assert_event_tags(parser, [('end', 'root')])
|
|
+ root = self._close_and_return_root(parser)
|
|
+ self.assertEqual(root.tag, 'root')
|
|
+
|
|
+ def test_simple_xml_chunk_22(self):
|
|
+ self.test_simple_xml(chunk_size=22)
|
|
|
|
def test_feed_while_iterating(self):
|
|
parser = self.etree.XMLPullParser()
|