def on_success_callable(context):
            assert context["dag_run"].dag_id == "test_dagrun_update_state_with_handle_callback_success"
        dag = DAG(
            dag_id="test_dagrun_update_state_with_handle_callback_success",
            start_date=datetime.datetime(2017, 1, 1),
            on_success_callback=on_success_callable,
        )
        DAG.bulk_write_to_db(dags=[dag], processor_subdir="/tmp/test", session=session)
        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(execute_callbacks=False)
        assert DagRunState.SUCCESS == dag_run.state
        
        assert callback == DagCallbackRequest(
            full_filepath=dag_run.dag.fileloc,
            dag_id="test_dagrun_update_state_with_handle_callback_success",
            run_id=dag_run.run_id,
            is_failure_callback=False,
            processor_subdir="/tmp/test",
            msg="success",
        )
    def test_dagrun_update_state_with_handle_callback_failure(self, session):