Fix unit test parsing to ensure external nodes continue to keep their…

… package name
dbt-labs · Jan 12, 2024 · 7fa5e5c · 7fa5e5c
1 parent ec0ed32
commit 7fa5e5c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/core/dbt/parser/unit_tests.py b/core/dbt/parser/unit_tests.py
@@ -118,7 +118,6 @@ def parse_unit_test_case(self, test_case: UnitTestDefinition):
 
             common_fields = {
                 "resource_type": NodeType.Model,
-                "package_name": test_case.package_name,
                 "original_file_path": original_input_node.original_file_path,
                 "config": ModelConfig(materialized="ephemeral"),
                 "database": original_input_node.database,
@@ -137,7 +136,8 @@ def parse_unit_test_case(self, test_case: UnitTestDefinition):
                 input_name = f"{unit_test_node.name}__{original_input_node.name}"
                 input_node = ModelNode(
                     **common_fields,
-                    unique_id=f"model.{test_case.package_name}.{input_name}",
+                    package_name=original_input_node.package_name,
+                    unique_id=f"model.{original_input_node.package_name}.{input_name}",
                     name=input_name,
                     path=original_input_node.path,
                 )
@@ -148,7 +148,8 @@ def parse_unit_test_case(self, test_case: UnitTestDefinition):
                 input_name = f"{unit_test_node.name}__{original_input_node.search_name}__{original_input_node.name}"
                 input_node = UnitTestSourceDefinition(
                     **common_fields,
-                    unique_id=f"model.{test_case.package_name}.{input_name}",
+                    package_name=original_input_node.package_name,
+                    unique_id=f"model.{original_input_node.package_name}.{input_name}",
                     name=original_input_node.name,  # must be the same name for source lookup to work
                     path=input_name + ".sql",  # for writing out compiled_code
                     source_name=original_input_node.source_name,  # needed for source lookup