update tests

dbt-labs · May 17, 2023 · b53b58c · b53b58c
1 parent ce26479
commit b53b58c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/tests/functional/adapter/test_constraints.py b/tests/functional/adapter/test_constraints.py
@@ -18,6 +18,10 @@
  my_model_incremental_wrong_order_sql,
  my_model_incremental_wrong_name_sql,
  my_incremental_model_sql,
+ model_fk_constraint_schema_yml,
+ my_model_wrong_order_depends_on_fk_sql,
+ foreign_key_model_sql,
+ my_model_incremental_wrong_order_depends_on_fk_sql,
 )
 
 # constraints are enforced via 'alter' statements that run after table creation
@@ -215,8 +219,9 @@ class TestSparkTableConstraintsDdlEnforcement(
  @pytest.fixture(scope="class")
  def models(self):
  return {
- "my_model.sql": my_model_wrong_order_sql,
- "constraints_schema.yml": constraints_yml,
+ "my_model.sql": my_model_wrong_order_depends_on_fk_sql,
+ "foreign_key_model.sql": foreign_key_model_sql,
+ "constraints_schema.yml": model_fk_constraint_schema_yml,
  }
 
 
@@ -227,8 +232,9 @@ class TestSparkIncrementalConstraintsDdlEnforcement(
  @pytest.fixture(scope="class")
  def models(self):
  return {
- "my_model.sql": my_model_incremental_wrong_order_sql,
- "constraints_schema.yml": constraints_yml,
+ "my_model.sql": my_model_incremental_wrong_order_depends_on_fk_sql,
+ "foreign_key_model.sql": foreign_key_model_sql,
+ "constraints_schema.yml": model_fk_constraint_schema_yml,
  }