tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-31 20:21:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the test alert sequence

Browse Source 
Also fixed the issue that MagicMock does not create asynchronous mocks by default.
This commit is contained in:
Anand Taralika
2024-05-13 22:43:12 -07:00
parent 3b0c7d2321
commit bd2e4cdfe0
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_alerting.py
+4 -3
View File
@@ -368,9 +368,10 @@ async def test_send_daily_reports_ignores_zero_values():
router.get_model_ids.return_value = ['model1', 'model2', 'model3']
slack_alerting = SlackAlerting(internal_usage_cache=MagicMock())
# model1: failed=None, latency=0; model2: failed=0, latency=0; model3: failed=10, latency=None
slack_alerting.internal_usage_cache.async_batch_get_cache = AsyncMock(return_value=[None, 0, 0, 0, 10, None])
# model1:failed=None, model2:failed=0, model3:failed=10, model1:latency=0; model2:latency=0; model3:latency=None
slack_alerting.internal_usage_cache.async_batch_get_cache = AsyncMock(return_value=[None, 0, 10, 0, 0, None])
slack_alerting.internal_usage_cache.async_batch_set_cache = AsyncMock()
router.get_model_info.side_effect = lambda x: {"litellm_params": {"model": x}}
with patch.object(slack_alerting, 'send_alert', new=AsyncMock()) as mock_send_alert: