test: Add test with context variable

Author: robwa

tests/templatetags/test_unicorn.py

@@ -1,26 +1,26 @@
-from django.template.base import Token, TokenType
+from django.template.base import Parser, Token, TokenType
 
 from django_unicorn.templatetags.unicorn import unicorn
 
 
 def test_unicorn():
     token = Token(TokenType.TEXT, "unicorn 'todo'")
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
 
-    assert unicorn_node.component_name == "todo"
+    assert unicorn_node.component_name.resolve({}) == "todo"
     assert len(unicorn_node.kwargs) == 0
 
 
 def test_unicorn_kwargs():
     token = Token(TokenType.TEXT, "unicorn 'todo' blob='blob'")
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
 
     assert unicorn_node.kwargs == {"blob": "blob"}
 
 
 def test_unicorn_args_and_kwargs():
     # args after the component name get ignored
     token = Token(TokenType.TEXT, "unicorn 'todo' '1' 2 hello='world' test=3 '4'")
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
 
     assert unicorn_node.kwargs == {"hello": "world", "test": 3}

tests/templatetags/test_unicorn_render.py

@@ -11,6 +11,10 @@
 from example.coffee.models import Flavor
 
 
+class FakeComponent(UnicornView):
+    template_name = "templates/test_component.html"
+
+
 class FakeComponentParent(UnicornView):
     template_name = "templates/test_component_parent.html"
 
@@ -66,7 +70,7 @@ def test_unicorn_render_kwarg():
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' test_kwarg='tested!'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     actual = unicorn_node.render(Context(context))
 
@@ -78,7 +82,7 @@ def test_unicorn_render_context_variable():
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' test_kwarg=test_var.nested",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {"test_var": {"nested": "variable!"}}
     actual = unicorn_node.render(Context(context))
 
@@ -90,7 +94,7 @@ def test_unicorn_render_with_invalid_html():
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargsWithHtmlEntity' test_kwarg=test_var.nested",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {"test_var": {"nested": "variable!"}}
     actual = unicorn_node.render(Context(context))
 
@@ -103,7 +107,7 @@ def test_unicorn_render_parent(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
     context = {"view": view}
     unicorn_node.render(Context(context))
@@ -121,7 +125,7 @@ def test_unicorn_render_parent_with_key(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view key='blob'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
     context = {"view": view}
     unicorn_node.render(Context(context))
@@ -138,7 +142,7 @@ def test_unicorn_render_parent_with_id(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view id='flob'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
     context = {"view": view}
     unicorn_node.render(Context(context))
@@ -155,7 +159,7 @@ def test_unicorn_render_parent_with_pk(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view pk=99",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
     context = {"view": view}
     unicorn_node.render(Context(context))
@@ -172,7 +176,7 @@ def test_unicorn_render_parent_with_model_id(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view model=model",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
 
     # Fake a model that only has an id
@@ -199,7 +203,7 @@ def test_unicorn_render_parent_with_model_pk(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view model=model",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
 
     flavor = Flavor(pk=187)
@@ -218,7 +222,7 @@ def test_unicorn_render_id_use_pk():
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentModel' model_id=model.id",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {"model": {"pk": 123}}
     actual = unicorn_node.render(Context(context))
 
@@ -231,7 +235,7 @@ def test_unicorn_render_component_one_script_tag(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -245,7 +249,7 @@ def test_unicorn_render_child_component_no_script_tag(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentKwargs' parent=view",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     view = FakeComponentParent(component_name="test", component_id="asdf")
     context = {"view": view}
     html = unicorn_node.render(Context(context))
@@ -259,7 +263,7 @@ def test_unicorn_render_parent_component_one_script_tag(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentParent'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -273,7 +277,7 @@ def test_unicorn_render_calls(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentCalls'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -288,7 +292,7 @@ def test_unicorn_render_calls_with_arg(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentCalls2'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -303,7 +307,7 @@ def test_unicorn_render_calls_no_mount_call(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentParent'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -318,7 +322,7 @@ def test_unicorn_render_hash(settings):
         TokenType.TEXT,
         "unicorn 'tests.templatetags.test_unicorn_render.FakeComponentParent'",
     )
-    unicorn_node = unicorn(None, token)
+    unicorn_node = unicorn(Parser([]), token)
     context = {}
     html = unicorn_node.render(Context(context))
 
@@ -331,3 +335,16 @@ def test_unicorn_render_hash(settings):
     rendered_content = html[:script_idx]
     expected_hash = generate_checksum(rendered_content)
     assert f'"hash":"{expected_hash}"' in html
+
+
+def test_unicorn_render_with_component_name_from_context():
+    token = Token(
+        TokenType.TEXT,
+        "unicorn component_name",
+    )
+    unicorn_node = unicorn(Parser([]), token)
+    context = {"component_name": "tests.templatetags.test_unicorn_render.FakeComponent"}
+    unicorn_node.render(Context(context))
+
+    # Success if no exception was raised so far
+    assert True