From: Aleš Mrázek <ales.mrazek@nic.cz>
Date: Wed, 5 Apr 2023 11:21:11 +0000 (+0200)
Subject: manager: datamodel: template: macro for get_tagset
X-Git-Tag: v6.0.1~11^2^2~16
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=7c1a9966236839f15b7069ea5dd96618c9d62a77;p=thirdparty%2Fknot-resolver.git

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 aa81ddd26..e327d1836 100644
--- 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 1ed97c998..3c2161a2b 100644
--- 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