From cba2b729b26331b381ce4a68ae4adda827bdb742 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 20 May 2024 15:37:35 -0700 Subject: [PATCH] test(test_alerting.py): testing fixes --- litellm/tests/test_alerting.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) diff --git a/litellm/tests/test_alerting.py b/litellm/tests/test_alerting.py index 7d277f3e5f..b1887f73d2 100644 --- a/litellm/tests/test_alerting.py +++ b/litellm/tests/test_alerting.py @@ -124,7 +124,9 @@ from datetime import datetime, timedelta @pytest.fixture def slack_alerting(): - return SlackAlerting(alerting_threshold=1, internal_usage_cache=DualCache()) + return SlackAlerting( + alerting_threshold=1, internal_usage_cache=DualCache(), alerting=["slack"] + ) # Test for hanging LLM responses @@ -162,7 +164,10 @@ async def test_budget_alerts_crossed(slack_alerting): user_current_spend = 101 with patch.object(slack_alerting, "send_alert", new=AsyncMock()) as mock_send_alert: await slack_alerting.budget_alerts( - "user_budget", user_max_budget, user_current_spend + "user_budget", + user_max_budget, + user_current_spend, + user_info=CallInfo(token="", spend=0), ) mock_send_alert.assert_awaited_once() @@ -174,12 +179,18 @@ async def test_budget_alerts_crossed_again(slack_alerting): user_current_spend = 101 with patch.object(slack_alerting, "send_alert", new=AsyncMock()) as mock_send_alert: await slack_alerting.budget_alerts( - "user_budget", user_max_budget, user_current_spend + "user_budget", + user_max_budget, + user_current_spend, + user_info=CallInfo(token="", spend=0), ) mock_send_alert.assert_awaited_once() mock_send_alert.reset_mock() await slack_alerting.budget_alerts( - "user_budget", user_max_budget, user_current_spend + "user_budget", + user_max_budget, + user_current_spend, + user_info=CallInfo(token="", spend=0), ) mock_send_alert.assert_not_awaited()