tests/test_macros.py

   1 # -*- coding: utf-8 -*-
   2 """
   3     unit test for the macros
   4     ~~~~~~~~~~~~~~~~~~~~~~~~
   5
   6     :copyright: 2007 by Armin Ronacher.
   7     :license: BSD, see LICENSE for more details.
   8 """
   9
  10 SIMPLE = '''\
  11 {% macro say_hello(name) %}Hello {{ name }}!{% endmacro %}
  12 {{ say_hello('Peter') }}\
  13 '''
  14
  15 SCOPING = '''\
  16 {% macro level1(data1) %}
  17 {% macro level2(data2) %}{{ data1 }}|{{ data2 }}{% endmacro %}
  18 {{ level2('bar') }}{% endmacro %}
  19 {{ level1('foo') }}\
  20 '''
  21
  22 ARGUMENTS = '''\
  23 {% macro m(a, b, c='c', d='d') %}{{ a }}|{{ b }}|{{ c }}|{{ d }}{% endmacro %}
  24 {{ m() }}|{{ m('a') }}|{{ m('a', 'b') }}|{{ m(1, 2, 3) }}\
  25 '''
  26
  27 VARARGS = '''\
  28 {% macro test() %}{{ varargs|join('|') }}{% endmacro %}\
  29 {{ test(1, 2, 3) }}\
  30 '''
  31
  32 SIMPLECALL = '''\
  33 {% macro test() %}[[{{ caller() }}]]{% endmacro %}\
  34 {% call test() %}data{% endcall %}\
  35 '''
  36
  37 COMPLEXCALL = '''\
  38 {% macro test() %}[[{{ caller('data') }}]]{% endmacro %}\
  39 {% call(data) test() %}{{ data }}{% endcall %}\
  40 '''
  41
  42 CALLERUNDEFINED = '''\
  43 {% set caller = 42 %}\
  44 {% macro test() %}{{ caller is not defined }}{% endmacro %}\
  45 {{ test() }}\
  46 '''
  47
  48 INCLUDETEMPLATE = '''{% macro test(foo) %}[{{ foo }}]{% endmacro %}'''
  49
  50
  51 def test_simple(env):
  52     tmpl = env.from_string(SIMPLE)
  53     assert tmpl.render() == 'Hello Peter!'
  54
  55
  56 def test_scoping(env):
  57     tmpl = env.from_string(SCOPING)
  58     assert tmpl.render() == 'foo|bar'
  59
  60
  61 def test_arguments(env):
  62     tmpl = env.from_string(ARGUMENTS)
  63     assert tmpl.render() == '||c|d|a||c|d|a|b|c|d|1|2|3|d'
  64
  65
  66 def test_varargs(env):
  67     tmpl = env.from_string(VARARGS)
  68     assert tmpl.render() == '1|2|3'
  69
  70
  71 def test_simple_call(env):
  72     tmpl = env.from_string(SIMPLECALL)
  73     assert tmpl.render() == '[[data]]'
  74
  75
  76 def test_complex_call(env):
  77     tmpl = env.from_string(COMPLEXCALL)
  78     assert tmpl.render() == '[[data]]'
  79
  80
  81 def test_caller_undefined(env):
  82     tmpl = env.from_string(CALLERUNDEFINED)
  83     assert tmpl.render() == 'True'
  84
  85
  86 def test_include(env):
  87     tmpl = env.from_string('{% from "include" import test %}{{ test("foo") }}')
  88     assert tmpl.render() == '[foo]'
  89
  90
  91 def test_macro_api(env):
  92     tmpl = env.from_string('{% macro foo(a, b) %}{% endmacro %}'
  93                            '{% macro bar() %}{{ varargs }}{{ kwargs }}{% endmacro %}'
  94                            '{% macro baz() %}{{ caller() }}{% endmacro %}')
  95     assert tmpl.module.foo.arguments == ('a', 'b')
  96     assert tmpl.module.foo.defaults == ()
  97     assert tmpl.module.foo.name == 'foo'
  98     assert not tmpl.module.foo.caller
  99     assert not tmpl.module.foo.catch_kwargs
 100     assert not tmpl.module.foo.catch_varargs
 101     assert tmpl.module.bar.arguments == ()
 102     assert tmpl.module.bar.defaults == ()
 103     assert not tmpl.module.bar.caller
 104     assert tmpl.module.bar.catch_kwargs
 105     assert tmpl.module.bar.catch_varargs
 106     assert tmpl.module.baz.caller