Mcknight/fix test store test (#832) (#920)

* add teardown_method for store_test_fail test * create pr * add teardown to delta test class as it also builds tests on alt schema (cherry picked from commit 99c5c5a) Co-authored-by: Matthew McKnight <[email protected]>
dbt-labs · Oct 24, 2023 · ab94c89 · ab94c89
1 parent 1ce809c
commit ab94c89
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tests/functional/adapter/store_test_failures_tests/test_store_test_failures.py b/tests/functional/adapter/store_test_failures_tests/test_store_test_failures.py
@@ -17,6 +17,17 @@ def project_config_update(self):
             "tests": {"+schema": TEST_AUDIT_SCHEMA_SUFFIX, "+store_failures": True},
         }
 
+    @pytest.fixture(scope="function", autouse=True)
+    def teardown_method(self, project):
+        yield
+        with project.adapter.connection_named("__test"):
+            relation = project.adapter.Relation.create(
+                database=project.database,
+                schema=f"{project.test_schema}_{TEST_AUDIT_SCHEMA_SUFFIX}",
+            )
+
+            project.adapter.drop_schema(relation)
+
     def test_store_and_assert(self, project):
         self.run_tests_store_one_failure(project)
         self.run_tests_store_failures_and_assert(project)
@@ -39,6 +50,17 @@ def project_config_update(self):
             },
         }
 
+    @pytest.fixture(scope="function", autouse=True)
+    def teardown_method(self, project):
+        yield
+        with project.adapter.connection_named("__test"):
+            relation = project.adapter.Relation.create(
+                database=project.database,
+                schema=f"{project.test_schema}_{TEST_AUDIT_SCHEMA_SUFFIX}",
+            )
+
+            project.adapter.drop_schema(relation)
+
     def test_store_and_assert_failure_with_delta(self, project):
         self.run_tests_store_one_failure(project)
         self.run_tests_store_failures_and_assert(project)