phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_alerting.py): fix tests

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-14 16:20:54 -07:00
parent e18002023a
commit 9e8eacbc77
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/tests/test_alerting.py
View file

@ -227,7 +227,9 @@ async def test_send_alert(slack_alerting):
slack_alerting.async_http_handler, "post", new=AsyncMock()
) as mock_post:
mock_post.return_value.status_code = 200
await slack_alerting.send_alert("Test message", "Low", "budget_alerts")
await slack_alerting.send_alert(
"Test message", "Low", "budget_alerts", alerting_metadata={}
)
mock_post.assert_awaited_once()
@ -286,7 +288,7 @@ async def test_daily_reports_completion(slack_alerting):
await asyncio.sleep(3)
response_val = await slack_alerting.send_daily_reports(router=router)
assert response_val == True
assert response_val is True
mock_send_alert.assert_awaited_once()
@ -311,7 +313,7 @@ async def test_daily_reports_completion(slack_alerting):
await asyncio.sleep(3)
response_val = await slack_alerting.send_daily_reports(router=router)
assert response_val == True
assert response_val is True
mock_send_alert.assert_awaited()