mirror of
https://github.com/python/cpython.git
synced 2025-07-23 11:15:24 +00:00
bpo-36227: ElementTree.tostring() default_namespace and xml_declaration arguments (GH-12225)
Add new keyword arguments "default_namespace" and "xml_declaration" to functions ET.tostring() and ET.tostringlist(), as known from ElementTree.write().
This commit is contained in:
Bernt Røskar Brenna
Stefan Behnel
parent
830b43d03c
commit
ffca16e25a
4 changed files with 152 additions and 9 deletions
|
|
@ -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>" %
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue