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

test(proxy): explain why the budget hook tests patch module globals

Browse source
This commit is contained in:
Keval Mahajan 2026-08-27 20:52:28 +05:30
parent 96db813262
commit 03c1348153
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/proxy/hooks/test_max_budget_limiter.py
View file

@ -250,10 +250,10 @@ async def test_skips_for_zero_cost_model_when_over_budget():
handler = _PROXY_MaxBudgetLimiter()
user_api_key_dict = _make_user_api_key_auth(user_max_budget=10.0)
with patch(
with patch( # test-quality-ok: hook resolves both collaborators as module globals, no injection seam
"litellm.proxy.auth.auth_checks._is_model_cost_zero",
return_value=True,
), patch(
), patch( # test-quality-ok: proxy module global, no injection seam
"litellm.proxy.proxy_server.get_current_spend",
new=AsyncMock(return_value=10.0),
) as mock_get_spend:
@ -274,10 +274,10 @@ async def test_still_rejects_priced_model_when_over_budget():
handler = _PROXY_MaxBudgetLimiter()
user_api_key_dict = _make_user_api_key_auth(user_max_budget=10.0)
with patch(
with patch( # test-quality-ok: hook resolves both collaborators as module globals, no injection seam
"litellm.proxy.auth.auth_checks._is_model_cost_zero",
return_value=False,
), patch(
), patch( # test-quality-ok: proxy module global, no injection seam
"litellm.proxy.proxy_server.get_current_spend",
new=AsyncMock(return_value=10.0),
), pytest.raises(HTTPException) as exc_info: