From 7917815f4c8163b309963d7d611c68c4228419e6 Mon Sep 17 00:00:00 2001 From: Damian Owsianny Date: Fri, 22 Nov 2024 15:23:53 +0100 Subject: [PATCH] Fix test_run_hooks tests --- tests/functional/adapter/hooks/test_run_hooks.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) 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)