7
7
from django .test import override_settings
8
8
from django_mermaid .templatetags import DEFAULT_THEME
9
9
10
+ import pytest
11
+
10
12
try :
11
13
import site
12
14
17
19
site_packages = sysconfig .get_paths ()["purelib" ]
18
20
19
21
20
- def test_tag_use_in_template (version ):
22
+ @pytest .mark .parametrize (
23
+ "template_code" ,
24
+ [
25
+ "{% load mermaid %}{% mermaid content %}" ,
26
+ "{% load mermaid %}{% startmermaid %}{{ content|safe }}{% endmermaid %}"
27
+ ]
28
+ )
29
+ def test_tag_use_in_template (version , template_code ):
21
30
theme = getattr (settings , "MERMAID_THEME" , DEFAULT_THEME )
22
- template = Template ("{% load mermaid %}{% mermaid content %}" )
31
+ template = Template (template_code )
23
32
template = template .render (Context ({"content" : "graph LR; A-->B;" }))
24
33
assert template == (
25
34
"<div class=\" mermaid\" >graph LR; A-->B;</div><script src=\" mermaid/%s/mermaid.js\" ></script>"
@@ -29,8 +38,15 @@ def test_tag_use_in_template(version):
29
38
30
39
31
40
@override_settings (MERMAID_THEME = "forest" )
32
- def test_tag_use_settings_theme (version ):
33
- template = Template ("{% load mermaid %}{% mermaid content %}" )
41
+ @pytest .mark .parametrize (
42
+ "template_code" ,
43
+ [
44
+ "{% load mermaid %}{% mermaid content %}" ,
45
+ "{% load mermaid %}{% startmermaid %}{{ content|safe }}{% endmermaid %}"
46
+ ]
47
+ )
48
+ def test_tag_use_settings_theme (version , template_code ):
49
+ template = Template (template_code )
34
50
template = template .render (Context ({"content" : "graph LR; A-->B;" }))
35
51
assert template == (
36
52
"<div class=\" mermaid\" >graph LR; A-->B;</div><script src=\" mermaid/%s/mermaid.js\" ></script>"
@@ -39,8 +55,15 @@ def test_tag_use_settings_theme(version):
39
55
)
40
56
41
57
42
- def test_tag_use_custom_theme (version ):
43
- template = Template ("{% load mermaid %}{% mermaid content \" dark\" %}" )
58
+ @pytest .mark .parametrize (
59
+ "template_code" ,
60
+ [
61
+ "{% load mermaid %}{% mermaid content \" dark\" %}" ,
62
+ "{% load mermaid %}{% startmermaid \" dark\" %}{{ content|safe }}{% endmermaid %}"
63
+ ]
64
+ )
65
+ def test_tag_use_custom_theme (version , template_code ):
66
+ template = Template (template_code )
44
67
template = template .render (Context ({"content" : "graph LR; A-->B;" }))
45
68
assert template == (
46
69
"<div class=\" mermaid\" >graph LR; A-->B;</div><script src=\" mermaid/%s/mermaid.js\" ></script>"
@@ -50,8 +73,15 @@ def test_tag_use_custom_theme(version):
50
73
51
74
52
75
@override_settings (MERMAID_THEME_VARIABLES = {"primaryColor" : "red" })
53
- def test_tag_use_custom_theme_variables (version ):
54
- template = Template ("{% load mermaid %}{% mermaid content \" dark\" %}" )
76
+ @pytest .mark .parametrize (
77
+ "template_code" ,
78
+ [
79
+ "{% load mermaid %}{% mermaid content \" dark\" %}" ,
80
+ "{% load mermaid %}{% startmermaid \" dark\" %}{{ content|safe }}{% endmermaid %}"
81
+ ]
82
+ )
83
+ def test_tag_use_custom_theme_variables (version , template_code ):
84
+ template = Template (template_code )
55
85
template = template .render (Context ({"content" : "graph LR; A-->B;" }))
56
86
assert template == (
57
87
"<div class=\" mermaid\" >graph LR; A-->B;</div><script src=\" mermaid/%s/mermaid.js\" ></script>"
@@ -61,8 +91,15 @@ def test_tag_use_custom_theme_variables(version):
61
91
62
92
63
93
@override_settings (MERMAID_THEME = "base" , MERMAID_THEME_VARIABLES = {"primaryColor" : "#efefef" })
64
- def test_tag_use_custom_theme_variables_with_base_theme (version ):
65
- template = Template ("{% load mermaid %}{% mermaid content %}" )
94
+ @pytest .mark .parametrize (
95
+ "template_code" ,
96
+ [
97
+ "{% load mermaid %}{% mermaid content %}" ,
98
+ "{% load mermaid %}{% startmermaid %}{{ content|safe }}{% endmermaid %}"
99
+ ]
100
+ )
101
+ def test_tag_use_custom_theme_variables_with_base_theme (version , template_code ):
102
+ template = Template (template_code )
66
103
template = template .render (Context ({"content" : "graph LR; A-->B;" }))
67
104
assert template == (
68
105
"<div class=\" mermaid\" >graph LR; A-->B;</div><script src=\" mermaid/%s/mermaid.js\" ></script>"
0 commit comments