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 109

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2025-02-15 23:57:04 -08:00
parent ce551946c4
commit 4a2e4133c3
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/proxy_admin_ui_tests/test_key_management.py
View file

@ -933,6 +933,7 @@ async def test_list_key_helper(prisma_client):
user_id=None,
team_id=None,
key_alias=None,
organization_id=None,
)
assert len(result["keys"]) == 2, "Should return exactly 2 keys"
assert result["total_count"] >= 5, "Should have at least 5 total keys"
@ -947,6 +948,7 @@ async def test_list_key_helper(prisma_client):
user_id=test_user_id,
team_id=None,
key_alias=None,
organization_id=None,
)
assert len(result["keys"]) == 3, "Should return exactly 3 keys for test user"
@ -958,6 +960,7 @@ async def test_list_key_helper(prisma_client):
user_id=None,
team_id=test_team_id,
key_alias=None,
organization_id=None,
)
assert len(result["keys"]) == 2, "Should return exactly 2 keys for test team"
@ -969,6 +972,7 @@ async def test_list_key_helper(prisma_client):
user_id=None,
team_id=None,
key_alias=test_key_alias,
organization_id=None,
)
assert len(result["keys"]) == 1, "Should return exactly 1 key with test alias"
@ -981,6 +985,7 @@ async def test_list_key_helper(prisma_client):
team_id=None,
key_alias=None,
return_full_object=True,
organization_id=None,
)
assert all(
isinstance(key, UserAPIKeyAuth) for key in result["keys"]
@ -1079,6 +1084,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
team_id=None,
key_alias=None,
return_full_object=True,
organization_id=None,
)
all_keys.extend(result["keys"])