Skip to content

Commit 7d29262

Browse files
SK-2548 fix the unit test cases
1 parent 5cedbab commit 7d29262

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

tests/utils/test__utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)