def test_unfinished_dag_runs_set_to_failed(self): dag = self._get_dummy_dag('dummy_dag') dag_run = dag.create_dagrun( run_id='test', state=State.RUNNING, ) job = BackfillJob(dag=dag, start_date=DEFAULT_DATE, end_date=DEFAULT_DATE + datetime.timedelta(days=8), ignore_first_depends_on_past=True) job._set_unfinished_dag_runs_to_failed([dag_run]) dag_run.refresh_from_db() self.assertEqual(State.FAILED, dag_run.state)
def test_dag_run_with_finished_tasks_set_to_success(self): dag = self._get_dummy_dag('dummy_dag') dag_run = dag.create_dagrun( run_id='test', state=State.RUNNING, ) for ti in dag_run.get_task_instances(): ti.set_state(State.SUCCESS) job = BackfillJob(dag=dag, start_date=DEFAULT_DATE, end_date=DEFAULT_DATE + datetime.timedelta(days=8), ignore_first_depends_on_past=True) job._set_unfinished_dag_runs_to_failed([dag_run]) dag_run.refresh_from_db() self.assertEqual(State.SUCCESS, dag_run.state)