def on_failure_callable(context):
            assert context["dag_run"].dag_id == "test_dagrun_failure_callback"
        dag = DAG(
            dag_id="test_dagrun_failure_callback",
            start_date=datetime.datetime(2017, 1, 1),
            on_failure_callback=on_failure_callable,
        )
        dag_task1 = EmptyOperator(task_id="test_state_succeeded1", dag=dag)
        dag_task2 = EmptyOperator(task_id="test_state_failed2", dag=dag)
        initial_task_states = {
            "test_state_succeeded1": TaskInstanceState.SUCCESS,
            "test_state_failed2": TaskInstanceState.FAILED,
        }
        dag_task1.set_downstream(dag_task2)
        
        dag = SerializedDAG.from_dict(SerializedDAG.to_dict(dag))
        dag_run = self.create_dag_run(dag=dag, task_states=initial_task_states, session=session)
        _, callback = dag_run.update_state()
        assert DagRunState.FAILED == dag_run.state
        
        assert callback is None
    def test_dagrun_update_state_with_handle_callback_success(self, session):