# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
from __future__ import annotations

import logging
from datetime import datetime, timedelta, timezone as dt_timezone
from json.decoder import JSONDecodeError
from typing import TYPE_CHECKING, cast
from unittest import mock
from unittest.mock import MagicMock, PropertyMock, patch

import pendulum
import pytest
import time_machine
from kubernetes.client.rest import ApiException
from urllib3.exceptions import HTTPError as BaseHTTPError

from airflow.providers.cncf.kubernetes.exceptions import KubernetesApiError
from airflow.providers.cncf.kubernetes.utils.pod_manager import (
    AsyncPodManager,
    PodLogsConsumer,
    PodManager,
    PodPhase,
    XComRetrievalError,
    _parse_log_level,
    detect_pod_terminate_early_issues,
    log_pod_event,
    parse_log_line,
)
from airflow.providers.common.compat.sdk import AirflowException, timezone

from unit.cncf.kubernetes.test_callbacks import MockKubernetesPodOperatorCallback, MockWrapper

if TYPE_CHECKING:
    from pendulum import DateTime


@pytest.fixture
def pod_factory():
    def _make(
        *,
        pod_phase: str = PodPhase.RUNNING,
        container_name: str = "base",
        terminated: bool = False,
        waiting_reason: str | None = None,
        waiting_message: str | None = None,
    ) -> mock.MagicMock:
        pod = mock.MagicMock()
        pod.status.phase = pod_phase
        cs = mock.MagicMock()
        cs.name = container_name
        cs.state = mock.MagicMock()
        cs.state.terminated = mock.MagicMock(finished_at=pendulum.now()) if terminated else None
        cs.state.waiting = (
            mock.MagicMock(reason=waiting_reason, message=waiting_message or "") if waiting_reason else None
        )
        pod.status.container_statuses = [cs]
        c_spec = mock.MagicMock(name=container_name)
        pod.spec.containers = [c_spec]
        return pod

    return _make


def test_parse_log_line():
    log_message = "This should return no timestamp"
    timestamp, line = parse_log_line(log_message)
    assert timestamp is None
    assert line == log_message

    real_timestamp = "2020-10-08T14:16:17.793417674Z"
    timestamp, line = parse_log_line(f"{real_timestamp} {log_message}")
    assert timestamp == pendulum.parse(real_timestamp)
    assert line == log_message


def test_log_pod_event():
    """Test logging a pod event."""
    mock_pod_manager = mock.Mock()
    mock_event = mock.Mock()
    mock_event.metadata.uid = "event-uid-1"
    mock_event.message = "Test event message"
    mock_event.involved_object.field_path = "Test field path"

    seen_events = set()

    log_pod_event(mock_pod_manager, mock_event, seen_events)

    assert "event-uid-1" in seen_events
    mock_pod_manager.log.info.assert_called_once_with(
        "The Pod has an Event: %s from %s", "Test event message", "Test field path"
    )


def test_log_pod_event_skips_duplicate():
    """Test that duplicate events are skipped."""
    mock_pod_manager = mock.Mock()
    mock_event = mock.Mock()
    mock_event.metadata.uid = "event-uid-1"
    mock_event.message = "Test event message"

    seen_events = {"event-uid-1"}  # Event already seen

    log_pod_event(mock_pod_manager, mock_event, seen_events)

    assert "event-uid-1" in seen_events
    mock_pod_manager.log.info.assert_not_called()


def test_log_pod_event_multiple_events():
    """Test logging multiple different events."""
    mock_pod_manager = mock.Mock()
    seen_events = set()

    # First event
    mock_event1 = mock.Mock()
    mock_event1.metadata.uid = "event-uid-1"
    mock_event1.message = "First message"
    mock_event1.involved_object.field_path = "Test field path 1"

    log_pod_event(mock_pod_manager, mock_event1, seen_events)
    assert "event-uid-1" in seen_events

    # Second event
    mock_event2 = mock.Mock()
    mock_event2.metadata.uid = "event-uid-2"
    mock_event2.message = "Second message"
    mock_event2.involved_object.field_path = "Test field path 2"

    log_pod_event(mock_pod_manager, mock_event2, seen_events)
    assert "event-uid-2" in seen_events
    assert len(seen_events) == 2
    assert mock_pod_manager.log.info.call_count == 2


def _pod_with_waiting_container(reason: str, message: str):
    pod = mock.MagicMock()
    container_status = mock.MagicMock()
    waiting_state = mock.MagicMock()
    waiting_state.reason = reason
    waiting_state.message = message
    container_status.state.waiting = waiting_state
    pod.status.container_statuses = [container_status]
    return pod


@pytest.mark.parametrize(
    ("reason", "message"),
    [
        ("ErrImagePull", "rpc error: 502 Bad Gateway from auth.docker.io"),
        ("ErrImagePull", "registry returned 503 Service Unavailable"),
        ("ErrImagePull", "got 504 Gateway Timeout"),
        # kubelet >= 1.32 appends the previous pull error to the
        # ImagePullBackOff message, so the 5xx patterns still match.
        ("ImagePullBackOff", 'Back-off pulling image "ubuntu:latest": 502 Bad Gateway'),
        ("ErrImagePull", "too many requests"),
        ("ErrImagePull", "pull QPS exceeded"),
    ],
)
def test_detect_pod_terminate_early_issues_returns_none_for_transient_messages(reason, message):
    """Transient registry/network errors should not trigger early termination.

    kubelet retries image pulls automatically; `startup_timeout` bounds the
    total wait. Returning None here keeps the monitoring loop going so those
    retries have a chance to succeed.
    """
    pod = _pod_with_waiting_container(reason, message)
    assert detect_pod_terminate_early_issues(pod) is None


@pytest.mark.parametrize(
    ("reason", "message"),
    [
        ("InvalidImageName", "Failed to apply default image tag"),
        ("ErrImageNeverPull", "Container image pull policy is Never"),
        ("ErrImagePull", "manifest unknown"),
        ("ImagePullBackOff", "unauthorized: authentication required"),
        # Bare ImagePullBackOff message (kubelet < 1.32) — must fail fast so
        # a genuinely missing image doesn't wait out startup_timeout.
        ("ImagePullBackOff", 'Back-off pulling image "nonexistent:latest"'),
    ],
)
def test_detect_pod_terminate_early_issues_returns_error_for_fatal_messages(reason, message):
    pod = _pod_with_waiting_container(reason, message)
    error = detect_pod_terminate_early_issues(pod)
    assert error is not None
    assert reason in error
    assert message in error


