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): verify reconnect cooldown timestamp set after attempt

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

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

@ -66,6 +66,27 @@ async def test_attempt_db_reconnect_should_skip_when_in_cooldown(mock_proxy_logg
client.health_check.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)
client._db_last_reconnect_attempt_ts = 0.0
client._db_reconnect_cooldown_seconds = 10
client.db.disconnect = AsyncMock(return_value=None)
client.db.connect = AsyncMock(return_value=None)
client.db.query_raw = AsyncMock(return_value=[{"result": 1}])
with patch(
"litellm.proxy.utils.time.time", side_effect=[100.0, 101.0, 150.0, 200.0]
):
result = await client.attempt_db_reconnect(
reason="unit_test_cooldown_timestamp_after_attempt",
timeout_seconds=0.1,
)
assert result is True
assert client._db_last_reconnect_attempt_ts == 200.0
@pytest.mark.asyncio
async def test_db_health_watchdog_should_trigger_reconnect_on_db_error(mock_proxy_logging):
client = PrismaClient(database_url="mock://test", proxy_logging_obj=mock_proxy_logging)