Fix post_model_hook in data and unit tests

dbt-labs · Oct 2, 2024 · 4155bad · 4155bad
1 parent 7940ad5
commit 4155bad
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/core/dbt/task/run.py b/core/dbt/task/run.py
@@ -427,7 +427,7 @@ def _execute_microbatch_model(
         else:
             return self._build_run_model_result(model, context)
 
-    def execute(self, model, manifest):
+    def execute(self, model: ModelNode, manifest: Manifest):
         context = generate_runtime_model_context(model, self.config, manifest)
 
         materialization_macro = manifest.find_materialization_macro_by_name(

diff --git a/core/dbt/task/test.py b/core/dbt/task/test.py
@@ -159,13 +159,15 @@ def execute_data_test(self, data_test: TestNode, manifest: Manifest) -> TestResu
                 "Invalid materialization context generated, missing config: {}".format(context)
             )
 
+        context_config = context["config"]
+
         # generate materialization macro
         macro_func = MacroGenerator(materialization_macro, context)
         try:
             # execute materialization macro
             macro_func()
         finally:
-            self.adapter.post_model_hook(context, hook_ctx)
+            self.adapter.post_model_hook(context_config, hook_ctx)
 
         # load results from context
         # could eventually be returned directly by materialization
@@ -238,6 +240,8 @@ def execute_unit_test(
                 "Invalid materialization context generated, missing config: {}".format(context)
             )
 
+        context_config = context["config"]
+
         # generate materialization macro
         macro_func = MacroGenerator(materialization_macro, context)
         try:
@@ -249,7 +253,7 @@ def execute_unit_test(
                 f"There may be an error in the unit test definition: check the data types.\n {e}"
             )
         finally:
-            self.adapter.post_model_hook(context, hook_ctx)
+            self.adapter.post_model_hook(context_config, hook_ctx)
 
         # load results from context
         # could eventually be returned directly by materialization