dstandish commented on a change in pull request #19904:
URL: https://github.com/apache/airflow/pull/19904#discussion_r759869597
##########
File path: tests/executors/test_kubernetes_executor.py
##########
@@ -687,6 +687,61 @@ def test_pending_pod_timeout_multi_namespace_mode(
)
mock_delete_pod.assert_called_once_with('foo90', 'anothernamespace')
+ def test_clear_not_launched_queued_tasks_not_launched(self, dag_maker,
create_dummy_dag, session):
+ """If a pod isn't found for a TI, reset the state to scheduled"""
+ mock_kube_client = mock.MagicMock()
+ mock_kube_client.list_namespaced_pod.return_value =
k8s.V1PodList(items=[])
+
+ create_dummy_dag(dag_id="test_clear", task_id="task1",
with_dagrun_type=None)
+ dag_run = dag_maker.create_dagrun()
+
+ ti = dag_run.task_instances[0]
+ ti.state = State.QUEUED
+ ti.queued_by_job_id = 1
+ session.flush()
+
+ executor = self.kubernetes_executor
+ executor.kube_client = mock_kube_client
+ executor.clear_not_launched_queued_tasks(session=session)
+
+ ti.refresh_from_db()
+ assert ti.state == State.SCHEDULED
+ assert mock_kube_client.list_namespaced_pod.call_count == 2
+ mock_kube_client.list_namespaced_pod.assert_any_call(
Review comment:
i tried `assert_has_calls` where you can specify the exact call list
(which would be ever so marginally more direct than the approach here but
apparently it does not work with MagicMock only Mock
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
