GH-10: Update test mocks and add two more for MERMAID_THEME_VARIABLES

Author: ArtyomVancyan

tests/test_tag.py

@@ -6,7 +6,6 @@
 from django.template import Context
 from django.template import Template
 from django.test import override_settings
-
 from django_mermaid.templatetags import DEFAULT_THEME
 
 
@@ -16,7 +15,8 @@ def test_tag_use_in_template(version):
     template = template.render(Context({"content": "graph LR; A-->B;"}))
     assert template == (
             "<div class=\"mermaid\">graph LR; A-->B;</div><script src=\"mermaid/%s/mermaid.js\"></script>"
-            "<script>mermaid.initialize({\"startOnLoad\": true, theme: \"%s\"});</script>" % (version, theme)
+            "<script>mermaid.initialize({\"startOnLoad\": true, \"theme\": \"%s\", \"themeVariables\": {}"
+            "});</script>" % (version, theme)
     )
 
 
@@ -26,7 +26,8 @@ def test_tag_use_settings_theme(version):
     template = template.render(Context({"content": "graph LR; A-->B;"}))
     assert template == (
             "<div class=\"mermaid\">graph LR; A-->B;</div><script src=\"mermaid/%s/mermaid.js\"></script>"
-            "<script>mermaid.initialize({\"startOnLoad\": true, theme: \"forest\"});</script>" % version
+            "<script>mermaid.initialize({\"startOnLoad\": true, \"theme\": \"forest\", \"themeVariables\""
+            ": {}});</script>" % version
     )
 
 
@@ -35,7 +36,30 @@ def test_tag_use_custom_theme(version):
     template = template.render(Context({"content": "graph LR; A-->B;"}))
     assert template == (
             "<div class=\"mermaid\">graph LR; A-->B;</div><script src=\"mermaid/%s/mermaid.js\"></script>"
-            "<script>mermaid.initialize({\"startOnLoad\": true, theme: \"dark\"});</script>" % version
+            "<script>mermaid.initialize({\"startOnLoad\": true, \"theme\": \"dark\", \"themeVariables\": "
+            "{}});</script>" % version
+    )
+
+
+@override_settings(MERMAID_THEME_VARIABLES={"primaryColor": "red"})
+def test_tag_use_custom_theme_variables(version):
+    template = Template("{% load mermaid %}{% mermaid content \"dark\" %}")
+    template = template.render(Context({"content": "graph LR; A-->B;"}))
+    assert template == (
+            "<div class=\"mermaid\">graph LR; A-->B;</div><script src=\"mermaid/%s/mermaid.js\"></script>"
+            "<script>mermaid.initialize({\"startOnLoad\": true, \"theme\": \"dark\", \"themeVariables\": "
+            "{}});</script>" % version
+    )
+
+
+@override_settings(MERMAID_THEME="base", MERMAID_THEME_VARIABLES={"primaryColor": "#efefef"})
+def test_tag_use_custom_theme_variables_with_base_theme(version):
+    template = Template("{% load mermaid %}{% mermaid content %}")
+    template = template.render(Context({"content": "graph LR; A-->B;"}))
+    assert template == (
+            "<div class=\"mermaid\">graph LR; A-->B;</div><script src=\"mermaid/%s/mermaid.js\"></script>"
+            "<script>mermaid.initialize({\"startOnLoad\": true, \"theme\": \"base\", \"themeVariables\": "
+            "{\"primaryColor\": \"#efefef\"}});</script>" % version
     )