Feature/support query tags in tests (#210)

* initial test materialization with query tagging

* set adapter to snowflake

* move over core materialization code for testing

* add query tagging macros

* revise with jtcohen's version

* add validation on query tags for tests

* add changelog entry

* update changelog again

* fix linting issue & update changelog with PR

* update changelog using changie

.changes/unreleased/Features-20220815-233505.yaml

@@ -0,0 +1,7 @@
+kind: Features
+body: Support query tags in dbt tests using the `+query_tag` config in `dbt_project.yml`
+time: 2022-08-15T23:35:05.967484-06:00
+custom:
+  Author: matt-winkler
+  Issue: "211"
+  PR: "210"

dbt/include/snowflake/macros/materializations/test.sql

@@ -0,0 +1,8 @@
+{%- materialization test, adapter='snowflake' -%}
+
+    {% set original_query_tag = set_query_tag() %}
+    {% set relations = materialization_test_default() %}
+    {% do unset_query_tag(original_query_tag) %}
+    {{ return(relations) }}
+
+{%- endmaterialization -%}

tests/integration/query_tag_tests/models/models.yml

@@ -0,0 +1,8 @@
+version: 2
+
+models:
+  - name: view_model_query_tag
+    columns:
+      - name: id
+        tests:
+          - unique

tests/integration/query_tag_tests/test_query_tags.py

@@ -33,6 +33,12 @@ def project_config(self):
                     'post-hook': '{{ check_query_tag() }}'
                 },
             },
+            'tests': {
+                'test': {
+                    'query_tag': self.prefix,
+                    'post-hook': '{{ check_query_tag() }}'
+                }
+            },
         }
 
     def build_all_with_query_tags(self):