scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(proxy): cover database transport error paths

Browse source
This commit is contained in:
jeevan6996 2026-09-01 00:03:36 +01:00
parent 27e810ea82
commit 2d19f0f65b
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/test_litellm/proxy/db/test_exception_handler.py
View file

@ -76,6 +76,18 @@ def test_is_database_transport_error_non_connection_prisma_errors(prisma_error):
assert PrismaDBExceptionHandler.is_database_transport_error(prisma_error) == False
@pytest.mark.parametrize(
"transport_error",
[
httpx.ConnectError("connection refused"),
ClientNotConnectedError(),
HTTPClientClosedError(),
],
)
def test_is_database_transport_error_connection_errors(transport_error):
assert PrismaDBExceptionHandler.is_database_transport_error(transport_error) is True
def test_is_database_connection_generic_errors():
"""
Test non-Prisma error cases for database connection checking
@ -97,6 +109,8 @@ def test_is_database_connection_generic_errors():
PrismaDBExceptionHandler.is_database_connection_error(db_proxy_exception)
== True
)
assert PrismaDBExceptionHandler.is_database_infrastructure_error(db_proxy_exception)
assert PrismaDBExceptionHandler.is_database_transport_error(db_proxy_exception)
# Test with non-DB error
regular_exception = Exception("Regular error")