ensure non null operationHandle for commandId creation

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

Author: varun-edachali-dbx

tests/unit/test_thrift_backend.py

@@ -595,6 +595,7 @@ def test_handle_execute_response_checks_operation_state_in_direct_results(self):
                         resultSet=None,
                         closeOperation=None,
                     ),
+                    operationHandle=self.operation_handle,
                 )
                 thrift_backend = ThriftDatabricksClient(
                     "foobar",
@@ -753,6 +754,7 @@ def test_direct_results_uses_display_message_if_available(self, tcli_service_cla
                 resultSet=None,
                 closeOperation=None,
             ),
+            operationHandle=self.operation_handle,
         )
 
         tcli_service_instance.ExecuteStatement.return_value = t_execute_resp
@@ -783,6 +785,7 @@ def test_handle_execute_response_checks_direct_results_for_error_statuses(self):
                     resultSet=None,
                     closeOperation=None,
                 ),
+                operationHandle=self.operation_handle,
             )
 
             resp_2 = resp_type(
@@ -795,6 +798,7 @@ def test_handle_execute_response_checks_direct_results_for_error_statuses(self):
                     resultSet=None,
                     closeOperation=None,
                 ),
+                operationHandle=self.operation_handle,
             )
 
             resp_3 = resp_type(
@@ -805,6 +809,7 @@ def test_handle_execute_response_checks_direct_results_for_error_statuses(self):
                     resultSet=ttypes.TFetchResultsResp(status=self.bad_status),
                     closeOperation=None,
                 ),
+                operationHandle=self.operation_handle,
             )
 
             resp_4 = resp_type(
@@ -815,6 +820,7 @@ def test_handle_execute_response_checks_direct_results_for_error_statuses(self):
                     resultSet=None,
                     closeOperation=ttypes.TCloseOperationResp(status=self.bad_status),
                 ),
+                operationHandle=self.operation_handle,
             )
 
             for error_resp in [resp_1, resp_2, resp_3, resp_4]: