Skip to content

Commit e96a078

Browse files
remove un-necessary tuple response
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 82ca1ee commit e96a078

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

tests/unit/test_thrift_backend.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -929,12 +929,9 @@ def test_handle_execute_response_can_handle_with_direct_results(self):
929929
auth_provider=AuthProvider(),
930930
ssl_options=SSLOptions(),
931931
)
932-
thrift_backend._results_message_to_execute_response = Mock(
933-
return_value=(Mock(), Mock())
934-
)
932+
thrift_backend._results_message_to_execute_response = Mock()
935933

936934
thrift_backend._handle_execute_response(execute_resp, Mock())
937-
938935
thrift_backend._results_message_to_execute_response.assert_called_with(
939936
execute_resp,
940937
ttypes.TOperationState.FINISHED_STATE,
@@ -1738,9 +1735,7 @@ def test_handle_execute_response_sets_active_op_handle(self):
17381735
thrift_backend = self._make_fake_thrift_backend()
17391736
thrift_backend._check_direct_results_for_error = Mock()
17401737
thrift_backend._wait_until_command_done = Mock()
1741-
thrift_backend._results_message_to_execute_response = Mock(
1742-
return_value=(Mock(), Mock())
1743-
)
1738+
thrift_backend._results_message_to_execute_response = Mock()
17441739

17451740
# Create a mock response with a real operation handle
17461741
mock_resp = Mock()

0 commit comments

Comments
 (0)