fix up tests error with attempts

apache · Aug 28, 2024 · 47781b0 · 47781b0
1 parent 38ad59f
commit 47781b0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/...-format-flink-common/src/test/java/org/apache/amoro/flink/read/TestMixedFormatSource.java b/...-format-flink-common/src/test/java/org/apache/amoro/flink/read/TestMixedFormatSource.java
@@ -234,6 +234,7 @@ public void testMixedFormatSource(FailoverType failoverType) throws Exception {
 
   @Test
   public void testDimTaskManagerFailover() throws Exception {
+    int restartAttempts = 10;
     List<RowData> updated = updateRecords();
     writeUpdate(updated);
     List<RowData> records = generateRecords(2, 1);
@@ -242,8 +243,8 @@ public void testDimTaskManagerFailover() throws Exception {
     MixedFormatSource<RowData> mixedFormatSource = initMixedFormatDimSource(true);
     StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
     // enable checkpoint
-    env.enableCheckpointing(1000);
-    env.setRestartStrategy(RestartStrategies.fixedDelayRestart(10, 0));
+    env.enableCheckpointing(3000);
+    env.setRestartStrategy(RestartStrategies.fixedDelayRestart(restartAttempts, 0));
 
     DataStream<RowData> input =
         env.fromSource(
@@ -262,8 +263,8 @@ public void testDimTaskManagerFailover() throws Exception {
         WatermarkAwareFailWrapper::continueProcessing,
         miniClusterResource.getMiniCluster());
 
-    while (WatermarkAwareFailWrapper.watermarkCounter.get() != PARALLELISM) {
-      Thread.sleep(1000);
+    while (WatermarkAwareFailWrapper.watermarkCounter.get() != restartAttempts) {
+      Thread.sleep(2000);
       LOG.info("wait for watermark after failover");
     }
     Assert.assertEquals(Long.MAX_VALUE, WatermarkAwareFailWrapper.getWatermarkAfterFailover());

diff --git a/amoro-mixed-format/amoro-mixed-format-spark/v3.2/amoro-mixed-format-spark-3.2/pom.xml b/amoro-mixed-format/amoro-mixed-format-spark/v3.2/amoro-mixed-format-spark-3.2/pom.xml