Fix post_model_hook in data and unit tests

core/dbt/task/base.py

@@ -6,7 +6,7 @@
 from contextlib import nullcontext
 from datetime import datetime
 from pathlib import Path
-from typing import Any, Dict, List, Optional, Set
+from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set
 
 import dbt.exceptions
 import dbt_common.exceptions.base

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(

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