fix: use robust query for relationship tests across dbt versions

Co-Authored-By: Yosef Arbiv <[email protected]>

Author: devin-ai-integration[bot]

integration_tests/tests/test_dbt_artifacts/test_test_owners.py

@@ -175,14 +175,27 @@ def test_relationship_test_uses_primary_model_owner_only(
                 select=f"{primary_model_name} {referenced_model_name}"
             )
 
-            tests = dbt_project.read_table(
+            # Query by parent_model_unique_id and filter by test_original_name in Python
+            # This is more robust across dbt versions (fusion vs latest_official)
+            all_tests = dbt_project.read_table(
                 "dbt_tests",
-                where=f"name LIKE '%relationships%' AND name LIKE '%{primary_model_name}%'",
-                raise_if_empty=True,
+                where=f"parent_model_unique_id LIKE '%{primary_model_name}%'",
+                raise_if_empty=False,
             )
 
-            assert len(tests) == 1, f"Expected 1 relationship test, got {len(tests)}"
-            test_row = tests[0]
+            # Filter for relationship tests
+            relationship_tests = [
+                t
+                for t in all_tests
+                if t.get("test_original_name") == "relationships"
+                or "relationships" in (t.get("short_name") or "").lower()
+                or "relationships" in (t.get("name") or "").lower()
+            ]
+
+            assert (
+                len(relationship_tests) == 1
+            ), f"Expected 1 relationship test, got {len(relationship_tests)}. All tests found: {[t.get('name') for t in all_tests]}"
+            test_row = relationship_tests[0]
             model_owners = _parse_model_owners(test_row.get("model_owners"))
 
             assert model_owners == [
@@ -261,14 +274,27 @@ def test_relationship_test_no_owner_on_primary_model(dbt_project: DbtProject, tm
                 select=f"{primary_model_name} {referenced_model_name}"
             )
 
-            tests = dbt_project.read_table(
+            # Query by parent_model_unique_id and filter by test_original_name in Python
+            # This is more robust across dbt versions (fusion vs latest_official)
+            all_tests = dbt_project.read_table(
                 "dbt_tests",
-                where=f"name LIKE '%relationships%' AND name LIKE '%{primary_model_name}%'",
-                raise_if_empty=True,
+                where=f"parent_model_unique_id LIKE '%{primary_model_name}%'",
+                raise_if_empty=False,
             )
 
-            assert len(tests) == 1, f"Expected 1 relationship test, got {len(tests)}"
-            test_row = tests[0]
+            # Filter for relationship tests
+            relationship_tests = [
+                t
+                for t in all_tests
+                if t.get("test_original_name") == "relationships"
+                or "relationships" in (t.get("short_name") or "").lower()
+                or "relationships" in (t.get("name") or "").lower()
+            ]
+
+            assert (
+                len(relationship_tests) == 1
+            ), f"Expected 1 relationship test, got {len(relationship_tests)}. All tests found: {[t.get('name') for t in all_tests]}"
+            test_row = relationship_tests[0]
             model_owners = _parse_model_owners(test_row.get("model_owners"))
 
             assert (