From 1d299923e34c96f2e96a5215ac196658f86ce1d1 Mon Sep 17 00:00:00 2001
From: Mike Alfare <13974384+mikealfare@users.noreply.github.com>
Date: Tue, 29 Oct 2024 12:35:37 -0400
Subject: [PATCH] Add test cases for generic tests config (#1172)

* add test cases for generic tests config

* update tests

---------

Co-authored-by: Colin Rogers <111200756+colin-rogers-dbt@users.noreply.github.com>
Co-authored-by: Colin <colin.rogers@dbtlabs.com>
---
 .../functional/generic_test_tests/__init__.py |  0
 tests/functional/generic_test_tests/_files.py | 90 +++++++++++++++++++
 .../functional/generic_test_tests/_models.py  |  0
 .../functional/generic_test_tests/_schemas.py |  0
 .../generic_test_tests/test_generic_tests.py  | 54 +++++++++++
 5 files changed, 144 insertions(+)
 create mode 100644 tests/functional/generic_test_tests/__init__.py
 create mode 100644 tests/functional/generic_test_tests/_files.py
 create mode 100644 tests/functional/generic_test_tests/_models.py
 create mode 100644 tests/functional/generic_test_tests/_schemas.py
 create mode 100644 tests/functional/generic_test_tests/test_generic_tests.py

diff --git a/tests/functional/generic_test_tests/__init__.py b/tests/functional/generic_test_tests/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/tests/functional/generic_test_tests/_files.py b/tests/functional/generic_test_tests/_files.py
new file mode 100644
index 000000000..a9743e43e
--- /dev/null
+++ b/tests/functional/generic_test_tests/_files.py
@@ -0,0 +1,90 @@
+SCHEMA__CONTROL = """
+version: 2
+models:
+  - name: colors
+    columns:
+      - name: color
+        data_tests:
+          - not_null
+"""
+
+
+SCHEMA__EXPLICIT_WAREHOUSE = """
+version: 2
+models:
+  - name: colors
+    columns:
+      - name: color
+        data_tests:
+          - not_null:
+              config:
+                snowflake_warehouse: DBT_TESTING_ALT
+"""
+
+
+SCHEMA__NOT_NULL = """
+version: 2
+models:
+  - name: facts
+    columns:
+      - name: value
+        data_tests:
+          - not_null:
+              config:
+                snowflake_warehouse: DBT_TESTING_ALT
+"""
+
+
+SCHEMA__RELATIONSHIPS = """
+version: 2
+models:
+  - name: facts
+    columns:
+      - name: color
+        data_tests:
+          - relationships:
+              to: ref('my_colors')
+              field: color
+              config:
+                snowflake_warehouse: DBT_TESTING_ALT
+"""
+
+
+SCHEMA__ACCEPTED_VALUES = """
+version: 2
+models:
+  - name: colors
+    columns:
+      - name: color
+        data_tests:
+          - accepted_values:
+              values: ['blue', 'red', 'green']
+              config:
+                snowflake_warehouse: DBT_TESTING_ALT
+"""
+
+
+SEED__COLORS = """
+color
+blue
+green
+red
+yellow
+""".strip()
+
+
+# record 10 is missing a value
+# record 7 has a color that's not on COLORS
+SEED__FACTS = """
+id,color,value
+1,blue,10
+2,red,20
+3,green,30
+4,yellow,40
+5,blue,50
+6,red,60
+7,orange,70
+8,green,80
+9,yellow,90
+10,blue,
+""".strip()
diff --git a/tests/functional/generic_test_tests/_models.py b/tests/functional/generic_test_tests/_models.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/tests/functional/generic_test_tests/_schemas.py b/tests/functional/generic_test_tests/_schemas.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/tests/functional/generic_test_tests/test_generic_tests.py b/tests/functional/generic_test_tests/test_generic_tests.py
new file mode 100644
index 000000000..a653a363b
--- /dev/null
+++ b/tests/functional/generic_test_tests/test_generic_tests.py
@@ -0,0 +1,54 @@
+import pytest
+
+from dbt.tests.util import run_dbt, run_dbt_and_capture
+
+from tests.functional.generic_test_tests import _files
+
+
+class TestWarehouseConfig:
+
+    @pytest.fixture(scope="class")
+    def seeds(self):
+        return {
+            "colors.csv": _files.SEED__COLORS,
+            "facts.csv": _files.SEED__FACTS,
+        }
+
+    @pytest.fixture(scope="class", autouse=True)
+    def setup(self, project):
+        run_dbt(["seed"])
+        run_dbt(["run"])
+        yield
+
+
+class TestWarehouseConfigControl(TestWarehouseConfig):
+
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {"schema.yml": _files.SCHEMA__CONTROL}
+
+    def test_expected_warehouse(self, project):
+        results, logs = run_dbt_and_capture(["test"])
+        assert len(results) == 1
+
+
+class TestWarehouseConfigExplicitWarehouse(TestWarehouseConfig):
+
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {"schema.yml": _files.SCHEMA__EXPLICIT_WAREHOUSE}
+
+    def test_expected_warehouse(self, project):
+        _, logs = run_dbt_and_capture(["test", "--log-level", "debug"])
+        assert "use warehouse " in logs
+
+
+class TestWarehouseConfigNotNull(TestWarehouseConfig):
+
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {"schema.yml": _files.SCHEMA__NOT_NULL}
+
+    def test_expected_warehouse(self, project):
+        _, logs = run_dbt_and_capture(["test", "--log-level", "debug"], expect_pass=False)
+        assert "use warehouse " in logs