def test_end(self):
ti = MagicMock(key="ti_key")
executor = DebugExecutor()
executor.tasks_to_run = [ti]
executor.running = {ti.key}
executor.end()
ti.set_state.assert_called_once_with(State.UPSTREAM_FAILED)
assert not executor.running
def test_run_task(self, task_instance_mock):
ti_key = "key"
job_id = " job_id"
task_instance_mock.key = ti_key
task_instance_mock.job_id = job_id
executor = DebugExecutor()
executor.running = {ti_key}
succeeded = executor._run_task(task_instance_mock)
assert succeeded
task_instance_mock._run_raw_task.assert_called_once_with(job_id=job_id)