scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix test_proxy_server_prisma_setup_invalid_url

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-05 19:53:57 -08:00
parent f00444a98c
commit 56a2c837b3
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/local_testing/test_proxy_server.py
View file

@ -1935,18 +1935,23 @@ async def test_proxy_server_prisma_setup_invalid_url():
from litellm.proxy.proxy_server import ProxyStartupEvent
from litellm.proxy.utils import ProxyLogging
from litellm.caching import DualCache
from litellm._logging import verbose_proxy_logger
import logging
verbose_proxy_logger.setLevel(logging.DEBUG)
user_api_key_cache = DualCache()
invalid_db_url = "postgresql://invalid:5432/nonexistent"
proxy_logging_obj = ProxyLogging(user_api_key_cache=user_api_key_cache)
# Test that it raises an exception with invalid URL
with pytest.raises(Exception) as exc_info:
try:
# Test that it raises an exception with invalid URL
await ProxyStartupEvent._setup_prisma_client(
database_url=invalid_db_url,
proxy_logging_obj=proxy_logging_obj,
user_api_key_cache=user_api_key_cache,
)
# Verify the error message indicates a connection failure
assert "Could not connect to the query engine" in str(exc_info.value)
assert False, "Expected an exception to be raised"
except Exception as e:
assert "Could not connect to the query engine" in str(e)