fixed e2e error

apache · Aug 10, 2023 · b7c9459 · b7c9459
1 parent eab4b96
commit b7c9459
Showing 1 changed file with 24 additions and 2 deletions.
diff --git a/...ctor-seatunnel-e2e-base/src/test/java/org/apache/seatunnel/engine/e2e/JobExecutionIT.java b/...ctor-seatunnel-e2e-base/src/test/java/org/apache/seatunnel/engine/e2e/JobExecutionIT.java
@@ -19,11 +19,15 @@
 
 import org.apache.seatunnel.common.config.Common;
 import org.apache.seatunnel.common.config.DeployMode;
+import org.apache.seatunnel.common.utils.RetryUtils;
 import org.apache.seatunnel.engine.client.SeaTunnelClient;
 import org.apache.seatunnel.engine.client.job.ClientJobProxy;
 import org.apache.seatunnel.engine.client.job.JobExecutionEnvironment;
+import org.apache.seatunnel.engine.common.Constant;
 import org.apache.seatunnel.engine.common.config.ConfigProvider;
 import org.apache.seatunnel.engine.common.config.JobConfig;
+import org.apache.seatunnel.engine.common.utils.ExceptionUtil;
+import org.apache.seatunnel.engine.common.utils.PassiveCompletableFuture;
 import org.apache.seatunnel.engine.core.job.JobResult;
 import org.apache.seatunnel.engine.core.job.JobStatus;
 import org.apache.seatunnel.engine.server.SeaTunnelServerStarter;
@@ -152,11 +156,29 @@ public void testGetErrorInfo() throws ExecutionException, InterruptedException {
         CompletableFuture<JobResult> completableFuture =
                 CompletableFuture.supplyAsync(
                         () -> {
-                            return clientJobProxy.doWaitForJobComplete().join();
+                            try {
+                                return RetryUtils.retryWithException(
+                                        () -> {
+                                            PassiveCompletableFuture<JobResult> jobFuture =
+                                                    clientJobProxy.doWaitForJobComplete();
+                                            return jobFuture.get();
+                                        },
+                                        new RetryUtils.RetryMaterial(
+                                                100000,
+                                                true,
+                                                exception ->
+                                                        ExceptionUtil.isOperationNeedRetryException(
+                                                                exception),
+                                                Constant.OPERATION_RETRY_SLEEP));
+                            } catch (Exception e) {
+                                throw new RuntimeException(e);
+                            }
                         });
 
-        JobResult result = completableFuture.get();
+        await().atMost(20000, TimeUnit.MILLISECONDS)
+                .untilAsserted(() -> Assertions.assertTrue(completableFuture.isDone()));
 
+        JobResult result = completableFuture.get();
         Assertions.assertEquals(result.getStatus(), JobStatus.FAILED);
         Assertions.assertTrue(result.getError().startsWith("java.lang.NumberFormatException"));
     }