scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-05 02:41:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 48

test: restore mock_new_budget.called assertion dropped in previous fix

Browse source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
RoyVivat 2026-04-02 15:14:04 -07:00
parent 34f2284579
commit b8a9605c3c
No known key found for this signature in database
GPG key ID: 59743472EC86530E
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py
View file

@ -1650,7 +1650,7 @@ async def test_upsert_team_member_budget_table_no_existing_budget():
"litellm.proxy.management_endpoints.budget_management_endpoints.new_budget",
new_callable=AsyncMock,
return_value=mock_budget_response,
),
) as mock_new_budget,
patch(
"litellm.proxy.proxy_server.prisma_client",
mock_prisma,
@ -1664,6 +1664,7 @@ async def test_upsert_team_member_budget_table_no_existing_budget():
team_member_budget_duration="45d",
)
assert mock_new_budget.called
assert "team_member_budget_id" in result["metadata"]
assert result["metadata"]["team_member_budget_id"] == "new_budget_456"