Add two more test cases

Author: ArtyomVancyan

tests/conftest.py

@@ -2,4 +2,13 @@
 
 
 def pytest_configure():
-    settings.configure(INSTALLED_APPS=["django_mermaid.apps.MermaidConfig"])
+    settings.configure(
+        INSTALLED_APPS=[
+            "django_mermaid.apps.MermaidConfig"
+        ],
+        TEMPLATES=[
+            {
+                "BACKEND": "django.template.backends.django.DjangoTemplates",
+            },
+        ]
+    )

tests/test_tag.py

@@ -1,3 +1,6 @@
+from django.template import Context
+from django.template import Template
+
 from django_mermaid.templatetags.mermaid import mermaid
 
 
@@ -6,3 +9,21 @@ def test_tag_renders():
         """<div class="mermaid">graph LR; A-->B;</div><script src="mermaid.js"></script>"""
         """<script>mermaid.initialize({"startOnLoad": true, theme: "default"});</script>"""
     )
+
+
+def test_tag_use_in_template():
+    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.js"></script>"""
+        """<script>mermaid.initialize({"startOnLoad": true, theme: "default"});</script>"""
+    )
+
+
+def test_tag_use_in_template_with_arguments():
+    template = Template("{% load mermaid %}{% mermaid content \"forest\" %}")
+    template = template.render(Context({"content": "graph LR; A-->B;"}))
+    assert template == (
+        """<div class="mermaid">graph LR; A-->B;</div><script src="mermaid.js"></script>"""
+        """<script>mermaid.initialize({"startOnLoad": true, theme: "forest"});</script>"""
+    )