command_handle -> command_id in ExecuteResponse

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

Author: varun-edachali-dbx

tests/unit/test_client.py

@@ -42,7 +42,7 @@ def new(cls):
             mock_result_set,
             description=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,
@@ -159,7 +159,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.result_set.ThriftResultSet" % 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,
             ),