My document

+ """) + self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) + + def test_otherwise_outside_choose(self): + """ + Verify that an `otherwise` directive outside of a `choose` directive is + reported as an error. + """ + tmpl = MarkupTemplate(""" +

+ """) + self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) + + def test_when_without_test(self): + """ + Verify that an `when` directive that doesn't have a `test` attribute + is reported as an error. + """ + tmpl = MarkupTemplate(""" +

+ foo +

+ """) + self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) + + def test_when_without_test_but_with_choose_value(self): + """ + Verify that an `when` directive that doesn't have a `test` attribute + works as expected as long as the parent `choose` directive has a test + expression. + """ + tmpl = MarkupTemplate(""" +

+ foo +

+ """) + self.assertEqual(""" + foo + """, str(tmpl.generate(foo='Yeah'))) + + def test_otherwise_without_test(self): + """ + Verify that an `otherwise` directive can be used without a `test` + attribute. + """ + tmpl = MarkupTemplate(""" +

+ foo +

+ """) + self.assertEqual(""" + foo + """, str(tmpl.generate())) + + def test_as_element(self): + """ + Verify that the directive can also be used as an element. + """ + tmpl = MarkupTemplate(""" + + 1 + 2 + 3 + + """) + self.assertEqual(""" + 1 + """, str(tmpl.generate())) + + def test_in_text_template(self): + """ + Verify that the directive works as expected in a text template. + """ + tmpl = TextTemplate("""#choose + #when 1 == 1 + 1 + #end + #when 2 == 2 + 2 + #end + #when 3 == 3 + 3 + #end + #end""") + self.assertEqual(""" 1\n""", str(tmpl.generate())) + + +class DefDirectiveTestCase(unittest.TestCase): + """Tests for the `py:def` template directive.""" + + def test_function_with_strip(self): + """ + Verify that a named template function with a strip directive actually + strips of the outer element. + """ + tmpl = MarkupTemplate(""" +

+ ${what} +

+ ${echo('foo')} + """) + self.assertEqual(""" + foo + """, str(tmpl.generate())) + + def test_exec_in_replace(self): + tmpl = MarkupTemplate("""

+ ${greeting}, ${name}! +

""") + self.assertEqual("""

+ hello, world! +

""", str(tmpl.generate())) + + def test_as_element(self): + """ + Verify that the directive can also be used as an element. + """ + tmpl = MarkupTemplate(""" + + ${what} + + ${echo('foo')} + """) + self.assertEqual(""" + foo + """, str(tmpl.generate())) + + def test_nested_defs(self): + """ + Verify that a template function defined inside a conditional block can + be called from outside that block. + """ + tmpl = MarkupTemplate(""" + + ${what} + + + ${what} + + ${echo('foo')} + """) + self.assertEqual(""" + foo + """, str(tmpl.generate(semantic=True))) + + def test_function_with_default_arg(self): + """ + Verify that keyword arguments work with `py:def` directives. + """ + tmpl = MarkupTemplate(""" + ${what} + ${echo('foo')} + """) + self.assertEqual(""" + foo + """, str(tmpl.generate())) + + def test_invocation_in_attribute(self): + tmpl = MarkupTemplate(""" + ${what or 'something'} +

bar

+ """) + self.assertEqual(""" +

bar

+ """, str(tmpl.generate())) + + def test_invocation_in_attribute_none(self): + tmpl = MarkupTemplate(""" + ${None} +

bar

+ """) + self.assertEqual(""" +

bar

+ """, str(tmpl.generate())) + + def test_function_raising_typeerror(self): + def badfunc(): + raise TypeError + tmpl = MarkupTemplate(""" +

+ ${badfunc()} +

