blob: a1d3a91715892f4b5961fea23de7bfc88bdb39c0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
# -*- coding: utf-8 -*-
# vim: set wrap textwidth=100
"""Test template-based shortcodes."""
from __future__ import unicode_literals
import pytest
from nikola import Nikola
class ShortcodeFakeSite(Nikola):
def _get_template_system(self):
if self._template_system is None:
# Load template plugin
self._template_system = self.plugin_manager.getPluginByName(
'jinja', "TemplateSystem").plugin_object
self._template_system.set_directories('.', 'cache')
self._template_system.set_site(self)
return self._template_system
template_system = property(_get_template_system)
@pytest.fixture(scope="module")
def fakesite():
s = ShortcodeFakeSite()
s.init_plugins()
s._template_system = None
return s
def test_mixedargs(fakesite):
TEST_TMPL = """
arg1: {{ _args[0] }}
arg2: {{ _args[1] }}
kwarg1: {{ kwarg1 }}
kwarg2: {{ kwarg2 }}
"""
fakesite.shortcode_registry['test1'] = \
fakesite._make_renderfunc(TEST_TMPL)
fakesite.shortcode_registry['test2'] = \
fakesite._make_renderfunc('Something completely different')
res = fakesite.apply_shortcodes(
'{{% test1 kwarg1=spamm arg1 kwarg2=foo,bar arg2 %}}')
assert res.strip() == """
arg1: arg1
arg2: arg2
kwarg1: spamm
kwarg2: foo,bar""".strip()
def test_onearg(fakesite):
fakesite.shortcode_registry['test1'] = \
fakesite._make_renderfunc('arg={{ _args[0] }}')
assert fakesite.apply_shortcodes('{{% test1 onearg %}}') == 'arg=onearg'
assert fakesite.apply_shortcodes('{{% test1 "one two" %}}') == 'arg=one two'
def test_kwarg(fakesite):
fakesite.shortcode_registry['test1'] = \
fakesite._make_renderfunc('foo={{ foo }}')
res = fakesite.apply_shortcodes('{{% test1 foo=bar %}}')
assert res == 'foo=bar'
res = fakesite.apply_shortcodes('{{% test1 foo="bar baz" %}}')
assert res == 'foo=bar baz'
res = fakesite.apply_shortcodes('{{% test1 foo="bar baz" spamm=ham %}}')
assert res == 'foo=bar baz'
def test_data(fakesite):
fakesite.shortcode_registry['test1'] = \
fakesite._make_renderfunc('data={{ data }}')
res = fakesite.apply_shortcodes('{{% test1 %}}spamm spamm{{% /test1 %}}')
assert res == 'data=spamm spamm'
res = fakesite.apply_shortcodes('{{% test1 spamm %}}')
assert res == 'data='
# surprise!
res = fakesite.apply_shortcodes('{{% test1 data=dummy %}}')
assert res == 'data='
|
