Skip to content

Commit 1392035

Browse files
committed
test(dag): assert full DDRQ missing-serialized log line in caplog
1 parent d4b65ba commit 1392035

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

tests/models/test_dag.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3096,7 +3096,10 @@ 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 "not found in the serialized_dag table" in caplog.text
3099+
assert (
3100+
"Dags have queued dataset events (DDRQs), but are not found in the serialized_dag table."
3101+
in caplog.text
3102+
)
31003103
assert orphan_dag_id in caplog.text
31013104
assert (
31023105
session.query(DatasetDagRunQueue)
@@ -3157,7 +3160,12 @@ def test_dags_needing_dagruns_missing_serialized_warning_lists_sorted_dag_ids(se
31573160

31583161
assert "ghost_a" not in dataset_triggered_dag_info
31593162
assert "ghost_z" not in dataset_triggered_dag_info
3160-
msg = next(r.message for r in caplog.records if "not found in the serialized_dag table" in r.message)
3163+
msg = next(
3164+
r.message
3165+
for r in caplog.records
3166+
if "Dags have queued dataset events (DDRQs), but are not found in the serialized_dag table."
3167+
in r.message
3168+
)
31613169
assert msg.index("ghost_a") < msg.index("ghost_z")
31623170
assert (
31633171
session.query(DatasetDagRunQueue)

0 commit comments

Comments
 (0)