Correct XML ampersand fixup

2014-01-20 22:11:34 +01:00 · 2014-01-20 22:11:34 +01:00 · 5aafe895fc
commit 5aafe895fc
parent b853d2e155
5 changed files with 25 additions and 10 deletions
--- a/test/test_utils.py
+++ b/test/test_utils.py
@ -16,6 +16,7 @@ from youtube_dl.utils import (
    DateRange,
    encodeFilename,
    find_xpath_attr,
+    fix_xml_ampersands,
    get_meta_content,
    orderedSet,
    parse_duration,
@ -200,5 +201,18 @@ class TestUtil(unittest.TestCase):
        self.assertEqual(parse_duration('9:12:43'), 33163)
        self.assertEqual(parse_duration('x:y'), None)

+    def test_fix_xml_ampersands(self):
+        self.assertEqual(
+            fix_xml_ampersands('"&x=y&z=a'), '"&amp;x=y&amp;z=a')
+        self.assertEqual(
+            fix_xml_ampersands('"&amp;x=y&wrong;&z=a'),
+            '"&amp;x=y&amp;wrong;&amp;z=a')
+        self.assertEqual(
+            fix_xml_ampersands('&amp;&apos;&gt;&lt;&quot;'),
+            '&amp;&apos;&gt;&lt;&quot;')
+        self.assertEqual(
+            fix_xml_ampersands('&#1234;&#x1abC;'), '&#1234;&#x1abC;')
+        self.assertEqual(fix_xml_ampersands('&#&#'), '&amp;#&amp;#')
+
 if __name__ == '__main__':
    unittest.main()