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):
929
929
auth_provider = AuthProvider (),
930
930
ssl_options = SSLOptions (),
931
931
)
932
- thrift_backend ._results_message_to_execute_response = Mock (
933
- return_value = (Mock (), Mock ())
934
- )
932
+ thrift_backend ._results_message_to_execute_response = Mock ()
935
933
936
934
thrift_backend ._handle_execute_response (execute_resp , Mock ())
937
-
938
935
thrift_backend ._results_message_to_execute_response .assert_called_with (
939
936
execute_resp ,
940
937
ttypes .TOperationState .FINISHED_STATE ,
@@ -1738,9 +1735,7 @@ def test_handle_execute_response_sets_active_op_handle(self):
1738
1735
thrift_backend = self ._make_fake_thrift_backend ()
1739
1736
thrift_backend ._check_direct_results_for_error = Mock ()
1740
1737
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 ()
1744
1739
1745
1740
# Create a mock response with a real operation handle
1746
1741
mock_resp = Mock ()
You can’t perform that action at this time.
0 commit comments