Update VolumeOperationProcessorTest mocks to use executeWithRetry

VolumeOperationProcessor uses executeWithRetry(), not execute().
Update test mocks to match actual usage:
- Replace when(databricksHttpClient.execute(any()))
- With when(databricksHttpClient.executeWithRetry(any(), any()))
- Also restore all 4 interface methods in ArrowResultChunkStatusTest
  (needed because convenience methods still exist in interface)

All volume and arrow tests pass.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: nishkarsh-db

src/test/java/com/databricks/jdbc/api/impl/arrow/ArrowResultChunkStatusTest.java

@@ -98,6 +98,21 @@ public CloseableHttpResponse execute(org.apache.http.client.methods.HttpUriReque
         return response(content, length);
       }
 
+      @Override
+      public CloseableHttpResponse execute(
+          org.apache.http.client.methods.HttpUriRequest request, boolean supportGzipEncoding)
+          throws DatabricksHttpException {
+        return response(content, length);
+      }
+
+      @Override
+      public CloseableHttpResponse executeWithRetry(
+          org.apache.http.client.methods.HttpUriRequest request,
+          com.databricks.jdbc.common.RequestType requestType)
+          throws DatabricksHttpException {
+        return response(content, length);
+      }
+
       @Override
       public CloseableHttpResponse executeWithRetry(
           org.apache.http.client.methods.HttpUriRequest request,

src/test/java/com/databricks/jdbc/api/impl/volume/VolumeOperationProcessorTest.java

@@ -36,7 +36,7 @@ void testExecuteGetOperationStream_UnsuccessfulHttpResponse() throws Exception {
             .getStreamReceiver((entity) -> {})
             .build();
 
-    when(databricksHttpClient.execute(any())).thenReturn(mockStream);
+    when(databricksHttpClient.executeWithRetry(any(), any())).thenReturn(mockStream);
     when(mockStream.getStatusLine()).thenReturn(mockStatusLine);
     when(mockStatusLine.getStatusCode()).thenReturn(400);
     volumeOperationProcessor.executeGetOperation();
@@ -56,7 +56,7 @@ void testExecuteGetOperationStream_HttpException() throws Exception {
 
     DatabricksHttpException mockException =
         new DatabricksHttpException("Test Exeception", DatabricksDriverErrorCode.INVALID_STATE);
-    doThrow(mockException).when(databricksHttpClient).execute(any());
+    doThrow(mockException).when(databricksHttpClient).executeWithRetry(any(), any());
 
     volumeOperationProcessor.executeGetOperation();
     assertEquals(volumeOperationProcessor.getStatus(), VolumeOperationStatus.FAILED);
@@ -74,7 +74,7 @@ void testExecutePutOperationStream_HttpException() throws Exception {
 
     DatabricksHttpException mockException =
         new DatabricksHttpException("Test Exeception", DatabricksDriverErrorCode.INVALID_STATE);
-    doThrow(mockException).when(databricksHttpClient).execute(any());
+    doThrow(mockException).when(databricksHttpClient).executeWithRetry(any(), any());
 
     volumeOperationProcessor.executePutOperation();
     assertEquals(volumeOperationProcessor.getStatus(), VolumeOperationStatus.FAILED);