class TestPodManager:
    def setup_method(self):
        self.mock_kube_client = mock.Mock()
        self.pod_manager = PodManager(
            kube_client=self.mock_kube_client,
            callbacks=[MockKubernetesPodOperatorCallback],
        )

    @pytest.mark.parametrize(
        ("message", "expected_level"),
        [
            ("ERROR: something went wrong", logging.ERROR),
            ("WARNING: low disk space", logging.WARNING),
            ("WARN: deprecated usage", logging.WARNING),
            ("DEBUG: entering function", logging.DEBUG),
            ("CRITICAL: system failure", logging.CRITICAL),
            ("FATAL: unrecoverable error", logging.CRITICAL),
            ("INFO: starting up", logging.INFO),
            ("[ERROR] bracketed prefix", logging.ERROR),
            ("plain log line with no level", logging.INFO),
            ("", logging.INFO),
        ],
    )
    def test_parse_log_level(self, message, expected_level):
        assert _parse_log_level(message) == expected_level

    def test_log_message_uses_detected_log_level(self):
        """_log_message should forward ERROR lines at ERROR level, not INFO."""
        with mock.patch.object(self.pod_manager.log, "log") as mock_log:
            self.pod_manager._log_message(
                message="ERROR: something failed",
                container_name="base",
                container_name_log_prefix_enabled=True,
                log_formatter=None,
            )
        mock_log.assert_called_once()
        assert mock_log.call_args[0][0] == logging.ERROR

    def test_log_message_defaults_to_info_for_plain_lines(self):
        """_log_message should use INFO for lines without a known level prefix."""
        with mock.patch.object(self.pod_manager.log, "log") as mock_log:
            self.pod_manager._log_message(
                message="Starting application",
                container_name="base",
                container_name_log_prefix_enabled=True,
                log_formatter=None,
            )
        mock_log.assert_called_once()
        assert mock_log.call_args[0][0] == logging.INFO

    def test_read_pod_logs_successfully_returns_logs(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.sentinel.logs
        logs = self.pod_manager.read_pod_logs(pod=mock.sentinel, container_name="base")
        assert isinstance(logs, PodLogsConsumer)
        assert logs.response == mock.sentinel.logs

    def test_read_pod_logs_retries_successfully(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod_log.side_effect = [
            BaseHTTPError("Boom"),
            mock.sentinel.logs,
        ]
        logs = self.pod_manager.read_pod_logs(pod=mock.sentinel, container_name="base")
        assert isinstance(logs, PodLogsConsumer)
        assert mock.sentinel.logs == logs.response
        self.mock_kube_client.read_namespaced_pod_log.assert_has_calls(
            [
                mock.call(
                    _preload_content=False,
                    container="base",
                    follow=True,
                    timestamps=False,
                    name=mock.sentinel.metadata.name,
                    namespace=mock.sentinel.metadata.namespace,
                ),
                mock.call(
                    _preload_content=False,
                    container="base",
                    follow=True,
                    timestamps=False,
                    name=mock.sentinel.metadata.name,
                    namespace=mock.sentinel.metadata.namespace,
                ),
            ]
        )

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    def test_fetch_container_logs_do_not_log_none(self, mock_container_is_running, caplog):
        MockWrapper.reset()
        caplog.set_level(logging.INFO)

        def consumer_iter():
            """This will simulate a container that hasn't produced any logs in the last read_timeout window"""
            yield from ()

        with mock.patch.object(PodLogsConsumer, "__iter__") as mock_consumer_iter:
            mock_consumer_iter.side_effect = consumer_iter
            mock_container_is_running.side_effect = [True, True, False]
            self.pod_manager.fetch_container_logs(mock.MagicMock(), "container-name", follow=True)
            assert "[container-name] None" not in (record.message for record in caplog.records)

    def test_read_pod_logs_retries_fails(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod_log.side_effect = [
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
        ]
        with pytest.raises(BaseHTTPError):
            self.pod_manager.read_pod_logs(pod=mock.sentinel, container_name="base")

    def test_read_pod_logs_successfully_with_tail_lines(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod_log.side_effect = [mock.sentinel.logs]
        logs = self.pod_manager.read_pod_logs(pod=mock.sentinel, container_name="base", tail_lines=100)
        assert isinstance(logs, PodLogsConsumer)
        assert mock.sentinel.logs == logs.response
        self.mock_kube_client.read_namespaced_pod_log.assert_has_calls(
            [
                mock.call(
                    _preload_content=False,
                    container="base",
                    follow=True,
                    timestamps=False,
                    name=mock.sentinel.metadata.name,
                    namespace=mock.sentinel.metadata.namespace,
                    tail_lines=100,
                ),
            ]
        )

    def test_read_pod_logs_successfully_with_since_seconds(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod_log.side_effect = [mock.sentinel.logs]
        logs = self.pod_manager.read_pod_logs(mock.sentinel, "base", since_seconds=2)
        assert isinstance(logs, PodLogsConsumer)
        assert mock.sentinel.logs == logs.response
        self.mock_kube_client.read_namespaced_pod_log.assert_has_calls(
            [
                mock.call(
                    _preload_content=False,
                    container="base",
                    follow=True,
                    timestamps=False,
                    name=mock.sentinel.metadata.name,
                    namespace=mock.sentinel.metadata.namespace,
                    since_seconds=2,
                ),
            ]
        )

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_watch_pod_events(self, mock_time_sleep):
        events = mock.MagicMock()
        events.items = []
        for id in ["event 1", "event 2"]:
            event = mock.MagicMock()
            event.message = f"test {id}"
            event.involved_object.field_path = f"object {id}"
            events.items.append(event)
        startup_check_interval = 10

        def mock_read_pod_events(*_, **__):
            self.pod_manager.stop_watching_events = True
            return events

        with (
            mock.patch.object(self.pod_manager, "read_pod_events", side_effect=mock_read_pod_events),
            mock.patch(
                "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.log",
                new_callable=PropertyMock,
            ) as log_mock,
        ):
            await self.pod_manager.watch_pod_events(pod=None, check_interval=startup_check_interval)

            log_mock.return_value.info.assert_any_call(
                "The Pod has an Event: %s from %s", "test event 1", "object event 1"
            )
            log_mock.return_value.info.assert_any_call(
                "The Pod has an Event: %s from %s", "test event 2", "object event 2"
            )
            mock_time_sleep.assert_called_once_with(startup_check_interval)

    def test_read_pod_events_successfully_returns_events(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.list_namespaced_event.return_value = mock.sentinel.events
        events = self.pod_manager.read_pod_events(mock.sentinel)
        assert mock.sentinel.events == events

    def test_read_pod_events_with_resource_version(self):
        """Test reading pod events with resource_version parameter."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        mock_events = mock.Mock()
        self.mock_kube_client.list_namespaced_event.return_value = mock_events

        events = self.pod_manager.read_pod_events(mock_pod, resource_version="12345")

        assert events == mock_events
        self.mock_kube_client.list_namespaced_event.assert_called_once_with(
            namespace="test-namespace",
            field_selector="involvedObject.name=test-pod",
            resource_version="12345",
            resource_version_match="NotOlderThan",
        )

    def test_read_pod_events_without_resource_version(self):
        """Test reading pod events without resource_version parameter."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        mock_events = mock.Mock()
        self.mock_kube_client.list_namespaced_event.return_value = mock_events

        events = self.pod_manager.read_pod_events(mock_pod)

        assert events == mock_events
        self.mock_kube_client.list_namespaced_event.assert_called_once_with(
            namespace="test-namespace",
            field_selector="involvedObject.name=test-pod",
            resource_version=None,
            resource_version_match=None,
        )

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_watch_pod_events_tracks_resource_version(self, mock_sleep):
        """Test that watch_pod_events tracks resource version."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        mock_event_1 = mock.Mock()
        mock_event_1.metadata.uid = "event-uid-1"
        mock_event_1.metadata.resource_version = "100"
        mock_event_1.message = "Event 1"
        mock_event_1.involved_object.field_path = "spec"

        mock_events_1 = mock.Mock()
        mock_events_1.items = [mock_event_1]

        mock_event_2 = mock.Mock()
        mock_event_2.metadata.uid = "event-uid-2"
        mock_event_2.metadata.resource_version = "101"
        mock_event_2.message = "Event 2"
        mock_event_2.involved_object.field_path = "spec"

        mock_events_2 = mock.Mock()
        mock_events_2.items = [mock_event_2]

        self.mock_kube_client.list_namespaced_event.side_effect = [mock_events_1, mock_events_2]
        self.pod_manager.stop_watching_events = False

        call_count = 0

        async def side_effect_sleep(*_, **__):
            nonlocal call_count
            call_count += 1
            if call_count >= 2:
                self.pod_manager.stop_watching_events = True

        mock_sleep.side_effect = side_effect_sleep

        await self.pod_manager.watch_pod_events(mock_pod, check_interval=1)

        # Check that resource_version was passed in second call
        calls = self.mock_kube_client.list_namespaced_event.call_args_list
        assert len(calls) == 2
        # First call should have no resource_version
        assert calls[0][1]["resource_version"] is None
        # Second call should use resource_version from first event
        assert calls[1][1]["resource_version"] == "100"

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_watch_pod_events_tracks_resource_version_when_first_response_has_no_events(
        self, mock_sleep
    ):
        """Test that watch_pod_events uses list metadata resource version when no events are returned."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        mock_events_1 = mock.Mock()
        mock_events_1.items = []
        mock_events_1.metadata = mock.Mock(resource_version="100")

        mock_event_2 = mock.Mock()
        mock_event_2.metadata.uid = "event-uid-2"
        mock_event_2.metadata.resource_version = "101"
        mock_event_2.message = "Event 2"
        mock_event_2.involved_object.field_path = "spec"

        mock_events_2 = mock.Mock()
        mock_events_2.items = [mock_event_2]
        mock_events_2.metadata = mock.Mock(resource_version="101")

        self.mock_kube_client.list_namespaced_event.side_effect = [mock_events_1, mock_events_2]
        self.pod_manager.stop_watching_events = False

        call_count = 0

        async def side_effect_sleep(*_, **__):
            nonlocal call_count
            call_count += 1
            if call_count >= 2:
                self.pod_manager.stop_watching_events = True

        mock_sleep.side_effect = side_effect_sleep

        await self.pod_manager.watch_pod_events(mock_pod, check_interval=1)

        calls = self.mock_kube_client.list_namespaced_event.call_args_list
        assert len(calls) == 2
        assert calls[0][1]["resource_version"] is None
        assert calls[1][1]["resource_version"] == "100"
        assert calls[1][1]["resource_version_match"] == "NotOlderThan"

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_watch_pod_events_deduplicates_events(self, mock_sleep):
        """Test that watch_pod_events deduplicates events."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        mock_event = mock.Mock()
        mock_event.metadata.uid = "event-uid-1"
        mock_event.metadata.resource_version = "100"
        mock_event.message = "Duplicate event"
        mock_event.involved_object.field_path = "spec"

        mock_events = mock.Mock()
        mock_events.items = [mock_event]

        # Will return the same event on each invocation
        self.mock_kube_client.list_namespaced_event.return_value = mock_events
        self.pod_manager.stop_watching_events = False

        call_count = 0

        async def side_effect_sleep(*_, **__):
            nonlocal call_count
            call_count += 1
            if call_count >= 2:
                # Stop after 2 iterations -> same event is returned 2 times
                self.pod_manager.stop_watching_events = True

        mock_sleep.side_effect = side_effect_sleep

        with mock.patch.object(self.pod_manager.log, "info") as mock_log_info:
            await self.pod_manager.watch_pod_events(mock_pod, check_interval=1)

            # Event should only be logged once despite being returned twice
            assert mock_log_info.call_count == 1
            mock_log_info.assert_called_with("The Pod has an Event: %s from %s", "Duplicate event", "spec")

    def test_read_pod_events_retries_successfully(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.list_namespaced_event.side_effect = [
            BaseHTTPError("Boom"),
            mock.sentinel.events,
        ]
        events = self.pod_manager.read_pod_events(mock.sentinel)
        assert mock.sentinel.events == events
        self.mock_kube_client.list_namespaced_event.assert_has_calls(
            [
                mock.call(
                    namespace=mock.sentinel.metadata.namespace,
                    field_selector=f"involvedObject.name={mock.sentinel.metadata.name}",
                    resource_version=None,
                    resource_version_match=None,
                ),
                mock.call(
                    namespace=mock.sentinel.metadata.namespace,
                    field_selector=f"involvedObject.name={mock.sentinel.metadata.name}",
                    resource_version=None,
                    resource_version_match=None,
                ),
            ]
        )

    def test_read_pod_events_retries_fails(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.list_namespaced_event.side_effect = [
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
        ]
        with pytest.raises(AirflowException):
            self.pod_manager.read_pod_events(mock.sentinel)

    def test_read_pod_returns_logs(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod.return_value = mock.sentinel.pod_info
        pod_info = self.pod_manager.read_pod(mock.sentinel)
        assert mock.sentinel.pod_info == pod_info

    def test_read_pod_retries_successfully(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod.side_effect = [
            BaseHTTPError("Boom"),
            mock.sentinel.pod_info,
        ]
        pod_info = self.pod_manager.read_pod(mock.sentinel)
        assert mock.sentinel.pod_info == pod_info
        self.mock_kube_client.read_namespaced_pod.assert_has_calls(
            [
                mock.call(mock.sentinel.metadata.name, mock.sentinel.metadata.namespace),
                mock.call(mock.sentinel.metadata.name, mock.sentinel.metadata.namespace),
            ]
        )

    def test_monitor_pod_empty_logs(self):
        mock.sentinel.metadata = mock.MagicMock()
        running_status = mock.MagicMock()
        running_status.configure_mock(**{"name": "base", "state.running": True})
        pod_info_running = mock.MagicMock(**{"status.container_statuses": [running_status]})
        pod_info_succeeded = mock.MagicMock(**{"status.phase": PodPhase.SUCCEEDED})

        def pod_state_gen():
            yield pod_info_running
            while True:
                yield pod_info_succeeded

        self.mock_kube_client.read_namespaced_pod.side_effect = pod_state_gen()
        mock_response = mock.MagicMock(stream=mock.MagicMock(return_value=iter(())))
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock_response
        self.pod_manager.fetch_container_logs(mock.sentinel, "base")

    def test_monitor_pod_logs_failures_non_fatal(self):
        mock.sentinel.metadata = mock.MagicMock()
        running_status = mock.MagicMock()
        running_status.configure_mock(**{"name": "base", "state.running": True})
        pod_info_running = mock.MagicMock(**{"status.container_statuses": [running_status]})
        pod_info_succeeded = mock.MagicMock(**{"status.phase": PodPhase.SUCCEEDED})

        def pod_state_gen():
            yield pod_info_running
            yield pod_info_running
            while True:
                yield pod_info_succeeded

        self.mock_kube_client.read_namespaced_pod.side_effect = pod_state_gen()

        def pod_log_gen():
            while True:
                yield BaseHTTPError("Boom")

        self.mock_kube_client.read_namespaced_pod_log.side_effect = pod_log_gen()

        self.pod_manager.fetch_container_logs(mock.sentinel, "base")

    def test_read_pod_retries_fails(self):
        mock.sentinel.metadata = mock.MagicMock()
        self.mock_kube_client.read_namespaced_pod.side_effect = [
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
            BaseHTTPError("Boom"),
        ]
        with pytest.raises(AirflowException):
            self.pod_manager.read_pod(mock.sentinel)

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.read_pod_logs")
    def test_fetch_container_logs_returning_last_timestamp(
        self, mock_read_pod_logs, mock_container_is_running
    ):
        timestamp_string = "2020-10-08T14:16:17.793417674Z"
        mock_read_pod_logs.return_value = [bytes(f"{timestamp_string} message", "utf-8"), b"notimestamp"]
        mock_container_is_running.side_effect = [True, False]

        status = self.pod_manager.fetch_container_logs(mock.MagicMock(), mock.MagicMock(), follow=True)

        assert status.last_log_time == cast("DateTime", pendulum.parse(timestamp_string))

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.read_pod_logs")
    def test_fetch_container_logs_invoke_progress_callback(
        self, mock_read_pod_logs, mock_container_is_running
    ):
        MockWrapper.reset()
        mock_callbacks = MockWrapper.mock_callbacks
        ts = "2020-10-08T14:16:17.793417674Z"
        message = "message"
        no_ts_message = "notimestamp"
        second_ts = "2020-10-08T15:16:18.793417674Z"
        second_message = "second message"
        mock_read_pod_logs.return_value = [
            bytes(f"{ts} {message}", "utf-8"),
            bytes(no_ts_message, "utf-8"),
            bytes(f"{second_ts} {second_message}", "utf-8"),
        ]
        mock_container_is_running.return_value = False
        mock_pod = mock.MagicMock()

        self.pod_manager.fetch_container_logs(mock_pod, "base", follow=True)
        mock_callbacks.progress_callback.assert_has_calls(
            [
                mock.call(
                    line=f"{message}\n{no_ts_message}",
                    client=self.pod_manager._client,
                    mode="sync",
                    container_name="base",
                    timestamp=pendulum.parse(ts),
                    pod=mock_pod,
                ),
                mock.call(
                    line=f"{second_message}",
                    client=self.pod_manager._client,
                    mode="sync",
                    container_name="base",
                    timestamp=pendulum.parse(second_ts),
                    pod=mock_pod,
                ),
            ]
        )

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    def test_fetch_container_logs_failures(self, mock_container_is_running):
        MockWrapper.reset()
        mock_callbacks = MockWrapper.mock_callbacks
        last_timestamp_string = "2020-10-08T14:18:17.793417674Z"
        messages = [
            bytes("2020-10-08T14:16:17.793417674Z message", "utf-8"),
            bytes("2020-10-08T14:17:17.793417674Z message", "utf-8"),
            None,
            bytes(f"{last_timestamp_string} message", "utf-8"),
        ]
        expected_call_count = len([message for message in messages if message is not None])

        def consumer_iter():
            while messages:
                message = messages.pop(0)
                if message is None:
                    raise BaseHTTPError("Boom")
                yield message

        with mock.patch.object(PodLogsConsumer, "__iter__") as mock_consumer_iter:
            mock_consumer_iter.side_effect = consumer_iter
            mock_container_is_running.side_effect = [True, True, False]
            status = self.pod_manager.fetch_container_logs(mock.MagicMock(), mock.MagicMock(), follow=True)
        assert status.last_log_time == cast("DateTime", pendulum.parse(last_timestamp_string))
        assert mock_callbacks.progress_callback.call_count == expected_call_count

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.read_pod_logs")
    def test_parse_multi_line_logs(self, mock_read_pod_logs, mock_container_is_running, caplog):
        log = (
            "2020-10-08T14:16:17.793417674Z message1 line1\n"
            "message1 line2\n"
            "message1 line3\n"
            "2020-10-08T14:16:18.793417674Z message2 line1\n"
            "message2 line2\n"
            "2020-10-08T14:16:19.793417674Z message3 line1\n"
        )
        mock_read_pod_logs.return_value = [bytes(log_line, "utf-8") for log_line in log.split("\n")]
        mock_container_is_running.return_value = False

        with caplog.at_level(logging.INFO):
            self.pod_manager.fetch_container_logs(mock.MagicMock(), mock.MagicMock(), follow=True)

        assert "message1 line1" in caplog.text
        assert "message1 line2" in caplog.text
        assert "message1 line3" in caplog.text
        assert "message2 line1" in caplog.text
        assert "message2 line2" in caplog.text
        assert "message3 line1" in caplog.text
        assert "ERROR" not in caplog.text

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.read_pod_logs")
    def test_container_log_times_tracks_last_timestamp(self, mock_read_pod_logs, mock_container_is_running):
        """Test that container_log_times dictionary tracks the last log timestamp for each container."""
        timestamp_string = "2020-10-08T14:16:17.793417674Z"
        mock_read_pod_logs.return_value = [bytes(f"{timestamp_string} message", "utf-8")]
        mock_container_is_running.return_value = False

        # Create a mock pod with namespace and name
        mock_pod = mock.MagicMock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        container_name = "test-container"

        # Ensure container_log_times is empty initially
        assert not self.pod_manager.container_log_times

        # Fetch logs which should populate container_log_times
        self.pod_manager.fetch_container_logs(mock_pod, container_name, follow=True)

        # Verify the timestamp was stored in container_log_times
        key = ("test-namespace", "test-pod", "test-container")
        assert key in self.pod_manager.container_log_times
        assert self.pod_manager.container_log_times[key] == cast("DateTime", pendulum.parse(timestamp_string))

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.read_pod_logs")
    def test_fetch_requested_container_logs_uses_since_time(
        self, mock_read_pod_logs, mock_container_is_running
    ):
        """Test that fetch_requested_container_logs passes since_time from container_log_times."""
        timestamp_string = "2020-10-08T14:16:17.793417674Z"
        mock_read_pod_logs.return_value = [bytes(f"{timestamp_string} message", "utf-8")]
        mock_container_is_running.return_value = False

        # Create a mock pod
        mock_pod = mock.MagicMock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        mock_container = mock.MagicMock()
        mock_container.name = "test-container"
        mock_pod.spec.containers = [mock_container]
        container_name = "test-container"

        # Pre-populate container_log_times with an earlier timestamp
        earlier_timestamp = pendulum.parse("2020-10-08T14:15:00.000000000Z")
        self.pod_manager.container_log_times[("test-namespace", "test-pod", "test-container")] = (
            earlier_timestamp
        )

        # Mock read_pod to return the mock_pod
        self.pod_manager.read_pod = mock.Mock(return_value=mock_pod)

        # Fetch logs - this should pass the earlier timestamp as since_time
        with mock.patch.object(
            self.pod_manager, "fetch_container_logs", wraps=self.pod_manager.fetch_container_logs
        ) as mock_fetch:
            self.pod_manager.fetch_requested_container_logs(mock_pod, containers=container_name)

            # Verify fetch_container_logs was called with since_time set to the earlier timestamp
            mock_fetch.assert_called_once()
            call_kwargs = mock_fetch.call_args[1]
            assert call_kwargs["since_time"] == earlier_timestamp

    @pytest.mark.parametrize("status", [409, 429])
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.run_pod_async")
    def test_start_pod_retries_on_409_or_429_error(self, mock_run_pod_async, status):
        mock_run_pod_async.side_effect = [
            ApiException(status=status),
            mock.MagicMock(),
        ]
        self.pod_manager.create_pod(mock.sentinel)
        assert mock_run_pod_async.call_count == 2

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.run_pod_async")
    def test_start_pod_fails_on_other_exception(self, mock_run_pod_async):
        mock_run_pod_async.side_effect = [ApiException(status=401)]
        with pytest.raises(ApiException):
            self.pod_manager.create_pod(mock.sentinel)

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.run_pod_async")
    def test_start_pod_retries_three_times(self, mock_run_pod_async):
        mock_run_pod_async.side_effect = [
            ApiException(status=409),
            ApiException(status=409),
            ApiException(status=409),
            ApiException(status=409),
            ApiException(status=409),
            ApiException(status=409),
        ]
        with pytest.raises(ApiException):
            self.pod_manager.create_pod(mock.sentinel)

        assert mock_run_pod_async.call_count == 5

    @pytest.mark.asyncio
    async def test_start_pod_raises_informative_error_on_scheduled_timeout(self):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        self.mock_kube_client.read_namespaced_pod.return_value = pod_response
        expected_msg = "Pod took too long to be scheduled on the cluster, giving up. More than 0s. Check the pod events in kubernetes."
        mock_pod = MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=0,
                startup_timeout=0,
            )

    @pytest.mark.asyncio
    async def test_start_pod_raises_informative_error_on_startup_timeout(self):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        condition = mock.MagicMock()
        condition.type = "PodScheduled"
        condition.status = "True"
        pod_response.status.conditions = [condition]

        self.mock_kube_client.read_namespaced_pod.return_value = pod_response
        expected_msg = "Pod took too long to start. More than 0s. Check the pod events in kubernetes."
        mock_pod = MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=0,
                startup_timeout=0,
            )

    @pytest.mark.asyncio
    @pytest.mark.parametrize("fail_reason", ["ErrImagePull", "ImagePullBackOff", "InvalidImageName"])
    async def test_start_pod_raises_fast_error_on_image_error(self, fail_reason):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        container_statuse = mock.MagicMock()
        waiting_state = mock.MagicMock()
        waiting_state.reason = fail_reason
        waiting_state.message = "Test error"
        container_statuse.state.waiting = waiting_state
        pod_response.status.container_statuses = [container_statuse]

        self.mock_kube_client.read_namespaced_pod.return_value = pod_response
        expected_msg = (
            f"Image cannot be pulled, unable to start: {waiting_state.reason}\n{waiting_state.message}"
        )
        mock_pod = MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=60,
                startup_timeout=60,
            )

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_start_pod_startup_interval_seconds(self, mock_time_sleep):
        condition_scheduled = mock.MagicMock()
        condition_scheduled.type = "PodScheduled"
        condition_scheduled.status = "True"

        pod_info_pending = mock.MagicMock(**{"status.phase": PodPhase.PENDING})
        pod_info_pending_scheduled = mock.MagicMock(
            **{"status.phase": PodPhase.PENDING, "status.conditions": [condition_scheduled]}
        )
        pod_info_succeeded = mock.MagicMock(**{"status.phase": PodPhase.SUCCEEDED})

        def pod_state_gen():
            yield pod_info_pending
            yield pod_info_pending_scheduled
            yield pod_info_pending_scheduled
            while True:
                yield pod_info_succeeded

        self.mock_kube_client.read_namespaced_pod.side_effect = pod_state_gen()
        startup_check_interval = 10  # Any value is fine, as time.sleep is mocked to do nothing
        schedule_timeout = 30
        startup_timeout = 60
        mock_pod = MagicMock()

        with mock.patch.object(self.pod_manager.log, "info") as mock_log_info:
            await self.pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=schedule_timeout,  # Never hit, any value is fine, as time.sleep is mocked to do nothing
                startup_timeout=startup_timeout,  # Never hit, any value is fine, as time.sleep is mocked to do nothing
                check_interval=startup_check_interval,
            )
            mock_time_sleep.assert_called_with(startup_check_interval)
            assert self.pod_manager.stop_watching_events is True
            assert mock_time_sleep.call_count == 3
            mock_log_info.assert_any_call(
                "::group::Waiting up to %ss to get the POD scheduled...", schedule_timeout
            )
            mock_log_info.assert_any_call("Waiting %ss to get the POD running...", startup_timeout)

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    def test_container_is_running(self, container_is_running_mock):
        mock_pod = MagicMock()
        self.pod_manager.read_pod = mock.MagicMock(return_value=mock_pod)
        self.pod_manager.container_is_running(None, "base")
        container_is_running_mock.assert_called_with(pod=mock_pod, container_name="base")

    @pytest.mark.parametrize("follow", [True, False])
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodLogsConsumer.logs_available")
    def test_fetch_container_done(self, logs_available, container_running, follow):
        """If container done, should exit, no matter setting of follow."""
        mock_pod = MagicMock()
        logs_available.return_value = False
        container_running.return_value = False
        ret = self.pod_manager.fetch_container_logs(pod=mock_pod, container_name="base", follow=follow)
        assert ret.last_log_time is None
        assert ret.running is False

    # adds all valid types for container_logs
    @pytest.mark.parametrize("follow", [True, False])
    @pytest.mark.parametrize("container_logs", ["base", "alpine", True, ["base", "alpine"]])
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    def test_fetch_requested_container_logs(self, container_is_running, container_logs, follow):
        mock_pod = MagicMock()
        self.pod_manager.read_pod = MagicMock()
        self.pod_manager.get_container_names = MagicMock()
        self.pod_manager.get_container_names.return_value = ["base", "alpine"]
        container_is_running.return_value = False
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2021-01-01 hi"])
        )

        ret_values = self.pod_manager.fetch_requested_container_logs(
            pod=mock_pod, containers=container_logs, follow_logs=follow
        )
        for ret in ret_values:
            assert ret.running is False

    # adds all invalid types for container_logs
    @pytest.mark.parametrize("container_logs", [1, None, 6.8, False])
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    def test_fetch_requested_container_logs_invalid(self, container_running, container_logs):
        mock_pod = MagicMock()
        self.pod_manager.read_pod = MagicMock()
        self.pod_manager.get_container_names = MagicMock()
        self.pod_manager.get_container_names.return_value = ["base", "alpine"]
        container_running.return_value = False
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2021-01-01 hi"])
        )

        ret_values = self.pod_manager.fetch_requested_container_logs(
            pod=mock_pod,
            containers=container_logs,
        )

        assert len(ret_values) == 0

    @time_machine.travel("2026-01-01 00:00:00", tick=False)
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodLogsConsumer.logs_available")
    def test_fetch_container_with_valid_since_time(self, logs_available, container_running):
        """Test that since_seconds is calculated correctly when since_time is a valid datetime."""
        mock_pod = MagicMock()
        since_time = datetime(2026, 1, 1, 0, 0, 0, tzinfo=dt_timezone.utc) - timedelta(seconds=30)
        logs_available.return_value = True
        container_running.return_value = False
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2026-01-01 hi"])
        )
        self.pod_manager.fetch_container_logs(pod=mock_pod, container_name="base", since_time=since_time)
        _, call_kwargs = self.mock_kube_client.read_namespaced_pod_log.call_args_list[0]
        assert call_kwargs["since_seconds"] == 30

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodLogsConsumer.logs_available")
    def test_fetch_container_with_invalid_since_time_falls_back_to_none(
        self, logs_available, container_running
    ):
        """Test that an invalid since_time is caught, warns, and since_seconds is passed as None."""
        mock_pod = MagicMock()
        logs_available.return_value = True
        container_running.return_value = False
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2026-01-01 hi"])
        )
        with mock.patch.object(self.pod_manager.log, "warning") as mock_warning:
            self.pod_manager.fetch_container_logs(
                pod=mock_pod, container_name="base", since_time="not-a-datetime"
            )
        _, call_kwargs = self.mock_kube_client.read_namespaced_pod_log.call_args_list[0]
        assert "since_seconds" not in call_kwargs
        mock_warning.assert_called_once_with(
            "Error calculating since_seconds with since_time %s. Using None instead.",
            "not-a-datetime",
        )

    @time_machine.travel("2026-01-01 00:00:00", tick=False)
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodLogsConsumer.logs_available")
    def test_fetch_container_with_string_since_time_converts_to_datetime(
        self, logs_available, container_running
    ):
        """Test that a valid ISO string since_time is converted to datetime and since_seconds is calculated."""
        mock_pod = MagicMock()
        since_time = "2025-12-31T23:59:30Z"
        logs_available.return_value = True
        container_running.return_value = False
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2026-01-01 hi"])
        )
        with patch.object(self.pod_manager.log, "warning") as mock_warning:
            self.pod_manager.fetch_container_logs(pod=mock_pod, container_name="base", since_time=since_time)
        _, call_kwargs = self.mock_kube_client.read_namespaced_pod_log.call_args_list[0]
        assert call_kwargs["since_seconds"] == 30
        mock_warning.assert_not_called()

    @pytest.mark.parametrize(
        ("follow", "is_running_calls", "exp_running"), [(True, 3, False), (False, 3, False)]
    )
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    def test_fetch_container_running_follow(
        self, container_running_mock, follow, is_running_calls, exp_running
    ):
        """
        When called with follow, should keep looping even after disconnections, if pod still running.
        When called with follow=False, should return immediately even though still running.
        """
        mock_pod = MagicMock()
        container_running_mock.side_effect = [True, True, False]
        self.mock_kube_client.read_namespaced_pod_log.return_value = mock.MagicMock(
            stream=mock.MagicMock(return_value=[b"2021-01-01 hi"])
        )
        ret = self.pod_manager.fetch_container_logs(pod=mock_pod, container_name="base", follow=follow)
        assert len(container_running_mock.call_args_list) == is_running_calls
        assert ret.last_log_time == pendulum.datetime(2021, 1, 1, tz="UTC")
        assert ret.running is exp_running

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.kubernetes_stream")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.extract_xcom_kill")
    @mock.patch(
        "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running",
        return_value=True,
    )
    def test_extract_xcom_success(
        self, mock_container_is_running, mock_exec_xcom_kill, mock_kubernetes_stream
    ):
        """test when valid json is retrieved from xcom sidecar container."""
        xcom_json = """{"a": "true"}"""
        mock_pod = MagicMock()
        mock_client = MagicMock()
        mock_client.peek_stderr.return_value = ""
        mock_client.read_all.return_value = xcom_json
        mock_kubernetes_stream.return_value = mock_client
        ret = self.pod_manager.extract_xcom(pod=mock_pod)
        assert ret == xcom_json
        assert mock_exec_xcom_kill.call_count == 1

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.kubernetes_stream")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.extract_xcom_kill")
    @mock.patch(
        "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running",
        return_value=True,
    )
    def test_extract_xcom_failure(
        self, mock_container_is_running, mock_exec_xcom_kill, mock_kubernetes_stream
    ):
        """test when invalid json is retrieved from xcom sidecar container."""
        xcom_json = """{"a": "tru"""  # codespell:ignore tru
        mock_pod = MagicMock()
        mock_client = MagicMock()
        mock_client.peek_stderr.return_value = ""
        mock_client.read_all.return_value = xcom_json
        mock_kubernetes_stream.return_value = mock_client
        with pytest.raises(JSONDecodeError):
            self.pod_manager.extract_xcom(pod=mock_pod)
        assert mock_exec_xcom_kill.call_count == 1

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.kubernetes_stream")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.extract_xcom_kill")
    @mock.patch(
        "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running",
        return_value=True,
    )
    def test_extract_xcom_empty(self, mock_container_is_running, mock_exec_xcom_kill, mock_kubernetes_stream):
        """test when __airflow_xcom_result_empty__ is retrieved from xcom sidecar container."""
        mock_pod = MagicMock()
        xcom_result = "__airflow_xcom_result_empty__"
        mock_client = MagicMock()
        mock_client.peek_stderr.return_value = ""
        mock_client.read_all.return_value = xcom_result
        mock_kubernetes_stream.return_value = mock_client
        ret = self.pod_manager.extract_xcom(pod=mock_pod)
        assert ret == xcom_result
        assert mock_exec_xcom_kill.call_count == 1

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.kubernetes_stream")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.extract_xcom_kill")
    @mock.patch(
        "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running",
        return_value=True,
    )
    def test_extract_xcom_none(self, mock_container_is_running, mock_exec_xcom_kill, mock_kubernetes_stream):
        """test when None is retrieved from xcom sidecar container."""
        mock_pod = MagicMock()
        mock_client = MagicMock()
        mock_client.peek_stderr.return_value = ""
        mock_client.read_all.return_value = None
        mock_kubernetes_stream.return_value = mock_client
        with pytest.raises(XComRetrievalError, match=r"Failed to extract xcom from pod:"):
            self.pod_manager.extract_xcom(pod=mock_pod)
        assert mock_exec_xcom_kill.call_count == 1

    @mock.patch(
        "airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running",
        return_value=False,
    )
    def test_extract_xcom_sidecar_terminated(self, mock_container_is_running):
        """test when None is retrieved from xcom sidecar container."""
        mock_pod = MagicMock()
        with pytest.raises(
            XComRetrievalError,
            match=r"container is not running! Not possible to read xcom from pod:",
        ):
            self.pod_manager.extract_xcom(pod=mock_pod)

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_terminated")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    def test_await_xcom_sidecar_container_timeout(
        self, mock_container_is_running, mock_container_is_terminated
    ):
        mock_pod = MagicMock()
        mock_container_is_running.return_value = False
        mock_container_is_terminated.return_value = False
        with pytest.raises(AirflowException):
            self.pod_manager.await_xcom_sidecar_container_start(pod=mock_pod, timeout=10, log_interval=5)
        mock_container_is_running.assert_any_call(mock_pod, "airflow-xcom-sidecar")
        mock_container_is_terminated.assert_any_call(mock_pod, "airflow-xcom-sidecar")

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_terminated")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    def test_await_xcom_sidecar_container_terminated(
        self, mock_container_is_running, mock_container_is_terminated
    ):
        mock_pod = MagicMock()
        mock_container_is_running.return_value = False
        mock_container_is_terminated.return_value = True
        match_error = (
            "Xcom sidecar container is already terminated! Not possible to read xcom output of task."
        )
        with pytest.raises(AirflowException, match=match_error):
            self.pod_manager.await_xcom_sidecar_container_start(pod=mock_pod, timeout=10, log_interval=5)

        mock_container_is_running.assert_any_call(mock_pod, "airflow-xcom-sidecar")
        mock_container_is_terminated.assert_any_call(mock_pod, "airflow-xcom-sidecar")

    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.PodManager.container_is_running")
    def test_await_xcom_sidecar_container_starts(self, mock_container_is_running):
        mock_pod = MagicMock()
        mock_container_is_running.return_value = True
        self.pod_manager.await_xcom_sidecar_container_start(pod=mock_pod)
        mock_container_is_running.assert_any_call(mock_pod, "airflow-xcom-sidecar")

    @mock.patch("time.sleep")
    def test_await_pod_completion_breaks_on_terminal_phase(self, mock_sleep, pod_factory):
        pending = pod_factory(pod_phase=PodPhase.PENDING)
        running = pod_factory(pod_phase=PodPhase.RUNNING)
        succeeded = pod_factory(pod_phase=PodPhase.SUCCEEDED)
        self.pod_manager.read_pod = mock.MagicMock(side_effect=[pending, running, succeeded])

        result = self.pod_manager.await_pod_completion(pod=mock.MagicMock(), istio_enabled=False)

        assert result is succeeded
        assert mock_sleep.call_count == 2

    @mock.patch("time.sleep")
    def test_await_pod_completion_breaks_on_istio_container_completed(self, mock_sleep, pod_factory):
        running1 = pod_factory(pod_phase=PodPhase.RUNNING, container_name="base", terminated=False)
        running2 = pod_factory(pod_phase=PodPhase.RUNNING, container_name="base", terminated=True)

        self.pod_manager.read_pod = mock.MagicMock(side_effect=[running1, running2])

        result = self.pod_manager.await_pod_completion(
            pod=mock.MagicMock(), istio_enabled=True, container_name="base"
        )

        assert result is running2
        assert mock_sleep.call_count == 1

    @mock.patch("time.sleep")
    def test_await_pod_completion_breaks_on_xcom_sidecar_container_completed(self, mock_sleep, pod_factory):
        """When do_xcom_push=True and base container has completed, stop waiting even if pod is still Running."""
        running1 = pod_factory(pod_phase=PodPhase.RUNNING, container_name="base", terminated=False)
        running2 = pod_factory(pod_phase=PodPhase.RUNNING, container_name="base", terminated=True)

        self.pod_manager.read_pod = mock.MagicMock(side_effect=[running1, running2])

        result = self.pod_manager.await_pod_completion(
            pod=mock.MagicMock(), istio_enabled=False, container_name="base", do_xcom_push=True
        )

        assert result is running2
        assert mock_sleep.call_count == 1

    @mock.patch("time.sleep")
    def test_await_pod_completion_waits_for_pod_phase_without_sidecars(self, mock_sleep, pod_factory):
        """Without istio or xcom sidecar, await_pod_completion waits for terminal pod phase."""
        running1 = pod_factory(pod_phase=PodPhase.RUNNING, container_name="base", terminated=True)
        succeeded = pod_factory(pod_phase=PodPhase.SUCCEEDED, container_name="base", terminated=True)

        self.pod_manager.read_pod = mock.MagicMock(side_effect=[running1, succeeded])

        result = self.pod_manager.await_pod_completion(
            pod=mock.MagicMock(), istio_enabled=False, container_name="base", do_xcom_push=False
        )

        assert result is succeeded
        assert mock_sleep.call_count == 1

    @mock.patch("time.sleep")
    def test_await_pod_completion_breaks_on_early_termination_issue(self, mock_sleep, pod_factory):
        running1 = pod_factory(pod_phase=PodPhase.PENDING, container_name="base")
        running2 = pod_factory(
            pod_phase=PodPhase.PENDING,
            container_name="base",
            waiting_reason="ImagePullBackOff",
            waiting_message="Back-off pulling image",
        )

        self.pod_manager.read_pod = mock.MagicMock(side_effect=[running1, running2])

        result = self.pod_manager.await_pod_completion(pod=mock.MagicMock(), istio_enabled=False)

        assert result is running2
        assert mock_sleep.call_count == 1


class TestAsyncPodManager:
    @pytest.fixture
    def mock_log_info(self):
        with mock.patch.object(self.async_pod_manager.log, "info") as mock_log_info:
            yield mock_log_info

    def setup_method(self):
        self.mock_async_hook = mock.AsyncMock()
        self.async_pod_manager = AsyncPodManager(
            async_hook=self.mock_async_hook,
            callbacks=[],
        )

    @pytest.mark.asyncio
    async def test_read_pod_events_with_resource_version(self):
        """Test async read_pod_events with resource_version parameter."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        mock_events = mock.Mock()

        self.mock_async_hook.get_pod_events.return_value = mock_events

        result = await self.async_pod_manager.read_pod_events(mock_pod, resource_version="12345")

        assert result == mock_events
        self.mock_async_hook.get_pod_events.assert_called_once_with(
            "test-pod", "test-namespace", resource_version="12345"
        )

    @pytest.mark.asyncio
    async def test_read_pod_events_without_resource_version(self):
        """Test async read_pod_events without resource_version parameter."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"
        mock_events = mock.Mock()

        self.mock_async_hook.get_pod_events.return_value = mock_events

        result = await self.async_pod_manager.read_pod_events(mock_pod)

        assert result == mock_events
        self.mock_async_hook.get_pod_events.assert_called_once_with(
            "test-pod", "test-namespace", resource_version=None
        )

    @pytest.mark.asyncio
    async def test_watch_pod_events_uses_hook_watch(self):
        """Test that watch_pod_events uses hook's watch_pod_events method."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        mock_event1 = mock.Mock()
        mock_event1.metadata.uid = "event-uid-1"
        mock_event1.metadata.resource_version = "100"
        mock_event1.message = "Event 1"
        mock_event1.involved_object.field_path = "spec"

        mock_event2 = mock.Mock()
        mock_event2.metadata.uid = "event-uid-2"
        mock_event2.metadata.resource_version = "101"
        mock_event2.message = "Event 2"
        mock_event2.involved_object.field_path = "spec"

        async def async_event_generator(*_, **__):
            yield mock_event1
            yield mock_event2
            self.async_pod_manager.stop_watching_events = True

        self.mock_async_hook.watch_pod_events = mock.Mock(side_effect=async_event_generator)

        with mock.patch.object(self.async_pod_manager.log, "info") as mock_log_info:
            await self.async_pod_manager.watch_pod_events(mock_pod, startup_check_interval=30)

            # Both events should be logged
            assert mock_log_info.call_count == 2
            calls = mock_log_info.call_args_list
            assert calls[0][0] == ("The Pod has an Event: %s from %s", "Event 1", "spec")
            assert calls[1][0] == ("The Pod has an Event: %s from %s", "Event 2", "spec")

        # Verify hook was called
        self.mock_async_hook.watch_pod_events.assert_called()

    @pytest.mark.asyncio
    async def test_watch_pod_events_tracks_resource_version(self):
        """Test that watch_pod_events tracks and updates resource version."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        # Create events for two iterations
        mock_event1 = mock.Mock()
        mock_event1.metadata.uid = "event-uid-1"
        mock_event1.metadata.resource_version = "100"
        mock_event1.message = "Event 1"
        mock_event1.involved_object.field_path = "spec"

        mock_event2 = mock.Mock()
        mock_event2.metadata.uid = "event-uid-2"
        mock_event2.metadata.resource_version = "101"
        mock_event2.message = "Event 2"
        mock_event2.involved_object.field_path = "spec"

        call_count = 0

        async def async_event_generator(*_, **__):
            nonlocal call_count
            call_count += 1
            if call_count == 1:
                # First iteration
                yield mock_event1
            else:
                # Second iteration
                yield mock_event2
                self.async_pod_manager.stop_watching_events = True

        self.mock_async_hook.watch_pod_events = mock.Mock(side_effect=async_event_generator)
        self.async_pod_manager.stop_watching_events = False

        await self.async_pod_manager.watch_pod_events(mock_pod, startup_check_interval=30)

        # Verify hook was called twice with updated resource_version
        assert self.mock_async_hook.watch_pod_events.call_count == 2
        calls = self.mock_async_hook.watch_pod_events.call_args_list

        # First call should have no resource_version
        assert calls[0][1]["resource_version"] is None
        # Second call should use resource_version from first event
        assert calls[1][1]["resource_version"] == "100"

    @pytest.mark.asyncio
    async def test_watch_pod_events_deduplicates_events(self):
        """Test that watch_pod_events deduplicates events across iterations."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        # Same event returned in two iterations
        mock_event = mock.Mock()
        mock_event.metadata.uid = "event-uid-1"
        mock_event.metadata.resource_version = "100"
        mock_event.message = "Duplicate event"
        mock_event.involved_object.field_path = "spec"

        call_count = 0

        async def async_event_generator(*_, **__):
            nonlocal call_count
            call_count += 1
            yield mock_event  # Return same event
            if call_count >= 2:
                self.async_pod_manager.stop_watching_events = True

        self.mock_async_hook.watch_pod_events = mock.Mock(side_effect=async_event_generator)
        self.async_pod_manager.stop_watching_events = False

        with mock.patch.object(self.async_pod_manager.log, "info") as mock_log_info:
            await self.async_pod_manager.watch_pod_events(mock_pod, startup_check_interval=30)

            # Event should only be logged once despite being returned twice
            assert mock_log_info.call_count == 1
            mock_log_info.assert_called_with("The Pod has an Event: %s from %s", "Duplicate event", "spec")

    @pytest.mark.asyncio
    async def test_watch_pod_events_handles_none_event(self):
        """Test that watch_pod_events handles None events gracefully."""
        mock_pod = mock.Mock()
        mock_pod.metadata.namespace = "test-namespace"
        mock_pod.metadata.name = "test-pod"

        async def async_event_generator(*_, **__):
            yield None  # None event should be skipped
            self.async_pod_manager.stop_watching_events = True

        self.mock_async_hook.watch_pod_events = mock.Mock(side_effect=async_event_generator)
        self.async_pod_manager.stop_watching_events = False

        with mock.patch.object(self.async_pod_manager.log, "info") as mock_log_info:
            await self.async_pod_manager.watch_pod_events(mock_pod, startup_check_interval=30)

            # No events should be logged for None
            mock_log_info.assert_not_called()

    @pytest.mark.asyncio
    async def test_start_pod_raises_informative_error_on_scheduled_timeout(self):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        self.mock_async_hook.get_pod.return_value = pod_response
        expected_msg = "Pod took too long to be scheduled on the cluster, giving up. More than 0s. Check the pod events in kubernetes."
        mock_pod = mock.MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.async_pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=0,
                startup_timeout=0,
            )
        self.mock_async_hook.get_pod.assert_called()

    @pytest.mark.asyncio
    async def test_start_pod_raises_informative_error_on_startup_timeout(self):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        condition = mock.MagicMock()
        condition.type = "PodScheduled"
        condition.status = "True"
        pod_response.status.conditions = [condition]
        self.mock_async_hook.get_pod.return_value = pod_response
        expected_msg = "Pod took too long to start. More than 0s. Check the pod events in kubernetes."
        mock_pod = mock.MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.async_pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=0,
                startup_timeout=0,
            )
        self.mock_async_hook.get_pod.assert_called()

    @pytest.mark.asyncio
    @pytest.mark.parametrize("fail_reason", ["ErrImagePull", "ImagePullBackOff", "InvalidImageName"])
    async def test_start_pod_raises_fast_error_on_image_error(self, fail_reason):
        pod_response = mock.MagicMock()
        pod_response.status.phase = "Pending"
        container_status = mock.MagicMock()
        waiting_state = mock.MagicMock()
        waiting_state.reason = fail_reason
        waiting_state.message = "Test error"
        container_status.state.waiting = waiting_state
        pod_response.status.container_statuses = [container_status]
        self.mock_async_hook.get_pod.return_value = pod_response
        expected_msg = (
            f"Image cannot be pulled, unable to start: {waiting_state.reason}\n{waiting_state.message}"
        )
        mock_pod = mock.MagicMock()
        with pytest.raises(AirflowException, match=expected_msg):
            await self.async_pod_manager.await_pod_start(
                pod=mock_pod,
                schedule_timeout=60,
                startup_timeout=60,
            )
        self.mock_async_hook.get_pod.assert_called()

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_start_pod_startup_interval_seconds(self, mock_time_sleep, mock_log_info):
        condition_scheduled = mock.MagicMock()
        condition_scheduled.type = "PodScheduled"
        condition_scheduled.status = "True"

        pod_info_pending = mock.MagicMock()
        pod_info_pending.status.phase = PodPhase.PENDING
        pod_info_pending.status.conditions = []

        pod_info_pending_scheduled = mock.MagicMock()
        pod_info_pending_scheduled.status.phase = PodPhase.PENDING
        pod_info_pending_scheduled.status.conditions = [condition_scheduled]

        pod_info_succeeded = mock.MagicMock()
        pod_info_succeeded.status.phase = PodPhase.SUCCEEDED

        # Simulate sequence of pod states
        self.mock_async_hook.get_pod.side_effect = [
            pod_info_pending,
            pod_info_pending_scheduled,
            pod_info_pending_scheduled,
            pod_info_succeeded,
        ]
        startup_check_interval = 10
        schedule_timeout = 30
        startup_timeout = 60
        mock_pod = mock.MagicMock()

        await self.async_pod_manager.await_pod_start(
            pod=mock_pod,
            schedule_timeout=schedule_timeout,
            startup_timeout=startup_timeout,
            check_interval=startup_check_interval,
        )
        assert mock_time_sleep.call_count == 3
        mock_log_info.assert_any_call(
            "::group::Waiting up to %ss to get the POD scheduled...", schedule_timeout
        )
        mock_log_info.assert_any_call("Waiting %ss to get the POD running...", startup_timeout)
        assert self.async_pod_manager.stop_watching_events is True

    @pytest.mark.asyncio
    @mock.patch("asyncio.sleep", new_callable=mock.AsyncMock)
    async def test_watch_pod_events(self, mock_time_sleep, mock_log_info):
        mock_pod = mock.MagicMock()
        mock_pod.metadata.name = "test-pod"
        mock_pod.metadata.namespace = "default"

        events = mock.MagicMock()
        events.items = []
        for id in ["event 1", "event 2"]:
            event = mock.MagicMock()
            event.message = f"test {id}"
            event.involved_object.field_path = f"object {id}"
            events.items.append(event)
        startup_check_interval = 10

        async def watch_events_generator(*_, **__):
            for event in events.items:
                yield event
            self.async_pod_manager.stop_watching_events = True

        self.mock_async_hook.watch_pod_events = mock.Mock(side_effect=watch_events_generator)

        await self.async_pod_manager.watch_pod_events(
            pod=mock_pod, startup_check_interval=startup_check_interval
        )
        mock_log_info.assert_any_call("The Pod has an Event: %s from %s", "test event 1", "object event 1")
        mock_log_info.assert_any_call("The Pod has an Event: %s from %s", "test event 2", "object event 2")

    @pytest.mark.asyncio
    @pytest.mark.parametrize(
        ("log_lines", "now", "expected_log_messages", "not_expected_log_messages"),
        [
            # Case 1: No logs
            ([], pendulum.now(), [], []),
            # Case 2: One log line with timestamp before now
            (
                [f"{pendulum.now().subtract(seconds=2).to_iso8601_string()} message"],
                pendulum.now(),
                ["message"],
                [],
            ),
            # Case 3: Log line with timestamp equal to now (should be skipped, so last_time is None)
            ([f"{pendulum.now().to_iso8601_string()} message"], pendulum.now(), [], ["message"]),
            # Case 4: Multiple log lines, last before now
            (
                [
                    f"{pendulum.now().subtract(seconds=3).to_iso8601_string()} msg1",
                    f"{pendulum.now().subtract(seconds=2).to_iso8601_string()} msg2",
                ],
                pendulum.now(),
                ["msg1", "msg2"],
                [],
            ),
            # Case 5: Log lines with continuation (no timestamp)
            (
                [
                    f"{pendulum.now().subtract(seconds=2).to_iso8601_string()} msg1",
                    "continued line",
                ],
                pendulum.now(),
                ["msg1\ncontinued line"],
                [],
            ),
            # Case 6: Log lines with continuation (no timestamp)
            (
                [
                    f"{pendulum.now().subtract(seconds=2).to_iso8601_string()} msg1",
                    f"{pendulum.now().to_iso8601_string()} msg2",
                ],
                pendulum.now(),
                ["msg1"],
                ["msg2"],
            ),
        ],
    )
    async def test_fetch_container_logs_before_current_sec_various_logs(
        self, log_lines, now, expected_log_messages, not_expected_log_messages
    ):
        pod = mock.MagicMock()
        container_name = "base"
        since_time = now.subtract(minutes=1)
        mock_async_hook = mock.AsyncMock()
        mock_async_hook.read_logs.return_value = log_lines

        with mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.pendulum.now", return_value=now):
            async_pod_manager = AsyncPodManager(
                async_hook=mock_async_hook,
                callbacks=[],
            )
            with mock.patch.object(async_pod_manager.log, "log") as mock_log:
                result = await async_pod_manager.fetch_container_logs_before_current_sec(
                    pod=pod, container_name=container_name, since_time=since_time
                )
                assert result == now

                for expected in expected_log_messages:
                    mock_log.assert_any_call(_parse_log_level(expected), "[%s] %s", container_name, expected)
                for not_expected in not_expected_log_messages:
                    unexpected_call = mock.call(
                        _parse_log_level(not_expected), "[%s] %s", container_name, not_expected
                    )
                    assert unexpected_call not in mock_log.mock_calls

    @pytest.mark.asyncio
    async def test_fetch_container_logs_before_current_sec_error_handling(self):
        pod = mock.MagicMock()
        container_name = "base"
        since_time = pendulum.now().subtract(minutes=1)

        async def fake_read_logs(**kwargs):
            raise KubernetesApiError("error")

        self.async_pod_manager._hook.read_logs = fake_read_logs
        with pytest.raises(KubernetesApiError):
            await self.async_pod_manager.fetch_container_logs_before_current_sec(
                pod=pod, container_name=container_name, since_time=since_time
            )


class TestPodLogsConsumer:
    @pytest.mark.parametrize(
        ("chunks", "expected_logs"),
        [
            ([b"message"], [b"message"]),
            ([b"message1\nmessage2"], [b"message1\n", b"message2"]),
            ([b"message1\n", b"message2"], [b"message1\n", b"message2"]),
            ([b"first_part", b"_second_part"], [b"first_part_second_part"]),
            ([b""], [b""]),
        ],
    )
    def test_chunks(self, chunks, expected_logs):
        with mock.patch.object(PodLogsConsumer, "logs_available") as logs_available:
            logs_available.return_value = True
            consumer = PodLogsConsumer(
                response=mock.MagicMock(stream=mock.MagicMock(return_value=chunks)),
                pod=mock.MagicMock(),
                pod_manager=mock.MagicMock(container_is_running=mock.MagicMock(return_value=True)),
                container_name="base",
            )
            assert list(consumer) == expected_logs

    def test_container_is_not_running(self):
        with mock.patch.object(PodLogsConsumer, "logs_available") as logs_available:
            logs_available.return_value = False
            consumer = PodLogsConsumer(
                response=mock.MagicMock(stream=mock.MagicMock(return_value=[b"message1", b"message2"])),
                pod=mock.MagicMock(),
                pod_manager=mock.MagicMock(container_is_running=mock.MagicMock(return_value=False)),
                container_name="base",
            )
            assert list(consumer) == []

    @pytest.mark.parametrize(
        (
            "container_run",
            "termination_time",
            "now_time",
            "post_termination_timeout",
            "expected_logs_available",
        ),
        [
            (
                False,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 1, 0, 0, tzinfo=timezone.utc),
                120,
                True,
            ),
            (
                False,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 2, 0, 0, tzinfo=timezone.utc),
                120,
                False,
            ),
            (
                False,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 5, 0, 0, tzinfo=timezone.utc),
                120,
                False,
            ),
            (
                True,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 1, 0, 0, tzinfo=timezone.utc),
                120,
                True,
            ),
            (
                True,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 2, 0, 0, tzinfo=timezone.utc),
                120,
                True,
            ),
            (
                True,
                datetime(2022, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2022, 1, 1, 0, 5, 0, 0, tzinfo=timezone.utc),
                120,
                True,
            ),
        ],
    )
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.container_is_running")
    @mock.patch("airflow.providers.cncf.kubernetes.utils.pod_manager.get_container_status")
    def test_logs_available(
        self,
        mock_get_container_status,
        mock_container_is_running,
        container_run,
        termination_time,
        now_time,
        post_termination_timeout,
        expected_logs_available,
    ):
        mock_container_is_running.return_value = container_run
        mock_get_container_status.return_value = mock.MagicMock(
            state=mock.MagicMock(terminated=mock.MagicMock(finished_at=termination_time))
        )
        with time_machine.travel(now_time):
            consumer = PodLogsConsumer(
                response=mock.MagicMock(),
                pod=mock.MagicMock(),
                pod_manager=mock.MagicMock(),
                container_name="base",
                post_termination_timeout=post_termination_timeout,
            )
            assert consumer.logs_available() == expected_logs_available

    @pytest.mark.parametrize(
        (
            "read_pod_cache_timeout",
            "mock_read_pod_at_0",
            "mock_read_pod_at_1",
            "mock_read_pods",
            "expected_read_pods",
        ),
        [
            (
                120,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 1, 0, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #0"],
            ),
            (
                120,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 2, 0, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #0"],
            ),
            (
                120,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 3, 0, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #1"],
            ),
            (
                2,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 0, 1, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #0"],
            ),
            (
                2,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 0, 2, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #0"],
            ),
            (
                2,
                datetime(2023, 1, 1, 0, 0, 0, 0, tzinfo=timezone.utc),
                datetime(2023, 1, 1, 0, 0, 3, 0, tzinfo=timezone.utc),
                ["Read pod #0", "Read pod #1"],
                ["Read pod #0", "Read pod #1"],
            ),
        ],
    )
    def test_read_pod(
        self,
        read_pod_cache_timeout,
        mock_read_pod_at_0,
        mock_read_pod_at_1,
        mock_read_pods,
        expected_read_pods,
    ):
        consumer = PodLogsConsumer(
            response=mock.MagicMock(),
            pod=mock.MagicMock(),
            pod_manager=mock.MagicMock(),
            container_name="base",
            read_pod_cache_timeout=read_pod_cache_timeout,
        )
        consumer.pod_manager.read_pod.side_effect = mock_read_pods
        # first read
        with time_machine.travel(mock_read_pod_at_0):
            assert consumer.read_pod() == expected_read_pods[0]

        # second read
        with time_machine.travel(mock_read_pod_at_1):
            assert consumer.read_pod() == expected_read_pods[1]
