tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-12 20:22:46 +00:00
Code Issues Packages Projects Releases Wiki 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
litellm/tests/test_alerting.py
+5 -3
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()