scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 49

test(auth): assert reconnect timeout argument in key lookup

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-20 15:53:15 -08:00
parent 14f529a0e9
commit 5e0fbf079c
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/proxy/auth/test_auth_checks.py
View file

@ -177,7 +177,10 @@ async def test_get_key_object_should_reconnect_once_on_db_connection_error():
assert key_obj.token == "hashed-token-1"
assert mock_prisma_client.get_data.await_count == 2
mock_prisma_client.attempt_db_reconnect.assert_awaited_once()
mock_prisma_client.attempt_db_reconnect.assert_awaited_once_with(
reason="auth_get_key_object_lookup_failure",
timeout_seconds=2.0,
)
@pytest.mark.asyncio
@ -199,7 +202,10 @@ async def test_get_key_object_should_raise_if_reconnect_fails_on_db_connection_e
user_api_key_cache=mock_cache,
)
mock_prisma_client.attempt_db_reconnect.assert_awaited_once()
mock_prisma_client.attempt_db_reconnect.assert_awaited_once_with(
reason="auth_get_key_object_lookup_failure",
timeout_seconds=2.0,
)
assert mock_prisma_client.get_data.await_count == 1