remove excess fetch_results mocks

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

tests/unit/test_client.py

@@ -213,7 +213,6 @@ def test_closing_result_set_hard_closes_commands(self):
         type(mock_connection).session = PropertyMock(return_value=mock_session)
 
         mock_thrift_backend.fetch_results.return_value = (Mock(), False)
-
         result_set = ThriftResultSet(
             mock_connection, mock_results_response, mock_thrift_backend
         )
@@ -479,7 +478,6 @@ def make_fake_row_slice(n_rows):
         mock_aq = Mock()
         mock_aq.next_n_rows.side_effect = make_fake_row_slice
         mock_thrift_backend.execute_command.return_value.arrow_queue = mock_aq
-        mock_thrift_backend.fetch_results.return_value = (mock_aq, True)
 
         cursor = client.Cursor(Mock(), mock_thrift_backend)
         cursor.execute("foo")

tests/unit/test_thrift_backend.py

@@ -1177,8 +1177,6 @@ def test_execute_statement_calls_client_and_handle_execute_response(
         )
         cursor_mock = Mock()
 
-        thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
         result = thrift_backend.execute_command(
             "foo", Mock(), 100, 200, Mock(), cursor_mock
         )
@@ -1214,8 +1212,6 @@ def test_get_catalogs_calls_client_and_handle_execute_response(
         )
         cursor_mock = Mock()
 
-        thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
         result = thrift_backend.get_catalogs(Mock(), 100, 200, cursor_mock)
         # Verify the result is a ResultSet
         self.assertIsInstance(result, ResultSet)
@@ -1247,8 +1243,6 @@ def test_get_schemas_calls_client_and_handle_execute_response(
         )
         cursor_mock = Mock()
 
-        thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
         result = thrift_backend.get_schemas(
             Mock(),
             100,
@@ -1289,8 +1283,6 @@ def test_get_tables_calls_client_and_handle_execute_response(
         )
         cursor_mock = Mock()
 
-        thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
         result = thrift_backend.get_tables(
             Mock(),
             100,
@@ -1335,8 +1327,6 @@ def test_get_columns_calls_client_and_handle_execute_response(
         )
         cursor_mock = Mock()
 
-        thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
         result = thrift_backend.get_columns(
             Mock(),
             100,
@@ -2261,8 +2251,6 @@ def test_execute_command_sets_complex_type_fields_correctly(
                 **complex_arg_types,
             )
 
-            thrift_backend.fetch_results = Mock(return_value=(Mock(), False))
-
             thrift_backend.execute_command(Mock(), Mock(), 100, 100, Mock(), Mock())
             t_execute_statement_req = tcli_service_instance.ExecuteStatement.call_args[
                 0