From b8a9605c3c2ac86763bbbfa5e5dcc74ec20891a4 Mon Sep 17 00:00:00 2001 From: RoyVivat Date: Thu, 2 Apr 2026 15:14:04 -0700 Subject: [PATCH] test: restore mock_new_budget.called assertion dropped in previous fix Co-Authored-By: Claude Sonnet 4.6 --- .../proxy/management_endpoints/test_team_endpoints.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py index b2a792a370a..5c97495f203 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -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"