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

test_list_key_helper_team_filtering

Browse source
This commit is contained in:
Ishaan Jaff 2025-02-15 16:41:47 -08:00
parent a0a1784997
commit 707b885a3f
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/proxy_admin_ui_tests/test_key_management.py
View file

@ -1011,6 +1011,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
from litellm.proxy.management_endpoints.key_management_endpoints import (
_list_key_helper,
)
import uuid
# Setup
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
@ -1024,7 +1025,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
for i in range(3):
key = await generate_key_fn(
data=GenerateKeyRequest(
key_alias=f"no_team_key_{i}",
key_alias=f"no_team_key_{i}.{uuid.uuid4()}",
),
user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN,
@ -1039,7 +1040,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
key = await generate_key_fn(
data=GenerateKeyRequest(
team_id="litellm-dashboard",
key_alias=f"dashboard_key_{i}",
key_alias=f"dashboard_key_{i}.{uuid.uuid4()}",
),
user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN,
@ -1055,7 +1056,7 @@ async def test_list_key_helper_team_filtering(prisma_client):
key = await generate_key_fn(
data=GenerateKeyRequest(
team_id=other_team_id,
key_alias=f"other_team_key_{i}",
key_alias=f"other_team_key_{i}.{uuid.uuid4()}",
),
user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN,