diff options
Diffstat (limited to 'Lib/test/test_xml_etree.py')
| -rw-r--r-- | Lib/test/test_xml_etree.py | 123 |
1 files changed, 123 insertions, 0 deletions
diff --git a/Lib/test/test_xml_etree.py b/Lib/test/test_xml_etree.py index 8a7ec00..bdcd4e0 100644 --- a/Lib/test/test_xml_etree.py +++ b/Lib/test/test_xml_etree.py @@ -9,6 +9,7 @@ import copy import functools import html import io +import locale import operator import pickle import sys @@ -756,6 +757,128 @@ class ElementTreeTest(unittest.TestCase): elem = ET.fromstring("<html><body>text</body></html>") self.assertEqual(ET.tostring(elem), b'<html><body>text</body></html>') + def test_tostring_default_namespace(self): + elem = ET.XML('<body xmlns="http://effbot.org/ns"><tag/></body>') + self.assertEqual( + ET.tostring(elem, encoding='unicode'), + '<ns0:body xmlns:ns0="http://effbot.org/ns"><ns0:tag /></ns0:body>' + ) + self.assertEqual( + ET.tostring(elem, encoding='unicode', default_namespace='http://effbot.org/ns'), + '<body xmlns="http://effbot.org/ns"><tag /></body>' + ) + + def test_tostring_default_namespace_different_namespace(self): + elem = ET.XML('<body xmlns="http://effbot.org/ns"><tag/></body>') + self.assertEqual( + ET.tostring(elem, encoding='unicode', default_namespace='foobar'), + '<ns1:body xmlns="foobar" xmlns:ns1="http://effbot.org/ns"><ns1:tag /></ns1:body>' + ) + + def test_tostring_default_namespace_original_no_namespace(self): + elem = ET.XML('<body><tag/></body>') + EXPECTED_MSG = '^cannot use non-qualified names with default_namespace option$' + with self.assertRaisesRegex(ValueError, EXPECTED_MSG): + ET.tostring(elem, encoding='unicode', default_namespace='foobar') + + def test_tostring_no_xml_declaration(self): + elem = ET.XML('<body><tag/></body>') + self.assertEqual( + ET.tostring(elem, encoding='unicode'), + '<body><tag /></body>' + ) + + def test_tostring_xml_declaration(self): + elem = ET.XML('<body><tag/></body>') + self.assertEqual( + ET.tostring(elem, encoding='utf8', xml_declaration=True), + b"<?xml version='1.0' encoding='utf8'?>\n<body><tag /></body>" + ) + + def test_tostring_xml_declaration_unicode_encoding(self): + elem = ET.XML('<body><tag/></body>') + preferredencoding = locale.getpreferredencoding() + self.assertEqual( + f"<?xml version='1.0' encoding='{preferredencoding}'?>\n<body><tag /></body>", + ET.tostring(elem, encoding='unicode', xml_declaration=True) + ) + + def test_tostring_xml_declaration_cases(self): + elem = ET.XML('<body><tag>ø</tag></body>') + preferredencoding = locale.getpreferredencoding() + TESTCASES = [ + # (expected_retval, encoding, xml_declaration) + # ... xml_declaration = None + (b'<body><tag>ø</tag></body>', None, None), + (b'<body><tag>\xc3\xb8</tag></body>', 'UTF-8', None), + (b'<body><tag>ø</tag></body>', 'US-ASCII', None), + (b"<?xml version='1.0' encoding='ISO-8859-1'?>\n" + b"<body><tag>\xf8</tag></body>", 'ISO-8859-1', None), + ('<body><tag>ø</tag></body>', 'unicode', None), + + # ... xml_declaration = False + (b"<body><tag>ø</tag></body>", None, False), + (b"<body><tag>\xc3\xb8</tag></body>", 'UTF-8', False), + (b"<body><tag>ø</tag></body>", 'US-ASCII', False), + (b"<body><tag>\xf8</tag></body>", 'ISO-8859-1', False), + ("<body><tag>ø</tag></body>", 'unicode', False), + + # ... xml_declaration = True + (b"<?xml version='1.0' encoding='us-ascii'?>\n" + b"<body><tag>ø</tag></body>", None, True), + (b"<?xml version='1.0' encoding='UTF-8'?>\n" + b"<body><tag>\xc3\xb8</tag></body>", 'UTF-8', True), + (b"<?xml version='1.0' encoding='US-ASCII'?>\n" + b"<body><tag>ø</tag></body>", 'US-ASCII', True), + (b"<?xml version='1.0' encoding='ISO-8859-1'?>\n" + b"<body><tag>\xf8</tag></body>", 'ISO-8859-1', True), + (f"<?xml version='1.0' encoding='{preferredencoding}'?>\n" + "<body><tag>ø</tag></body>", 'unicode', True), + + ] + for expected_retval, encoding, xml_declaration in TESTCASES: + with self.subTest(f'encoding={encoding} ' + f'xml_declaration={xml_declaration}'): + self.assertEqual( + ET.tostring( + elem, + encoding=encoding, + xml_declaration=xml_declaration + ), + expected_retval + ) + + def test_tostringlist_default_namespace(self): + elem = ET.XML('<body xmlns="http://effbot.org/ns"><tag/></body>') + self.assertEqual( + ''.join(ET.tostringlist(elem, encoding='unicode')), + '<ns0:body xmlns:ns0="http://effbot.org/ns"><ns0:tag /></ns0:body>' + ) + self.assertEqual( + ''.join(ET.tostringlist(elem, encoding='unicode', default_namespace='http://effbot.org/ns')), + '<body xmlns="http://effbot.org/ns"><tag /></body>' + ) + + def test_tostringlist_xml_declaration(self): + elem = ET.XML('<body><tag/></body>') + self.assertEqual( + ''.join(ET.tostringlist(elem, encoding='unicode')), + '<body><tag /></body>' + ) + self.assertEqual( + b''.join(ET.tostringlist(elem, xml_declaration=True)), + b"<?xml version='1.0' encoding='us-ascii'?>\n<body><tag /></body>" + ) + + preferredencoding = locale.getpreferredencoding() + stringlist = ET.tostringlist(elem, encoding='unicode', xml_declaration=True) + self.assertEqual( + ''.join(stringlist), + f"<?xml version='1.0' encoding='{preferredencoding}'?>\n<body><tag /></body>" + ) + self.assertRegex(stringlist[0], r"^<\?xml version='1.0' encoding='.+'?>") + self.assertEqual(['<body', '>', '<tag', ' />', '</body>'], stringlist[1:]) + def test_encoding(self): def check(encoding, body=''): xml = ("<?xml version='1.0' encoding='%s'?><xml>%s</xml>" % |