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

test(scim): preserve rotation grace period transaction writes

Browse source
This commit is contained in:
Joshua Valluru 2026-09-28 21:56:46 -07:00
parent d362a91d2b
commit f4d6091b6c
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/test_litellm/proxy/management_endpoints/test_key_management_endpoints.py
View file

@ -21239,3 +21239,22 @@ async def test_route_write_rechecks_binding_and_preserves_supported_updates(boun
response = await write
assert response["data"]["allowed_routes"] == routes
tx.litellm_budgettable.update.assert_not_awaited()
@pytest.mark.asyncio
@pytest.mark.parametrize("transactional", [False, True])
async def test_rotation_grace_period_is_written_to_the_selected_database(transactional):
from litellm.proxy.management_endpoints.key_management_endpoints import _insert_deprecated_key
client = _make_regenerate_mock_prisma()
transaction = AsyncMock()
before = datetime.now(timezone.utc)
await _insert_deprecated_key(client, "old-token", "new-token", "1h", tx=transaction if transactional else None)
selected = transaction if transactional else client.db
unused = client.db if transactional else transaction
saved = selected.litellm_deprecatedverificationtoken.upsert.call_args.kwargs
assert saved["where"] == {"token": "old-token"}
assert saved["data"]["create"]["active_token_id"] == "new-token"
assert saved["data"]["update"]["active_token_id"] == "new-token"
assert before + timedelta(hours=1) <= saved["data"]["create"]["revoke_at"] <= datetime.now(timezone.utc) + timedelta(hours=1)
unused.litellm_deprecatedverificationtoken.upsert.assert_not_awaited()