File tree Expand file tree Collapse file tree 1 file changed +2
-7
lines changed
Expand file tree Collapse file tree 1 file changed +2
-7
lines changed Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments