Fix tests

Author: pawankartik-elastic

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClusterCancellationIT.java

@@ -163,8 +163,8 @@ public void testCancel() throws Exception {
         } finally {
             SimplePauseFieldPlugin.allowEmitting.countDown();
         }
-        Exception error = expectThrows(Exception.class, requestFuture::actionGet);
-        assertThat(error.getMessage(), containsString("proxy timeout"));
+        RemoteComputeException rce = expectThrows(RemoteComputeException.class, requestFuture::actionGet);
+        assertThat(rce.getCause().getMessage(), containsString("proxy timeout"));
     }
 
     public void testSameRemoteClusters() throws Exception {

x-pack/plugin/esql/src/internalClusterTest/java/org/elasticsearch/xpack/esql/action/CrossClusterQueryWithPartialResultsIT.java

@@ -194,7 +194,8 @@ public void sendResponse(Exception exception) {
             request.includeCCSMetadata(randomBoolean());
             {
                 request.allowPartialResults(false);
-                Exception error = expectThrows(Exception.class, () -> runQuery(request).close());
+                RemoteComputeException rce = expectThrows(RemoteComputeException.class, () -> runQuery(request).close());
+                var error = (Exception) rce.getCause();
                 var unwrapped = ExceptionsHelper.unwrap(error, simulatedFailure.getClass());
                 assertNotNull(unwrapped);
                 assertThat(unwrapped.getMessage(), equalTo(simulatedFailure.getMessage()));
@@ -241,7 +242,8 @@ public void testFailToStartRequestOnRemoteCluster() throws Exception {
             request.includeCCSMetadata(randomBoolean());
             {
                 request.allowPartialResults(false);
-                var error = expectThrows(Exception.class, () -> runQuery(request).close());
+                var rce = expectThrows(RemoteComputeException.class, () -> runQuery(request).close());
+                var error = (Exception) rce.getCause();
                 EsqlTestUtils.assertEsqlFailure(error);
                 var unwrapped = ExceptionsHelper.unwrap(error, simulatedFailure.getClass());
                 assertNotNull(unwrapped);