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

test(proxy): cover deleted key lookup fallback

Browse source
This commit is contained in:
AxelRay 2026-09-23 16:02:36 +08:00
parent 2a14059c11
commit ed3e209501
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/proxy/management_endpoints/test_common_daily_activity.py
View file

@ -346,6 +346,18 @@ async def test_get_user_api_key_filter_scopes_to_active_and_deleted_user_keys():
assert await get_user_api_key_filter(mock_prisma, "target-user", "unrelated-key") == []
@pytest.mark.asyncio
async def test_get_user_api_key_filter_handles_deleted_lookup_failure_and_owned_key():
mock_prisma = MagicMock()
active_key = SimpleNamespace(token="active-key", user_id="target-user")
mock_prisma.db.litellm_verificationtoken.find_many = AsyncMock(return_value=[active_key])
mock_prisma.db.litellm_deletedverificationtoken.find_many = AsyncMock(
side_effect=RuntimeError("deleted-key table unavailable")
)
assert await get_user_api_key_filter(mock_prisma, "target-user", "active-key") == ["active-key"]
@pytest.mark.asyncio
async def test_get_api_key_metadata_falls_back_to_deleted_keys():
"""Test that get_api_key_metadata should fall back to deleted keys table for missing keys."""