+++ /dev/null
-# XXX TypeErrors on calling handlers, or on bad return values from a\r
-# handler, are obscure and unhelpful.\r
-\r
-import StringIO, sys\r
-import unittest\r
-\r
-from xml.parsers import expat\r
-\r
-from test import test_support\r
-from test.test_support import sortdict, run_unittest\r
-\r
-\r
-class SetAttributeTest(unittest.TestCase):\r
- def setUp(self):\r
- self.parser = expat.ParserCreate(namespace_separator='!')\r
- self.set_get_pairs = [\r
- [0, 0],\r
- [1, 1],\r
- [2, 1],\r
- [0, 0],\r
- ]\r
-\r
- def test_returns_unicode(self):\r
- for x, y in self.set_get_pairs:\r
- self.parser.returns_unicode = x\r
- self.assertEqual(self.parser.returns_unicode, y)\r
-\r
- def test_ordered_attributes(self):\r
- for x, y in self.set_get_pairs:\r
- self.parser.ordered_attributes = x\r
- self.assertEqual(self.parser.ordered_attributes, y)\r
-\r
- def test_specified_attributes(self):\r
- for x, y in self.set_get_pairs:\r
- self.parser.specified_attributes = x\r
- self.assertEqual(self.parser.specified_attributes, y)\r
-\r
-\r
-data = '''\\r
-<?xml version="1.0" encoding="iso-8859-1" standalone="no"?>\r
-<?xml-stylesheet href="stylesheet.css"?>\r
-<!-- comment data -->\r
-<!DOCTYPE quotations SYSTEM "quotations.dtd" [\r
-<!ELEMENT root ANY>\r
-<!NOTATION notation SYSTEM "notation.jpeg">\r
-<!ENTITY acirc "â">\r
-<!ENTITY external_entity SYSTEM "entity.file">\r
-<!ENTITY unparsed_entity SYSTEM "entity.file" NDATA notation>\r
-%unparsed_entity;\r
-]>\r
-\r
-<root attr1="value1" attr2="value2ὀ">\r
-<myns:subelement xmlns:myns="http://www.python.org/namespace">\r
- Contents of subelements\r
-</myns:subelement>\r
-<sub2><![CDATA[contents of CDATA section]]></sub2>\r
-&external_entity;\r
-</root>\r
-'''\r
-\r
-\r
-# Produce UTF-8 output\r
-class ParseTest(unittest.TestCase):\r
- class Outputter:\r
- def __init__(self):\r
- self.out = []\r
-\r
- def StartElementHandler(self, name, attrs):\r
- self.out.append('Start element: ' + repr(name) + ' ' +\r
- sortdict(attrs))\r
-\r
- def EndElementHandler(self, name):\r
- self.out.append('End element: ' + repr(name))\r
-\r
- def CharacterDataHandler(self, data):\r
- data = data.strip()\r
- if data:\r
- self.out.append('Character data: ' + repr(data))\r
-\r
- def ProcessingInstructionHandler(self, target, data):\r
- self.out.append('PI: ' + repr(target) + ' ' + repr(data))\r
-\r
- def StartNamespaceDeclHandler(self, prefix, uri):\r
- self.out.append('NS decl: ' + repr(prefix) + ' ' + repr(uri))\r
-\r
- def EndNamespaceDeclHandler(self, prefix):\r
- self.out.append('End of NS decl: ' + repr(prefix))\r
-\r
- def StartCdataSectionHandler(self):\r
- self.out.append('Start of CDATA section')\r
-\r
- def EndCdataSectionHandler(self):\r
- self.out.append('End of CDATA section')\r
-\r
- def CommentHandler(self, text):\r
- self.out.append('Comment: ' + repr(text))\r
-\r
- def NotationDeclHandler(self, *args):\r
- name, base, sysid, pubid = args\r
- self.out.append('Notation declared: %s' %(args,))\r
-\r
- def UnparsedEntityDeclHandler(self, *args):\r
- entityName, base, systemId, publicId, notationName = args\r
- self.out.append('Unparsed entity decl: %s' %(args,))\r
-\r
- def NotStandaloneHandler(self, userData):\r
- self.out.append('Not standalone')\r
- return 1\r
-\r
- def ExternalEntityRefHandler(self, *args):\r
- context, base, sysId, pubId = args\r
- self.out.append('External entity ref: %s' %(args[1:],))\r
- return 1\r
-\r
- def DefaultHandler(self, userData):\r
- pass\r
-\r
- def DefaultHandlerExpand(self, userData):\r
- pass\r
-\r
- handler_names = [\r
- 'StartElementHandler', 'EndElementHandler',\r
- 'CharacterDataHandler', 'ProcessingInstructionHandler',\r
- 'UnparsedEntityDeclHandler', 'NotationDeclHandler',\r
- 'StartNamespaceDeclHandler', 'EndNamespaceDeclHandler',\r
- 'CommentHandler', 'StartCdataSectionHandler',\r
- 'EndCdataSectionHandler',\r
- 'DefaultHandler', 'DefaultHandlerExpand',\r
- #'NotStandaloneHandler',\r
- 'ExternalEntityRefHandler'\r
- ]\r
-\r
- def test_utf8(self):\r
-\r
- out = self.Outputter()\r
- parser = expat.ParserCreate(namespace_separator='!')\r
- for name in self.handler_names:\r
- setattr(parser, name, getattr(out, name))\r
- parser.returns_unicode = 0\r
- parser.Parse(data, 1)\r
-\r
- # Verify output\r
- op = out.out\r
- self.assertEqual(op[0], 'PI: \'xml-stylesheet\' \'href="stylesheet.css"\'')\r
- self.assertEqual(op[1], "Comment: ' comment data '")\r
- self.assertEqual(op[2], "Notation declared: ('notation', None, 'notation.jpeg', None)")\r
- self.assertEqual(op[3], "Unparsed entity decl: ('unparsed_entity', None, 'entity.file', None, 'notation')")\r
- self.assertEqual(op[4], "Start element: 'root' {'attr1': 'value1', 'attr2': 'value2\\xe1\\xbd\\x80'}")\r
- self.assertEqual(op[5], "NS decl: 'myns' 'http://www.python.org/namespace'")\r
- self.assertEqual(op[6], "Start element: 'http://www.python.org/namespace!subelement' {}")\r
- self.assertEqual(op[7], "Character data: 'Contents of subelements'")\r
- self.assertEqual(op[8], "End element: 'http://www.python.org/namespace!subelement'")\r
- self.assertEqual(op[9], "End of NS decl: 'myns'")\r
- self.assertEqual(op[10], "Start element: 'sub2' {}")\r
- self.assertEqual(op[11], 'Start of CDATA section')\r
- self.assertEqual(op[12], "Character data: 'contents of CDATA section'")\r
- self.assertEqual(op[13], 'End of CDATA section')\r
- self.assertEqual(op[14], "End element: 'sub2'")\r
- self.assertEqual(op[15], "External entity ref: (None, 'entity.file', None)")\r
- self.assertEqual(op[16], "End element: 'root'")\r
-\r
- def test_unicode(self):\r
- # Try the parse again, this time producing Unicode output\r
- out = self.Outputter()\r
- parser = expat.ParserCreate(namespace_separator='!')\r
- parser.returns_unicode = 1\r
- for name in self.handler_names:\r
- setattr(parser, name, getattr(out, name))\r
-\r
- parser.Parse(data, 1)\r
-\r
- op = out.out\r
- self.assertEqual(op[0], 'PI: u\'xml-stylesheet\' u\'href="stylesheet.css"\'')\r
- self.assertEqual(op[1], "Comment: u' comment data '")\r
- self.assertEqual(op[2], "Notation declared: (u'notation', None, u'notation.jpeg', None)")\r
- self.assertEqual(op[3], "Unparsed entity decl: (u'unparsed_entity', None, u'entity.file', None, u'notation')")\r
- self.assertEqual(op[4], "Start element: u'root' {u'attr1': u'value1', u'attr2': u'value2\\u1f40'}")\r
- self.assertEqual(op[5], "NS decl: u'myns' u'http://www.python.org/namespace'")\r
- self.assertEqual(op[6], "Start element: u'http://www.python.org/namespace!subelement' {}")\r
- self.assertEqual(op[7], "Character data: u'Contents of subelements'")\r
- self.assertEqual(op[8], "End element: u'http://www.python.org/namespace!subelement'")\r
- self.assertEqual(op[9], "End of NS decl: u'myns'")\r
- self.assertEqual(op[10], "Start element: u'sub2' {}")\r
- self.assertEqual(op[11], 'Start of CDATA section')\r
- self.assertEqual(op[12], "Character data: u'contents of CDATA section'")\r
- self.assertEqual(op[13], 'End of CDATA section')\r
- self.assertEqual(op[14], "End element: u'sub2'")\r
- self.assertEqual(op[15], "External entity ref: (None, u'entity.file', None)")\r
- self.assertEqual(op[16], "End element: u'root'")\r
-\r
- def test_parse_file(self):\r
- # Try parsing a file\r
- out = self.Outputter()\r
- parser = expat.ParserCreate(namespace_separator='!')\r
- parser.returns_unicode = 1\r
- for name in self.handler_names:\r
- setattr(parser, name, getattr(out, name))\r
- file = StringIO.StringIO(data)\r
-\r
- parser.ParseFile(file)\r
-\r
- op = out.out\r
- self.assertEqual(op[0], 'PI: u\'xml-stylesheet\' u\'href="stylesheet.css"\'')\r
- self.assertEqual(op[1], "Comment: u' comment data '")\r
- self.assertEqual(op[2], "Notation declared: (u'notation', None, u'notation.jpeg', None)")\r
- self.assertEqual(op[3], "Unparsed entity decl: (u'unparsed_entity', None, u'entity.file', None, u'notation')")\r
- self.assertEqual(op[4], "Start element: u'root' {u'attr1': u'value1', u'attr2': u'value2\\u1f40'}")\r
- self.assertEqual(op[5], "NS decl: u'myns' u'http://www.python.org/namespace'")\r
- self.assertEqual(op[6], "Start element: u'http://www.python.org/namespace!subelement' {}")\r
- self.assertEqual(op[7], "Character data: u'Contents of subelements'")\r
- self.assertEqual(op[8], "End element: u'http://www.python.org/namespace!subelement'")\r
- self.assertEqual(op[9], "End of NS decl: u'myns'")\r
- self.assertEqual(op[10], "Start element: u'sub2' {}")\r
- self.assertEqual(op[11], 'Start of CDATA section')\r
- self.assertEqual(op[12], "Character data: u'contents of CDATA section'")\r
- self.assertEqual(op[13], 'End of CDATA section')\r
- self.assertEqual(op[14], "End element: u'sub2'")\r
- self.assertEqual(op[15], "External entity ref: (None, u'entity.file', None)")\r
- self.assertEqual(op[16], "End element: u'root'")\r
-\r
- # Issue 4877: expat.ParseFile causes segfault on a closed file.\r
- fp = open(test_support.TESTFN, 'wb')\r
- try:\r
- fp.close()\r
- parser = expat.ParserCreate()\r
- with self.assertRaises(ValueError):\r
- parser.ParseFile(fp)\r
- finally:\r
- test_support.unlink(test_support.TESTFN)\r
-\r
-\r
-class NamespaceSeparatorTest(unittest.TestCase):\r
- def test_legal(self):\r
- # Tests that make sure we get errors when the namespace_separator value\r
- # is illegal, and that we don't for good values:\r
- expat.ParserCreate()\r
- expat.ParserCreate(namespace_separator=None)\r
- expat.ParserCreate(namespace_separator=' ')\r
-\r
- def test_illegal(self):\r
- try:\r
- expat.ParserCreate(namespace_separator=42)\r
- self.fail()\r
- except TypeError, e:\r
- self.assertEqual(str(e),\r
- 'ParserCreate() argument 2 must be string or None, not int')\r
-\r
- try:\r
- expat.ParserCreate(namespace_separator='too long')\r
- self.fail()\r
- except ValueError, e:\r
- self.assertEqual(str(e),\r
- 'namespace_separator must be at most one character, omitted, or None')\r
-\r
- def test_zero_length(self):\r
- # ParserCreate() needs to accept a namespace_separator of zero length\r
- # to satisfy the requirements of RDF applications that are required\r
- # to simply glue together the namespace URI and the localname. Though\r
- # considered a wart of the RDF specifications, it needs to be supported.\r
- #\r
- # See XML-SIG mailing list thread starting with\r
- # http://mail.python.org/pipermail/xml-sig/2001-April/005202.html\r
- #\r
- expat.ParserCreate(namespace_separator='') # too short\r
-\r
-\r
-class InterningTest(unittest.TestCase):\r
- def test(self):\r
- # Test the interning machinery.\r
- p = expat.ParserCreate()\r
- L = []\r
- def collector(name, *args):\r
- L.append(name)\r
- p.StartElementHandler = collector\r
- p.EndElementHandler = collector\r
- p.Parse("<e> <e/> <e></e> </e>", 1)\r
- tag = L[0]\r
- self.assertEqual(len(L), 6)\r
- for entry in L:\r
- # L should have the same string repeated over and over.\r
- self.assertTrue(tag is entry)\r
-\r
-\r
-class BufferTextTest(unittest.TestCase):\r
- def setUp(self):\r
- self.stuff = []\r
- self.parser = expat.ParserCreate()\r
- self.parser.buffer_text = 1\r
- self.parser.CharacterDataHandler = self.CharacterDataHandler\r
-\r
- def check(self, expected, label):\r
- self.assertEqual(self.stuff, expected,\r
- "%s\nstuff = %r\nexpected = %r"\r
- % (label, self.stuff, map(unicode, expected)))\r
-\r
- def CharacterDataHandler(self, text):\r
- self.stuff.append(text)\r
-\r
- def StartElementHandler(self, name, attrs):\r
- self.stuff.append("<%s>" % name)\r
- bt = attrs.get("buffer-text")\r
- if bt == "yes":\r
- self.parser.buffer_text = 1\r
- elif bt == "no":\r
- self.parser.buffer_text = 0\r
-\r
- def EndElementHandler(self, name):\r
- self.stuff.append("</%s>" % name)\r
-\r
- def CommentHandler(self, data):\r
- self.stuff.append("<!--%s-->" % data)\r
-\r
- def setHandlers(self, handlers=[]):\r
- for name in handlers:\r
- setattr(self.parser, name, getattr(self, name))\r
-\r
- def test_default_to_disabled(self):\r
- parser = expat.ParserCreate()\r
- self.assertFalse(parser.buffer_text)\r
-\r
- def test_buffering_enabled(self):\r
- # Make sure buffering is turned on\r
- self.assertTrue(self.parser.buffer_text)\r
- self.parser.Parse("<a>1<b/>2<c/>3</a>", 1)\r
- self.assertEqual(self.stuff, ['123'],\r
- "buffered text not properly collapsed")\r
-\r
- def test1(self):\r
- # XXX This test exposes more detail of Expat's text chunking than we\r
- # XXX like, but it tests what we need to concisely.\r
- self.setHandlers(["StartElementHandler"])\r
- self.parser.Parse("<a>1<b buffer-text='no'/>2\n3<c buffer-text='yes'/>4\n5</a>", 1)\r
- self.assertEqual(self.stuff,\r
- ["<a>", "1", "<b>", "2", "\n", "3", "<c>", "4\n5"],\r
- "buffering control not reacting as expected")\r
-\r
- def test2(self):\r
- self.parser.Parse("<a>1<b/><2><c/> \n 3</a>", 1)\r
- self.assertEqual(self.stuff, ["1<2> \n 3"],\r
- "buffered text not properly collapsed")\r
-\r
- def test3(self):\r
- self.setHandlers(["StartElementHandler"])\r
- self.parser.Parse("<a>1<b/>2<c/>3</a>", 1)\r
- self.assertEqual(self.stuff, ["<a>", "1", "<b>", "2", "<c>", "3"],\r
- "buffered text not properly split")\r
-\r
- def test4(self):\r
- self.setHandlers(["StartElementHandler", "EndElementHandler"])\r
- self.parser.CharacterDataHandler = None\r
- self.parser.Parse("<a>1<b/>2<c/>3</a>", 1)\r
- self.assertEqual(self.stuff,\r
- ["<a>", "<b>", "</b>", "<c>", "</c>", "</a>"])\r
-\r
- def test5(self):\r
- self.setHandlers(["StartElementHandler", "EndElementHandler"])\r
- self.parser.Parse("<a>1<b></b>2<c/>3</a>", 1)\r
- self.assertEqual(self.stuff,\r
- ["<a>", "1", "<b>", "</b>", "2", "<c>", "</c>", "3", "</a>"])\r
-\r
- def test6(self):\r
- self.setHandlers(["CommentHandler", "EndElementHandler",\r
- "StartElementHandler"])\r
- self.parser.Parse("<a>1<b/>2<c></c>345</a> ", 1)\r
- self.assertEqual(self.stuff,\r
- ["<a>", "1", "<b>", "</b>", "2", "<c>", "</c>", "345", "</a>"],\r
- "buffered text not properly split")\r
-\r
- def test7(self):\r
- self.setHandlers(["CommentHandler", "EndElementHandler",\r
- "StartElementHandler"])\r
- self.parser.Parse("<a>1<b/>2<c></c>3<!--abc-->4<!--def-->5</a> ", 1)\r
- self.assertEqual(self.stuff,\r
- ["<a>", "1", "<b>", "</b>", "2", "<c>", "</c>", "3",\r
- "<!--abc-->", "4", "<!--def-->", "5", "</a>"],\r
- "buffered text not properly split")\r
-\r
-\r
-# Test handling of exception from callback:\r
-class HandlerExceptionTest(unittest.TestCase):\r
- def StartElementHandler(self, name, attrs):\r
- raise RuntimeError(name)\r
-\r
- def test(self):\r
- parser = expat.ParserCreate()\r
- parser.StartElementHandler = self.StartElementHandler\r
- try:\r
- parser.Parse("<a><b><c/></b></a>", 1)\r
- self.fail()\r
- except RuntimeError, e:\r
- self.assertEqual(e.args[0], 'a',\r
- "Expected RuntimeError for element 'a', but" + \\r
- " found %r" % e.args[0])\r
-\r
-\r
-# Test Current* members:\r
-class PositionTest(unittest.TestCase):\r
- def StartElementHandler(self, name, attrs):\r
- self.check_pos('s')\r
-\r
- def EndElementHandler(self, name):\r
- self.check_pos('e')\r
-\r
- def check_pos(self, event):\r
- pos = (event,\r
- self.parser.CurrentByteIndex,\r
- self.parser.CurrentLineNumber,\r
- self.parser.CurrentColumnNumber)\r
- self.assertTrue(self.upto < len(self.expected_list),\r
- 'too many parser events')\r
- expected = self.expected_list[self.upto]\r
- self.assertEqual(pos, expected,\r
- 'Expected position %s, got position %s' %(pos, expected))\r
- self.upto += 1\r
-\r
- def test(self):\r
- self.parser = expat.ParserCreate()\r
- self.parser.StartElementHandler = self.StartElementHandler\r
- self.parser.EndElementHandler = self.EndElementHandler\r
- self.upto = 0\r
- self.expected_list = [('s', 0, 1, 0), ('s', 5, 2, 1), ('s', 11, 3, 2),\r
- ('e', 15, 3, 6), ('e', 17, 4, 1), ('e', 22, 5, 0)]\r
-\r
- xml = '<a>\n <b>\n <c/>\n </b>\n</a>'\r
- self.parser.Parse(xml, 1)\r
-\r
-\r
-class sf1296433Test(unittest.TestCase):\r
- def test_parse_only_xml_data(self):\r
- # http://python.org/sf/1296433\r
- #\r
- xml = "<?xml version='1.0' encoding='iso8859'?><s>%s</s>" % ('a' * 1025)\r
- # this one doesn't crash\r
- #xml = "<?xml version='1.0'?><s>%s</s>" % ('a' * 10000)\r
-\r
- class SpecificException(Exception):\r
- pass\r
-\r
- def handler(text):\r
- raise SpecificException\r
-\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = handler\r
-\r
- self.assertRaises(Exception, parser.Parse, xml)\r
-\r
-class ChardataBufferTest(unittest.TestCase):\r
- """\r
- test setting of chardata buffer size\r
- """\r
-\r
- def test_1025_bytes(self):\r
- self.assertEqual(self.small_buffer_test(1025), 2)\r
-\r
- def test_1000_bytes(self):\r
- self.assertEqual(self.small_buffer_test(1000), 1)\r
-\r
- def test_wrong_size(self):\r
- parser = expat.ParserCreate()\r
- parser.buffer_text = 1\r
- def f(size):\r
- parser.buffer_size = size\r
-\r
- self.assertRaises(TypeError, f, sys.maxint+1)\r
- self.assertRaises(ValueError, f, -1)\r
- self.assertRaises(ValueError, f, 0)\r
-\r
- def test_unchanged_size(self):\r
- xml1 = ("<?xml version='1.0' encoding='iso8859'?><s>%s" % ('a' * 512))\r
- xml2 = 'a'*512 + '</s>'\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = self.counting_handler\r
- parser.buffer_size = 512\r
- parser.buffer_text = 1\r
-\r
- # Feed 512 bytes of character data: the handler should be called\r
- # once.\r
- self.n = 0\r
- parser.Parse(xml1)\r
- self.assertEqual(self.n, 1)\r
-\r
- # Reassign to buffer_size, but assign the same size.\r
- parser.buffer_size = parser.buffer_size\r
- self.assertEqual(self.n, 1)\r
-\r
- # Try parsing rest of the document\r
- parser.Parse(xml2)\r
- self.assertEqual(self.n, 2)\r
-\r
-\r
- def test_disabling_buffer(self):\r
- xml1 = "<?xml version='1.0' encoding='iso8859'?><a>%s" % ('a' * 512)\r
- xml2 = ('b' * 1024)\r
- xml3 = "%s</a>" % ('c' * 1024)\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = self.counting_handler\r
- parser.buffer_text = 1\r
- parser.buffer_size = 1024\r
- self.assertEqual(parser.buffer_size, 1024)\r
-\r
- # Parse one chunk of XML\r
- self.n = 0\r
- parser.Parse(xml1, 0)\r
- self.assertEqual(parser.buffer_size, 1024)\r
- self.assertEqual(self.n, 1)\r
-\r
- # Turn off buffering and parse the next chunk.\r
- parser.buffer_text = 0\r
- self.assertFalse(parser.buffer_text)\r
- self.assertEqual(parser.buffer_size, 1024)\r
- for i in range(10):\r
- parser.Parse(xml2, 0)\r
- self.assertEqual(self.n, 11)\r
-\r
- parser.buffer_text = 1\r
- self.assertTrue(parser.buffer_text)\r
- self.assertEqual(parser.buffer_size, 1024)\r
- parser.Parse(xml3, 1)\r
- self.assertEqual(self.n, 12)\r
-\r
-\r
-\r
- def make_document(self, bytes):\r
- return ("<?xml version='1.0'?><tag>" + bytes * 'a' + '</tag>')\r
-\r
- def counting_handler(self, text):\r
- self.n += 1\r
-\r
- def small_buffer_test(self, buffer_len):\r
- xml = "<?xml version='1.0' encoding='iso8859'?><s>%s</s>" % ('a' * buffer_len)\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = self.counting_handler\r
- parser.buffer_size = 1024\r
- parser.buffer_text = 1\r
-\r
- self.n = 0\r
- parser.Parse(xml)\r
- return self.n\r
-\r
- def test_change_size_1(self):\r
- xml1 = "<?xml version='1.0' encoding='iso8859'?><a><s>%s" % ('a' * 1024)\r
- xml2 = "aaa</s><s>%s</s></a>" % ('a' * 1025)\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = self.counting_handler\r
- parser.buffer_text = 1\r
- parser.buffer_size = 1024\r
- self.assertEqual(parser.buffer_size, 1024)\r
-\r
- self.n = 0\r
- parser.Parse(xml1, 0)\r
- parser.buffer_size *= 2\r
- self.assertEqual(parser.buffer_size, 2048)\r
- parser.Parse(xml2, 1)\r
- self.assertEqual(self.n, 2)\r
-\r
- def test_change_size_2(self):\r
- xml1 = "<?xml version='1.0' encoding='iso8859'?><a>a<s>%s" % ('a' * 1023)\r
- xml2 = "aaa</s><s>%s</s></a>" % ('a' * 1025)\r
- parser = expat.ParserCreate()\r
- parser.CharacterDataHandler = self.counting_handler\r
- parser.buffer_text = 1\r
- parser.buffer_size = 2048\r
- self.assertEqual(parser.buffer_size, 2048)\r
-\r
- self.n=0\r
- parser.Parse(xml1, 0)\r
- parser.buffer_size //= 2\r
- self.assertEqual(parser.buffer_size, 1024)\r
- parser.Parse(xml2, 1)\r
- self.assertEqual(self.n, 4)\r
-\r
-class MalformedInputText(unittest.TestCase):\r
- def test1(self):\r
- xml = "\0\r\n"\r
- parser = expat.ParserCreate()\r
- try:\r
- parser.Parse(xml, True)\r
- self.fail()\r
- except expat.ExpatError as e:\r
- self.assertEqual(str(e), 'unclosed token: line 2, column 0')\r
-\r
- def test2(self):\r
- xml = "<?xml version\xc2\x85='1.0'?>\r\n"\r
- parser = expat.ParserCreate()\r
- try:\r
- parser.Parse(xml, True)\r
- self.fail()\r
- except expat.ExpatError as e:\r
- self.assertEqual(str(e), 'XML declaration not well-formed: line 1, column 14')\r
-\r
-def test_main():\r
- run_unittest(SetAttributeTest,\r
- ParseTest,\r
- NamespaceSeparatorTest,\r
- InterningTest,\r
- BufferTextTest,\r
- HandlerExceptionTest,\r
- PositionTest,\r
- sf1296433Test,\r
- ChardataBufferTest,\r
- MalformedInputText)\r
-\r
-if __name__ == "__main__":\r
- test_main()\r