From 3447c3529a273a984f442690a27654c55e759e79 Mon Sep 17 00:00:00 2001 From: agustin18 Date: Mon, 28 Sep 2026 20:04:59 +0000 Subject: [PATCH] test(timeout): test timeout extension against regressions with delayed call Signed-off-by: agustin18 --- tests/unit/test_timeout.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_timeout.py b/tests/unit/test_timeout.py index 160dd49f3eb..4f8a247f90c 100644 --- a/tests/unit/test_timeout.py +++ b/tests/unit/test_timeout.py @@ -25,12 +25,16 @@ async def test_async_timeout_decorator_enforces_per_call_timeout(timeout_arg: st @pytest.mark.asyncio -async def test_async_timeout_decorator_extends_duration_with_per_call_timeout() -> None: +@pytest.mark.parametrize("timeout_arg", ["request_timeout", "force_timeout"]) +async def test_async_timeout_decorator_extends_duration_with_per_call_timeout( + timeout_arg: str, +) -> None: @timeout(timeout_duration=0.001) - async def fast_func(**kwargs): + async def delayed_func(**kwargs): + await asyncio.sleep(0.01) return "completed" - result: Final = await fast_func(request_timeout=60.0, model="test-model") + result: Final = await delayed_func(**{timeout_arg: 60.0, "model": "test-model"}) assert result == "completed"