From 56a2c837b32860f0076a0848658b7a0f80b5b72c Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 5 Nov 2024 19:53:57 -0800 Subject: [PATCH] fix test_proxy_server_prisma_setup_invalid_url --- tests/local_testing/test_proxy_server.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/tests/local_testing/test_proxy_server.py b/tests/local_testing/test_proxy_server.py index 40f02be4f5d..7d652cacef2 100644 --- a/tests/local_testing/test_proxy_server.py +++ b/tests/local_testing/test_proxy_server.py @@ -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)