diff --git a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py index 632f5faa57ffa..0d6f353310b4b 100644 --- a/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py +++ b/providers/cncf/kubernetes/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py @@ -19,7 +19,6 @@ from __future__ import annotations import enum -import itertools import json import math import time @@ -119,9 +118,12 @@ def get_xcom_sidecar_container_resources(self) -> str | None: def get_container_status(pod: V1Pod, container_name: str) -> V1ContainerStatus | None: """Retrieve container status.""" if pod and pod.status: - container_statuses = itertools.chain( - pod.status.container_statuses, pod.status.init_container_statuses - ) + container_statuses = [] + if pod.status.container_statuses: + container_statuses.extend(pod.status.container_statuses) + if pod.status.init_container_statuses: + container_statuses.extend(pod.status.init_container_statuses) + else: container_statuses = None diff --git a/providers/cncf/kubernetes/tests/unit/cncf/kubernetes/utils/test_pod_manager.py b/providers/cncf/kubernetes/tests/unit/cncf/kubernetes/utils/test_pod_manager.py index 0ba6f0e1bcf0a..320470cf48a8f 100644 --- a/providers/cncf/kubernetes/tests/unit/cncf/kubernetes/utils/test_pod_manager.py +++ b/providers/cncf/kubernetes/tests/unit/cncf/kubernetes/utils/test_pod_manager.py @@ -888,6 +888,12 @@ def container(name, succeeded): pod_mock_list.append(pytest.param(p, False, id="None remote_pod.status")) p = RemotePodMock() p.status = RemotePodMock() + p.status.container_statuses = None + p.status.init_container_statuses = [] + + pod_mock_list.append(pytest.param(p, False, id="None remote_pod.status.container_statuses")) + p = RemotePodMock() + p.status = RemotePodMock() p.status.container_statuses = [] p.status.init_container_statuses = [] pod_mock_list.append(pytest.param(p, False, id="empty remote_pod.status.container_statuses"))