tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-01 22:22:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): add missing start_db_health_watchdog_task mock (#21804)

Browse Source 
* fix(tests): add missing start_db_health_watchdog_task mock in test_proxy_server_prisma_setup

* fix(tests): add missing start_db_health_watchdog_task mock in test_health_check_not_called_when_disabled
This commit is contained in:
Ishaan Jaff
2026-02-21 12:31:52 -08:00
committed by GitHub
parent 0b69c21eca
commit daa682e125
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/proxy_unit_tests/test_proxy_server.py
+1
View File
@@ -2205,6 +2205,7 @@ async def test_proxy_server_prisma_setup():
mock_client._set_spend_logs_row_count_in_proxy_state = (
AsyncMock()
) # Mock the _set_spend_logs_row_count_in_proxy_state method
mock_client.start_db_health_watchdog_task = AsyncMock()
# Mock the db attribute with start_token_refresh_task for RDS IAM token refresh
mock_db = MagicMock()
mock_db.start_token_refresh_task = AsyncMock()
tests/proxy_unit_tests/test_proxy_utils.py
+1
View File
@@ -1623,6 +1623,7 @@ async def test_health_check_not_called_when_disabled(monkeypatch):
mock_prisma.health_check = AsyncMock()
mock_prisma.check_view_exists = AsyncMock()
mock_prisma._set_spend_logs_row_count_in_proxy_state = AsyncMock()
mock_prisma.start_db_health_watchdog_task = AsyncMock()
# Mock the db attribute with start_token_refresh_task for RDS IAM token refresh
mock_db = MagicMock()
mock_db.start_token_refresh_task = AsyncMock()