phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test - auth exception handler

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-26 18:28:28 -07:00
parent 15b1a8afb0
commit 5242c5fbab
1 changed files with 27 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/litellm/proxy/auth/test_auth_exception_handler.py
View file

@ -101,9 +101,33 @@ def test_should_allow_request_on_db_unavailable_false():
assert handler.should_allow_request_on_db_unavailable() == False assert handler.should_allow_request_on_db_unavailable() == False
# Test _handle_authentication_error method
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_handle_authentication_error_db_unavailable(): @pytest.mark.parametrize(
"prisma_error",
[
PrismaError(),
DataError(data={"user_facing_error": {"meta": {"table": "test_table"}}}),
UniqueViolationError(
data={"user_facing_error": {"meta": {"table": "test_table"}}}
),
ForeignKeyViolationError(
data={"user_facing_error": {"meta": {"table": "test_table"}}}
),
MissingRequiredValueError(
data={"user_facing_error": {"meta": {"table": "test_table"}}}
),
RawQueryError(data={"user_facing_error": {"meta": {"table": "test_table"}}}),
TableNotFoundError(
data={"user_facing_error": {"meta": {"table": "test_table"}}}
),
RecordNotFoundError(
data={"user_facing_error": {"meta": {"table": "test_table"}}}
),
HTTPClientClosedError(),
ClientNotConnectedError(),
],
)
async def test_handle_authentication_error_db_unavailable(prisma_error):
handler = UserAPIKeyAuthExceptionHandler() handler = UserAPIKeyAuthExceptionHandler()
# Mock request and other dependencies # Mock request and other dependencies
@ -118,9 +142,8 @@ async def test_handle_authentication_error_db_unavailable():
"litellm.proxy.proxy_server.general_settings", "litellm.proxy.proxy_server.general_settings",
{"allow_requests_on_db_unavailable": True}, {"allow_requests_on_db_unavailable": True},
): ):
db_error = prisma_errors.PrismaError()
result = await handler._handle_authentication_error( result = await handler._handle_authentication_error(
db_error, prisma_error,
mock_request, mock_request,
mock_request_data, mock_request_data,
mock_route, mock_route,