179 lines
7.8 KiB
Python
179 lines
7.8 KiB
Python
# -*- coding: utf-8 -*-
|
||
from __future__ import unicode_literals
|
||
|
||
from unittest import skipUnless
|
||
import warnings
|
||
|
||
from django.test import SimpleTestCase
|
||
from django.utils import six, text
|
||
|
||
IS_WIDE_BUILD = (len('\U0001F4A9') == 1)
|
||
|
||
|
||
class TestUtilsText(SimpleTestCase):
|
||
|
||
def test_truncate_chars(self):
|
||
truncator = text.Truncator(
|
||
'The quick brown fox jumped over the lazy dog.'
|
||
)
|
||
self.assertEqual('The quick brown fox jumped over the lazy dog.',
|
||
truncator.chars(100)),
|
||
self.assertEqual('The quick brown fox ...',
|
||
truncator.chars(23)),
|
||
self.assertEqual('The quick brown fo.....',
|
||
truncator.chars(23, '.....')),
|
||
|
||
# Ensure that we normalize our unicode data first
|
||
nfc = text.Truncator('o\xfco\xfco\xfco\xfc')
|
||
nfd = text.Truncator('ou\u0308ou\u0308ou\u0308ou\u0308')
|
||
self.assertEqual('oüoüoüoü', nfc.chars(8))
|
||
self.assertEqual('oüoüoüoü', nfd.chars(8))
|
||
self.assertEqual('oü...', nfc.chars(5))
|
||
self.assertEqual('oü...', nfd.chars(5))
|
||
|
||
# Ensure the final length is calculated correctly when there are
|
||
# combining characters with no precomposed form, and that combining
|
||
# characters are not split up.
|
||
truncator = text.Truncator('-B\u030AB\u030A----8')
|
||
self.assertEqual('-B\u030A...', truncator.chars(5))
|
||
self.assertEqual('-B\u030AB\u030A-...', truncator.chars(7))
|
||
self.assertEqual('-B\u030AB\u030A----8', truncator.chars(8))
|
||
|
||
# Ensure the length of the end text is correctly calculated when it
|
||
# contains combining characters with no precomposed form.
|
||
truncator = text.Truncator('-----')
|
||
self.assertEqual('---B\u030A', truncator.chars(4, 'B\u030A'))
|
||
self.assertEqual('-----', truncator.chars(5, 'B\u030A'))
|
||
|
||
# Make a best effort to shorten to the desired length, but requesting
|
||
# a length shorter than the ellipsis shouldn't break
|
||
self.assertEqual('...', text.Truncator('asdf').chars(1))
|
||
|
||
def test_truncate_words(self):
|
||
truncator = text.Truncator('The quick brown fox jumped over the lazy '
|
||
'dog.')
|
||
self.assertEqual('The quick brown fox jumped over the lazy dog.',
|
||
truncator.words(10))
|
||
self.assertEqual('The quick brown fox...', truncator.words(4))
|
||
self.assertEqual('The quick brown fox[snip]',
|
||
truncator.words(4, '[snip]'))
|
||
|
||
def test_truncate_html_words(self):
|
||
truncator = text.Truncator('<p id="par"><strong><em>The quick brown fox'
|
||
' jumped over the lazy dog.</em></strong></p>')
|
||
self.assertEqual('<p id="par"><strong><em>The quick brown fox jumped over'
|
||
' the lazy dog.</em></strong></p>', truncator.words(10, html=True))
|
||
self.assertEqual('<p id="par"><strong><em>The quick brown fox...</em>'
|
||
'</strong></p>', truncator.words(4, html=True))
|
||
self.assertEqual('<p id="par"><strong><em>The quick brown fox....</em>'
|
||
'</strong></p>', truncator.words(4, '....', html=True))
|
||
self.assertEqual('<p id="par"><strong><em>The quick brown fox</em>'
|
||
'</strong></p>', truncator.words(4, '', html=True))
|
||
|
||
# Test with new line inside tag
|
||
truncator = text.Truncator('<p>The quick <a href="xyz.html"\n'
|
||
'id="mylink">brown fox</a> jumped over the lazy dog.</p>')
|
||
self.assertEqual('<p>The quick <a href="xyz.html"\n'
|
||
'id="mylink">brown...</a></p>', truncator.words(3, '...', html=True))
|
||
|
||
# Test self-closing tags
|
||
truncator = text.Truncator('<br/>The <hr />quick brown fox jumped over'
|
||
' the lazy dog.')
|
||
self.assertEqual('<br/>The <hr />quick brown...',
|
||
truncator.words(3, '...', html=True))
|
||
truncator = text.Truncator('<br>The <hr/>quick <em>brown fox</em> '
|
||
'jumped over the lazy dog.')
|
||
self.assertEqual('<br>The <hr/>quick <em>brown...</em>',
|
||
truncator.words(3, '...', html=True))
|
||
|
||
# Test html entities
|
||
truncator = text.Truncator('<i>Buenos días!'
|
||
' ¿Cómo está?</i>')
|
||
self.assertEqual('<i>Buenos días! ¿Cómo...</i>',
|
||
truncator.words(3, '...', html=True))
|
||
truncator = text.Truncator('<p>I <3 python, what about you?</p>')
|
||
self.assertEqual('<p>I <3 python...</p>',
|
||
truncator.words(3, '...', html=True))
|
||
|
||
def test_wrap(self):
|
||
digits = '1234 67 9'
|
||
self.assertEqual(text.wrap(digits, 100), '1234 67 9')
|
||
self.assertEqual(text.wrap(digits, 9), '1234 67 9')
|
||
self.assertEqual(text.wrap(digits, 8), '1234 67\n9')
|
||
|
||
self.assertEqual(text.wrap('short\na long line', 7),
|
||
'short\na long\nline')
|
||
|
||
self.assertEqual(text.wrap('do-not-break-long-words please? ok', 8),
|
||
'do-not-break-long-words\nplease?\nok')
|
||
|
||
long_word = 'l%sng' % ('o' * 20)
|
||
self.assertEqual(text.wrap(long_word, 20), long_word)
|
||
self.assertEqual(text.wrap('a %s word' % long_word, 10),
|
||
'a\n%s\nword' % long_word)
|
||
|
||
def test_normalize_newlines(self):
|
||
self.assertEqual(text.normalize_newlines("abc\ndef\rghi\r\n"),
|
||
"abc\ndef\nghi\n")
|
||
self.assertEqual(text.normalize_newlines("\n\r\r\n\r"), "\n\n\n\n")
|
||
self.assertEqual(text.normalize_newlines("abcdefghi"), "abcdefghi")
|
||
self.assertEqual(text.normalize_newlines(""), "")
|
||
|
||
def test_normalize_newlines_bytes(self):
|
||
"""normalize_newlines should be able to handle bytes too"""
|
||
normalized = text.normalize_newlines(b"abc\ndef\rghi\r\n")
|
||
self.assertEqual(normalized, "abc\ndef\nghi\n")
|
||
self.assertIsInstance(normalized, six.text_type)
|
||
|
||
def test_slugify(self):
|
||
items = (
|
||
('Hello, World!', 'hello-world'),
|
||
('spam & eggs', 'spam-eggs'),
|
||
)
|
||
for value, output in items:
|
||
self.assertEqual(text.slugify(value), output)
|
||
|
||
def test_unescape_entities(self):
|
||
items = [
|
||
('', ''),
|
||
('foo', 'foo'),
|
||
('&', '&'),
|
||
('&', '&'),
|
||
('&', '&'),
|
||
('foo & bar', 'foo & bar'),
|
||
('foo & bar', 'foo & bar'),
|
||
]
|
||
for value, output in items:
|
||
self.assertEqual(text.unescape_entities(value), output)
|
||
|
||
def test_get_valid_filename(self):
|
||
filename = "^&'@{}[],$=!-#()%+~_123.txt"
|
||
self.assertEqual(text.get_valid_filename(filename), "-_123.txt")
|
||
|
||
def test_javascript_quote(self):
|
||
input = "<script>alert('Hello \\xff.\n Welcome\there\r');</script>"
|
||
output = r"<script>alert(\'Hello \\xff.\n Welcome\there\r\');<\/script>"
|
||
with warnings.catch_warnings():
|
||
self.assertEqual(text.javascript_quote(input), output)
|
||
|
||
# Exercising quote_double_quotes keyword argument
|
||
input = '"Text"'
|
||
with warnings.catch_warnings():
|
||
self.assertEqual(text.javascript_quote(input), '"Text"')
|
||
self.assertEqual(text.javascript_quote(input, quote_double_quotes=True),
|
||
'"Text"')
|
||
|
||
@skipUnless(IS_WIDE_BUILD, 'Not running in a wide build of Python')
|
||
def test_javascript_quote_unicode(self):
|
||
input = "<script>alert('Hello \\xff.\n Wel𝕃come\there\r');</script>"
|
||
output = r"<script>alert(\'Hello \\xff.\n Wel𝕃come\there\r\');<\/script>"
|
||
with warnings.catch_warnings():
|
||
self.assertEqual(text.javascript_quote(input), output)
|
||
|
||
def test_deprecation(self):
|
||
with warnings.catch_warnings(record=True) as w:
|
||
warnings.simplefilter("always")
|
||
text.javascript_quote('thingy')
|
||
self.assertEqual(len(w), 1)
|
||
self.assertIn('escapejs()', repr(w[0].message))
|