104 lines
4.2 KiB
Python
104 lines
4.2 KiB
Python
"""
|
||
Tests for the html module functions.
|
||
"""
|
||
|
||
import html
|
||
import unittest
|
||
|
||
|
||
class HtmlTests(unittest.TestCase):
|
||
def test_escape(self):
|
||
self.assertEqual(
|
||
html.escape('\'<script>"&foo;"</script>\''),
|
||
''<script>"&foo;"</script>'')
|
||
self.assertEqual(
|
||
html.escape('\'<script>"&foo;"</script>\'', False),
|
||
'\'<script>"&foo;"</script>\'')
|
||
|
||
def test_unescape(self):
|
||
numeric_formats = ['&#%d', '&#%d;', '&#x%x', '&#x%x;']
|
||
errmsg = 'unescape(%r) should have returned %r'
|
||
def check(text, expected):
|
||
self.assertEqual(html.unescape(text), expected,
|
||
msg=errmsg % (text, expected))
|
||
def check_num(num, expected):
|
||
for format in numeric_formats:
|
||
text = format % num
|
||
self.assertEqual(html.unescape(text), expected,
|
||
msg=errmsg % (text, expected))
|
||
# check text with no character references
|
||
check('no character references', 'no character references')
|
||
# check & followed by invalid chars
|
||
check('&\n&\t& &&', '&\n&\t& &&')
|
||
# check & followed by numbers and letters
|
||
check('&0 &9 &a &0; &9; &a;', '&0 &9 &a &0; &9; &a;')
|
||
# check incomplete entities at the end of the string
|
||
for x in ['&', '&#', '&#x', '&#X', '&#y', '&#xy', '&#Xy']:
|
||
check(x, x)
|
||
check(x+';', x+';')
|
||
# check several combinations of numeric character references,
|
||
# possibly followed by different characters
|
||
formats = ['&#%d', '&#%07d', '&#%d;', '&#%07d;',
|
||
'&#x%x', '&#x%06x', '&#x%x;', '&#x%06x;',
|
||
'&#x%X', '&#x%06X', '&#X%x;', '&#X%06x;']
|
||
for num, char in zip([65, 97, 34, 38, 0x2603, 0x101234],
|
||
['A', 'a', '"', '&', '\u2603', '\U00101234']):
|
||
for s in formats:
|
||
check(s % num, char)
|
||
for end in [' ', 'X']:
|
||
check((s+end) % num, char+end)
|
||
# check invalid code points
|
||
for cp in [0xD800, 0xDB00, 0xDC00, 0xDFFF, 0x110000]:
|
||
check_num(cp, '\uFFFD')
|
||
# check more invalid code points
|
||
for cp in [0x1, 0xb, 0xe, 0x7f, 0xfffe, 0xffff, 0x10fffe, 0x10ffff]:
|
||
check_num(cp, '')
|
||
# check invalid numbers
|
||
for num, ch in zip([0x0d, 0x80, 0x95, 0x9d], '\r\u20ac\u2022\x9d'):
|
||
check_num(num, ch)
|
||
# check small numbers
|
||
check_num(0, '\uFFFD')
|
||
check_num(9, '\t')
|
||
# check a big number
|
||
check_num(1000000000000000000, '\uFFFD')
|
||
# check that multiple trailing semicolons are handled correctly
|
||
for e in ['";', '";', '";', '";']:
|
||
check(e, '";')
|
||
# check that semicolons in the middle don't create problems
|
||
for e in ['"quot;', '"quot;', '"quot;', '"quot;']:
|
||
check(e, '"quot;')
|
||
# check triple adjacent charrefs
|
||
for e in ['"', '"', '"', '"']:
|
||
check(e*3, '"""')
|
||
check((e+';')*3, '"""')
|
||
# check that the case is respected
|
||
for e in ['&', '&', '&', '&']:
|
||
check(e, '&')
|
||
for e in ['&Amp', '&Amp;']:
|
||
check(e, e)
|
||
# check that non-existent named entities are returned unchanged
|
||
check('&svadilfari;', '&svadilfari;')
|
||
# the following examples are in the html5 specs
|
||
check('¬it', '¬it')
|
||
check('¬it;', '¬it;')
|
||
check('¬in', '¬in')
|
||
check('∉', '∉')
|
||
# a similar example with a long name
|
||
check('¬ReallyAnExistingNamedCharacterReference;',
|
||
'¬ReallyAnExistingNamedCharacterReference;')
|
||
# longest valid name
|
||
check('∳', '∳')
|
||
# check a charref that maps to two unicode chars
|
||
check('∾̳', '\u223E\u0333')
|
||
check('&acE', '&acE')
|
||
# see #12888
|
||
check('{ ' * 1050, '{ ' * 1050)
|
||
# see #15156
|
||
check('ÉricÉric&alphacentauriαcentauri',
|
||
'ÉricÉric&alphacentauriαcentauri')
|
||
check('&co;', '&co;')
|
||
|
||
|
||
if __name__ == '__main__':
|
||
unittest.main()
|