From b53c34280a2781575622c660df3ea2cc796d4db3 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Fri, 20 Feb 2026 17:44:28 -0800 Subject: [PATCH] test(auth): assert reconnect lock timeout argument --- tests/test_litellm/proxy/auth/test_auth_checks.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py index 4c2909384bd..1d8d1be58c7 100644 --- a/tests/test_litellm/proxy/auth/test_auth_checks.py +++ b/tests/test_litellm/proxy/auth/test_auth_checks.py @@ -180,6 +180,7 @@ async def test_get_key_object_should_reconnect_once_on_db_connection_error(): mock_prisma_client.attempt_db_reconnect.assert_awaited_once_with( reason="auth_get_key_object_lookup_failure", timeout_seconds=2.0, + lock_timeout_seconds=0.1, ) @@ -205,6 +206,7 @@ async def test_get_key_object_should_raise_if_reconnect_fails_on_db_connection_e mock_prisma_client.attempt_db_reconnect.assert_awaited_once_with( reason="auth_get_key_object_lookup_failure", timeout_seconds=2.0, + lock_timeout_seconds=0.1, ) assert mock_prisma_client.get_data.await_count == 1