diff --git a/tests/proxy_behavior/management/test_team_member_reset_budget.py b/tests/proxy_behavior/management/test_team_member_reset_budget.py index 42f327c33ef..1e55b8b6b15 100644 --- a/tests/proxy_behavior/management/test_team_member_reset_budget.py +++ b/tests/proxy_behavior/management/test_team_member_reset_budget.py @@ -117,8 +117,6 @@ async def test_team_member_reset_budget_authz_matrix( async def test_team_member_reset_budget_leaves_shared_default_row_untouched(proxy_client, prisma, scratch, world): - """Relinking must point the member at the shared row, not copy or edit it, so a later - /team/update to team_member_budget reaches this member again.""" member_id = scratch.tag("member") default_budget_id = await _seed_team_with_default_budget(prisma, world, "alpha", scratch.prefix, scratch) await _seed_custom_member(prisma, scratch.prefix, member_id, scratch) @@ -161,8 +159,6 @@ async def test_team_member_reset_budget_without_team_default_detaches_member(pro async def test_team_member_reset_budget_with_deleted_team_default_detaches_member(proxy_client, prisma, scratch, world): - """metadata.team_member_budget_id can outlive its budget row; a stale id must not be - relinked to (the FK would fail) and must read as no budget, not as the team default.""" member_id = scratch.tag("member") await create_scratch_team( prisma, 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 484c054fa54..1f19163933a 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_team_endpoints.py @@ -14443,10 +14443,6 @@ def _team_with_default_budget(team_id: str, budget_id: str) -> LiteLLM_TeamTable @pytest.mark.asyncio async def test_reset_team_member_budget_fn_relinks_custom_member_to_team_default(monkeypatch): - """An admin undoing a per-member budget must put the membership back on the team's shared - default row (a connect, not a copy) so later /team/update changes reach the member again, - and must drop the cached membership so the old cap stops being enforced. The shared row and - the member's tracked spend are never written.""" from litellm.proxy.common_utils.user_api_key_cache import UserApiKeyCache mock_prisma_client = MagicMock() @@ -14498,9 +14494,6 @@ async def test_reset_team_member_budget_fn_relinks_custom_member_to_team_default async def test_reset_team_member_budget_fn_detaches_member_when_team_has_no_usable_default( monkeypatch, team_obj, default_row ): - """With no shared default to link to, reset leaves the member exactly where a freshly added - member would be: no budget row at all, reported as budget_source='none', rather than - connecting to a budget_id that does not exist or leaving the custom cap in place.""" mock_prisma_client = MagicMock() membership_row = LiteLLM_TeamMembership(user_id="member-1", team_id="team-1", budget_id="custom-b1") mock_prisma_client.db.litellm_teammembership.find_unique = AsyncMock(return_value=membership_row) @@ -14604,9 +14597,6 @@ async def _team_info_budget_sources( @pytest.mark.asyncio async def test_team_info_reports_whether_each_member_follows_the_team_default_budget(): - """/team/info must tell the caller which members still follow the team's shared member budget - and which carry their own row, since budget_id alone only means something to a reader who - also knows the team's team_member_budget_id.""" sources = await _team_info_budget_sources( team_row=_team_with_default_budget("team-1", "team-default-b"), memberships=[ @@ -14626,9 +14616,6 @@ async def test_team_info_reports_whether_each_member_follows_the_team_default_bu @pytest.mark.asyncio async def test_team_info_reports_no_budget_source_when_team_has_no_default(): - """A team that never set team_member_budget has nothing for members to inherit, so an - unlinked member is 'none' rather than 'team_default', while a member with their own row is - still 'custom'.""" sources = await _team_info_budget_sources( team_row=LiteLLM_TeamTable(team_id="team-1"), memberships=[ @@ -14646,9 +14633,6 @@ async def test_team_info_reports_no_budget_source_when_team_has_no_default(): @pytest.mark.asyncio async def test_team_info_reports_no_budget_source_when_team_default_row_was_deleted(): - """If the budget row named by team_member_budget_id was removed via /budget/delete, nothing is - enforced for unlinked members any more, so /team/info must not keep advertising a team default - that no longer exists.""" sources = await _team_info_budget_sources( team_row=_team_with_default_budget("team-1", "deleted-b"), memberships=[