Asquator commented on code in PR #61274:
URL: https://github.com/apache/airflow/pull/61274#discussion_r2867804484
##########
airflow-core/tests/unit/models/test_dagrun.py:
##########
@@ -433,9 +433,16 @@ def on_success_callable(context):
}
dag_run = self.create_dag_run(dag=dag,
task_states=initial_task_states, session=session)
- with mock.patch.object(dag_run, "handle_dag_callback") as
handle_dag_callback:
+ with mock.patch.object(dag_run, "execute_dag_callbacks") as
execute_dag_callbacks:
_, callback = dag_run.update_state()
- assert handle_dag_callback.mock_calls == [mock.call(dag=dag,
success=True, reason="success")]
+ assert execute_dag_callbacks.mock_calls == [
Review Comment:
Added a test for deadlocked tasks.
As for timeouts, it's tested in `test_scheduler_job.py`. Let me know if I
missed something.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
