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
|
"""Test template-based shortcodes."""
import pytest
from nikola import Nikola
def test_mixedargs(site):
test_template = """
arg1: {{ _args[0] }}
arg2: {{ _args[1] }}
kwarg1: {{ kwarg1 }}
kwarg2: {{ kwarg2 }}
"""
site.shortcode_registry["test1"] = site._make_renderfunc(test_template)
site.shortcode_registry["test2"] = site._make_renderfunc(
"Something completely different"
)
res = site.apply_shortcodes("{{% test1 kwarg1=spamm arg1 kwarg2=foo,bar arg2 %}}")[
0
]
assert res.strip() == """
arg1: arg1
arg2: arg2
kwarg1: spamm
kwarg2: foo,bar""".strip()
@pytest.mark.parametrize(
"template, data, expected_result",
[
# one argument
("arg={{ _args[0] }}", "{{% test1 onearg %}}", "arg=onearg"),
("arg={{ _args[0] }}", '{{% test1 "one two" %}}', "arg=one two"),
# keyword arguments
("foo={{ foo }}", "{{% test1 foo=bar %}}", "foo=bar"),
("foo={{ foo }}", '{{% test1 foo="bar baz" %}}', "foo=bar baz"),
("foo={{ foo }}", '{{% test1 foo="bar baz" spamm=ham %}}', "foo=bar baz"),
# data
(
"data={{ data }}",
"{{% test1 %}}spamm spamm{{% /test1 %}}",
"data=spamm spamm",
),
("data={{ data }}", "{{% test1 spamm %}}", "data="),
("data={{ data }}", "{{% test1 data=dummy %}}", "data="),
],
)
def test_applying_shortcode(site, template, data, expected_result):
site.shortcode_registry["test1"] = site._make_renderfunc(template)
assert site.apply_shortcodes(data)[0] == expected_result
@pytest.fixture(scope="module")
def site():
s = ShortcodeFakeSite()
s.init_plugins()
s._template_system = None
return s
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)
|
