76e88cae2a
Twenty-three tests imported helpers from sibling test modules by dotted path (tests.run_agent.test_run_agent, tests.cli.test_cli_init, tests.gateway.test_42039_duplicate_user_message); those follow the moves and renames. The run_agent autouse fixture that zeroes jittered_backoff now covers all of tests/agent, where one pre-existing test asserts the real backoff is positive — it opts out via a real_retry_backoff marker, the same pattern real_concurrent_gate and real_agent_prewarm use.
40 lines
1.7 KiB
Python
40 lines
1.7 KiB
Python
"""Fast-path fixtures shared across tests/agent/.
|
|
|
|
Many tests in this directory exercise the retry/backoff paths in the
|
|
agent loop. Production code uses ``jittered_backoff(base_delay=5.0)``
|
|
with a ``while time.time() < sleep_end`` loop — a single retry test
|
|
spends 5+ seconds of real wall-clock time on backoff waits.
|
|
|
|
Mocking ``jittered_backoff`` to return 0.0 collapses the while-loop
|
|
to a no-op (``time.time() < time.time() + 0`` is false immediately),
|
|
which handles the most common case without touching ``time.sleep``.
|
|
|
|
We deliberately DO NOT mock ``time.sleep`` here — some tests
|
|
(test_interrupt_propagation, test_primary_runtime_restore, etc.) use
|
|
the real ``time.sleep`` for threading coordination or assert that it
|
|
was called with specific values. Tests that want to additionally
|
|
fast-path direct ``time.sleep(N)`` calls in production code should
|
|
monkeypatch ``run_agent.time.sleep`` locally (see
|
|
``test_anthropic_error_handling.py`` for the pattern).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _fast_retry_backoff(request, monkeypatch):
|
|
"""Short-circuit retry backoff for all tests in this directory.
|
|
|
|
Tests that assert on the real backoff value opt out with
|
|
``@pytest.mark.real_retry_backoff``.
|
|
"""
|
|
if request.node.get_closest_marker("real_retry_backoff"):
|
|
return
|
|
# The agent.turn_* retry paths import ``jittered_backoff`` lazily from
|
|
# ``agent.retry_utils``; patch it there so rate-limit / invalid-response /
|
|
# server-error retries don't burn real wall-clock seconds.
|
|
from agent import retry_utils as _retry_utils
|
|
monkeypatch.setattr(_retry_utils, "jittered_backoff", lambda *a, **k: 0.0)
|