From 3b22fb4215596596f6142bab25d201580102cc40 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 23 Aug 2025 10:49:03 -0700 Subject: [PATCH] test: update tests --- .../test_internal_user_endpoints.py | 24 ------------------- 1 file changed, 24 deletions(-) diff --git a/tests/enterprise/litellm_enterprise/proxy/management_endpoints/test_internal_user_endpoints.py b/tests/enterprise/litellm_enterprise/proxy/management_endpoints/test_internal_user_endpoints.py index d69c613a4f5..69c3b4cb59a 100644 --- a/tests/enterprise/litellm_enterprise/proxy/management_endpoints/test_internal_user_endpoints.py +++ b/tests/enterprise/litellm_enterprise/proxy/management_endpoints/test_internal_user_endpoints.py @@ -159,27 +159,3 @@ class TestAvailableEnterpriseUsers: CommonProxyErrors.db_not_connected_error.value in response.json()["detail"]["error"] ) - - @pytest.mark.asyncio - async def test_available_users_not_premium_user( - self, client, mock_user_api_key_auth - ): - """Test when premium_user is None (not a premium user)""" - from litellm.proxy._types import CommonProxyErrors - - with patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma, patch( - "litellm.proxy.proxy_server.premium_user", - None, - ): - # Override the dependency - client.app.dependency_overrides[mock_user_api_key_auth] = lambda: { - "user_id": "test_user" - } - - response = client.get("/user/available_users") - - assert response.status_code == 500 - assert ( - CommonProxyErrors.not_premium_user.value - in response.json()["detail"]["error"] - )