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 114

test: update tests

Browse source
This commit is contained in:
Krrish Dholakia 2025-08-23 10:49:03 -07:00
parent dba78046b9
commit 3b22fb4215
1 changed files with 0 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
tests/enterprise/litellm_enterprise/proxy/management_endpoints/test_internal_user_endpoints.py
View file

@ -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"]
)