Refactor query tag tests (#57)

* Refactor query tag tests

* Try this

* Try storing failures, too

* Keep trying

* Try it this way

tests/integration/query_tag_tests/macros/check_tag.sql

@@ -0,0 +1,10 @@
+{% macro check_query_tag() %}
+
+  {% if execute %}
+    {% set query_tag = get_current_query_tag() %}
+    {% if query_tag != var("query_tag") %}
+      {{ exceptions.raise_compiler_error("Query tag not used!") }}
+    {% endif %}
+  {% endif %}
+
+{% endmacro %}

tests/integration/query_tag_tests/models/incremental_model_query_tag.sql

@@ -0,0 +1,3 @@
+{{ config(materialized = 'incremental', unique_key = 'id') }}
+
+select 1 as id

tests/integration/query_tag_tests/models/table_model_query_tag.sql

@@ -0,0 +1,3 @@
+{{ config(materialized = 'table') }}
+
+select 1 as id

tests/integration/query_tag_tests/models/view_model_query_tag.sql

@@ -0,0 +1,3 @@
+{{ config(materialized = 'view') }}
+
+select 1 as id

tests/integration/query_tag_tests/seeds/seed_query_tag.csv

@@ -0,0 +1,2 @@
+id
+1

tests/integration/simple_snapshot_test/check-snapshots-query-tag/snapshot.sql → tests/integration/query_tag_tests/snapshots/snapshot_query_tag.sql

@@ -6,8 +6,7 @@
             unique_key='id',
             strategy='check',
             check_cols=['color'],
-            query_tag=var('query_tag')
         )
     }}
-    select * from {{target.database}}.{{schema}}.seed
+    select 1 as id, 'blue' as color
 {% endsnapshot %}

tests/integration/query_tag_tests/test_query_tags.py

@@ -0,0 +1,46 @@
+import os
+import csv
+from tests.integration.base import DBTIntegrationTest, use_profile
+
+class TestSeedWithQueryTag(DBTIntegrationTest):
+    @property
+    def schema(self):
+        return "query_tag"
+
+    @property
+    def models(self):
+        return "models"
+
+    @property
+    def project_config(self):
+        return {
+            'config-version': 2,
+            'models': {
+                'test': {
+                    'query_tag': self.prefix,
+                    'post-hook': '{{ check_query_tag() }}'
+                },
+            },
+            'seeds': {
+                'test': {
+                    'query_tag': self.prefix,
+                    'post-hook': '{{ check_query_tag() }}'
+                },
+            },
+            'snapshots': {
+                'test': {
+                    'query_tag': self.prefix,
+                    'post-hook': '{{ check_query_tag() }}'
+                },
+            },
+        }
+
+    def build_all_with_query_tags(self):
+        self.run_dbt(['build', '--vars', '{{"query_tag": "{}"}}'.format(self.prefix)])
+
+    @use_profile('snowflake')
+    def test__snowflake__build_tagged_twice(self):
+        self.build_all_with_query_tags()
+        self.build_all_with_query_tags()
+
+