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

delete deprecated code test

Browse source
This commit is contained in:
Ishaan Jaff 2025-02-20 18:53:59 -08:00
parent d23d4305fb
commit da510651e0
1 changed files with 0 additions and 42 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
tests/proxy_admin_ui_tests/test_key_management.py
View file

@ -738,48 +738,6 @@ def test_prepare_metadata_fields(
assert updated_non_default_values == expected_result
@pytest.mark.asyncio
async def test_user_info_as_proxy_admin(prisma_client):
"""
Test /user/info endpoint as a proxy admin without passing a user ID.
Verifies that the endpoint returns all teams and keys.
"""
litellm.set_verbose = True
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)
setattr(litellm.proxy.proxy_server, "master_key", "sk-1234")
await litellm.proxy.proxy_server.prisma_client.connect()
# Call user_info as a proxy admin without a user_id
user_info_response = await user_info(
user_id=None,
user_api_key_dict=UserAPIKeyAuth(
user_role=LitellmUserRoles.PROXY_ADMIN,
api_key="sk-1234",
user_id="admin",
),
)
print("user info response: ", user_info_response.model_dump_json(indent=4))
# Verify response
assert user_info_response.user_id is None
assert user_info_response.user_info is None
# Verify that teams and keys are returned
assert user_info_response.teams is not None
assert len(user_info_response.teams) > 0, "Expected at least one team in response"
# assert that the teams are sorted by team_alias
team_aliases = [
getattr(team, "team_alias", "") or "" for team in user_info_response.teams
]
print("Team aliases order in response=", team_aliases)
assert team_aliases == sorted(team_aliases), "Teams are not sorted by team_alias"
assert user_info_response.keys is not None
assert len(user_info_response.keys) > 0, "Expected at least one key in response"
@pytest.mark.asyncio
async def test_key_update_with_model_specific_params(prisma_client):
setattr(litellm.proxy.proxy_server, "prisma_client", prisma_client)