From ab004e5ae32a45fb687e1f3a4df0c771c71f8fdd Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Tue, 29 Sep 2026 19:12:22 +0000 Subject: [PATCH] fix(test): handle fallback authority in team budget tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/test_litellm/proxy/auth/test_auth_checks.py | 3 ++- .../management_helpers/test_management_helpers_utils.py | 9 ++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/proxy/auth/test_auth_checks.py b/tests/test_litellm/proxy/auth/test_auth_checks.py index 582327b3ad0..ee316933e68 100644 --- a/tests/test_litellm/proxy/auth/test_auth_checks.py +++ b/tests/test_litellm/proxy/auth/test_auth_checks.py @@ -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() diff --git a/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py b/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py index 922504ecc58..64f9b3b1540 100644 --- a/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py +++ b/tests/test_litellm/proxy/management_helpers/test_management_helpers_utils.py @@ -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: