Integration test when excluding columns from union_relations

dbt-labs · May 13, 2022 · 854d334 · 854d334
1 parent 892d111
commit 854d334
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/integration_tests/data/sql/data_union_exclude_expected.csv b/integration_tests/data/sql/data_union_exclude_expected.csv
@@ -0,0 +1,6 @@
+id,favorite_color,favorite_number
+1,,pi
+2,,e
+3,,4
+1,"green",7
+2,"pink",13
diff --git a/integration_tests/models/sql/schema.yml b/integration_tests/models/sql/schema.yml
@@ -163,6 +163,11 @@ models:
         tests:
           - dbt_utils.not_constant
 
+  - name: test_union_exclude
+    tests:
+      - dbt_utils.equality:
+          compare_model: ref('data_union_exclude_expected')
+
   - name: test_get_relations_by_pattern
     tests:
       - dbt_utils.equality:

diff --git a/integration_tests/models/sql/test_union_exclude.sql b/integration_tests/models/sql/test_union_exclude.sql
@@ -0,0 +1,4 @@
+select
+    {{ dbt_utils.star(ref("test_union_exclude_base"), except=["_dbt_source_relation"]) }}
+
+from {{ ref("test_union_exclude_base") }}
diff --git a/integration_tests/models/sql/test_union_exclude_base.sql b/integration_tests/models/sql/test_union_exclude_base.sql
@@ -0,0 +1,8 @@
+
+{{ dbt_utils.union_relations(
+    relations=[
+        ref('data_union_table_1'),
+        ref('data_union_table_2'),
+    ],
+    exclude=['name']
+) }}