call correct method in test_fall_back_to_hive_schema

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

Author: varun-edachali-dbx

tests/unit/test_thrift_backend.py

@@ -965,7 +965,6 @@ def test_use_arrow_schema_if_available(self, tcli_service_class):
 
         thrift_backend = self._create_thrift_client()
 
-        # Call the real _results_message_to_execute_response method
         execute_response, _ = thrift_backend._handle_execute_response(
             t_execute_resp, Mock()
         )
@@ -1000,10 +999,7 @@ def test_fall_back_to_hive_schema_if_no_arrow_schema(self, tcli_service_class):
         thrift_backend = self._create_thrift_client()
         thrift_backend._hive_schema_to_arrow_schema = Mock()
 
-        # Call the real _results_message_to_execute_response method
-        thrift_backend._results_message_to_execute_response(
-            t_execute_resp, ttypes.TOperationState.FINISHED_STATE
-        )
+        thrift_backend._handle_execute_response(t_execute_resp, Mock())
 
         self.assertEqual(
             hive_schema_mock,