From da510651e02639e4c3b8a0d8b5f9f6dd154b0cba Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Thu, 20 Feb 2025 18:53:59 -0800 Subject: [PATCH] delete deprecated code test --- .../test_key_management.py | 42 ------------------- 1 file changed, 42 deletions(-) diff --git a/tests/proxy_admin_ui_tests/test_key_management.py b/tests/proxy_admin_ui_tests/test_key_management.py index fbdfd78a882..69353440705 100644 --- a/tests/proxy_admin_ui_tests/test_key_management.py +++ b/tests/proxy_admin_ui_tests/test_key_management.py @@ -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)