chore: add jinja test

phil65 · Sep 23, 2023 · 849d74c · 849d74c
1 parent 37f863e
commit 849d74c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/mknodes/jinja/environment.py b/mknodes/jinja/environment.py
@@ -28,7 +28,7 @@ def __init__(self, *, undefined: str = "silent", load_templates: bool = False):
         super().__init__(undefined=behavior, loader=loader, trim_blocks=True)
         self.filters.update(jinjahelpers.ENVIRONMENT_FILTERS)
         self.globals.update(jinjahelpers.ENVIRONMENT_GLOBALS)
-        self.rendered_nodes: set[mk.MkNode] = set()
+        self.rendered_nodes: list[mk.MkNode] = list()
 
     def collect_reqs(self, ctx, **kwargs):
         pass
@@ -153,7 +153,7 @@ def set_mknodes_filters(self, parent: mk.MkNode | None = None):
 
             def wrapped(ctx, *args, fn=fn, **kwargs):
                 node = fn(*args, **kwargs)
-                self.rendered_nodes.add(node)
+                self.rendered_nodes.append(node)
                 return node
 
             new = functools.partial(wrapped, fn=fn)

diff --git a/tests/test_jinja.py b/tests/test_jinja.py
@@ -0,0 +1,10 @@
+from __future__ import annotations
+
+import mknodes
+
+
+def test_if_mknodes_parent_is_set(project):
+    page = mknodes.MkPage()
+    project._root += page
+    page.env.render_string(r"{{ 'test' | MkHeader }}")
+    assert page.env.rendered_nodes[-1].parent == page