diff --git a/tests/functional/adapter/hooks/test_run_hooks.py b/tests/functional/adapter/hooks/test_run_hooks.py index ec3fb9f0..0008a95f 100644 --- 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)