+ """) + self.assertRaises(TypeError, list, tmpl.generate(badfunc=badfunc)) + + def test_def_in_matched(self): + tmpl = MarkupTemplate(""" + ${select('*')} + + + ${maketitle(True)} + + """) + self.assertEqual(""" + True + """, str(tmpl.generate())) + + def test_in_text_template(self): + """ + Verify that the directive works as expected in a text template. + """ + tmpl = TextTemplate(""" + #def echo(greeting, name='world') + ${greeting}, ${name}! + #end + ${echo('Hi', name='you')} + """) + self.assertEqual(""" Hi, you! + """, str(tmpl.generate())) + + +class ForDirectiveTestCase(unittest.TestCase): + """Tests for the `py:for` template directive.""" + + def test_loop_with_strip(self): + """ + Verify that the combining the `py:for` directive with `py:strip` works + correctly. + """ + tmpl = MarkupTemplate(""" +
+ ${item} +
+ """) + self.assertEqual(""" + 1 + 2 + 3 + 4 + 5 + """, str(tmpl.generate(items=range(1, 6)))) + + def test_as_element(self): + """ + Verify that the directive can also be used as an element. + """ + tmpl = MarkupTemplate(""" + + ${item} + + """) + self.assertEqual(""" + 1 + 2 + 3 + 4 + 5 + """, str(tmpl.generate(items=range(1, 6)))) + + def test_multi_assignment(self): + """ + Verify that assignment to tuples works correctly. + """ + tmpl = MarkupTemplate(""" + +
key=$k, value=$v
+ + """) + self.assertEqual(""" +
key=a, value=1
+
key=b, value=2
+ """, str(tmpl.generate(items=dict(a=1, b=2).items()))) + + def test_nested_assignment(self): + """ + Verify that assignment to nested tuples works correctly. + """ + tmpl = MarkupTemplate(""" + +
$idx: key=$k, value=$v
+ + """) + self.assertEqual(""" +
0: key=a, value=1
+
1: key=b, value=2
+ """, str(tmpl.generate(items=enumerate(dict(a=1, b=2).items())))) + + def test_not_iterable(self): + """ + Verify that assignment to nested tuples works correctly. + """ + tmpl = MarkupTemplate(""" + + $item + + """, filename='test.html') + try: + list(tmpl.generate(foo=12)) + except TemplateRuntimeError, e: + self.assertEqual('test.html', e.filename) + if sys.version_info[:2] >= (2, 4): + self.assertEqual(2, e.lineno) + + +class IfDirectiveTestCase(unittest.TestCase): + """Tests for the `py:if` template directive.""" + + def test_loop_with_strip(self): + """ + Verify that the combining the `py:if` directive with `py:strip` works + correctly. + """ + tmpl = MarkupTemplate(""" + ${bar} + """) + self.assertEqual(""" + Hello + """, str(tmpl.generate(foo=True, bar='Hello'))) + + def test_as_element(self): + """ + Verify that the directive can also be used as an element. + """ + tmpl = MarkupTemplate(""" + ${bar} + """) + self.assertEqual(""" + Hello + """, str(tmpl.generate(foo=True, bar='Hello'))) + + +class MatchDirectiveTestCase(unittest.TestCase): + """Tests for the `py:match` template directive.""" + + def test_with_strip(self): + """ + Verify that a match template can produce the same kind of element that + it matched without entering an infinite recursion. + """ + tmpl = MarkupTemplate(""" + +
${select('text()')}
+ + Hey Joe + """) + self.assertEqual(""" +
Hey Joe
+ """, str(tmpl.generate())) + + def test_without_strip(self): + """ + Verify that a match template can produce the same kind of element that + it matched without entering an infinite recursion. + """ + tmpl = MarkupTemplate(""" + +
${select('text()')}
+ + Hey Joe + """) + self.assertEqual(""" + +
Hey Joe
+ + """, str(tmpl.generate())) + + def test_as_element(self): + """ + Verify that the directive can also be used as an element. + """ + tmpl = MarkupTemplate(""" + +
${select('text()')}
+ + Hey Joe + """) + self.assertEqual(""" +
Hey Joe
+ """, str(tmpl.generate())) + + def test_recursive_match_1(self): + """ + Match directives are applied recursively, meaning that they are also + applied to any content they may have produced themselves: + """ + tmpl = MarkupTemplate(""" + +
+ ${select('*')} +
+ + + + + + + """) + self.assertEqual(""" + +
+ + +
+
+ + +
+ + """, str(tmpl.generate())) + + def test_recursive_match_2(self): + """ + When two or more match templates match the same element and also + themselves output the element they match, avoiding recursion is even + more complex, but should work. + """ + tmpl = MarkupTemplate(""" + +
+ ${select('*')} + + + ${select('*')} +
+ + +
Foo
+ + """) + self.assertEqual(""" + +
Foo
+
+ + """, str(tmpl.generate())) + + def test_select_all_attrs(self): + tmpl = MarkupTemplate(""" +
+ ${select('text()')} +
+ Hey Joe + """) + self.assertEqual(""" +
+ Hey Joe +
+ """, str(tmpl.generate())) + + def test_select_all_attrs_empty(self): + tmpl = MarkupTemplate(""" +
+ ${select('text()')} +
+ Hey Joe + """) + self.assertEqual(""" +
+ Hey Joe +
+ """, str(tmpl.generate())) + + def test_select_all_attrs_in_body(self): + tmpl = MarkupTemplate(""" +
+ Hey ${select('text()')} ${select('@*')} +
+ Joe + """) + self.assertEqual(""" +
+ Hey Joe Cool +
+ """, str(tmpl.generate())) + + def test_def_in_match(self): + tmpl = MarkupTemplate(""" + + ${select('*')} + ${maketitle(True)} + """) + self.assertEqual(""" + True + """, str(tmpl.generate())) + + def test_match_with_xpath_variable(self): + tmpl = MarkupTemplate("""
+ + Hello ${select('@name')} + + +
""") + self.assertEqual("""
+ + Hello Dude + +
""", str(tmpl.generate(tagname='greeting'))) + self.assertEqual("""
+ +
""", str(tmpl.generate(tagname='sayhello'))) + + def test_content_directive_in_match(self): + tmpl = MarkupTemplate(""" +
I said something.
+ bar + """) + self.assertEqual(""" +
I said bar.
+ """, str(tmpl.generate())) + + def test_cascaded_matches(self): + tmpl = MarkupTemplate(""" + ${select('*')} + ${select('title')} + ${select('*')}
+ Welcome to Markup +
Are you ready to mark up?
+ """) + self.assertEqual(""" + Welcome to Markup +
Are you ready to mark up?
+ """, str(tmpl.generate())) + + def test_multiple_matches(self): + tmpl = MarkupTemplate(""" + +
+ ${field.capitalize()} + +
+ """) + fields = ['hello_%s' % i for i in range(5)] + values = dict([('hello_%s' % i, i) for i in range(5)]) + self.assertEqual(""" +
+ Hello_0 + +
+ Hello_1 + +
+ Hello_2 + +
+ Hello_3 + +
+ Hello_4 + +
+ """, str(tmpl.generate(fields=fields, values=values))) + + def test_namespace_context(self): + tmpl = MarkupTemplate(""" +
Foo
+ + """) + # FIXME: there should be a way to strip out unwanted/unused namespaces, + # such as the "x" in this example + self.assertEqual(""" +
Foo
+ """, str(tmpl.generate())) + + def test_match_with_position_predicate(self): + tmpl = MarkupTemplate(""" +
${select('*|text()')}
+ +
Foo
+
Bar
+ + """) + self.assertEqual(""" + +
Foo
+
Bar
+ + """, str(tmpl.generate())) + + def test_match_with_closure(self): + tmpl = MarkupTemplate(""" +
${select('*|text()')}
+ +
Foo
+
Bar
+ + """) + self.assertEqual(""" + +
Foo
+
Bar
+ + """, str(tmpl.generate())) + + def test_match_without_closure(self): + tmpl = MarkupTemplate(""" +
${select('*|text()')}
+ +
Foo
+
Bar
+ + """) + self.assertEqual(""" + +
Foo
+
Bar
+ + """, str(tmpl.generate())) + + # FIXME + #def test_match_after_step(self): + # tmpl = MarkupTemplate("""
+ # + # Hello ${select('@name')} + # + # + #
""") + # self.assertEqual("""
+ # + # Hello Dude + # + #
""", str(tmpl.generate())) + + +class StripDirectiveTestCase(unittest.TestCase): + """Tests for the `py:strip` template directive.""" + + def test_strip_false(self): + tmpl = MarkupTemplate("""
+
foo
+
""") + self.assertEqual("""
+
foo
+
""", str(tmpl.generate())) + + def test_strip_empty(self): + tmpl = MarkupTemplate("""
+
foo
+
""") + self.assertEqual("""
+ foo +
""", str(tmpl.generate())) + + +class WithDirectiveTestCase(unittest.TestCase): + """Tests for the `py:with` template directive.""" + + def test_shadowing(self): + tmpl = MarkupTemplate("""
+ ${x} + + ${x} +
""") + self.assertEqual("""
+ 42 + 84 + 42 +
""", str(tmpl.generate(x=42))) + + def test_as_element(self): + tmpl = MarkupTemplate("""
+ ${x} +
""") + self.assertEqual("""
+ 84 +
""", str(tmpl.generate(x=42))) + + def test_multiple_vars_same_name(self): + tmpl = MarkupTemplate("""
+ + $foo + +
""") + self.assertEqual("""
+ baz +
""", str(tmpl.generate(x=42))) + + def test_multiple_vars_single_assignment(self): + tmpl = MarkupTemplate("""
+ ${x} ${y} ${z} +
""") + self.assertEqual("""
+ 1 1 1 +
""", str(tmpl.generate(x=42))) + + def test_nested_vars_single_assignment(self): + tmpl = MarkupTemplate("""
+ ${x} ${y} ${z} +
""") + self.assertEqual("""
+ 1 2 3 +
""", str(tmpl.generate(x=42))) + + def test_multiple_vars_trailing_semicolon(self): + tmpl = MarkupTemplate("""
+ ${x} ${y} +
""") + self.assertEqual("""
+ 84 42 +
""", str(tmpl.generate(x=42))) + + def test_semicolon_escape(self): + tmpl = MarkupTemplate("""
+ + ${x} + ${y} + +
""") + self.assertEqual("""
+ here is a semicolon: ; + here are two semicolons: ;; +
""", str(tmpl.generate())) + + def test_unicode_expr(self): + tmpl = MarkupTemplate("""
+ + $weeks + +
""") + self.assertEqual("""
+ + 一二三四五六日 + +
""", str(tmpl.generate())) + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(doctest.DocTestSuite(directives)) + suite.addTest(unittest.makeSuite(AttrsDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(ChooseDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(DefDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(ForDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(IfDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(MatchDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(StripDirectiveTestCase, 'test')) + suite.addTest(unittest.makeSuite(WithDirectiveTestCase, 'test')) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/genshi/template/tests/eval.py b/genshi/template/tests/eval.py new file mode 100644 --- /dev/null +++ b/genshi/template/tests/eval.py @@ -0,0 +1,389 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2006 Edgewall Software +# All rights reserved. +# +# This software is licensed as described in the file COPYING, which +# you should have received as part of this distribution. The terms +# are also available at http://genshi.edgewall.org/wiki/License. +# +# This software consists of voluntary contributions made by many +# individuals. For the exact contribution history, see the revision +# history and logs, available at http://genshi.edgewall.org/log/. + +import doctest +import sys +import unittest + +from genshi.template.eval import Expression, Undefined + + +class ExpressionTestCase(unittest.TestCase): + + def test_name_lookup(self): + self.assertEqual('bar', Expression('foo').evaluate({'foo': 'bar'})) + self.assertEqual(id, Expression('id').evaluate({}, nocall=True)) + self.assertEqual('bar', Expression('id').evaluate({'id': 'bar'})) + self.assertEqual(None, Expression('id').evaluate({'id': None}, + nocall=True)) + + def test_str_literal(self): + self.assertEqual('foo', Expression('"foo"').evaluate({})) + self.assertEqual('foo', Expression('"""foo"""').evaluate({})) + self.assertEqual('foo', Expression("'foo'").evaluate({})) + self.assertEqual('foo', Expression("'''foo'''").evaluate({})) + self.assertEqual('foo', Expression("u'foo'").evaluate({})) + self.assertEqual('foo', Expression("r'foo'").evaluate({})) + + def test_str_literal_non_ascii(self): + expr = Expression(u"u'\xfe'") + self.assertEqual(u'þ', expr.evaluate({})) + expr = Expression("u'\xfe'") + self.assertEqual(u'þ', expr.evaluate({})) + expr = Expression("'\xc3\xbe'") + self.assertEqual(u'þ', expr.evaluate({})) + + def test_num_literal(self): + self.assertEqual(42, Expression("42").evaluate({})) + self.assertEqual(42L, Expression("42L").evaluate({})) + self.assertEqual(.42, Expression(".42").evaluate({})) + self.assertEqual(07, Expression("07").evaluate({})) + self.assertEqual(0xF2, Expression("0xF2").evaluate({})) + self.assertEqual(0XF2, Expression("0XF2").evaluate({})) + + def test_dict_literal(self): + self.assertEqual({}, Expression("{}").evaluate({})) + self.assertEqual({'key': True}, + Expression("{'key': value}").evaluate({'value': True})) + + def test_list_literal(self): + self.assertEqual([], Expression("[]").evaluate({})) + self.assertEqual([1, 2, 3], Expression("[1, 2, 3]").evaluate({})) + self.assertEqual([True], + Expression("[value]").evaluate({'value': True})) + + def test_tuple_literal(self): + self.assertEqual((), Expression("()").evaluate({})) + self.assertEqual((1, 2, 3), Expression("(1, 2, 3)").evaluate({})) + self.assertEqual((True,), + Expression("(value,)").evaluate({'value': True})) + + def test_unaryop_pos(self): + self.assertEqual(1, Expression("+1").evaluate({})) + self.assertEqual(1, Expression("+x").evaluate({'x': 1})) + + def test_unaryop_neg(self): + self.assertEqual(-1, Expression("-1").evaluate({})) + self.assertEqual(-1, Expression("-x").evaluate({'x': 1})) + + def test_unaryop_not(self): + self.assertEqual(False, Expression("not True").evaluate({})) + self.assertEqual(False, Expression("not x").evaluate({'x': True})) + + def test_unaryop_inv(self): + self.assertEqual(-2, Expression("~1").evaluate({})) + self.assertEqual(-2, Expression("~x").evaluate({'x': 1})) + + def test_binop_add(self): + self.assertEqual(3, Expression("2 + 1").evaluate({})) + self.assertEqual(3, Expression("x + y").evaluate({'x': 2, 'y': 1})) + + def test_binop_sub(self): + self.assertEqual(1, Expression("2 - 1").evaluate({})) + self.assertEqual(1, Expression("x - y").evaluate({'x': 1, 'y': 1})) + + def test_binop_sub(self): + self.assertEqual(1, Expression("2 - 1").evaluate({})) + self.assertEqual(1, Expression("x - y").evaluate({'x': 2, 'y': 1})) + + def test_binop_mul(self): + self.assertEqual(4, Expression("2 * 2").evaluate({})) + self.assertEqual(4, Expression("x * y").evaluate({'x': 2, 'y': 2})) + + def test_binop_pow(self): + self.assertEqual(4, Expression("2 ** 2").evaluate({})) + self.assertEqual(4, Expression("x ** y").evaluate({'x': 2, 'y': 2})) + + def test_binop_div(self): + self.assertEqual(2, Expression("4 / 2").evaluate({})) + self.assertEqual(2, Expression("x / y").evaluate({'x': 4, 'y': 2})) + + def test_binop_floordiv(self): + self.assertEqual(1, Expression("3 // 2").evaluate({})) + self.assertEqual(1, Expression("x // y").evaluate({'x': 3, 'y': 2})) + + def test_binop_mod(self): + self.assertEqual(1, Expression("3 % 2").evaluate({})) + self.assertEqual(1, Expression("x % y").evaluate({'x': 3, 'y': 2})) + + def test_binop_and(self): + self.assertEqual(0, Expression("1 & 0").evaluate({})) + self.assertEqual(0, Expression("x & y").evaluate({'x': 1, 'y': 0})) + + def test_binop_or(self): + self.assertEqual(1, Expression("1 | 0").evaluate({})) + self.assertEqual(1, Expression("x | y").evaluate({'x': 1, 'y': 0})) + + def test_binop_contains(self): + self.assertEqual(True, Expression("1 in (1, 2, 3)").evaluate({})) + self.assertEqual(True, Expression("x in y").evaluate({'x': 1, + 'y': (1, 2, 3)})) + + def test_binop_not_contains(self): + self.assertEqual(True, Expression("4 not in (1, 2, 3)").evaluate({})) + self.assertEqual(True, Expression("x not in y").evaluate({'x': 4, + 'y': (1, 2, 3)})) + + def test_binop_is(self): + self.assertEqual(True, Expression("1 is 1").evaluate({})) + self.assertEqual(True, Expression("x is y").evaluate({'x': 1, 'y': 1})) + self.assertEqual(False, Expression("1 is 2").evaluate({})) + self.assertEqual(False, Expression("x is y").evaluate({'x': 1, 'y': 2})) + + def test_binop_is_not(self): + self.assertEqual(True, Expression("1 is not 2").evaluate({})) + self.assertEqual(True, Expression("x is not y").evaluate({'x': 1, + 'y': 2})) + self.assertEqual(False, Expression("1 is not 1").evaluate({})) + self.assertEqual(False, Expression("x is not y").evaluate({'x': 1, + 'y': 1})) + + def test_boolop_and(self): + self.assertEqual(False, Expression("True and False").evaluate({})) + self.assertEqual(False, Expression("x and y").evaluate({'x': True, + 'y': False})) + + def test_boolop_or(self): + self.assertEqual(True, Expression("True or False").evaluate({})) + self.assertEqual(True, Expression("x or y").evaluate({'x': True, + 'y': False})) + + def test_compare_eq(self): + self.assertEqual(True, Expression("1 == 1").evaluate({})) + self.assertEqual(True, Expression("x == y").evaluate({'x': 1, 'y': 1})) + + def test_compare_ne(self): + self.assertEqual(False, Expression("1 != 1").evaluate({})) + self.assertEqual(False, Expression("x != y").evaluate({'x': 1, 'y': 1})) + self.assertEqual(False, Expression("1 <> 1").evaluate({})) + self.assertEqual(False, Expression("x <> y").evaluate({'x': 1, 'y': 1})) + + def test_compare_lt(self): + self.assertEqual(True, Expression("1 < 2").evaluate({})) + self.assertEqual(True, Expression("x < y").evaluate({'x': 1, 'y': 2})) + + def test_compare_le(self): + self.assertEqual(True, Expression("1 <= 1").evaluate({})) + self.assertEqual(True, Expression("x <= y").evaluate({'x': 1, 'y': 1})) + + def test_compare_gt(self): + self.assertEqual(True, Expression("2 > 1").evaluate({})) + self.assertEqual(True, Expression("x > y").evaluate({'x': 2, 'y': 1})) + + def test_compare_ge(self): + self.assertEqual(True, Expression("1 >= 1").evaluate({})) + self.assertEqual(True, Expression("x >= y").evaluate({'x': 1, 'y': 1})) + + def test_compare_multi(self): + self.assertEqual(True, Expression("1 != 3 == 3").evaluate({})) + self.assertEqual(True, Expression("x != y == y").evaluate({'x': 1, + 'y': 3})) + + def test_call_function(self): + self.assertEqual(42, Expression("foo()").evaluate({'foo': lambda: 42})) + data = {'foo': 'bar'} + self.assertEqual('BAR', Expression("foo.upper()").evaluate(data)) + data = {'foo': {'bar': range(42)}} + self.assertEqual(42, Expression("len(foo.bar)").evaluate(data)) + + def test_call_keywords(self): + self.assertEqual(42, Expression("foo(x=bar)").evaluate({'foo': lambda x: x, + 'bar': 42})) + + def test_call_star_args(self): + self.assertEqual(42, Expression("foo(*bar)").evaluate({'foo': lambda x: x, + 'bar': [42]})) + + def test_call_dstar_args(self): + def foo(x): + return x + self.assertEqual(42, Expression("foo(**bar)").evaluate({'foo': foo, + 'bar': {"x": 42}})) + + def test_call_function_without_params(self): + self.assertEqual(42, Expression("foo").evaluate({'foo': lambda: 42})) + data = {'foo': 'bar'} + self.assertEqual('BAR', Expression("foo.upper").evaluate(data)) + data = {'foo': {'bar': range(42)}} + + def test_lambda(self): + # Define a custom `sorted` function cause the builtin isn't available + # on Python 2.3 + def sorted(items, compfunc): + items.sort(compfunc) + return items + data = {'items': [{'name': 'b', 'value': 0}, {'name': 'a', 'value': 1}], + 'sorted': sorted} + expr = Expression("sorted(items, lambda a, b: cmp(a.name, b.name))") + self.assertEqual([{'name': 'a', 'value': 1}, {'name': 'b', 'value': 0}], + expr.evaluate(data)) + + def test_list_comprehension(self): + expr = Expression("[n for n in numbers if n < 2]") + self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) + + expr = Expression("[(i, n + 1) for i, n in enumerate(numbers)]") + self.assertEqual([(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)], + expr.evaluate({'numbers': range(5)})) + + expr = Expression("[offset + n for n in numbers]") + self.assertEqual([2, 3, 4, 5, 6], + expr.evaluate({'numbers': range(5), 'offset': 2})) + + def test_list_comprehension_with_getattr(self): + items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] + expr = Expression("[i.name for i in items if i.value > 1]") + self.assertEqual(['b'], expr.evaluate({'items': items})) + + def test_list_comprehension_with_getitem(self): + items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] + expr = Expression("[i['name'] for i in items if i['value'] > 1]") + self.assertEqual(['b'], expr.evaluate({'items': items})) + + if sys.version_info >= (2, 4): + # Generator expressions only supported in Python 2.4 and up + + def test_generator_expression(self): + expr = Expression("list(n for n in numbers if n < 2)") + self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) + + expr = Expression("list((i, n + 1) for i, n in enumerate(numbers))") + self.assertEqual([(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)], + expr.evaluate({'numbers': range(5)})) + + expr = Expression("list(offset + n for n in numbers)") + self.assertEqual([2, 3, 4, 5, 6], + expr.evaluate({'numbers': range(5), 'offset': 2})) + + def test_generator_expression_with_getattr(self): + items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] + expr = Expression("list(i.name for i in items if i.value > 1)") + self.assertEqual(['b'], expr.evaluate({'items': items})) + + def test_generator_expression_with_getitem(self): + items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] + expr = Expression("list(i['name'] for i in items if i['value'] > 1)") + self.assertEqual(['b'], expr.evaluate({'items': items})) + + def test_slice(self): + expr = Expression("numbers[0:2]") + self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) + + def test_slice_with_vars(self): + expr = Expression("numbers[start:end]") + self.assertEqual([0, 1], expr.evaluate({'numbers': range(5), 'start': 0, + 'end': 2})) + + def test_slice_copy(self): + expr = Expression("numbers[:]") + self.assertEqual([0, 1, 2, 3, 4], expr.evaluate({'numbers': range(5)})) + + def test_slice_stride(self): + expr = Expression("numbers[::stride]") + self.assertEqual([0, 2, 4], expr.evaluate({'numbers': range(5), + 'stride': 2})) + + def test_slice_negative_start(self): + expr = Expression("numbers[-1:]") + self.assertEqual([4], expr.evaluate({'numbers': range(5)})) + + def test_slice_negative_end(self): + expr = Expression("numbers[:-1]") + self.assertEqual([0, 1, 2, 3], expr.evaluate({'numbers': range(5)})) + + def test_error_access_undefined(self): + expr = Expression("nothing", filename='index.html', lineno=50) + self.assertEqual(Undefined, type(expr.evaluate({}))) + + def test_error_call_undefined(self): + expr = Expression("nothing()", filename='index.html', lineno=50) + try: + expr.evaluate({}) + self.fail('Expected NameError') + except NameError, e: + exc_type, exc_value, exc_traceback = sys.exc_info() + frame = exc_traceback.tb_next + frames = [] + while frame.tb_next: + frame = frame.tb_next + frames.append(frame) + self.assertEqual('Variable "nothing" is not defined', str(e)) + self.assertEqual('', + frames[-3].tb_frame.f_code.co_name) + self.assertEqual('index.html', + frames[-3].tb_frame.f_code.co_filename) + self.assertEqual(50, frames[-3].tb_lineno) + + def test_error_getattr_undefined(self): + expr = Expression("nothing.nil", filename='index.html', lineno=50) + try: + expr.evaluate({}) + self.fail('Expected NameError') + except NameError, e: + exc_type, exc_value, exc_traceback = sys.exc_info() + frame = exc_traceback.tb_next + frames = [] + while frame.tb_next: + frame = frame.tb_next + frames.append(frame) + self.assertEqual('Variable "nothing" is not defined', str(e)) + self.assertEqual('', + frames[-3].tb_frame.f_code.co_name) + self.assertEqual('index.html', + frames[-3].tb_frame.f_code.co_filename) + self.assertEqual(50, frames[-3].tb_lineno) + + def test_error_getitem_undefined(self): + expr = Expression("nothing[0]", filename='index.html', lineno=50) + try: + expr.evaluate({}) + self.fail('Expected NameError') + except NameError, e: + exc_type, exc_value, exc_traceback = sys.exc_info() + frame = exc_traceback.tb_next + frames = [] + while frame.tb_next: + frame = frame.tb_next + frames.append(frame) + self.assertEqual('Variable "nothing" is not defined', str(e)) + self.assertEqual('', + frames[-3].tb_frame.f_code.co_name) + self.assertEqual('index.html', + frames[-3].tb_frame.f_code.co_filename) + self.assertEqual(50, frames[-3].tb_lineno) + + def test_error_getattr_nested_undefined(self): + expr = Expression("nothing.nil", filename='index.html', lineno=50) + val = expr.evaluate({'nothing': object()}) + assert isinstance(val, Undefined) + self.assertEqual("nil", val._name) + + def test_error_getitem_nested_undefined_string(self): + expr = Expression("nothing['bla']", filename='index.html', lineno=50) + val = expr.evaluate({'nothing': object()}) + assert isinstance(val, Undefined) + self.assertEqual("bla", val._name) + + def test_error_getitem_nested_undefined_int(self): + expr = Expression("nothing[0]", filename='index.html', lineno=50) + self.assertRaises(TypeError, expr.evaluate, {'nothing': object()}) + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(doctest.DocTestSuite(Expression.__module__)) + suite.addTest(unittest.makeSuite(ExpressionTestCase, 'test')) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/genshi/template/tests/loader.py b/genshi/template/tests/loader.py new file mode 100644 --- /dev/null +++ b/genshi/template/tests/loader.py @@ -0,0 +1,200 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2006 Edgewall Software +# All rights reserved. +# +# This software is licensed as described in the file COPYING, which +# you should have received as part of this distribution. The terms +# are also available at http://genshi.edgewall.org/wiki/License. +# +# This software consists of voluntary contributions made by many +# individuals. For the exact contribution history, see the revision +# history and logs, available at http://genshi.edgewall.org/log/. + +import doctest +import os +import shutil +import tempfile +import unittest + +from genshi.template.loader import TemplateLoader +from genshi.template.markup import MarkupTemplate + + +class TemplateLoaderTestCase(unittest.TestCase): + """Tests for the template loader.""" + + def setUp(self): + self.dirname = tempfile.mkdtemp(suffix='markup_test') + + def tearDown(self): + shutil.rmtree(self.dirname) + + def test_search_path_empty(self): + loader = TemplateLoader() + self.assertEqual([], loader.search_path) + + def test_search_path_as_string(self): + loader = TemplateLoader(self.dirname) + self.assertEqual([self.dirname], loader.search_path) + + def test_relative_include_samedir(self): + file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') + try: + file2.write(""" + + """) + finally: + file2.close() + + loader = TemplateLoader([self.dirname]) + tmpl = loader.load('tmpl2.html') + self.assertEqual(""" +
Included
+ """, tmpl.generate().render()) + + def test_relative_include_subdir(self): + os.mkdir(os.path.join(self.dirname, 'sub')) + file1 = open(os.path.join(self.dirname, 'sub', 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') + try: + file2.write(""" + + """) + finally: + file2.close() + + loader = TemplateLoader([self.dirname]) + tmpl = loader.load('tmpl2.html') + self.assertEqual(""" +
Included
+ """, tmpl.generate().render()) + + def test_relative_include_parentdir(self): + file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + os.mkdir(os.path.join(self.dirname, 'sub')) + file2 = open(os.path.join(self.dirname, 'sub', 'tmpl2.html'), 'w') + try: + file2.write(""" + + """) + finally: + file2.close() + + loader = TemplateLoader([self.dirname]) + tmpl = loader.load('sub/tmpl2.html') + self.assertEqual(""" +
Included
+ """, tmpl.generate().render()) + + def test_relative_include_without_search_path(self): + file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') + try: + file2.write(""" + + """) + finally: + file2.close() + + loader = TemplateLoader() + tmpl = loader.load(os.path.join(self.dirname, 'tmpl2.html')) + self.assertEqual(""" +
Included
+ """, tmpl.generate().render()) + + def test_relative_include_without_search_path_nested(self): + file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') + try: + file2.write("""
+ +
""") + finally: + file2.close() + + file3 = open(os.path.join(self.dirname, 'tmpl3.html'), 'w') + try: + file3.write(""" + + """) + finally: + file3.close() + + loader = TemplateLoader() + tmpl = loader.load(os.path.join(self.dirname, 'tmpl3.html')) + self.assertEqual(""" +
+
Included
+
+ """, tmpl.generate().render()) + + def test_relative_include_from_inmemory_template(self): + file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') + try: + file1.write("""
Included
""") + finally: + file1.close() + + loader = TemplateLoader([self.dirname]) + tmpl2 = MarkupTemplate(""" + + """, filename='subdir/tmpl2.html', loader=loader) + + self.assertEqual(""" +
Included
+ """, tmpl2.generate().render()) + + def test_load_with_default_encoding(self): + f = open(os.path.join(self.dirname, 'tmpl.html'), 'w') + try: + f.write(u'
\xf6
'.encode('iso-8859-1')) + finally: + f.close() + loader = TemplateLoader([self.dirname], default_encoding='iso-8859-1') + loader.load('tmpl.html') + + def test_load_with_explicit_encoding(self): + f = open(os.path.join(self.dirname, 'tmpl.html'), 'w') + try: + f.write(u'
\xf6
'.encode('iso-8859-1')) + finally: + f.close() + loader = TemplateLoader([self.dirname], default_encoding='utf-8') + loader.load('tmpl.html', encoding='iso-8859-1') + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(doctest.DocTestSuite(TemplateLoader.__module__)) + suite.addTest(unittest.makeSuite(TemplateLoaderTestCase, 'test')) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/genshi/template/tests/markup.py b/genshi/template/tests/markup.py new file mode 100644 --- /dev/null +++ b/genshi/template/tests/markup.py @@ -0,0 +1,190 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2006 Edgewall Software +# All rights reserved. +# +# This software is licensed as described in the file COPYING, which +# you should have received as part of this distribution. The terms +# are also available at http://genshi.edgewall.org/wiki/License. +# +# This software consists of voluntary contributions made by many +# individuals. For the exact contribution history, see the revision +# history and logs, available at http://genshi.edgewall.org/log/. + +import doctest +import sys +import unittest + +from genshi.core import Markup +from genshi.template.core import BadDirectiveError, TemplateSyntaxError +from genshi.template.markup import MarkupTemplate + + +class MarkupTemplateTestCase(unittest.TestCase): + """Tests for markup template processing.""" + + def test_interpolate_mixed3(self): + tmpl = MarkupTemplate(' ${var} $var') + self.assertEqual(' 42 42', str(tmpl.generate(var=42))) + + def test_interpolate_leading_trailing_space(self): + tmpl = MarkupTemplate('${ foo }') + self.assertEqual('bar', str(tmpl.generate(foo='bar'))) + + def test_interpolate_multiline(self): + tmpl = MarkupTemplate("""${dict( + bar = 'baz' + )[foo]}""") + self.assertEqual('baz', str(tmpl.generate(foo='bar'))) + + def test_interpolate_non_string_attrs(self): + tmpl = MarkupTemplate('') + self.assertEqual('', str(tmpl.generate())) + + def test_interpolate_list_result(self): + tmpl = MarkupTemplate('$foo') + self.assertEqual('buzz', str(tmpl.generate(foo=('buzz',)))) + + def test_empty_attr(self): + tmpl = MarkupTemplate('') + self.assertEqual('', str(tmpl.generate())) + + def test_bad_directive_error(self): + xml = '
' + try: + tmpl = MarkupTemplate(xml, filename='test.html') + except BadDirectiveError, e: + self.assertEqual('test.html', e.filename) + if sys.version_info[:2] >= (2, 4): + self.assertEqual(1, e.lineno) + + def test_directive_value_syntax_error(self): + xml = """
""" + try: + tmpl = MarkupTemplate(xml, filename='test.html') + self.fail('Expected SyntaxError') + except TemplateSyntaxError, e: + self.assertEqual('test.html', e.filename) + if sys.version_info[:2] >= (2, 4): + self.assertEqual(1, e.lineno) + + def test_expression_syntax_error(self): + xml = """
+ Foo ${bar"} +
""" + try: + tmpl = MarkupTemplate(xml, filename='test.html') + self.fail('Expected SyntaxError') + except TemplateSyntaxError, e: + self.assertEqual('test.html', e.filename) + if sys.version_info[:2] >= (2, 4): + self.assertEqual(2, e.lineno) + + def test_expression_syntax_error_multi_line(self): + xml = """
+ + ${bar"} + +
""" + try: + tmpl = MarkupTemplate(xml, filename='test.html') + self.fail('Expected SyntaxError') + except TemplateSyntaxError, e: + self.assertEqual('test.html', e.filename) + if sys.version_info[:2] >= (2, 4): + self.assertEqual(3, e.lineno) + + def test_markup_noescape(self): + """ + Verify that outputting context data that is a `Markup` instance is not + escaped. + """ + tmpl = MarkupTemplate("""
+ $myvar +
""") + self.assertEqual("""
+ foo +
""", str(tmpl.generate(myvar=Markup('foo')))) + + def test_text_noescape_quotes(self): + """ + Verify that outputting context data in text nodes doesn't escape quotes. + """ + tmpl = MarkupTemplate("""
+ $myvar +
""") + self.assertEqual("""
+ "foo" +
""", str(tmpl.generate(myvar='"foo"'))) + + def test_attr_escape_quotes(self): + """ + Verify that outputting context data in attribtes escapes quotes. + """ + tmpl = MarkupTemplate("""
+ +
""") + self.assertEqual("""
+ +
""", str(tmpl.generate(myvar='"foo"'))) + + def test_directive_element(self): + tmpl = MarkupTemplate("""
+ bar +
""") + self.assertEqual("""
+ bar +
""", str(tmpl.generate(myvar='"foo"'))) + + def test_normal_comment(self): + tmpl = MarkupTemplate("""
+ +
""") + self.assertEqual("""
+ +
""", str(tmpl.generate())) + + def test_template_comment(self): + tmpl = MarkupTemplate("""
+ + +
""") + self.assertEqual("""
+
""", str(tmpl.generate())) + + def test_parse_with_same_namespace_nested(self): + tmpl = MarkupTemplate("""
+ + +
""") + self.assertEqual("""
+ + +
""", str(tmpl.generate())) + + def test_latin1_encoded_with_xmldecl(self): + tmpl = MarkupTemplate(u""" +
+ \xf6 +
""".encode('iso-8859-1'), encoding='iso-8859-1') + self.assertEqual(u"""
+ \xf6 +
""", unicode(tmpl.generate())) + + def test_latin1_encoded_explicit_encoding(self): + tmpl = MarkupTemplate(u"""
+ \xf6 +
""".encode('iso-8859-1'), encoding='iso-8859-1') + self.assertEqual(u"""
+ \xf6 +
""", unicode(tmpl.generate())) + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(doctest.DocTestSuite(MarkupTemplate.__module__)) + suite.addTest(unittest.makeSuite(MarkupTemplateTestCase, 'test')) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/genshi/template/tests/text.py b/genshi/template/tests/text.py new file mode 100644 --- /dev/null +++ b/genshi/template/tests/text.py @@ -0,0 +1,69 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2006 Edgewall Software +# All rights reserved. +# +# This software is licensed as described in the file COPYING, which +# you should have received as part of this distribution. The terms +# are also available at http://genshi.edgewall.org/wiki/License. +# +# This software consists of voluntary contributions made by many +# individuals. For the exact contribution history, see the revision +# history and logs, available at http://genshi.edgewall.org/log/. + +import doctest +import unittest + +from genshi.template.text import TextTemplate + + +class TextTemplateTestCase(unittest.TestCase): + """Tests for text template processing.""" + + def test_escaping(self): + tmpl = TextTemplate('\\#escaped') + self.assertEqual('#escaped', str(tmpl.generate())) + + def test_comment(self): + tmpl = TextTemplate('## a comment') + self.assertEqual('', str(tmpl.generate())) + + def test_comment_escaping(self): + tmpl = TextTemplate('\\## escaped comment') + self.assertEqual('## escaped comment', str(tmpl.generate())) + + def test_end_with_args(self): + tmpl = TextTemplate(""" + #if foo + bar + #end 'if foo'""") + self.assertEqual('', str(tmpl.generate())) + + def test_latin1_encoded(self): + text = u'$foo\xf6$bar'.encode('iso-8859-1') + tmpl = TextTemplate(text, encoding='iso-8859-1') + self.assertEqual(u'x\xf6y', unicode(tmpl.generate(foo='x', bar='y'))) + + # FIXME + #def test_empty_lines(self): + # tmpl = TextTemplate("""Your items: + # + # #for item in items + # * ${item} + # + # #end""") + # self.assertEqual("""Your items: + # * 0 + # * 1 + # * 2 + # """, tmpl.generate(items=range(3)).render('text')) + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(doctest.DocTestSuite(TextTemplate.__module__)) + suite.addTest(unittest.makeSuite(TextTemplateTestCase, 'test')) + return suite + +if __name__ == '__main__': + unittest.main(defaultTest='suite') diff --git a/genshi/template/text.py b/genshi/template/text.py new file mode 100644 --- /dev/null +++ b/genshi/template/text.py @@ -0,0 +1,107 @@ +# -*- coding: utf-8 -*- +# +# Copyright (C) 2006 Edgewall Software +# All rights reserved. +# +# This software is licensed as described in the file COPYING, which +# you should have received as part of this distribution. The terms +# are also available at http://genshi.edgewall.org/wiki/License. +# +# This software consists of voluntary contributions made by many +# individuals. For the exact contribution history, see the revision +# history and logs, available at http://genshi.edgewall.org/log/. + +"""Plain text templating engine.""" + +import re + +from genshi.template.core import BadDirectiveError, Template, SUB +from genshi.template.directives import * + + +class TextTemplate(Template): + """Implementation of a simple text-based template engine. + + >>> tmpl = TextTemplate('''Dear $name, + ... + ... We have the following items for you: + ... #for item in items + ... * $item + ... #end + ... + ... All the best, + ... Foobar''') + >>> print tmpl.generate(name='Joe', items=[1, 2, 3]).render('text') + Dear Joe, + + We have the following items for you: + * 1 + * 2 + * 3 + + All the best, + Foobar + """ + directives = [('def', DefDirective), + ('when', WhenDirective), + ('otherwise', OtherwiseDirective), + ('for', ForDirective), + ('if', IfDirective), + ('choose', ChooseDirective), + ('with', WithDirective)] + + _DIRECTIVE_RE = re.compile(r'^\s*(? offset: + text = source[offset:start] + for kind, data, pos in self._interpolate(text, self.basedir, + self.filename, lineno): + stream.append((kind, data, pos)) + lineno += len(text.splitlines()) + + text = source[start:end].lstrip()[1:] + lineno += len(text.splitlines()) + directive = text.split(None, 1) + if len(directive) > 1: + command, value = directive + else: + command, value = directive[0], None + + if command == 'end': + depth -= 1 + if depth in dirmap: + directive, start_offset = dirmap.pop(depth) + substream = stream[start_offset:] + stream[start_offset:] = [(SUB, ([directive], substream), + (self.filepath, lineno, 0))] + elif command != '#': + cls = self._dir_by_name.get(command) + if cls is None: + raise BadDirectiveError(command) + directive = cls(value, None, self.filepath, lineno, 0) + dirmap[depth] = (directive, len(stream)) + depth += 1 + + offset = end + + if offset < len(source): + text = source[offset:].replace('\\#', '#') + for kind, data, pos in self._interpolate(text, self.basedir, + self.filename, lineno): + stream.append((kind, data, pos)) + + return stream diff --git a/genshi/tests/__init__.py b/genshi/tests/__init__.py --- a/genshi/tests/__init__.py +++ b/genshi/tests/__init__.py @@ -11,18 +11,17 @@ # individuals. For the exact contribution history, see the revision # history and logs, available at http://genshi.edgewall.org/log/. -import doctest import unittest def suite(): import genshi - from genshi.tests import builder, core, eval, filters, input, output, \ - path, template, util + from genshi.tests import builder, core, filters, input, output, path, \ + util + from genshi.template import tests as template + suite = unittest.TestSuite() - suite.addTest(doctest.DocTestSuite(genshi)) suite.addTest(builder.suite()) suite.addTest(core.suite()) - suite.addTest(eval.suite()) suite.addTest(filters.suite()) suite.addTest(input.suite()) suite.addTest(output.suite()) diff --git a/genshi/tests/eval.py b/genshi/tests/eval.py deleted file mode 100644 --- a/genshi/tests/eval.py +++ /dev/null @@ -1,389 +0,0 @@ -# -*- coding: utf-8 -*- -# -# Copyright (C) 2006 Edgewall Software -# All rights reserved. -# -# This software is licensed as described in the file COPYING, which -# you should have received as part of this distribution. The terms -# are also available at http://genshi.edgewall.org/wiki/License. -# -# This software consists of voluntary contributions made by many -# individuals. For the exact contribution history, see the revision -# history and logs, available at http://genshi.edgewall.org/log/. - -import doctest -import sys -import unittest - -from genshi.eval import Expression, Undefined - - -class ExpressionTestCase(unittest.TestCase): - - def test_name_lookup(self): - self.assertEqual('bar', Expression('foo').evaluate({'foo': 'bar'})) - self.assertEqual(id, Expression('id').evaluate({}, nocall=True)) - self.assertEqual('bar', Expression('id').evaluate({'id': 'bar'})) - self.assertEqual(None, Expression('id').evaluate({'id': None}, - nocall=True)) - - def test_str_literal(self): - self.assertEqual('foo', Expression('"foo"').evaluate({})) - self.assertEqual('foo', Expression('"""foo"""').evaluate({})) - self.assertEqual('foo', Expression("'foo'").evaluate({})) - self.assertEqual('foo', Expression("'''foo'''").evaluate({})) - self.assertEqual('foo', Expression("u'foo'").evaluate({})) - self.assertEqual('foo', Expression("r'foo'").evaluate({})) - - def test_str_literal_non_ascii(self): - expr = Expression(u"u'\xfe'") - self.assertEqual(u'þ', expr.evaluate({})) - expr = Expression("u'\xfe'") - self.assertEqual(u'þ', expr.evaluate({})) - expr = Expression("'\xc3\xbe'") - self.assertEqual(u'þ', expr.evaluate({})) - - def test_num_literal(self): - self.assertEqual(42, Expression("42").evaluate({})) - self.assertEqual(42L, Expression("42L").evaluate({})) - self.assertEqual(.42, Expression(".42").evaluate({})) - self.assertEqual(07, Expression("07").evaluate({})) - self.assertEqual(0xF2, Expression("0xF2").evaluate({})) - self.assertEqual(0XF2, Expression("0XF2").evaluate({})) - - def test_dict_literal(self): - self.assertEqual({}, Expression("{}").evaluate({})) - self.assertEqual({'key': True}, - Expression("{'key': value}").evaluate({'value': True})) - - def test_list_literal(self): - self.assertEqual([], Expression("[]").evaluate({})) - self.assertEqual([1, 2, 3], Expression("[1, 2, 3]").evaluate({})) - self.assertEqual([True], - Expression("[value]").evaluate({'value': True})) - - def test_tuple_literal(self): - self.assertEqual((), Expression("()").evaluate({})) - self.assertEqual((1, 2, 3), Expression("(1, 2, 3)").evaluate({})) - self.assertEqual((True,), - Expression("(value,)").evaluate({'value': True})) - - def test_unaryop_pos(self): - self.assertEqual(1, Expression("+1").evaluate({})) - self.assertEqual(1, Expression("+x").evaluate({'x': 1})) - - def test_unaryop_neg(self): - self.assertEqual(-1, Expression("-1").evaluate({})) - self.assertEqual(-1, Expression("-x").evaluate({'x': 1})) - - def test_unaryop_not(self): - self.assertEqual(False, Expression("not True").evaluate({})) - self.assertEqual(False, Expression("not x").evaluate({'x': True})) - - def test_unaryop_inv(self): - self.assertEqual(-2, Expression("~1").evaluate({})) - self.assertEqual(-2, Expression("~x").evaluate({'x': 1})) - - def test_binop_add(self): - self.assertEqual(3, Expression("2 + 1").evaluate({})) - self.assertEqual(3, Expression("x + y").evaluate({'x': 2, 'y': 1})) - - def test_binop_sub(self): - self.assertEqual(1, Expression("2 - 1").evaluate({})) - self.assertEqual(1, Expression("x - y").evaluate({'x': 1, 'y': 1})) - - def test_binop_sub(self): - self.assertEqual(1, Expression("2 - 1").evaluate({})) - self.assertEqual(1, Expression("x - y").evaluate({'x': 2, 'y': 1})) - - def test_binop_mul(self): - self.assertEqual(4, Expression("2 * 2").evaluate({})) - self.assertEqual(4, Expression("x * y").evaluate({'x': 2, 'y': 2})) - - def test_binop_pow(self): - self.assertEqual(4, Expression("2 ** 2").evaluate({})) - self.assertEqual(4, Expression("x ** y").evaluate({'x': 2, 'y': 2})) - - def test_binop_div(self): - self.assertEqual(2, Expression("4 / 2").evaluate({})) - self.assertEqual(2, Expression("x / y").evaluate({'x': 4, 'y': 2})) - - def test_binop_floordiv(self): - self.assertEqual(1, Expression("3 // 2").evaluate({})) - self.assertEqual(1, Expression("x // y").evaluate({'x': 3, 'y': 2})) - - def test_binop_mod(self): - self.assertEqual(1, Expression("3 % 2").evaluate({})) - self.assertEqual(1, Expression("x % y").evaluate({'x': 3, 'y': 2})) - - def test_binop_and(self): - self.assertEqual(0, Expression("1 & 0").evaluate({})) - self.assertEqual(0, Expression("x & y").evaluate({'x': 1, 'y': 0})) - - def test_binop_or(self): - self.assertEqual(1, Expression("1 | 0").evaluate({})) - self.assertEqual(1, Expression("x | y").evaluate({'x': 1, 'y': 0})) - - def test_binop_contains(self): - self.assertEqual(True, Expression("1 in (1, 2, 3)").evaluate({})) - self.assertEqual(True, Expression("x in y").evaluate({'x': 1, - 'y': (1, 2, 3)})) - - def test_binop_not_contains(self): - self.assertEqual(True, Expression("4 not in (1, 2, 3)").evaluate({})) - self.assertEqual(True, Expression("x not in y").evaluate({'x': 4, - 'y': (1, 2, 3)})) - - def test_binop_is(self): - self.assertEqual(True, Expression("1 is 1").evaluate({})) - self.assertEqual(True, Expression("x is y").evaluate({'x': 1, 'y': 1})) - self.assertEqual(False, Expression("1 is 2").evaluate({})) - self.assertEqual(False, Expression("x is y").evaluate({'x': 1, 'y': 2})) - - def test_binop_is_not(self): - self.assertEqual(True, Expression("1 is not 2").evaluate({})) - self.assertEqual(True, Expression("x is not y").evaluate({'x': 1, - 'y': 2})) - self.assertEqual(False, Expression("1 is not 1").evaluate({})) - self.assertEqual(False, Expression("x is not y").evaluate({'x': 1, - 'y': 1})) - - def test_boolop_and(self): - self.assertEqual(False, Expression("True and False").evaluate({})) - self.assertEqual(False, Expression("x and y").evaluate({'x': True, - 'y': False})) - - def test_boolop_or(self): - self.assertEqual(True, Expression("True or False").evaluate({})) - self.assertEqual(True, Expression("x or y").evaluate({'x': True, - 'y': False})) - - def test_compare_eq(self): - self.assertEqual(True, Expression("1 == 1").evaluate({})) - self.assertEqual(True, Expression("x == y").evaluate({'x': 1, 'y': 1})) - - def test_compare_ne(self): - self.assertEqual(False, Expression("1 != 1").evaluate({})) - self.assertEqual(False, Expression("x != y").evaluate({'x': 1, 'y': 1})) - self.assertEqual(False, Expression("1 <> 1").evaluate({})) - self.assertEqual(False, Expression("x <> y").evaluate({'x': 1, 'y': 1})) - - def test_compare_lt(self): - self.assertEqual(True, Expression("1 < 2").evaluate({})) - self.assertEqual(True, Expression("x < y").evaluate({'x': 1, 'y': 2})) - - def test_compare_le(self): - self.assertEqual(True, Expression("1 <= 1").evaluate({})) - self.assertEqual(True, Expression("x <= y").evaluate({'x': 1, 'y': 1})) - - def test_compare_gt(self): - self.assertEqual(True, Expression("2 > 1").evaluate({})) - self.assertEqual(True, Expression("x > y").evaluate({'x': 2, 'y': 1})) - - def test_compare_ge(self): - self.assertEqual(True, Expression("1 >= 1").evaluate({})) - self.assertEqual(True, Expression("x >= y").evaluate({'x': 1, 'y': 1})) - - def test_compare_multi(self): - self.assertEqual(True, Expression("1 != 3 == 3").evaluate({})) - self.assertEqual(True, Expression("x != y == y").evaluate({'x': 1, - 'y': 3})) - - def test_call_function(self): - self.assertEqual(42, Expression("foo()").evaluate({'foo': lambda: 42})) - data = {'foo': 'bar'} - self.assertEqual('BAR', Expression("foo.upper()").evaluate(data)) - data = {'foo': {'bar': range(42)}} - self.assertEqual(42, Expression("len(foo.bar)").evaluate(data)) - - def test_call_keywords(self): - self.assertEqual(42, Expression("foo(x=bar)").evaluate({'foo': lambda x: x, - 'bar': 42})) - - def test_call_star_args(self): - self.assertEqual(42, Expression("foo(*bar)").evaluate({'foo': lambda x: x, - 'bar': [42]})) - - def test_call_dstar_args(self): - def foo(x): - return x - self.assertEqual(42, Expression("foo(**bar)").evaluate({'foo': foo, - 'bar': {"x": 42}})) - - def test_call_function_without_params(self): - self.assertEqual(42, Expression("foo").evaluate({'foo': lambda: 42})) - data = {'foo': 'bar'} - self.assertEqual('BAR', Expression("foo.upper").evaluate(data)) - data = {'foo': {'bar': range(42)}} - - def test_lambda(self): - # Define a custom `sorted` function cause the builtin isn't available - # on Python 2.3 - def sorted(items, compfunc): - items.sort(compfunc) - return items - data = {'items': [{'name': 'b', 'value': 0}, {'name': 'a', 'value': 1}], - 'sorted': sorted} - expr = Expression("sorted(items, lambda a, b: cmp(a.name, b.name))") - self.assertEqual([{'name': 'a', 'value': 1}, {'name': 'b', 'value': 0}], - expr.evaluate(data)) - - def test_list_comprehension(self): - expr = Expression("[n for n in numbers if n < 2]") - self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) - - expr = Expression("[(i, n + 1) for i, n in enumerate(numbers)]") - self.assertEqual([(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)], - expr.evaluate({'numbers': range(5)})) - - expr = Expression("[offset + n for n in numbers]") - self.assertEqual([2, 3, 4, 5, 6], - expr.evaluate({'numbers': range(5), 'offset': 2})) - - def test_list_comprehension_with_getattr(self): - items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] - expr = Expression("[i.name for i in items if i.value > 1]") - self.assertEqual(['b'], expr.evaluate({'items': items})) - - def test_list_comprehension_with_getitem(self): - items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] - expr = Expression("[i['name'] for i in items if i['value'] > 1]") - self.assertEqual(['b'], expr.evaluate({'items': items})) - - if sys.version_info >= (2, 4): - # Generator expressions only supported in Python 2.4 and up - - def test_generator_expression(self): - expr = Expression("list(n for n in numbers if n < 2)") - self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) - - expr = Expression("list((i, n + 1) for i, n in enumerate(numbers))") - self.assertEqual([(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)], - expr.evaluate({'numbers': range(5)})) - - expr = Expression("list(offset + n for n in numbers)") - self.assertEqual([2, 3, 4, 5, 6], - expr.evaluate({'numbers': range(5), 'offset': 2})) - - def test_generator_expression_with_getattr(self): - items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] - expr = Expression("list(i.name for i in items if i.value > 1)") - self.assertEqual(['b'], expr.evaluate({'items': items})) - - def test_generator_expression_with_getitem(self): - items = [{'name': 'a', 'value': 1}, {'name': 'b', 'value': 2}] - expr = Expression("list(i['name'] for i in items if i['value'] > 1)") - self.assertEqual(['b'], expr.evaluate({'items': items})) - - def test_slice(self): - expr = Expression("numbers[0:2]") - self.assertEqual([0, 1], expr.evaluate({'numbers': range(5)})) - - def test_slice_with_vars(self): - expr = Expression("numbers[start:end]") - self.assertEqual([0, 1], expr.evaluate({'numbers': range(5), 'start': 0, - 'end': 2})) - - def test_slice_copy(self): - expr = Expression("numbers[:]") - self.assertEqual([0, 1, 2, 3, 4], expr.evaluate({'numbers': range(5)})) - - def test_slice_stride(self): - expr = Expression("numbers[::stride]") - self.assertEqual([0, 2, 4], expr.evaluate({'numbers': range(5), - 'stride': 2})) - - def test_slice_negative_start(self): - expr = Expression("numbers[-1:]") - self.assertEqual([4], expr.evaluate({'numbers': range(5)})) - - def test_slice_negative_end(self): - expr = Expression("numbers[:-1]") - self.assertEqual([0, 1, 2, 3], expr.evaluate({'numbers': range(5)})) - - def test_error_access_undefined(self): - expr = Expression("nothing", filename='index.html', lineno=50) - self.assertEqual(Undefined, type(expr.evaluate({}))) - - def test_error_call_undefined(self): - expr = Expression("nothing()", filename='index.html', lineno=50) - try: - expr.evaluate({}) - self.fail('Expected NameError') - except NameError, e: - exc_type, exc_value, exc_traceback = sys.exc_info() - frame = exc_traceback.tb_next - frames = [] - while frame.tb_next: - frame = frame.tb_next - frames.append(frame) - self.assertEqual('Variable "nothing" is not defined', str(e)) - self.assertEqual('', - frames[-3].tb_frame.f_code.co_name) - self.assertEqual('index.html', - frames[-3].tb_frame.f_code.co_filename) - self.assertEqual(50, frames[-3].tb_lineno) - - def test_error_getattr_undefined(self): - expr = Expression("nothing.nil", filename='index.html', lineno=50) - try: - expr.evaluate({}) - self.fail('Expected NameError') - except NameError, e: - exc_type, exc_value, exc_traceback = sys.exc_info() - frame = exc_traceback.tb_next - frames = [] - while frame.tb_next: - frame = frame.tb_next - frames.append(frame) - self.assertEqual('Variable "nothing" is not defined', str(e)) - self.assertEqual('', - frames[-3].tb_frame.f_code.co_name) - self.assertEqual('index.html', - frames[-3].tb_frame.f_code.co_filename) - self.assertEqual(50, frames[-3].tb_lineno) - - def test_error_getitem_undefined(self): - expr = Expression("nothing[0]", filename='index.html', lineno=50) - try: - expr.evaluate({}) - self.fail('Expected NameError') - except NameError, e: - exc_type, exc_value, exc_traceback = sys.exc_info() - frame = exc_traceback.tb_next - frames = [] - while frame.tb_next: - frame = frame.tb_next - frames.append(frame) - self.assertEqual('Variable "nothing" is not defined', str(e)) - self.assertEqual('', - frames[-3].tb_frame.f_code.co_name) - self.assertEqual('index.html', - frames[-3].tb_frame.f_code.co_filename) - self.assertEqual(50, frames[-3].tb_lineno) - - def test_error_getattr_nested_undefined(self): - expr = Expression("nothing.nil", filename='index.html', lineno=50) - val = expr.evaluate({'nothing': object()}) - assert isinstance(val, Undefined) - self.assertEqual("nil", val._name) - - def test_error_getitem_nested_undefined_string(self): - expr = Expression("nothing['bla']", filename='index.html', lineno=50) - val = expr.evaluate({'nothing': object()}) - assert isinstance(val, Undefined) - self.assertEqual("bla", val._name) - - def test_error_getitem_nested_undefined_int(self): - expr = Expression("nothing[0]", filename='index.html', lineno=50) - self.assertRaises(TypeError, expr.evaluate, {'nothing': object()}) - - -def suite(): - suite = unittest.TestSuite() - suite.addTest(unittest.makeSuite(ExpressionTestCase, 'test')) - suite.addTest(doctest.DocTestSuite(Expression.__module__)) - return suite - -if __name__ == '__main__': - unittest.main(defaultTest='suite') diff --git a/genshi/tests/template.py b/genshi/tests/template.py deleted file mode 100644 --- a/genshi/tests/template.py +++ /dev/null @@ -1,1377 +0,0 @@ -# -*- coding: utf-8 -*- -# -# Copyright (C) 2006 Edgewall Software -# All rights reserved. -# -# This software is licensed as described in the file COPYING, which -# you should have received as part of this distribution. The terms -# are also available at http://genshi.edgewall.org/wiki/License. -# -# This software consists of voluntary contributions made by many -# individuals. For the exact contribution history, see the revision -# history and logs, available at http://genshi.edgewall.org/log/. - -import doctest -import os -import unittest -import shutil -import sys -import tempfile - -from genshi import template -from genshi.core import Markup, Stream -from genshi.template import BadDirectiveError, MarkupTemplate, Template, \ - TemplateLoader, TemplateRuntimeError, \ - TemplateSyntaxError, TextTemplate - - -class AttrsDirectiveTestCase(unittest.TestCase): - """Tests for the `py:attrs` template directive.""" - - def test_combined_with_loop(self): - """ - Verify that the directive has access to the loop variables. - """ - tmpl = MarkupTemplate(""" - - """) - items = [{'id': 1, 'class': 'foo'}, {'id': 2, 'class': 'bar'}] - self.assertEqual(""" - - """, str(tmpl.generate(items=items))) - - def test_update_existing_attr(self): - """ - Verify that an attribute value that evaluates to `None` removes an - existing attribute of that name. - """ - tmpl = MarkupTemplate(""" - - """) - self.assertEqual(""" - - """, str(tmpl.generate())) - - def test_remove_existing_attr(self): - """ - Verify that an attribute value that evaluates to `None` removes an - existing attribute of that name. - """ - tmpl = MarkupTemplate(""" - - """) - self.assertEqual(""" - - """, str(tmpl.generate())) - - -class ChooseDirectiveTestCase(unittest.TestCase): - """Tests for the `py:choose` template directive and the complementary - directives `py:when` and `py:otherwise`.""" - - def test_multiple_true_whens(self): - """ - Verify that, if multiple `py:when` bodies match, only the first is - output. - """ - tmpl = MarkupTemplate("""
- 1 - 2 - 3 -
""") - self.assertEqual("""
- 1 -
""", str(tmpl.generate())) - - def test_otherwise(self): - tmpl = MarkupTemplate("""
- hidden - hello -
""") - self.assertEqual("""
- hello -
""", str(tmpl.generate())) - - def test_nesting(self): - """ - Verify that `py:choose` blocks can be nested: - """ - tmpl = MarkupTemplate(""" -
-
- 2 - 3 -
-
- """) - self.assertEqual(""" -
-
- 3 -
-
- """, str(tmpl.generate())) - - def test_complex_nesting(self): - """ - Verify more complex nesting. - """ - tmpl = MarkupTemplate(""" -
-
- OK - FAIL -
-
- """) - self.assertEqual(""" -
-
- OK -
-
- """, str(tmpl.generate())) - - def test_complex_nesting_otherwise(self): - """ - Verify more complex nesting using otherwise. - """ - tmpl = MarkupTemplate(""" -
-
- FAIL - OK -
-
- """) - self.assertEqual(""" -
-
- OK -
-
- """, str(tmpl.generate())) - - def test_when_with_strip(self): - """ - Verify that a when directive with a strip directive actually strips of - the outer element. - """ - tmpl = MarkupTemplate(""" -
- foo -
- """) - self.assertEqual(""" - foo - """, str(tmpl.generate())) - - def test_when_outside_choose(self): - """ - Verify that a `when` directive outside of a `choose` directive is - reported as an error. - """ - tmpl = MarkupTemplate(""" -
- """) - self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) - - def test_otherwise_outside_choose(self): - """ - Verify that an `otherwise` directive outside of a `choose` directive is - reported as an error. - """ - tmpl = MarkupTemplate(""" -
- """) - self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) - - def test_when_without_test(self): - """ - Verify that an `when` directive that doesn't have a `test` attribute - is reported as an error. - """ - tmpl = MarkupTemplate(""" -
- foo -
- """) - self.assertRaises(TemplateRuntimeError, str, tmpl.generate()) - - def test_when_without_test_but_with_choose_value(self): - """ - Verify that an `when` directive that doesn't have a `test` attribute - works as expected as long as the parent `choose` directive has a test - expression. - """ - tmpl = MarkupTemplate(""" -
- foo -
- """) - self.assertEqual(""" - foo - """, str(tmpl.generate(foo='Yeah'))) - - def test_otherwise_without_test(self): - """ - Verify that an `otherwise` directive can be used without a `test` - attribute. - """ - tmpl = MarkupTemplate(""" -
- foo -
- """) - self.assertEqual(""" - foo - """, str(tmpl.generate())) - - def test_as_element(self): - """ - Verify that the directive can also be used as an element. - """ - tmpl = MarkupTemplate(""" - - 1 - 2 - 3 - - """) - self.assertEqual(""" - 1 - """, str(tmpl.generate())) - - def test_in_text_template(self): - """ - Verify that the directive works as expected in a text template. - """ - tmpl = TextTemplate("""#choose - #when 1 == 1 - 1 - #end - #when 2 == 2 - 2 - #end - #when 3 == 3 - 3 - #end - #end""") - self.assertEqual(""" 1\n""", str(tmpl.generate())) - - -class DefDirectiveTestCase(unittest.TestCase): - """Tests for the `py:def` template directive.""" - - def test_function_with_strip(self): - """ - Verify that a named template function with a strip directive actually - strips of the outer element. - """ - tmpl = MarkupTemplate(""" -
- ${what} -
- ${echo('foo')} - """) - self.assertEqual(""" - foo - """, str(tmpl.generate())) - - def test_exec_in_replace(self): - tmpl = MarkupTemplate("""
-
- ${greeting}, ${name}! -
-
-
""") - self.assertEqual("""
-
- hello, world! -
-
""", str(tmpl.generate())) - - def test_as_element(self): - """ - Verify that the directive can also be used as an element. - """ - tmpl = MarkupTemplate(""" - - ${what} - - ${echo('foo')} - """) - self.assertEqual(""" - foo - """, str(tmpl.generate())) - - def test_nested_defs(self): - """ - Verify that a template function defined inside a conditional block can - be called from outside that block. - """ - tmpl = MarkupTemplate(""" - - ${what} - - - ${what} - - ${echo('foo')} - """) - self.assertEqual(""" - foo - """, str(tmpl.generate(semantic=True))) - - def test_function_with_default_arg(self): - """ - Verify that keyword arguments work with `py:def` directives. - """ - tmpl = MarkupTemplate(""" - ${what} - ${echo('foo')} - """) - self.assertEqual(""" - foo - """, str(tmpl.generate())) - - def test_invocation_in_attribute(self): - tmpl = MarkupTemplate(""" - ${what or 'something'} -
bar
- """) - self.assertEqual(""" -
bar
- """, str(tmpl.generate())) - - def test_invocation_in_attribute_none(self): - tmpl = MarkupTemplate(""" - ${None} -
bar
- """) - self.assertEqual(""" -
bar
- """, str(tmpl.generate())) - - def test_function_raising_typeerror(self): - def badfunc(): - raise TypeError - tmpl = MarkupTemplate(""" -
- ${badfunc()} -
-
- """) - self.assertRaises(TypeError, list, tmpl.generate(badfunc=badfunc)) - - def test_def_in_matched(self): - tmpl = MarkupTemplate(""" - ${select('*')} - - - ${maketitle(True)} - - """) - self.assertEqual(""" - True - """, str(tmpl.generate())) - - def test_in_text_template(self): - """ - Verify that the directive works as expected in a text template. - """ - tmpl = TextTemplate(""" - #def echo(greeting, name='world') - ${greeting}, ${name}! - #end - ${echo('Hi', name='you')} - """) - self.assertEqual(""" Hi, you! - """, str(tmpl.generate())) - - -class ForDirectiveTestCase(unittest.TestCase): - """Tests for the `py:for` template directive.""" - - def test_loop_with_strip(self): - """ - Verify that the combining the `py:for` directive with `py:strip` works - correctly. - """ - tmpl = MarkupTemplate(""" -
- ${item} -
- """) - self.assertEqual(""" - 1 - 2 - 3 - 4 - 5 - """, str(tmpl.generate(items=range(1, 6)))) - - def test_as_element(self): - """ - Verify that the directive can also be used as an element. - """ - tmpl = MarkupTemplate(""" - - ${item} - - """) - self.assertEqual(""" - 1 - 2 - 3 - 4 - 5 - """, str(tmpl.generate(items=range(1, 6)))) - - def test_multi_assignment(self): - """ - Verify that assignment to tuples works correctly. - """ - tmpl = MarkupTemplate(""" - -
key=$k, value=$v
- - """) - self.assertEqual(""" -
key=a, value=1
-
key=b, value=2
- """, str(tmpl.generate(items=dict(a=1, b=2).items()))) - - def test_nested_assignment(self): - """ - Verify that assignment to nested tuples works correctly. - """ - tmpl = MarkupTemplate(""" - -
$idx: key=$k, value=$v
- - """) - self.assertEqual(""" -
0: key=a, value=1
-
1: key=b, value=2
- """, str(tmpl.generate(items=enumerate(dict(a=1, b=2).items())))) - - def test_not_iterable(self): - """ - Verify that assignment to nested tuples works correctly. - """ - tmpl = MarkupTemplate(""" - - $item - - """, filename='test.html') - try: - list(tmpl.generate(foo=12)) - except TemplateRuntimeError, e: - self.assertEqual('test.html', e.filename) - if sys.version_info[:2] >= (2, 4): - self.assertEqual(2, e.lineno) - - -class IfDirectiveTestCase(unittest.TestCase): - """Tests for the `py:if` template directive.""" - - def test_loop_with_strip(self): - """ - Verify that the combining the `py:if` directive with `py:strip` works - correctly. - """ - tmpl = MarkupTemplate(""" - ${bar} - """) - self.assertEqual(""" - Hello - """, str(tmpl.generate(foo=True, bar='Hello'))) - - def test_as_element(self): - """ - Verify that the directive can also be used as an element. - """ - tmpl = MarkupTemplate(""" - ${bar} - """) - self.assertEqual(""" - Hello - """, str(tmpl.generate(foo=True, bar='Hello'))) - - -class MatchDirectiveTestCase(unittest.TestCase): - """Tests for the `py:match` template directive.""" - - def test_with_strip(self): - """ - Verify that a match template can produce the same kind of element that - it matched without entering an infinite recursion. - """ - tmpl = MarkupTemplate(""" - -
${select('text()')}
- - Hey Joe - """) - self.assertEqual(""" -
Hey Joe
- """, str(tmpl.generate())) - - def test_without_strip(self): - """ - Verify that a match template can produce the same kind of element that - it matched without entering an infinite recursion. - """ - tmpl = MarkupTemplate(""" - -
${select('text()')}
- - Hey Joe - """) - self.assertEqual(""" - -
Hey Joe
- - """, str(tmpl.generate())) - - def test_as_element(self): - """ - Verify that the directive can also be used as an element. - """ - tmpl = MarkupTemplate(""" - -
${select('text()')}
- - Hey Joe - """) - self.assertEqual(""" -
Hey Joe
- """, str(tmpl.generate())) - - def test_recursive_match_1(self): - """ - Match directives are applied recursively, meaning that they are also - applied to any content they may have produced themselves: - """ - tmpl = MarkupTemplate(""" - -
- ${select('*')} -
- - - - - - - """) - self.assertEqual(""" - -
- - -
-
- - -
- - """, str(tmpl.generate())) - - def test_recursive_match_2(self): - """ - When two or more match templates match the same element and also - themselves output the element they match, avoiding recursion is even - more complex, but should work. - """ - tmpl = MarkupTemplate(""" - -
- ${select('*')} - - - ${select('*')} -
- - -
Foo
- - """) - self.assertEqual(""" - -
Foo
-
- - """, str(tmpl.generate())) - - def test_select_all_attrs(self): - tmpl = MarkupTemplate(""" -
- ${select('text()')} -
- Hey Joe - """) - self.assertEqual(""" -
- Hey Joe -
- """, str(tmpl.generate())) - - def test_select_all_attrs_empty(self): - tmpl = MarkupTemplate(""" -
- ${select('text()')} -
- Hey Joe - """) - self.assertEqual(""" -
- Hey Joe -
- """, str(tmpl.generate())) - - def test_select_all_attrs_in_body(self): - tmpl = MarkupTemplate(""" -
- Hey ${select('text()')} ${select('@*')} -
- Joe - """) - self.assertEqual(""" -
- Hey Joe Cool -
- """, str(tmpl.generate())) - - def test_def_in_match(self): - tmpl = MarkupTemplate(""" - - ${select('*')} - ${maketitle(True)} - """) - self.assertEqual(""" - True - """, str(tmpl.generate())) - - def test_match_with_xpath_variable(self): - tmpl = MarkupTemplate("""
- - Hello ${select('@name')} - - -
""") - self.assertEqual("""
- - Hello Dude - -
""", str(tmpl.generate(tagname='greeting'))) - self.assertEqual("""
- -
""", str(tmpl.generate(tagname='sayhello'))) - - def test_content_directive_in_match(self): - tmpl = MarkupTemplate(""" -
I said something.
- bar - """) - self.assertEqual(""" -
I said bar.
- """, str(tmpl.generate())) - - def test_cascaded_matches(self): - tmpl = MarkupTemplate(""" - ${select('*')} - ${select('title')} - ${select('*')}
- Welcome to Markup -
Are you ready to mark up?
- """) - self.assertEqual(""" - Welcome to Markup -
Are you ready to mark up?
- """, str(tmpl.generate())) - - def test_multiple_matches(self): - tmpl = MarkupTemplate(""" - -
- ${field.capitalize()} - -
- """) - fields = ['hello_%s' % i for i in range(5)] - values = dict([('hello_%s' % i, i) for i in range(5)]) - self.assertEqual(""" -
- Hello_0 - -
- Hello_1 - -
- Hello_2 - -
- Hello_3 - -
- Hello_4 - -
- """, str(tmpl.generate(fields=fields, values=values))) - - def test_namespace_context(self): - tmpl = MarkupTemplate(""" -
Foo
- - """) - # FIXME: there should be a way to strip out unwanted/unused namespaces, - # such as the "x" in this example - self.assertEqual(""" -
Foo
- """, str(tmpl.generate())) - - def test_match_with_position_predicate(self): - tmpl = MarkupTemplate(""" -
${select('*|text()')}
- -
Foo
-
Bar
- - """) - self.assertEqual(""" - -
Foo
-
Bar
- - """, str(tmpl.generate())) - - def test_match_with_closure(self): - tmpl = MarkupTemplate(""" -
${select('*|text()')}
- -
Foo
-
Bar
- - """) - self.assertEqual(""" - -
Foo
-
Bar
- - """, str(tmpl.generate())) - - def test_match_without_closure(self): - tmpl = MarkupTemplate(""" -
${select('*|text()')}
- -
Foo
-
Bar
- - """) - self.assertEqual(""" - -
Foo
-
Bar
- - """, str(tmpl.generate())) - - # FIXME - #def test_match_after_step(self): - # tmpl = MarkupTemplate("""
- # - # Hello ${select('@name')} - # - # - #
""") - # self.assertEqual("""
- # - # Hello Dude - # - #
""", str(tmpl.generate())) - - -class StripDirectiveTestCase(unittest.TestCase): - """Tests for the `py:strip` template directive.""" - - def test_strip_false(self): - tmpl = MarkupTemplate("""
-
foo
-
""") - self.assertEqual("""
-
foo
-
""", str(tmpl.generate())) - - def test_strip_empty(self): - tmpl = MarkupTemplate("""
-
foo
-
""") - self.assertEqual("""
- foo -
""", str(tmpl.generate())) - - -class WithDirectiveTestCase(unittest.TestCase): - """Tests for the `py:with` template directive.""" - - def test_shadowing(self): - tmpl = MarkupTemplate("""
- ${x} - - ${x} -
""") - self.assertEqual("""
- 42 - 84 - 42 -
""", str(tmpl.generate(x=42))) - - def test_as_element(self): - tmpl = MarkupTemplate("""
- ${x} -
""") - self.assertEqual("""
- 84 -
""", str(tmpl.generate(x=42))) - - def test_multiple_vars_same_name(self): - tmpl = MarkupTemplate("""
- - $foo - -
""") - self.assertEqual("""
- baz -
""", str(tmpl.generate(x=42))) - - def test_multiple_vars_single_assignment(self): - tmpl = MarkupTemplate("""
- ${x} ${y} ${z} -
""") - self.assertEqual("""
- 1 1 1 -
""", str(tmpl.generate(x=42))) - - def test_nested_vars_single_assignment(self): - tmpl = MarkupTemplate("""
- ${x} ${y} ${z} -
""") - self.assertEqual("""
- 1 2 3 -
""", str(tmpl.generate(x=42))) - - def test_multiple_vars_trailing_semicolon(self): - tmpl = MarkupTemplate("""
- ${x} ${y} -
""") - self.assertEqual("""
- 84 42 -
""", str(tmpl.generate(x=42))) - - def test_semicolon_escape(self): - tmpl = MarkupTemplate("""
- - ${x} - ${y} - -
""") - self.assertEqual("""
- here is a semicolon: ; - here are two semicolons: ;; -
""", str(tmpl.generate())) - - def test_unicode_expr(self): - tmpl = MarkupTemplate("""
- - $weeks - -
""") - self.assertEqual("""
- - 一二三四五六日 - -
""", str(tmpl.generate())) - - -class TemplateTestCase(unittest.TestCase): - """Tests for basic template processing, expression evaluation and error - reporting. - """ - - def test_interpolate_string(self): - parts = list(Template._interpolate('bla')) - self.assertEqual(1, len(parts)) - self.assertEqual(Stream.TEXT, parts[0][0]) - self.assertEqual('bla', parts[0][1]) - - def test_interpolate_simple(self): - parts = list(Template._interpolate('${bla}')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('bla', parts[0][1].source) - - def test_interpolate_escaped(self): - parts = list(Template._interpolate('$${bla}')) - self.assertEqual(1, len(parts)) - self.assertEqual(Stream.TEXT, parts[0][0]) - self.assertEqual('${bla}', parts[0][1]) - - def test_interpolate_short(self): - parts = list(Template._interpolate('$bla')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('bla', parts[0][1].source) - - def test_interpolate_short_starting_with_underscore(self): - parts = list(Template._interpolate('$_bla')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('_bla', parts[0][1].source) - - def test_interpolate_short_containing_underscore(self): - parts = list(Template._interpolate('$foo_bar')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('foo_bar', parts[0][1].source) - - def test_interpolate_short_starting_with_dot(self): - parts = list(Template._interpolate('$.bla')) - self.assertEqual(1, len(parts)) - self.assertEqual(Stream.TEXT, parts[0][0]) - self.assertEqual('$.bla', parts[0][1]) - - def test_interpolate_short_containing_dot(self): - parts = list(Template._interpolate('$foo.bar')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('foo.bar', parts[0][1].source) - - def test_interpolate_short_starting_with_digit(self): - parts = list(Template._interpolate('$0bla')) - self.assertEqual(1, len(parts)) - self.assertEqual(Stream.TEXT, parts[0][0]) - self.assertEqual('$0bla', parts[0][1]) - - def test_interpolate_short_containing_digit(self): - parts = list(Template._interpolate('$foo0')) - self.assertEqual(1, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('foo0', parts[0][1].source) - - def test_interpolate_mixed1(self): - parts = list(Template._interpolate('$foo bar $baz')) - self.assertEqual(3, len(parts)) - self.assertEqual(Template.EXPR, parts[0][0]) - self.assertEqual('foo', parts[0][1].source) - self.assertEqual(Stream.TEXT, parts[1][0]) - self.assertEqual(' bar ', parts[1][1]) - self.assertEqual(Template.EXPR, parts[2][0]) - self.assertEqual('baz', parts[2][1].source) - - def test_interpolate_mixed2(self): - parts = list(Template._interpolate('foo $bar baz')) - self.assertEqual(3, len(parts)) - self.assertEqual(Stream.TEXT, parts[0][0]) - self.assertEqual('foo ', parts[0][1]) - self.assertEqual(Template.EXPR, parts[1][0]) - self.assertEqual('bar', parts[1][1].source) - self.assertEqual(Stream.TEXT, parts[2][0]) - self.assertEqual(' baz', parts[2][1]) - - -class MarkupTemplateTestCase(unittest.TestCase): - """Tests for markup template processing.""" - - def test_interpolate_mixed3(self): - tmpl = MarkupTemplate(' ${var} $var') - self.assertEqual(' 42 42', str(tmpl.generate(var=42))) - - def test_interpolate_leading_trailing_space(self): - tmpl = MarkupTemplate('${ foo }') - self.assertEqual('bar', str(tmpl.generate(foo='bar'))) - - def test_interpolate_multiline(self): - tmpl = MarkupTemplate("""${dict( - bar = 'baz' - )[foo]}""") - self.assertEqual('baz', str(tmpl.generate(foo='bar'))) - - def test_interpolate_non_string_attrs(self): - tmpl = MarkupTemplate('') - self.assertEqual('', str(tmpl.generate())) - - def test_interpolate_list_result(self): - tmpl = MarkupTemplate('$foo') - self.assertEqual('buzz', str(tmpl.generate(foo=('buzz',)))) - - def test_empty_attr(self): - tmpl = MarkupTemplate('') - self.assertEqual('', str(tmpl.generate())) - - def test_bad_directive_error(self): - xml = '
' - try: - tmpl = MarkupTemplate(xml, filename='test.html') - except BadDirectiveError, e: - self.assertEqual('test.html', e.filename) - if sys.version_info[:2] >= (2, 4): - self.assertEqual(1, e.lineno) - - def test_directive_value_syntax_error(self): - xml = """
""" - try: - tmpl = MarkupTemplate(xml, filename='test.html') - self.fail('Expected SyntaxError') - except TemplateSyntaxError, e: - self.assertEqual('test.html', e.filename) - if sys.version_info[:2] >= (2, 4): - self.assertEqual(1, e.lineno) - - def test_expression_syntax_error(self): - xml = """
- Foo ${bar"} -
""" - try: - tmpl = MarkupTemplate(xml, filename='test.html') - self.fail('Expected SyntaxError') - except TemplateSyntaxError, e: - self.assertEqual('test.html', e.filename) - if sys.version_info[:2] >= (2, 4): - self.assertEqual(2, e.lineno) - - def test_expression_syntax_error_multi_line(self): - xml = """
- - ${bar"} - -
""" - try: - tmpl = MarkupTemplate(xml, filename='test.html') - self.fail('Expected SyntaxError') - except TemplateSyntaxError, e: - self.assertEqual('test.html', e.filename) - if sys.version_info[:2] >= (2, 4): - self.assertEqual(3, e.lineno) - - def test_markup_noescape(self): - """ - Verify that outputting context data that is a `Markup` instance is not - escaped. - """ - tmpl = MarkupTemplate("""
- $myvar -
""") - self.assertEqual("""
- foo -
""", str(tmpl.generate(myvar=Markup('foo')))) - - def test_text_noescape_quotes(self): - """ - Verify that outputting context data in text nodes doesn't escape quotes. - """ - tmpl = MarkupTemplate("""
- $myvar -
""") - self.assertEqual("""
- "foo" -
""", str(tmpl.generate(myvar='"foo"'))) - - def test_attr_escape_quotes(self): - """ - Verify that outputting context data in attribtes escapes quotes. - """ - tmpl = MarkupTemplate("""
- -
""") - self.assertEqual("""
- -
""", str(tmpl.generate(myvar='"foo"'))) - - def test_directive_element(self): - tmpl = MarkupTemplate("""
- bar -
""") - self.assertEqual("""
- bar -
""", str(tmpl.generate(myvar='"foo"'))) - - def test_normal_comment(self): - tmpl = MarkupTemplate("""
- -
""") - self.assertEqual("""
- -
""", str(tmpl.generate())) - - def test_template_comment(self): - tmpl = MarkupTemplate("""
- - -
""") - self.assertEqual("""
-
""", str(tmpl.generate())) - - def test_parse_with_same_namespace_nested(self): - tmpl = MarkupTemplate("""
- - -
""") - self.assertEqual("""
- - -
""", str(tmpl.generate())) - - def test_latin1_encoded_with_xmldecl(self): - tmpl = MarkupTemplate(u""" -
- \xf6 -
""".encode('iso-8859-1'), encoding='iso-8859-1') - self.assertEqual(u"""
- \xf6 -
""", unicode(tmpl.generate())) - - def test_latin1_encoded_explicit_encoding(self): - tmpl = MarkupTemplate(u"""
- \xf6 -
""".encode('iso-8859-1'), encoding='iso-8859-1') - self.assertEqual(u"""
- \xf6 -
""", unicode(tmpl.generate())) - - -class TextTemplateTestCase(unittest.TestCase): - """Tests for text template processing.""" - - def test_escaping(self): - tmpl = TextTemplate('\\#escaped') - self.assertEqual('#escaped', str(tmpl.generate())) - - def test_comment(self): - tmpl = TextTemplate('## a comment') - self.assertEqual('', str(tmpl.generate())) - - def test_comment_escaping(self): - tmpl = TextTemplate('\\## escaped comment') - self.assertEqual('## escaped comment', str(tmpl.generate())) - - def test_end_with_args(self): - tmpl = TextTemplate(""" - #if foo - bar - #end 'if foo'""") - self.assertEqual('', str(tmpl.generate())) - - def test_latin1_encoded(self): - text = u'$foo\xf6$bar'.encode('iso-8859-1') - tmpl = TextTemplate(text, encoding='iso-8859-1') - self.assertEqual(u'x\xf6y', unicode(tmpl.generate(foo='x', bar='y'))) - - # FIXME - #def test_empty_lines(self): - # tmpl = TextTemplate("""Your items: - # - # #for item in items - # * ${item} - # - # #end""") - # self.assertEqual("""Your items: - # * 0 - # * 1 - # * 2 - # """, tmpl.generate(items=range(3)).render('text')) - - -class TemplateLoaderTestCase(unittest.TestCase): - """Tests for the template loader.""" - - def setUp(self): - self.dirname = tempfile.mkdtemp(suffix='markup_test') - - def tearDown(self): - shutil.rmtree(self.dirname) - - def test_search_path_empty(self): - loader = TemplateLoader() - self.assertEqual([], loader.search_path) - - def test_search_path_as_string(self): - loader = TemplateLoader(self.dirname) - self.assertEqual([self.dirname], loader.search_path) - - def test_relative_include_samedir(self): - file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') - try: - file2.write(""" - - """) - finally: - file2.close() - - loader = TemplateLoader([self.dirname]) - tmpl = loader.load('tmpl2.html') - self.assertEqual(""" -
Included
- """, tmpl.generate().render()) - - def test_relative_include_subdir(self): - os.mkdir(os.path.join(self.dirname, 'sub')) - file1 = open(os.path.join(self.dirname, 'sub', 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') - try: - file2.write(""" - - """) - finally: - file2.close() - - loader = TemplateLoader([self.dirname]) - tmpl = loader.load('tmpl2.html') - self.assertEqual(""" -
Included
- """, tmpl.generate().render()) - - def test_relative_include_parentdir(self): - file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - os.mkdir(os.path.join(self.dirname, 'sub')) - file2 = open(os.path.join(self.dirname, 'sub', 'tmpl2.html'), 'w') - try: - file2.write(""" - - """) - finally: - file2.close() - - loader = TemplateLoader([self.dirname]) - tmpl = loader.load('sub/tmpl2.html') - self.assertEqual(""" -
Included
- """, tmpl.generate().render()) - - def test_relative_include_without_search_path(self): - file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') - try: - file2.write(""" - - """) - finally: - file2.close() - - loader = TemplateLoader() - tmpl = loader.load(os.path.join(self.dirname, 'tmpl2.html')) - self.assertEqual(""" -
Included
- """, tmpl.generate().render()) - - def test_relative_include_without_search_path_nested(self): - file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - file2 = open(os.path.join(self.dirname, 'tmpl2.html'), 'w') - try: - file2.write("""
- -
""") - finally: - file2.close() - - file3 = open(os.path.join(self.dirname, 'tmpl3.html'), 'w') - try: - file3.write(""" - - """) - finally: - file3.close() - - loader = TemplateLoader() - tmpl = loader.load(os.path.join(self.dirname, 'tmpl3.html')) - self.assertEqual(""" -
-
Included
-
- """, tmpl.generate().render()) - - def test_relative_include_from_inmemory_template(self): - file1 = open(os.path.join(self.dirname, 'tmpl1.html'), 'w') - try: - file1.write("""
Included
""") - finally: - file1.close() - - loader = TemplateLoader([self.dirname]) - tmpl2 = MarkupTemplate(""" - - """, filename='subdir/tmpl2.html', loader=loader) - - self.assertEqual(""" -
Included
- """, tmpl2.generate().render()) - - def test_load_with_default_encoding(self): - f = open(os.path.join(self.dirname, 'tmpl.html'), 'w') - try: - f.write(u'
\xf6
'.encode('iso-8859-1')) - finally: - f.close() - loader = TemplateLoader([self.dirname], default_encoding='iso-8859-1') - loader.load('tmpl.html') - - def test_load_with_explicit_encoding(self): - f = open(os.path.join(self.dirname, 'tmpl.html'), 'w') - try: - f.write(u'
\xf6
'.encode('iso-8859-1')) - finally: - f.close() - loader = TemplateLoader([self.dirname], default_encoding='utf-8') - loader.load('tmpl.html', encoding='iso-8859-1') - - -def suite(): - suite = unittest.TestSuite() - suite.addTest(doctest.DocTestSuite(template)) - suite.addTest(unittest.makeSuite(AttrsDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(ChooseDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(DefDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(ForDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(IfDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(MatchDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(StripDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(WithDirectiveTestCase, 'test')) - suite.addTest(unittest.makeSuite(TemplateTestCase, 'test')) - suite.addTest(unittest.makeSuite(MarkupTemplateTestCase, 'test')) - suite.addTest(unittest.makeSuite(TextTemplateTestCase, 'test')) - suite.addTest(unittest.makeSuite(TemplateLoaderTestCase, 'test')) - return suite - -if __name__ == '__main__': - unittest.main(defaultTest='suite') diff --git a/setup.py b/setup.py --- a/setup.py +++ b/setup.py @@ -46,14 +46,14 @@ 'Topic :: Text Processing :: Markup :: XML' ], keywords = ['python.templating.engines'], - packages = ['genshi'], + packages = ['genshi', 'genshi.template'], test_suite = 'genshi.tests.suite', extras_require = {'plugin': ['setuptools>=0.6a2']}, entry_points = """ [python.templating.engines] - genshi = genshi.plugin:MarkupTemplateEnginePlugin[plugin] - genshi-markup = genshi.plugin:MarkupTemplateEnginePlugin[plugin] - genshi-text = genshi.plugin:TextTemplateEnginePlugin[plugin] + genshi = genshi.template.plugin:MarkupTemplateEnginePlugin[plugin] + genshi-markup = genshi.template.plugin:MarkupTemplateEnginePlugin[plugin] + genshi-text = genshi.template.plugin:TextTemplateEnginePlugin[plugin] """, )