redundancies + op_handle -> command_id

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

Author: varun-edachali-dbx

src/databricks/sql/client.py

@@ -876,7 +876,7 @@ def get_async_execution_result(self):
         operation_state = self.get_query_state()
         if operation_state == CommandState.SUCCEEDED:
             self.active_result_set = self.backend.get_execution_result(
-                self.active_op_handle, self
+                self.active_command_id, self
             )
 
             if self.active_result_set and self.active_result_set.is_staging_operation:

src/databricks/sql/utils.py

@@ -74,7 +74,7 @@ def build_queue(
         Returns:
             ResultSetQueue
         """
-        # Handle Thrift result data
+
         if (
             row_set_type == TSparkRowSetType.ARROW_BASED_SET
             and t_row_set is not None

tests/unit/test_result_set_queue_factories.py

@@ -3,32 +3,17 @@
 """
 
 import unittest
-from unittest.mock import MagicMock, patch
-import sys
-from typing import Dict, List, Any, Optional
-
-# Add the necessary path to import the modules
-sys.path.append("/home/varun.edachali/conn/databricks-sql-python/src")
-
-try:
-    import pyarrow
-except ImportError:
-    pyarrow = None
+from unittest.mock import MagicMock
 
 from databricks.sql.utils import (
-    ThriftResultSetQueueFactory,
     SeaResultSetQueueFactory,
     JsonQueue,
-    ArrowQueue,
-    ColumnQueue,
-    CloudFetchQueue,
 )
-from databricks.sql.thrift_api.TCLIService.ttypes import TSparkRowSetType, TRowSet
 from databricks.sql.backend.models import ResultData
 
 
 class TestResultSetQueueFactories(unittest.TestCase):
-    """Tests for the ThriftResultSetQueueFactory and SeaResultSetQueueFactory classes."""
+    """Tests for the SeaResultSetQueueFactory classes."""
 
     def test_sea_result_set_queue_factory_with_data(self):
         """Test SeaResultSetQueueFactory with data."""