@@ -615,7 +615,7 @@ def test_handle_exception_connect_error(self, mock_log_and_reject_error):
615615 handle_exception (mock_error , mock_logger )
616616
617617 mock_log_and_reject_error .assert_called_once_with (
618- SkyflowMessages . Error . GENERIC_API_ERROR . value ,
618+ 'Connection refused' ,
619619 SkyflowMessages .ErrorCodes .INVALID_INPUT .value ,
620620 None ,
621621 logger = mock_logger
@@ -688,7 +688,7 @@ def test_handle_exception_generic_error_with_status(self, mock_log_and_reject_er
688688 handle_exception (mock_error , mock_logger )
689689
690690 mock_log_and_reject_error .assert_called_once_with (
691- SkyflowMessages . Error . GENERIC_API_ERROR . value ,
691+ str ( mock_error ) ,
692692 503 ,
693693 "1234" ,
694694 logger = mock_logger
@@ -706,7 +706,7 @@ def test_handle_exception_no_content_type(self, mock_log_and_reject_error):
706706 handle_exception (mock_error , mock_logger )
707707
708708 mock_log_and_reject_error .assert_called_once_with (
709- SkyflowMessages . Error . GENERIC_API_ERROR . value ,
709+ str ( mock_error ) ,
710710 500 ,
711711 "1234" ,
712712 logger = mock_logger
@@ -812,7 +812,7 @@ def test_handle_generic_error_with_status(self, mock_log_and_reject_error):
812812 handle_generic_error_with_status (mock_error , request_id , status , mock_logger )
813813
814814 mock_log_and_reject_error .assert_called_once_with (
815- SkyflowMessages . Error . GENERIC_API_ERROR . value ,
815+ str ( mock_error ) ,
816816 503 ,
817817 request_id ,
818818 logger = mock_logger
0 commit comments