scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

address greptile issues - Add async health check tests for AktoLogger

Browse source
This commit is contained in:
rzeta-10 2026-03-23 22:48:18 +05:30
parent 179af2e7d1
commit acfa0f4531
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/logging_callback_tests/test_akto_logger.py
View file

@ -335,3 +335,29 @@ async def test_async_log_failure_swallows_errors(logger):
start_time=None,
end_time=None,
)
# ── Health check ──
@pytest.mark.asyncio
async def test_async_health_check_healthy(logger):
logger.async_http_handler.get = AsyncMock(return_value=MagicMock(status_code=200))
result = await logger.async_health_check()
assert result["status"] == "healthy"
assert result["error_message"] is None
@pytest.mark.asyncio
async def test_async_health_check_unhealthy(logger):
logger.async_http_handler.get = AsyncMock(return_value=MagicMock(status_code=503))
result = await logger.async_health_check()
assert result["status"] == "unhealthy"
assert "503" in result["error_message"]
@pytest.mark.asyncio
async def test_async_health_check_exception(logger):
logger.async_http_handler.get = AsyncMock(side_effect=httpx.ConnectError("refused"))
result = await logger.async_health_check()
assert result["status"] == "unhealthy"