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

fix(test): handle fallback authority in team budget tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-29 19:12:22 +00:00
parent 1b3e01aa4c
commit ab004e5ae3
2 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_litellm/proxy/auth/test_auth_checks.py
View file

@ -8045,7 +8045,7 @@ async def test_check_team_member_budget_missing_membership_is_verified_with_unav
patch.object(proxy_server, "prisma_client", prisma_client),
patch.object(proxy_server, "spend_counter_cache", _unavailable_spend_counter_cache()),
):
await _check_team_member_budget(
result: Final = await _check_team_member_budget(
team_object=LiteLLM_TeamTable(
team_id="test-team",
metadata={"team_member_budget_id": "default-budget-100"},
@ -8059,6 +8059,7 @@ async def test_check_team_member_budget_missing_membership_is_verified_with_unav
team_membership_loaded=True,
)
assert result is None
membership_find_unique.assert_awaited()

9
tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py
View file

@ -884,7 +884,14 @@ async def test_team_update_reaches_inherited_members_but_not_overridden_ones():
team_member_budget=1.0,
)
async def spend_from_membership(counter_key: str, fallback_spend: float, max_budget: float | None = None) -> float:
async def spend_from_membership(
counter_key: str,
fallback_spend: float,
max_budget: float | None = None,
*,
fallback_authoritative: bool,
) -> float:
assert not fallback_authoritative
return fallback_spend
async def check(user_id: str, spend: float) -> None: