account for new exec resp

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

Author: varun-edachali-dbx

tests/unit/test_client.py

@@ -121,10 +121,8 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):
 
                 # Verify initial state
                 self.assertEqual(real_result_set.has_been_closed_server_side, closed)
-                expected_op_state = (
-                    CommandState.CLOSED if closed else CommandState.SUCCEEDED
-                )
-                self.assertEqual(real_result_set.op_state, expected_op_state)
+                expected_status = CommandState.CLOSED if closed else CommandState.SUCCEEDED
+                self.assertEqual(real_result_set.status, expected_status)
 
                 # Mock execute_command to return our real result set
                 cursor.backend.execute_command = Mock(return_value=real_result_set)
@@ -146,8 +144,8 @@ def test_closing_connection_closes_commands(self, mock_thrift_client_class):
                 # 1. has_been_closed_server_side should always be True after close()
                 self.assertTrue(real_result_set.has_been_closed_server_side)
 
-                # 2. op_state should always be CLOSED after close()
-                self.assertEqual(real_result_set.op_state, CommandState.CLOSED)
+                # 2. status should always be CLOSED after close()
+                self.assertEqual(real_result_set.status, CommandState.CLOSED)
 
                 # 3. Backend close_command should be called appropriately
                 if not closed:
@@ -556,7 +554,7 @@ def test_cursor_keeps_connection_alive(self, mock_client_class):
         self.assertEqual(instance.close_session.call_count, 0)
         cursor.close()
 
-    @patch("%s.utils.ExecuteResponse" % PACKAGE_NAME, autospec=True)
+    @patch("%s.backend.types.ExecuteResponse" % PACKAGE_NAME)
     @patch("%s.client.Cursor._handle_staging_operation" % PACKAGE_NAME)
     @patch("%s.session.ThriftDatabricksClient" % PACKAGE_NAME)
     def test_staging_operation_response_is_handled(
@@ -678,7 +676,7 @@ def test_resultset_close_handles_cursor_already_closed_error(self):
         """Test that ResultSet.close() handles CursorAlreadyClosedError properly."""
         result_set = client.ThriftResultSet.__new__(client.ThriftResultSet)
         result_set.backend = Mock()
-        result_set.backend.CLOSED_OP_STATE = "CLOSED"
+        result_set.backend.CLOSED_OP_STATE = CommandState.CLOSED
         result_set.connection = Mock()
         result_set.connection.open = True
         result_set.op_state = "RUNNING"