feat: add test-level pii tag support for consistent behavior with show_sample_rows

Author: joostboon

macros/edr/materializations/test/test.sql

@@ -80,6 +80,7 @@
     {% elif elementary.get_config_var("enable_samples_on_show_sample_rows_tags") %}
         {% set sample_limit = 0 %}
     {% elif elementary.is_pii_table(flattened_test) %} {% set sample_limit = 0 %}
+    {% elif elementary.is_pii_test(flattened_test) %} {% set sample_limit = 0 %}
     {% elif elementary.should_disable_sampling_for_pii(flattened_test) %}
         {% set sample_limit = 0 %}
     {% endif %}

macros/edr/system/system_utils/is_pii_test.sql

@@ -0,0 +1,19 @@
+{% macro is_pii_test(flattened_test) %}
+    {% if not elementary.get_config_var("disable_samples_on_pii_tags") %}
+        {% do return(false) %}
+    {% endif %}
+
+    {% set raw_pii_tags = elementary.get_config_var("pii_tags") %}
+    {% if raw_pii_tags is string %} {% set pii_tags = [raw_pii_tags | lower] %}
+    {% else %} {% set pii_tags = (raw_pii_tags or []) | map("lower") | list %}
+    {% endif %}
+
+    {% set raw_test_tags = elementary.insensitive_get_dict_value(
+        flattened_test, "tags", []
+    ) %}
+    {% if raw_test_tags is string %} {% set test_tags = [raw_test_tags | lower] %}
+    {% else %} {% set test_tags = (raw_test_tags or []) | map("lower") | list %}
+    {% endif %}
+
+    {% do return(elementary.lists_intersection(test_tags, pii_tags) | length > 0) %}
+{% endmacro %}