def test_sync(self, run_task_mock): run_task_mock.return_value = True executor = DebugExecutor() ti1 = MagicMock(key="t1") ti2 = MagicMock(key="t2") executor.tasks_to_run = [ti1, ti2] executor.sync() assert not executor.tasks_to_run run_task_mock.assert_has_calls([mock.call(ti1), mock.call(ti2)])
def test_fail_fast(self, change_state_mock): with mock.patch.dict("os.environ", {"AIRFLOW__DEBUG__FAIL_FAST": "True"}): executor = DebugExecutor() ti1 = MagicMock(key="t1") ti2 = MagicMock(key="t2") ti1._run_raw_task.side_effect = Exception executor.tasks_to_run = [ti1, ti2] executor.sync() assert executor.fail_fast assert not executor.tasks_to_run change_state_mock.assert_has_calls([ mock.call(ti1.key, State.FAILED), mock.call(ti2.key, State.UPSTREAM_FAILED), ])