From 4fb51162057a404198564cdf2287b5990d866100 Mon Sep 17 00:00:00 2001 From: Vlada Dusek Date: Sun, 11 Oct 2026 23:12:21 +0200 Subject: [PATCH] test: serve the whole-log fallback in the streamed log mocks --- tests/unit/test_logging.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/unit/test_logging.py b/tests/unit/test_logging.py index ea076875..883723ed 100644 --- a/tests/unit/test_logging.py +++ b/tests/unit/test_logging.py @@ -227,6 +227,10 @@ def mock_api(httpserver: HTTPServer) -> None: httpserver.expect_request( f'/v2/actor-runs/{_MOCKED_RUN_ID}/log', method='GET', query_string='stream=true&raw=true' ).respond_with_handler(_streaming_log_handler) + # A streamed log stopped before its first chunk falls back to reading the whole log. + httpserver.expect_request( + f'/v2/actor-runs/{_MOCKED_RUN_ID}/log', method='GET', query_string='raw=true' + ).respond_with_data(b''.join(_MOCKED_ACTOR_LOGS), content_type='application/octet-stream') @pytest.fixture @@ -1137,6 +1141,9 @@ def _status_handler(request: Request) -> Response: httpserver.expect_request( f'/v2/actor-runs/{_MOCKED_RUN_ID}/log', method='GET', query_string='stream=true&raw=true' ).respond_with_handler(_streaming_log_handler) + httpserver.expect_request( + f'/v2/actor-runs/{_MOCKED_RUN_ID}/log', method='GET', query_string='raw=true' + ).respond_with_data(b''.join(_MOCKED_ACTOR_LOGS), content_type='application/octet-stream') @pytest.mark.usefixtures('mock_api_failing_status_poll', 'propagate_stream_logs')