Fix test_run_hooks tests

starburstdata · Nov 22, 2024 · 27f9590 · 27f9590
1 parent dee895a
commit 27f9590
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tests/functional/adapter/hooks/test_run_hooks.py b/tests/functional/adapter/hooks/test_run_hooks.py
@@ -1,8 +1,12 @@
 import pytest
-from dbt.tests.adapter.hooks import test_run_hooks as core_base
+from dbt.tests.adapter.hooks.test_run_hooks import (
+    BaseAfterRunHooks,
+    BasePrePostRunHooks,
+)
+from dbt.tests.util import run_dbt
 
 
-class TestPrePostRunHooksTrino(core_base.TestPrePostRunHooks):
+class TestPrePostRunHooksTrino(BasePrePostRunHooks):
     @pytest.fixture(scope="class")
     def project_config_update(self):
         return {
@@ -50,5 +54,6 @@ def check_hooks(self, state, project, host):
         ), "invocation_id was not set"
 
 
-class TestAfterRunHooksTrino(core_base.TestAfterRunHooks):
-    pass
+class TestAfterRunHooksTrino(BaseAfterRunHooks):
+    def test_missing_column_pre_hook(self, project):
+        run_dbt(["run"], expect_pass=False)