command_handle -> command_id in ExecuteResponse

Author: varun-edachali-dbx

src/databricks/sql/client.py

@@ -1230,7 +1230,7 @@ def __init__(
         :param use_cloud_fetch: Whether to use cloud fetch for retrieving results
         """
         self.connection = connection
-        self.command_id = execute_response.command_handle
+        self.command_id = execute_response.command_id
         self.op_state = execute_response.status
         self.has_been_closed_server_side = execute_response.has_been_closed_server_side
         self.has_more_rows = execute_response.has_more_rows

tests/unit/test_client.py

@@ -42,7 +42,7 @@ def new(cls):
             description=None,
             arrow_queue=None,
             is_staging_operation=False,
-            command_handle=b"\x22",
+            command_id=None,
             has_been_closed_server_side=True,
             has_more_rows=True,
             lz4_compressed=True,
@@ -155,7 +155,7 @@ def test_closing_result_set_hard_closes_commands(self):
         result_set.close()
 
         mock_thrift_backend.close_command.assert_called_once_with(
-            mock_results_response.command_handle
+            mock_results_response.command_id
         )
 
     @patch("%s.client.ResultSet" % PACKAGE_NAME)

tests/unit/test_fetches.py

@@ -47,7 +47,7 @@ def make_dummy_result_set_from_initial_results(initial_results):
                 has_more_rows=False,
                 description=Mock(),
                 lz4_compressed=Mock(),
-                command_handle=None,
+                command_id=None,
                 arrow_queue=arrow_queue,
                 arrow_schema_bytes=schema.serialize().to_pybytes(),
                 is_staging_operation=False,
@@ -96,7 +96,7 @@ def fetch_results(
                     for col_id in range(num_cols)
                 ],
                 lz4_compressed=Mock(),
-                command_handle=None,
+                command_id=None,
                 arrow_queue=None,
                 arrow_schema_bytes=None,
                 is_staging_operation=False,

tests/unit/test_fetches_bench.py

@@ -37,7 +37,7 @@ def make_dummy_result_set_from_initial_results(arrow_table):
                 has_been_closed_server_side=True,
                 has_more_rows=False,
                 description=Mock(),
-                command_handle=None,
+                command_id=None,
                 arrow_queue=arrow_queue,
                 arrow_schema=arrow_table.schema,
             ),