manager: datamodel: template: macro for get_tagset

diff --git a/manager/knot_resolver_manager/datamodel/templates/macros/policy_macros.lua.j2 b/manager/knot_resolver_manager/datamodel/templates/macros/policy_macros.lua.j2
index aa81ddd26ab8b49d9e961ace8504e6c2ba1ad158..e327d183680f2b7c2a4132ba92d9f962274a5b27 100644 (file)
--- a/manager/knot_resolver_manager/datamodel/templates/macros/policy_macros.lua.j2
+++ b/manager/knot_resolver_manager/datamodel/templates/macros/policy_macros.lua.j2
@@ -54,6 +54,10 @@ policy.FLAGS({
 policy.TAGS_ASSIGN({{ string_table(tags) }})
 {%- endmacro %}
 
+{% macro policy_get_tagset(tags) -%}
+policy.get_tagset({{ string_table(tags) }})
+{%- endmacro %}
+
 
 {# Filters #}
 

diff --git a/manager/tests/unit/datamodel/templates/test_policy_macros.py b/manager/tests/unit/datamodel/templates/test_policy_macros.py
index 1ed97c9987bc3c695d71393d0c6695c8779d0f68..3c2161a2b9d72c11e57ae2030b9b45b23cc24fab 100644 (file)
--- a/manager/tests/unit/datamodel/templates/test_policy_macros.py
+++ b/manager/tests/unit/datamodel/templates/test_policy_macros.py
@@ -38,6 +38,16 @@ def test_policy_tags_assign():
     assert tmpl.render(tags=tags) == "policy.TAGS_ASSIGN({" + ",".join([f"'{x}'" for x in tags]) + ",})"
 
 
+def test_policy_get_tagset():
+    tags: List[str] = ["t01", "t02", "t03"]
+    tmpl_str = """{% from 'macros/policy_macros.lua.j2' import policy_get_tagset %}
+{{ policy_get_tagset(tags) }}"""
+
+    tmpl = template_from_str(tmpl_str)
+    assert tmpl.render(tags=tags[1]) == f"policy.get_tagset('{tags[1]}')"
+    assert tmpl.render(tags=tags) == "policy.get_tagset({" + ",".join([f"'{x}'" for x in tags]) + ",})"
+
+
 # Filters