143 lines
4.9 KiB
Python
143 lines
4.9 KiB
Python
"""
|
|
Testing some internals of the template processing. These are *not* examples to be copied in user code.
|
|
"""
|
|
from __future__ import unicode_literals
|
|
|
|
from unittest import TestCase
|
|
|
|
from django.template import Library, Template, TemplateSyntaxError
|
|
from django.template.base import (
|
|
FilterExpression, Parser, TokenParser, Variable,
|
|
)
|
|
from django.test import override_settings
|
|
from django.utils import six
|
|
|
|
|
|
class ParserTests(TestCase):
|
|
def test_token_parsing(self):
|
|
# Tests for TokenParser behavior in the face of quoted strings with
|
|
# spaces.
|
|
|
|
p = TokenParser("tag thevar|filter sometag")
|
|
self.assertEqual(p.tagname, "tag")
|
|
self.assertEqual(p.value(), "thevar|filter")
|
|
self.assertTrue(p.more())
|
|
self.assertEqual(p.tag(), "sometag")
|
|
self.assertFalse(p.more())
|
|
|
|
p = TokenParser('tag "a value"|filter sometag')
|
|
self.assertEqual(p.tagname, "tag")
|
|
self.assertEqual(p.value(), '"a value"|filter')
|
|
self.assertTrue(p.more())
|
|
self.assertEqual(p.tag(), "sometag")
|
|
self.assertFalse(p.more())
|
|
|
|
p = TokenParser("tag 'a value'|filter sometag")
|
|
self.assertEqual(p.tagname, "tag")
|
|
self.assertEqual(p.value(), "'a value'|filter")
|
|
self.assertTrue(p.more())
|
|
self.assertEqual(p.tag(), "sometag")
|
|
self.assertFalse(p.more())
|
|
|
|
def test_filter_parsing(self):
|
|
c = {"article": {"section": "News"}}
|
|
p = Parser("")
|
|
|
|
def fe_test(s, val):
|
|
self.assertEqual(FilterExpression(s, p).resolve(c), val)
|
|
|
|
fe_test("article.section", "News")
|
|
fe_test("article.section|upper", "NEWS")
|
|
fe_test('"News"', "News")
|
|
fe_test("'News'", "News")
|
|
fe_test(r'"Some \"Good\" News"', 'Some "Good" News')
|
|
fe_test(r'"Some \"Good\" News"', 'Some "Good" News')
|
|
fe_test(r"'Some \'Bad\' News'", "Some 'Bad' News")
|
|
|
|
fe = FilterExpression(r'"Some \"Good\" News"', p)
|
|
self.assertEqual(fe.filters, [])
|
|
self.assertEqual(fe.var, 'Some "Good" News')
|
|
|
|
# Filtered variables should reject access of attributes beginning with
|
|
# underscores.
|
|
self.assertRaises(TemplateSyntaxError, FilterExpression, "article._hidden|upper", p)
|
|
|
|
def test_variable_parsing(self):
|
|
c = {"article": {"section": "News"}}
|
|
self.assertEqual(Variable("article.section").resolve(c), "News")
|
|
self.assertEqual(Variable('"News"').resolve(c), "News")
|
|
self.assertEqual(Variable("'News'").resolve(c), "News")
|
|
|
|
# Translated strings are handled correctly.
|
|
self.assertEqual(Variable("_(article.section)").resolve(c), "News")
|
|
self.assertEqual(Variable('_("Good News")').resolve(c), "Good News")
|
|
self.assertEqual(Variable("_('Better News')").resolve(c), "Better News")
|
|
|
|
# Escaped quotes work correctly as well.
|
|
self.assertEqual(
|
|
Variable(r'"Some \"Good\" News"').resolve(c), 'Some "Good" News'
|
|
)
|
|
self.assertEqual(
|
|
Variable(r"'Some \'Better\' News'").resolve(c), "Some 'Better' News"
|
|
)
|
|
|
|
# Variables should reject access of attributes beginning with
|
|
# underscores.
|
|
self.assertRaises(TemplateSyntaxError, Variable, "article._hidden")
|
|
|
|
# Variables should raise on non string type
|
|
with six.assertRaisesRegex(self, TypeError, "Variable must be a string or number, got <(class|type) 'dict'>"):
|
|
Variable({})
|
|
|
|
@override_settings(DEBUG=True)
|
|
def test_compile_filter_error(self):
|
|
# regression test for #19819
|
|
msg = "Could not parse the remainder: '@bar' from 'foo@bar'"
|
|
with six.assertRaisesRegex(self, TemplateSyntaxError, msg) as cm:
|
|
Template("{% if 1 %}{{ foo@bar }}{% endif %}")
|
|
self.assertEqual(cm.exception.django_template_source[1], (10, 23))
|
|
|
|
def test_filter_args_count(self):
|
|
p = Parser("")
|
|
l = Library()
|
|
|
|
@l.filter
|
|
def no_arguments(value):
|
|
pass
|
|
|
|
@l.filter
|
|
def one_argument(value, arg):
|
|
pass
|
|
|
|
@l.filter
|
|
def one_opt_argument(value, arg=False):
|
|
pass
|
|
|
|
@l.filter
|
|
def two_arguments(value, arg, arg2):
|
|
pass
|
|
|
|
@l.filter
|
|
def two_one_opt_arg(value, arg, arg2=False):
|
|
pass
|
|
p.add_library(l)
|
|
for expr in (
|
|
'1|no_arguments:"1"',
|
|
'1|two_arguments',
|
|
'1|two_arguments:"1"',
|
|
'1|two_one_opt_arg',
|
|
):
|
|
with self.assertRaises(TemplateSyntaxError):
|
|
FilterExpression(expr, p)
|
|
for expr in (
|
|
# Correct number of arguments
|
|
'1|no_arguments',
|
|
'1|one_argument:"1"',
|
|
# One optional
|
|
'1|one_opt_argument',
|
|
'1|one_opt_argument:"1"',
|
|
# Not supplying all
|
|
'1|two_one_opt_arg:"1"',
|
|
):
|
|
FilterExpression(expr, p)
|