assert dr.state == DagRunState.RUNNING
        assert dr2.state == DagRunState.RUNNING
        ti2_op1.set_state(state=TaskInstanceState.RUNNING, session=session)
        dr.update_state()
        dr2.update_state()
        assert dr.state == DagRunState.RUNNING
        assert dr2.state == DagRunState.RUNNING
    def test_dagrun_success_callback(self, session):
        def on_success_callable(context):
            assert context["dag_run"].dag_id == "test_dagrun_success_callback"
        dag = DAG(
            dag_id="test_dagrun_success_callback",
            start_date=datetime.datetime(2017, 1, 1),
            on_success_callback=on_success_callable,
        )
        dag_task1 = EmptyOperator(task_id="test_state_succeeded1", dag=dag)
        dag_task2 = EmptyOperator(task_id="test_state_succeeded2", dag=dag)
        dag_task1.set_downstream(dag_task2)
        initial_task_states = {
            "test_state_succeeded1": TaskInstanceState.SUCCESS,
            "test_state_succeeded2": TaskInstanceState.SUCCESS,
        }
        
        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.SUCCESS == dag_run.state
        
        assert callback is None
    def test_dagrun_failure_callback(self, session):