Add dedicated tests for jinja templates

md_pdf/build/jinja.py

@@ -10,7 +10,6 @@ def render_content(content, context):
             'jinja2.ext.with_',
             'jinja2.ext.loopcontrols'
         ]
-
     )
     template = env.from_string(content)
     return template.render(**context)

tests/test_jinja.py

@@ -0,0 +1,27 @@
+from tests import BaseTestCase
+from md_pdf.build.jinja import render_content
+
+
+class ContentRendererTestCase(BaseTestCase):
+    def test_renders_template(self):
+        html = 'test {{ test }}'
+        output = render_content(html, self.extend_config({
+            'test': 'content'
+        }))
+        self.assertEqual(output, 'test content')
+
+    def test_changes_nothing(self):
+        html = 'test test'
+        output = render_content(html, self.extend_config({
+            'test': 'content'
+        }))
+        self.assertEqual(output, html)
+
+    def test_with_block(self):
+        html = """
+        {% with test = 'test' %}
+            {{ test }} thing
+        {% endwith %}
+        """
+        output = render_content(html, self.BASE_VALID_CONFIG)
+        self.assertIn('test thing', output)

tests/test_parser.py

@@ -48,10 +48,3 @@ class RenderTemplateTestCase(BaseTestCase):
             'test': 'content'
         }))
         self.assertEqual(output, 'test content')
-
-    def test_changes_nothing(self):
-        html = 'test test'
-        output = content.render_template(html, self.extend_config({
-            'test': 'content'
-        }))
-        self.assertEqual(output, html)