Skip to content

Commit d4b65ba

Browse files
committed
test(dag): align DDRQ missing-serialized log assertions with dag.py
dags_needing_dagruns now logs that dags are not found in serialized_dag instead of the older DDRQ/SerializedDagModel wording.
1 parent efd3111 commit d4b65ba

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

tests/models/test_dag.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3096,7 +3096,7 @@ def test_dags_needing_dagruns_skips_ddrq_when_serialized_dag_missing(self, sessi
30963096
_query, dataset_triggered_dag_info = DagModel.dags_needing_dagruns(session)
30973097

30983098
assert orphan_dag_id not in dataset_triggered_dag_info
3099-
assert "DAGs in DDRQ but missing SerializedDagModel" in caplog.text
3099+
assert "not found in the serialized_dag table" in caplog.text
31003100
assert orphan_dag_id in caplog.text
31013101
assert (
31023102
session.query(DatasetDagRunQueue)
@@ -3157,9 +3157,7 @@ def test_dags_needing_dagruns_missing_serialized_warning_lists_sorted_dag_ids(se
31573157

31583158
assert "ghost_a" not in dataset_triggered_dag_info
31593159
assert "ghost_z" not in dataset_triggered_dag_info
3160-
msg = next(
3161-
r.message for r in caplog.records if "DAGs in DDRQ but missing SerializedDagModel" in r.message
3162-
)
3160+
msg = next(r.message for r in caplog.records if "not found in the serialized_dag table" in r.message)
31633161
assert msg.index("ghost_a") < msg.index("ghost_z")
31643162
assert (
31653163
session.query(DatasetDagRunQueue)

0 commit comments

Comments
 (0)