add sql_header tests for constraints

.changes/unreleased/Fixes-20230601-120430.yaml

@@ -0,0 +1,6 @@
+kind: Fixes
+body: test model constraints with sql headers
+time: 2023-06-01T12:04:30.876751-04:00
+custom:
+  Author: michelleark
+  Issue: "7714"

dev-requirements.txt

@@ -1,7 +1,7 @@
 # install latest changes in dbt-core
 # TODO: how to automate switching from develop to version branches?
-git+https://github.com/dbt-labs/dbt-core.git#egg=dbt-core&subdirectory=core
-git+https://github.com/dbt-labs/dbt-core.git#egg=dbt-tests-adapter&subdirectory=tests/adapter
+git+https://github.com/dbt-labs/dbt-core.git@7714/model-contracts-sql-header#egg=dbt-core&subdirectory=core
+git+https://github.com/dbt-labs/dbt-core.git@7714/model-contracts-sql-header#egg=dbt-tests-adapter&subdirectory=tests/adapter
 
 # if version 1.x or greater -> pin to major version
 # if version 0.x -> pin to minor

tests/functional/adapter/test_constraints.py

@@ -3,6 +3,8 @@
     BaseTableConstraintsColumnsEqual,
     BaseViewConstraintsColumnsEqual,
     BaseIncrementalConstraintsColumnsEqual,
+    BaseTableContractSqlHeader,
+    BaseIncrementalContractSqlHeader,
     BaseConstraintsRuntimeDdlEnforcement,
     BaseConstraintsRollback,
     BaseIncrementalConstraintsRuntimeDdlEnforcement,
@@ -20,6 +22,7 @@
     my_model_incremental_wrong_name_sql,
     model_schema_yml,
     constrained_model_schema_yml,
+    model_contract_header_schema_yml,
 )
 
 _expected_sql_bigquery = """
@@ -45,6 +48,37 @@
 # - does not support a data type named 'text' (TODO handle this via type translation/aliasing!)
 constraints_yml = model_schema_yml.replace("text", "string")
 model_constraints_yml = constrained_model_schema_yml.replace("text", "string")
+model_contract_header_schema_yml = model_contract_header_schema_yml.replace("text", "string")
+
+
+my_model_contract_sql_header_sql = """
+{{
+  config(
+    materialized = "table"
+  )
+}}
+
+{% call set_sql_header(config) %}
+DECLARE DEMO STRING DEFAULT 'hello world';
+{% endcall %}
+
+SELECT DEMO as column_name
+"""
+
+my_model_incremental_contract_sql_header_sql = """
+{{
+  config(
+    materialized = "incremental",
+    on_schema_change="append_new_columns"
+  )
+}}
+
+{% call set_sql_header(config) %}
+DECLARE DEMO STRING DEFAULT 'hello world';
+{% endcall %}
+
+SELECT DEMO as column_name
+"""
 
 
 class BigQueryColumnEqualSetup:
@@ -113,6 +147,24 @@ def models(self):
         }
 
 
+class TestBigQueryTableContractsSqlHeader(BaseTableContractSqlHeader):
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "my_model_contract_sql_header.sql": my_model_contract_sql_header_sql,
+            "constraints_schema.yml": model_contract_header_schema_yml,
+        }
+
+
+class TestBigQueryIncrementalContractsSqlHeader(BaseIncrementalContractSqlHeader):
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "my_model_contract_sql_header.sql": my_model_incremental_contract_sql_header_sql,
+            "constraints_schema.yml": model_contract_header_schema_yml,
+        }
+
+
 class TestBigQueryTableConstraintsRuntimeDdlEnforcement(BaseConstraintsRuntimeDdlEnforcement):
     @pytest.fixture(scope="class")
     def models(self):