tests.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667
  1. # -*- coding: utf-8 -*-
  2. from __future__ import unicode_literals
  3. import datetime
  4. import decimal
  5. import unittest
  6. from django.template.defaultfilters import *
  7. from django.test import TestCase
  8. from django.utils import six
  9. from django.utils import translation
  10. from django.utils.safestring import SafeData
  11. from django.utils.encoding import python_2_unicode_compatible
  12. from i18n import TransRealMixin
  13. class DefaultFiltersTests(TestCase):
  14. def test_floatformat(self):
  15. self.assertEqual(floatformat(7.7), '7.7')
  16. self.assertEqual(floatformat(7.0), '7')
  17. self.assertEqual(floatformat(0.7), '0.7')
  18. self.assertEqual(floatformat(0.07), '0.1')
  19. self.assertEqual(floatformat(0.007), '0.0')
  20. self.assertEqual(floatformat(0.0), '0')
  21. self.assertEqual(floatformat(7.7, 3), '7.700')
  22. self.assertEqual(floatformat(6.000000, 3), '6.000')
  23. self.assertEqual(floatformat(6.200000, 3), '6.200')
  24. self.assertEqual(floatformat(6.200000, -3), '6.200')
  25. self.assertEqual(floatformat(13.1031, -3), '13.103')
  26. self.assertEqual(floatformat(11.1197, -2), '11.12')
  27. self.assertEqual(floatformat(11.0000, -2), '11')
  28. self.assertEqual(floatformat(11.000001, -2), '11.00')
  29. self.assertEqual(floatformat(8.2798, 3), '8.280')
  30. self.assertEqual(floatformat(5555.555, 2), '5555.56')
  31. self.assertEqual(floatformat(001.3000, 2), '1.30')
  32. self.assertEqual(floatformat(0.12345, 2), '0.12')
  33. self.assertEqual(floatformat(decimal.Decimal('555.555'), 2), '555.56')
  34. self.assertEqual(floatformat(decimal.Decimal('09.000')), '9')
  35. self.assertEqual(floatformat('foo'), '')
  36. self.assertEqual(floatformat(13.1031, 'bar'), '13.1031')
  37. self.assertEqual(floatformat(18.125, 2), '18.13')
  38. self.assertEqual(floatformat('foo', 'bar'), '')
  39. self.assertEqual(floatformat('¿Cómo esta usted?'), '')
  40. self.assertEqual(floatformat(None), '')
  41. # Check that we're not converting to scientific notation.
  42. self.assertEqual(floatformat(0, 6), '0.000000')
  43. self.assertEqual(floatformat(0, 7), '0.0000000')
  44. self.assertEqual(floatformat(0, 10), '0.0000000000')
  45. self.assertEqual(floatformat(0.000000000000000000015, 20),
  46. '0.00000000000000000002')
  47. pos_inf = float(1e30000)
  48. self.assertEqual(floatformat(pos_inf), six.text_type(pos_inf))
  49. neg_inf = float(-1e30000)
  50. self.assertEqual(floatformat(neg_inf), six.text_type(neg_inf))
  51. nan = pos_inf / pos_inf
  52. self.assertEqual(floatformat(nan), six.text_type(nan))
  53. class FloatWrapper(object):
  54. def __init__(self, value):
  55. self.value = value
  56. def __float__(self):
  57. return self.value
  58. self.assertEqual(floatformat(FloatWrapper(11.000001), -2), '11.00')
  59. # Regression for #15789
  60. decimal_ctx = decimal.getcontext()
  61. old_prec, decimal_ctx.prec = decimal_ctx.prec, 2
  62. try:
  63. self.assertEqual(floatformat(1.2345, 2), '1.23')
  64. self.assertEqual(floatformat(15.2042, -3), '15.204')
  65. self.assertEqual(floatformat(1.2345, '2'), '1.23')
  66. self.assertEqual(floatformat(15.2042, '-3'), '15.204')
  67. self.assertEqual(floatformat(decimal.Decimal('1.2345'), 2), '1.23')
  68. self.assertEqual(floatformat(decimal.Decimal('15.2042'), -3), '15.204')
  69. finally:
  70. decimal_ctx.prec = old_prec
  71. def test_floatformat_py2_fail(self):
  72. self.assertEqual(floatformat(1.00000000000000015, 16), '1.0000000000000002')
  73. # The test above fails because of Python 2's float handling. Floats with
  74. # many zeroes after the decimal point should be passed in as another type
  75. # such as unicode or Decimal.
  76. if not six.PY3:
  77. test_floatformat_py2_fail = unittest.expectedFailure(test_floatformat_py2_fail)
  78. def test_addslashes(self):
  79. self.assertEqual(addslashes('"double quotes" and \'single quotes\''),
  80. '\\"double quotes\\" and \\\'single quotes\\\'')
  81. self.assertEqual(addslashes(r'\ : backslashes, too'),
  82. '\\\\ : backslashes, too')
  83. def test_capfirst(self):
  84. self.assertEqual(capfirst('hello world'), 'Hello world')
  85. def test_escapejs(self):
  86. self.assertEqual(escapejs_filter('"double quotes" and \'single quotes\''),
  87. '\\u0022double quotes\\u0022 and \\u0027single quotes\\u0027')
  88. self.assertEqual(escapejs_filter(r'\ : backslashes, too'),
  89. '\\u005C : backslashes, too')
  90. self.assertEqual(escapejs_filter('and lots of whitespace: \r\n\t\v\f\b'),
  91. 'and lots of whitespace: \\u000D\\u000A\\u0009\\u000B\\u000C\\u0008')
  92. self.assertEqual(escapejs_filter(r'<script>and this</script>'),
  93. '\\u003Cscript\\u003Eand this\\u003C/script\\u003E')
  94. self.assertEqual(
  95. escapejs_filter('paragraph separator:\u2029and line separator:\u2028'),
  96. 'paragraph separator:\\u2029and line separator:\\u2028')
  97. def test_fix_ampersands(self):
  98. self.assertEqual(fix_ampersands_filter('Jack & Jill & Jeroboam'),
  99. 'Jack &amp; Jill &amp; Jeroboam')
  100. def test_linenumbers(self):
  101. self.assertEqual(linenumbers('line 1\nline 2'),
  102. '1. line 1\n2. line 2')
  103. self.assertEqual(linenumbers('\n'.join(['x'] * 10)),
  104. '01. x\n02. x\n03. x\n04. x\n05. x\n06. x\n07. '\
  105. 'x\n08. x\n09. x\n10. x')
  106. def test_lower(self):
  107. self.assertEqual(lower('TEST'), 'test')
  108. # uppercase E umlaut
  109. self.assertEqual(lower('\xcb'), '\xeb')
  110. def test_make_list(self):
  111. self.assertEqual(make_list('abc'), ['a', 'b', 'c'])
  112. self.assertEqual(make_list(1234), ['1', '2', '3', '4'])
  113. def test_slugify(self):
  114. self.assertEqual(slugify(' Jack & Jill like numbers 1,2,3 and 4 and'\
  115. ' silly characters ?%.$!/'),
  116. 'jack-jill-like-numbers-123-and-4-and-silly-characters')
  117. self.assertEqual(slugify("Un \xe9l\xe9phant \xe0 l'or\xe9e du bois"),
  118. 'un-elephant-a-loree-du-bois')
  119. def test_stringformat(self):
  120. self.assertEqual(stringformat(1, '03d'), '001')
  121. self.assertEqual(stringformat(1, 'z'), '')
  122. def test_title(self):
  123. self.assertEqual(title('a nice title, isn\'t it?'),
  124. "A Nice Title, Isn't It?")
  125. self.assertEqual(title('discoth\xe8que'), 'Discoth\xe8que')
  126. def test_truncatewords(self):
  127. self.assertEqual(
  128. truncatewords('A sentence with a few words in it', 1), 'A ...')
  129. self.assertEqual(
  130. truncatewords('A sentence with a few words in it', 5),
  131. 'A sentence with a few ...')
  132. self.assertEqual(
  133. truncatewords('A sentence with a few words in it', 100),
  134. 'A sentence with a few words in it')
  135. self.assertEqual(
  136. truncatewords('A sentence with a few words in it',
  137. 'not a number'), 'A sentence with a few words in it')
  138. def test_truncatewords_html(self):
  139. self.assertEqual(truncatewords_html(
  140. '<p>one <a href="#">two - three <br>four</a> five</p>', 0), '')
  141. self.assertEqual(truncatewords_html('<p>one <a href="#">two - '\
  142. 'three <br>four</a> five</p>', 2),
  143. '<p>one <a href="#">two ...</a></p>')
  144. self.assertEqual(truncatewords_html(
  145. '<p>one <a href="#">two - three <br>four</a> five</p>', 4),
  146. '<p>one <a href="#">two - three <br>four ...</a></p>')
  147. self.assertEqual(truncatewords_html(
  148. '<p>one <a href="#">two - three <br>four</a> five</p>', 5),
  149. '<p>one <a href="#">two - three <br>four</a> five</p>')
  150. self.assertEqual(truncatewords_html(
  151. '<p>one <a href="#">two - three <br>four</a> five</p>', 100),
  152. '<p>one <a href="#">two - three <br>four</a> five</p>')
  153. self.assertEqual(truncatewords_html(
  154. '\xc5ngstr\xf6m was here', 1), '\xc5ngstr\xf6m ...')
  155. def test_upper(self):
  156. self.assertEqual(upper('Mixed case input'), 'MIXED CASE INPUT')
  157. # lowercase e umlaut
  158. self.assertEqual(upper('\xeb'), '\xcb')
  159. def test_urlencode(self):
  160. self.assertEqual(urlencode('fran\xe7ois & jill'),
  161. 'fran%C3%A7ois%20%26%20jill')
  162. self.assertEqual(urlencode(1), '1')
  163. def test_iriencode(self):
  164. self.assertEqual(iriencode('S\xf8r-Tr\xf8ndelag'),
  165. 'S%C3%B8r-Tr%C3%B8ndelag')
  166. self.assertEqual(iriencode(urlencode('fran\xe7ois & jill')),
  167. 'fran%C3%A7ois%20%26%20jill')
  168. def test_urlizetrunc(self):
  169. self.assertEqual(urlizetrunc('http://short.com/', 20), '<a href='\
  170. '"http://short.com/" rel="nofollow">http://short.com/</a>')
  171. self.assertEqual(urlizetrunc('http://www.google.co.uk/search?hl=en'\
  172. '&q=some+long+url&btnG=Search&meta=', 20), '<a href="http://'\
  173. 'www.google.co.uk/search?hl=en&q=some+long+url&btnG=Search&'\
  174. 'meta=" rel="nofollow">http://www.google...</a>')
  175. self.assertEqual(urlizetrunc('http://www.google.co.uk/search?hl=en'\
  176. '&q=some+long+url&btnG=Search&meta=', 20), '<a href="http://'\
  177. 'www.google.co.uk/search?hl=en&q=some+long+url&btnG=Search'\
  178. '&meta=" rel="nofollow">http://www.google...</a>')
  179. # Check truncating of URIs which are the exact length
  180. uri = 'http://31characteruri.com/test/'
  181. self.assertEqual(len(uri), 31)
  182. self.assertEqual(urlizetrunc(uri, 31),
  183. '<a href="http://31characteruri.com/test/" rel="nofollow">'\
  184. 'http://31characteruri.com/test/</a>')
  185. self.assertEqual(urlizetrunc(uri, 30),
  186. '<a href="http://31characteruri.com/test/" rel="nofollow">'\
  187. 'http://31characteruri.com/t...</a>')
  188. self.assertEqual(urlizetrunc(uri, 2),
  189. '<a href="http://31characteruri.com/test/"'\
  190. ' rel="nofollow">...</a>')
  191. def test_urlize(self):
  192. # Check normal urlize
  193. self.assertEqual(urlize('http://google.com'),
  194. '<a href="http://google.com" rel="nofollow">http://google.com</a>')
  195. self.assertEqual(urlize('http://google.com/'),
  196. '<a href="http://google.com/" rel="nofollow">http://google.com/</a>')
  197. self.assertEqual(urlize('www.google.com'),
  198. '<a href="http://www.google.com" rel="nofollow">www.google.com</a>')
  199. self.assertEqual(urlize('djangoproject.org'),
  200. '<a href="http://djangoproject.org" rel="nofollow">djangoproject.org</a>')
  201. self.assertEqual(urlize('info@djangoproject.org'),
  202. '<a href="mailto:info@djangoproject.org">info@djangoproject.org</a>')
  203. # Check urlize with https addresses
  204. self.assertEqual(urlize('https://google.com'),
  205. '<a href="https://google.com" rel="nofollow">https://google.com</a>')
  206. # Check urlize doesn't overquote already quoted urls - see #9655
  207. self.assertEqual(urlize('http://hi.baidu.com/%D6%D8%D0%C2%BF'),
  208. '<a href="http://hi.baidu.com/%D6%D8%D0%C2%BF" rel="nofollow">'
  209. 'http://hi.baidu.com/%D6%D8%D0%C2%BF</a>')
  210. self.assertEqual(urlize('www.mystore.com/30%OffCoupons!'),
  211. '<a href="http://www.mystore.com/30%25OffCoupons!" rel="nofollow">'
  212. 'www.mystore.com/30%OffCoupons!</a>')
  213. self.assertEqual(urlize('http://en.wikipedia.org/wiki/Caf%C3%A9'),
  214. '<a href="http://en.wikipedia.org/wiki/Caf%C3%A9" rel="nofollow">'
  215. 'http://en.wikipedia.org/wiki/Caf%C3%A9</a>')
  216. self.assertEqual(urlize('http://en.wikipedia.org/wiki/Café'),
  217. '<a href="http://en.wikipedia.org/wiki/Caf%C3%A9" rel="nofollow">'
  218. 'http://en.wikipedia.org/wiki/Café</a>')
  219. # Check urlize keeps balanced parentheses - see #11911
  220. self.assertEqual(urlize('http://en.wikipedia.org/wiki/Django_(web_framework)'),
  221. '<a href="http://en.wikipedia.org/wiki/Django_(web_framework)" rel="nofollow">'
  222. 'http://en.wikipedia.org/wiki/Django_(web_framework)</a>')
  223. self.assertEqual(urlize('(see http://en.wikipedia.org/wiki/Django_(web_framework))'),
  224. '(see <a href="http://en.wikipedia.org/wiki/Django_(web_framework)" rel="nofollow">'
  225. 'http://en.wikipedia.org/wiki/Django_(web_framework)</a>)')
  226. # Check urlize adds nofollow properly - see #12183
  227. self.assertEqual(urlize('foo@bar.com or www.bar.com'),
  228. '<a href="mailto:foo@bar.com">foo@bar.com</a> or '
  229. '<a href="http://www.bar.com" rel="nofollow">www.bar.com</a>')
  230. # Check urlize handles IDN correctly - see #13704
  231. self.assertEqual(urlize('http://c✶.ws'),
  232. '<a href="http://xn--c-lgq.ws" rel="nofollow">http://c✶.ws</a>')
  233. self.assertEqual(urlize('www.c✶.ws'),
  234. '<a href="http://www.xn--c-lgq.ws" rel="nofollow">www.c✶.ws</a>')
  235. self.assertEqual(urlize('c✶.org'),
  236. '<a href="http://xn--c-lgq.org" rel="nofollow">c✶.org</a>')
  237. self.assertEqual(urlize('info@c✶.org'),
  238. '<a href="mailto:info@xn--c-lgq.org">info@c✶.org</a>')
  239. # Check urlize doesn't highlight malformed URIs - see #16395
  240. self.assertEqual(urlize('http:///www.google.com'),
  241. 'http:///www.google.com')
  242. self.assertEqual(urlize('http://.google.com'),
  243. 'http://.google.com')
  244. self.assertEqual(urlize('http://@foo.com'),
  245. 'http://@foo.com')
  246. # Check urlize accepts more TLDs - see #16656
  247. self.assertEqual(urlize('usa.gov'),
  248. '<a href="http://usa.gov" rel="nofollow">usa.gov</a>')
  249. # Check urlize don't crash on invalid email with dot-starting domain - see #17592
  250. self.assertEqual(urlize('email@.stream.ru'),
  251. 'email@.stream.ru')
  252. # Check urlize accepts uppercased URL schemes - see #18071
  253. self.assertEqual(urlize('HTTPS://github.com/'),
  254. '<a href="https://github.com/" rel="nofollow">HTTPS://github.com/</a>')
  255. # Check urlize trims trailing period when followed by parenthesis - see #18644
  256. self.assertEqual(urlize('(Go to http://www.example.com/foo.)'),
  257. '(Go to <a href="http://www.example.com/foo" rel="nofollow">http://www.example.com/foo</a>.)')
  258. # Check urlize handles brackets properly (#19070)
  259. self.assertEqual(urlize('[see www.example.com]'),
  260. '[see <a href="http://www.example.com" rel="nofollow">www.example.com</a>]' )
  261. self.assertEqual(urlize('see test[at[example.com'),
  262. 'see <a href="http://test[at[example.com" rel="nofollow">test[at[example.com</a>' )
  263. self.assertEqual(urlize('[http://168.192.0.1](http://168.192.0.1)'),
  264. '[<a href="http://168.192.0.1](http://168.192.0.1)" rel="nofollow">http://168.192.0.1](http://168.192.0.1)</a>')
  265. # Check urlize works with IPv4/IPv6 addresses
  266. self.assertEqual(urlize('http://192.168.0.15/api/9'),
  267. '<a href="http://192.168.0.15/api/9" rel="nofollow">http://192.168.0.15/api/9</a>')
  268. self.assertEqual(urlize('http://[2001:db8:cafe::2]/api/9'),
  269. '<a href="http://[2001:db8:cafe::2]/api/9" rel="nofollow">http://[2001:db8:cafe::2]/api/9</a>')
  270. def test_wordcount(self):
  271. self.assertEqual(wordcount(''), 0)
  272. self.assertEqual(wordcount('oneword'), 1)
  273. self.assertEqual(wordcount('lots of words'), 3)
  274. self.assertEqual(wordwrap('this is a long paragraph of text that '\
  275. 'really needs to be wrapped I\'m afraid', 14),
  276. "this is a long\nparagraph of\ntext that\nreally needs\nto be "\
  277. "wrapped\nI'm afraid")
  278. self.assertEqual(wordwrap('this is a short paragraph of text.\n '\
  279. 'But this line should be indented', 14),
  280. 'this is a\nshort\nparagraph of\ntext.\n But this\nline '\
  281. 'should be\nindented')
  282. self.assertEqual(wordwrap('this is a short paragraph of text.\n '\
  283. 'But this line should be indented',15), 'this is a short\n'\
  284. 'paragraph of\ntext.\n But this line\nshould be\nindented')
  285. def test_rjust(self):
  286. self.assertEqual(ljust('test', 10), 'test ')
  287. self.assertEqual(ljust('test', 3), 'test')
  288. self.assertEqual(rjust('test', 10), ' test')
  289. self.assertEqual(rjust('test', 3), 'test')
  290. def test_center(self):
  291. self.assertEqual(center('test', 6), ' test ')
  292. def test_cut(self):
  293. self.assertEqual(cut('a string to be mangled', 'a'),
  294. ' string to be mngled')
  295. self.assertEqual(cut('a string to be mangled', 'ng'),
  296. 'a stri to be maled')
  297. self.assertEqual(cut('a string to be mangled', 'strings'),
  298. 'a string to be mangled')
  299. def test_force_escape(self):
  300. escaped = force_escape('<some html & special characters > here')
  301. self.assertEqual(
  302. escaped, '&lt;some html &amp; special characters &gt; here')
  303. self.assertIsInstance(escaped, SafeData)
  304. self.assertEqual(
  305. force_escape('<some html & special characters > here ĐÅ€£'),
  306. '&lt;some html &amp; special characters &gt; here'\
  307. ' \u0110\xc5\u20ac\xa3')
  308. def test_linebreaks(self):
  309. self.assertEqual(linebreaks_filter('line 1'), '<p>line 1</p>')
  310. self.assertEqual(linebreaks_filter('line 1\nline 2'),
  311. '<p>line 1<br />line 2</p>')
  312. self.assertEqual(linebreaks_filter('line 1\rline 2'),
  313. '<p>line 1<br />line 2</p>')
  314. self.assertEqual(linebreaks_filter('line 1\r\nline 2'),
  315. '<p>line 1<br />line 2</p>')
  316. def test_linebreaksbr(self):
  317. self.assertEqual(linebreaksbr('line 1\nline 2'),
  318. 'line 1<br />line 2')
  319. self.assertEqual(linebreaksbr('line 1\rline 2'),
  320. 'line 1<br />line 2')
  321. self.assertEqual(linebreaksbr('line 1\r\nline 2'),
  322. 'line 1<br />line 2')
  323. def test_removetags(self):
  324. self.assertEqual(removetags('some <b>html</b> with <script>alert'\
  325. '("You smell")</script> disallowed <img /> tags', 'script img'),
  326. 'some <b>html</b> with alert("You smell") disallowed tags')
  327. self.assertEqual(striptags('some <b>html</b> with <script>alert'\
  328. '("You smell")</script> disallowed <img /> tags'),
  329. 'some html with alert("You smell") disallowed tags')
  330. def test_dictsort(self):
  331. sorted_dicts = dictsort([{'age': 23, 'name': 'Barbara-Ann'},
  332. {'age': 63, 'name': 'Ra Ra Rasputin'},
  333. {'name': 'Jonny B Goode', 'age': 18}], 'age')
  334. self.assertEqual([sorted(dict.items()) for dict in sorted_dicts],
  335. [[('age', 18), ('name', 'Jonny B Goode')],
  336. [('age', 23), ('name', 'Barbara-Ann')],
  337. [('age', 63), ('name', 'Ra Ra Rasputin')]])
  338. # If it gets passed a list of something else different from
  339. # dictionaries it should fail silently
  340. self.assertEqual(dictsort([1, 2, 3], 'age'), '')
  341. self.assertEqual(dictsort('Hello!', 'age'), '')
  342. self.assertEqual(dictsort({'a': 1}, 'age'), '')
  343. self.assertEqual(dictsort(1, 'age'), '')
  344. def test_dictsortreversed(self):
  345. sorted_dicts = dictsortreversed([{'age': 23, 'name': 'Barbara-Ann'},
  346. {'age': 63, 'name': 'Ra Ra Rasputin'},
  347. {'name': 'Jonny B Goode', 'age': 18}],
  348. 'age')
  349. self.assertEqual([sorted(dict.items()) for dict in sorted_dicts],
  350. [[('age', 63), ('name', 'Ra Ra Rasputin')],
  351. [('age', 23), ('name', 'Barbara-Ann')],
  352. [('age', 18), ('name', 'Jonny B Goode')]])
  353. # If it gets passed a list of something else different from
  354. # dictionaries it should fail silently
  355. self.assertEqual(dictsortreversed([1, 2, 3], 'age'), '')
  356. self.assertEqual(dictsortreversed('Hello!', 'age'), '')
  357. self.assertEqual(dictsortreversed({'a': 1}, 'age'), '')
  358. self.assertEqual(dictsortreversed(1, 'age'), '')
  359. def test_first(self):
  360. self.assertEqual(first([0,1,2]), 0)
  361. self.assertEqual(first(''), '')
  362. self.assertEqual(first('test'), 't')
  363. def test_join(self):
  364. self.assertEqual(join([0,1,2], 'glue'), '0glue1glue2')
  365. def test_length(self):
  366. self.assertEqual(length('1234'), 4)
  367. self.assertEqual(length([1,2,3,4]), 4)
  368. self.assertEqual(length_is([], 0), True)
  369. self.assertEqual(length_is([], 1), False)
  370. self.assertEqual(length_is('a', 1), True)
  371. self.assertEqual(length_is('a', 10), False)
  372. def test_slice(self):
  373. self.assertEqual(slice_filter('abcdefg', '0'), '')
  374. self.assertEqual(slice_filter('abcdefg', '1'), 'a')
  375. self.assertEqual(slice_filter('abcdefg', '-1'), 'abcdef')
  376. self.assertEqual(slice_filter('abcdefg', '1:2'), 'b')
  377. self.assertEqual(slice_filter('abcdefg', '1:3'), 'bc')
  378. self.assertEqual(slice_filter('abcdefg', '0::2'), 'aceg')
  379. def test_unordered_list(self):
  380. self.assertEqual(unordered_list(['item 1', 'item 2']),
  381. '\t<li>item 1</li>\n\t<li>item 2</li>')
  382. self.assertEqual(unordered_list(['item 1', ['item 1.1']]),
  383. '\t<li>item 1\n\t<ul>\n\t\t<li>item 1.1</li>\n\t</ul>\n\t</li>')
  384. self.assertEqual(
  385. unordered_list(['item 1', ['item 1.1', 'item1.2'], 'item 2']),
  386. '\t<li>item 1\n\t<ul>\n\t\t<li>item 1.1</li>\n\t\t<li>item1.2'\
  387. '</li>\n\t</ul>\n\t</li>\n\t<li>item 2</li>')
  388. self.assertEqual(
  389. unordered_list(['item 1', ['item 1.1', ['item 1.1.1',
  390. ['item 1.1.1.1']]]]),
  391. '\t<li>item 1\n\t<ul>\n\t\t<li>item 1.1\n\t\t<ul>\n\t\t\t<li>'\
  392. 'item 1.1.1\n\t\t\t<ul>\n\t\t\t\t<li>item 1.1.1.1</li>\n\t\t\t'\
  393. '</ul>\n\t\t\t</li>\n\t\t</ul>\n\t\t</li>\n\t</ul>\n\t</li>')
  394. self.assertEqual(unordered_list(
  395. ['States', ['Kansas', ['Lawrence', 'Topeka'], 'Illinois']]),
  396. '\t<li>States\n\t<ul>\n\t\t<li>Kansas\n\t\t<ul>\n\t\t\t<li>'\
  397. 'Lawrence</li>\n\t\t\t<li>Topeka</li>\n\t\t</ul>\n\t\t</li>'\
  398. '\n\t\t<li>Illinois</li>\n\t</ul>\n\t</li>')
  399. @python_2_unicode_compatible
  400. class ULItem(object):
  401. def __init__(self, title):
  402. self.title = title
  403. def __str__(self):
  404. return 'ulitem-%s' % str(self.title)
  405. a = ULItem('a')
  406. b = ULItem('b')
  407. self.assertEqual(unordered_list([a,b]),
  408. '\t<li>ulitem-a</li>\n\t<li>ulitem-b</li>')
  409. # Old format for unordered lists should still work
  410. self.assertEqual(unordered_list(['item 1', []]), '\t<li>item 1</li>')
  411. self.assertEqual(unordered_list(['item 1', [['item 1.1', []]]]),
  412. '\t<li>item 1\n\t<ul>\n\t\t<li>item 1.1</li>\n\t</ul>\n\t</li>')
  413. self.assertEqual(unordered_list(['item 1', [['item 1.1', []],
  414. ['item 1.2', []]]]), '\t<li>item 1\n\t<ul>\n\t\t<li>item 1.1'\
  415. '</li>\n\t\t<li>item 1.2</li>\n\t</ul>\n\t</li>')
  416. self.assertEqual(unordered_list(['States', [['Kansas', [['Lawrence',
  417. []], ['Topeka', []]]], ['Illinois', []]]]), '\t<li>States\n\t'\
  418. '<ul>\n\t\t<li>Kansas\n\t\t<ul>\n\t\t\t<li>Lawrence</li>'\
  419. '\n\t\t\t<li>Topeka</li>\n\t\t</ul>\n\t\t</li>\n\t\t<li>'\
  420. 'Illinois</li>\n\t</ul>\n\t</li>')
  421. def test_add(self):
  422. self.assertEqual(add('1', '2'), 3)
  423. def test_get_digit(self):
  424. self.assertEqual(get_digit(123, 1), 3)
  425. self.assertEqual(get_digit(123, 2), 2)
  426. self.assertEqual(get_digit(123, 3), 1)
  427. self.assertEqual(get_digit(123, 4), 0)
  428. self.assertEqual(get_digit(123, 0), 123)
  429. self.assertEqual(get_digit('xyz', 0), 'xyz')
  430. def test_date(self):
  431. # real testing of date() is in dateformat.py
  432. self.assertEqual(date(datetime.datetime(2005, 12, 29), "d F Y"),
  433. '29 December 2005')
  434. self.assertEqual(date(datetime.datetime(2005, 12, 29), r'jS \o\f F'),
  435. '29th of December')
  436. def test_time(self):
  437. # real testing of time() is done in dateformat.py
  438. self.assertEqual(time(datetime.time(13), "h"), '01')
  439. self.assertEqual(time(datetime.time(0), "h"), '12')
  440. def test_timesince(self):
  441. # real testing is done in timesince.py, where we can provide our own 'now'
  442. # NOTE: \xa0 avoids wrapping between value and unit
  443. self.assertEqual(
  444. timesince_filter(datetime.datetime.now() - datetime.timedelta(1)),
  445. '1\xa0day')
  446. self.assertEqual(
  447. timesince_filter(datetime.datetime(2005, 12, 29),
  448. datetime.datetime(2005, 12, 30)),
  449. '1\xa0day')
  450. def test_timeuntil(self):
  451. # NOTE: \xa0 avoids wrapping between value and unit
  452. self.assertEqual(
  453. timeuntil_filter(datetime.datetime.now() + datetime.timedelta(1, 1)),
  454. '1\xa0day')
  455. self.assertEqual(
  456. timeuntil_filter(datetime.datetime(2005, 12, 30),
  457. datetime.datetime(2005, 12, 29)),
  458. '1\xa0day')
  459. def test_default(self):
  460. self.assertEqual(default("val", "default"), 'val')
  461. self.assertEqual(default(None, "default"), 'default')
  462. self.assertEqual(default('', "default"), 'default')
  463. def test_if_none(self):
  464. self.assertEqual(default_if_none("val", "default"), 'val')
  465. self.assertEqual(default_if_none(None, "default"), 'default')
  466. self.assertEqual(default_if_none('', "default"), '')
  467. def test_divisibleby(self):
  468. self.assertEqual(divisibleby(4, 2), True)
  469. self.assertEqual(divisibleby(4, 3), False)
  470. def test_yesno(self):
  471. self.assertEqual(yesno(True), 'yes')
  472. self.assertEqual(yesno(False), 'no')
  473. self.assertEqual(yesno(None), 'maybe')
  474. self.assertEqual(yesno(True, 'certainly,get out of town,perhaps'),
  475. 'certainly')
  476. self.assertEqual(yesno(False, 'certainly,get out of town,perhaps'),
  477. 'get out of town')
  478. self.assertEqual(yesno(None, 'certainly,get out of town,perhaps'),
  479. 'perhaps')
  480. self.assertEqual(yesno(None, 'certainly,get out of town'),
  481. 'get out of town')
  482. def test_filesizeformat(self):
  483. # NOTE: \xa0 avoids wrapping between value and unit
  484. self.assertEqual(filesizeformat(1023), '1023\xa0bytes')
  485. self.assertEqual(filesizeformat(1024), '1.0\xa0KB')
  486. self.assertEqual(filesizeformat(10*1024), '10.0\xa0KB')
  487. self.assertEqual(filesizeformat(1024*1024-1), '1024.0\xa0KB')
  488. self.assertEqual(filesizeformat(1024*1024), '1.0\xa0MB')
  489. self.assertEqual(filesizeformat(1024*1024*50), '50.0\xa0MB')
  490. self.assertEqual(filesizeformat(1024*1024*1024-1), '1024.0\xa0MB')
  491. self.assertEqual(filesizeformat(1024*1024*1024), '1.0\xa0GB')
  492. self.assertEqual(filesizeformat(1024*1024*1024*1024), '1.0\xa0TB')
  493. self.assertEqual(filesizeformat(1024*1024*1024*1024*1024), '1.0\xa0PB')
  494. self.assertEqual(filesizeformat(1024*1024*1024*1024*1024*2000),
  495. '2000.0\xa0PB')
  496. self.assertEqual(filesizeformat(complex(1,-1)), '0\xa0bytes')
  497. self.assertEqual(filesizeformat(""), '0\xa0bytes')
  498. self.assertEqual(filesizeformat("\N{GREEK SMALL LETTER ALPHA}"),
  499. '0\xa0bytes')
  500. def test_pluralize(self):
  501. self.assertEqual(pluralize(1), '')
  502. self.assertEqual(pluralize(0), 's')
  503. self.assertEqual(pluralize(2), 's')
  504. self.assertEqual(pluralize([1]), '')
  505. self.assertEqual(pluralize([]), 's')
  506. self.assertEqual(pluralize([1,2,3]), 's')
  507. self.assertEqual(pluralize(1,'es'), '')
  508. self.assertEqual(pluralize(0,'es'), 'es')
  509. self.assertEqual(pluralize(2,'es'), 'es')
  510. self.assertEqual(pluralize(1,'y,ies'), 'y')
  511. self.assertEqual(pluralize(0,'y,ies'), 'ies')
  512. self.assertEqual(pluralize(2,'y,ies'), 'ies')
  513. self.assertEqual(pluralize(0,'y,ies,error'), '')
  514. def test_phone2numeric(self):
  515. self.assertEqual(phone2numeric_filter('0800 flowers'), '0800 3569377')
  516. def test_non_string_input(self):
  517. # Filters shouldn't break if passed non-strings
  518. self.assertEqual(addslashes(123), '123')
  519. self.assertEqual(linenumbers(123), '1. 123')
  520. self.assertEqual(lower(123), '123')
  521. self.assertEqual(make_list(123), ['1', '2', '3'])
  522. self.assertEqual(slugify(123), '123')
  523. self.assertEqual(title(123), '123')
  524. self.assertEqual(truncatewords(123, 2), '123')
  525. self.assertEqual(upper(123), '123')
  526. self.assertEqual(urlencode(123), '123')
  527. self.assertEqual(urlize(123), '123')
  528. self.assertEqual(urlizetrunc(123, 1), '123')
  529. self.assertEqual(wordcount(123), 1)
  530. self.assertEqual(wordwrap(123, 2), '123')
  531. self.assertEqual(ljust('123', 4), '123 ')
  532. self.assertEqual(rjust('123', 4), ' 123')
  533. self.assertEqual(center('123', 5), ' 123 ')
  534. self.assertEqual(center('123', 6), ' 123 ')
  535. self.assertEqual(cut(123, '2'), '13')
  536. self.assertEqual(escape(123), '123')
  537. self.assertEqual(linebreaks_filter(123), '<p>123</p>')
  538. self.assertEqual(linebreaksbr(123), '123')
  539. self.assertEqual(removetags(123, 'a'), '123')
  540. self.assertEqual(striptags(123), '123')
  541. class DefaultFiltersI18NTests(TransRealMixin, TestCase):
  542. def test_localized_filesizeformat(self):
  543. # NOTE: \xa0 avoids wrapping between value and unit
  544. with self.settings(USE_L10N=True):
  545. with translation.override('de', deactivate=True):
  546. self.assertEqual(filesizeformat(1023), '1023\xa0Bytes')
  547. self.assertEqual(filesizeformat(1024), '1,0\xa0KB')
  548. self.assertEqual(filesizeformat(10*1024), '10,0\xa0KB')
  549. self.assertEqual(filesizeformat(1024*1024-1), '1024,0\xa0KB')
  550. self.assertEqual(filesizeformat(1024*1024), '1,0\xa0MB')
  551. self.assertEqual(filesizeformat(1024*1024*50), '50,0\xa0MB')
  552. self.assertEqual(filesizeformat(1024*1024*1024-1), '1024,0\xa0MB')
  553. self.assertEqual(filesizeformat(1024*1024*1024), '1,0\xa0GB')
  554. self.assertEqual(filesizeformat(1024*1024*1024*1024), '1,0\xa0TB')
  555. self.assertEqual(filesizeformat(1024*1024*1024*1024*1024),
  556. '1,0\xa0PB')
  557. self.assertEqual(filesizeformat(1024*1024*1024*1024*1024*2000),
  558. '2000,0\xa0PB')
  559. self.assertEqual(filesizeformat(complex(1,-1)), '0\xa0Bytes')
  560. self.assertEqual(filesizeformat(""), '0\xa0Bytes')
  561. self.assertEqual(filesizeformat("\N{GREEK SMALL LETTER ALPHA}"),
  562. '0\xa0Bytes')