Skip to content

Commit eed73ab

Browse files
committed
Added distinct test results test.
1 parent f1ff266 commit eed73ab

File tree

1 file changed

+23
-8
lines changed

1 file changed

+23
-8
lines changed

tests/integration/report/test_report.py

Lines changed: 23 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,14 +28,7 @@ def test_distinct_test_results():
2828
test_result_ids = set()
2929
for test_results in report_data["test_results"].values():
3030
for test_result in test_results:
31-
metadata = test_result["metadata"]
32-
test_result_id = (
33-
metadata["test_unique_id"],
34-
metadata["table_name"],
35-
metadata["column_name"],
36-
metadata["test_type"],
37-
metadata["test_sub_type"],
38-
)
31+
test_result_id = get_test_result_id(test_result["metadata"])
3932
assert test_result_id not in test_result_ids
4033
test_result_ids.add(test_result_id)
4134

@@ -77,6 +70,28 @@ def test_sidebar(report_data_fixture):
7770
)
7871

7972

73+
def get_test_result_id(test_metadata):
74+
test_type_id_map = {
75+
"anomaly_detection": (
76+
test_metadata["model_unique_id"],
77+
test_metadata["column_name"],
78+
test_metadata["test_unique_id"],
79+
test_metadata["test_sub_type"],
80+
),
81+
"schema_change": (
82+
test_metadata["model_unique_id"],
83+
test_metadata["test_unique_id"],
84+
),
85+
"dbt_test": (
86+
test_metadata["model_unique_id"],
87+
test_metadata["column_name"],
88+
test_metadata["test_unique_id"],
89+
test_metadata["test_sub_type"],
90+
),
91+
}
92+
return test_type_id_map[test_metadata["test_type"]]
93+
94+
8095
def assert_totals(data_totals: Totals, fixture_totals: Totals):
8196
assert data_totals.keys() == fixture_totals.keys()
8297
for total_key in fixture_totals:

0 commit comments

Comments
 (0)