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

test(proxy): lock plain member updates to the merge-patch contract without a team default snapshot

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-18 01:28:49 +00:00
parent 569ef3ea25
commit 3926e74e9d
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/test_litellm/proxy/common_utils/test_upsert_budget_membership.py
View file

@ -280,6 +280,29 @@ async def test_create_from_temp_pair_skips_zero_team_default_cap(mock_tx, fake_u
mock_tx.litellm_teammembership.upsert.assert_awaited_once()
@pytest.mark.asyncio
async def test_create_from_plain_patch_does_not_snapshot_team_default(mock_tx, fake_user):
mock_tx.litellm_budgettable.find_unique = AsyncMock(
return_value=budget_row(budget_id="team-default-budget-1", max_budget=0.4, rpm_limit=10)
)
await _upsert_budget_and_membership(
mock_tx,
team_id="team-default",
user_id="user-unlinked",
existing_budget_id=None,
user_api_key_dict=fake_user,
budget_patch={"tpm_limit": 500},
team_default_budget_id="team-default-budget-1",
)
mock_tx.litellm_budgettable.find_unique.assert_not_awaited()
data = mock_tx.litellm_budgettable.create.await_args.kwargs["data"]
assert data["tpm_limit"] == 500
assert "max_budget" not in data
assert "rpm_limit" not in data
mock_tx.litellm_teammembership.upsert.assert_awaited_once()
# TEST: clone-on-write when the membership still points at the team's shared
# default budget. Editing this member must fork a private budget instead of
# mutating the shared row, and cloning a duration must seed a fresh reset time.