From bd2e4cdfe00811d67bd8c14400f9bfeee9fec72d Mon Sep 17 00:00:00 2001 From: Anand Taralika Date: Mon, 13 May 2024 22:43:12 -0700 Subject: [PATCH] Fixed the test alert sequence Also fixed the issue that MagicMock does not create asynchronous mocks by default. --- litellm/tests/test_alerting.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_alerting.py b/litellm/tests/test_alerting.py index b3518e012f..6770618423 100644 --- a/litellm/tests/test_alerting.py +++ b/litellm/tests/test_alerting.py @@ -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: