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(proxy): add reconnect lock-timeout behavior coverage

Browse source
This commit is contained in:
Ishaan Jaffer 2026-02-20 17:44:23 -08:00
parent 3bb2423ab8
commit f2883192cd
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/test_litellm/proxy/db/test_prisma_self_heal.py
View file

@ -66,6 +66,32 @@ async def test_attempt_db_reconnect_should_skip_when_in_cooldown(mock_proxy_logg
client.db.query_raw.assert_not_called()
@pytest.mark.asyncio
async def test_attempt_db_reconnect_should_skip_when_lock_timeout_expires(
mock_proxy_logging,
):
client = PrismaClient(database_url="mock://test", proxy_logging_obj=mock_proxy_logging)
client.db.disconnect = AsyncMock(return_value=None)
client.db.connect = AsyncMock(return_value=None)
client.db.query_raw = AsyncMock(return_value=[{"result": 1}])
await client._db_reconnect_lock.acquire()
try:
result = await client.attempt_db_reconnect(
reason="unit_test_reconnect_lock_timeout",
force=True,
timeout_seconds=0.1,
lock_timeout_seconds=0.01,
)
finally:
client._db_reconnect_lock.release()
assert result is False
client.db.disconnect.assert_not_called()
client.db.connect.assert_not_called()
client.db.query_raw.assert_not_called()
@pytest.mark.asyncio
async def test_attempt_db_reconnect_should_set_cooldown_after_attempt(mock_proxy_logging):
client = PrismaClient(database_url="mock://test", proxy_logging_obj=mock_proxy_logging)