(flink) branch master updated: [FLINK-34336][test] Fix the bug that AutoRescalingITCase may hang sometimes

[fanrui]

This is an automated email from the ASF dual-hosted git repository.

fanrui pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink.git


The following commit(s) were added to refs/heads/master by this push:
     new e2e3de2d48e [FLINK-34336][test] Fix the bug that AutoRescalingITCase 
may hang sometimes
e2e3de2d48e is described below

commit e2e3de2d48e3f02b746bdbdcb4da7b0477986a11
Author: Rui Fan <1996fan...@gmail.com>
AuthorDate: Fri Feb 2 13:08:01 2024 +0800

    [FLINK-34336][test] Fix the bug that AutoRescalingITCase may hang sometimes
---
 .../org/apache/flink/test/checkpointing/AutoRescalingITCase.java     | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/AutoRescalingITCase.java
 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/AutoRescalingITCase.java
index ba13faca294..404e129eb4e 100644
--- 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/AutoRescalingITCase.java
+++ 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/AutoRescalingITCase.java
@@ -351,7 +351,7 @@ public class AutoRescalingITCase extends TestLogger {
 
             restClusterClient.updateJobResourceRequirements(jobID, 
builder.build()).join();
 
-            waitForRunningTasks(restClusterClient, jobID, parallelism2);
+            waitForRunningTasks(restClusterClient, jobID, 2 * parallelism2);
             waitForAvailableSlots(restClusterClient, totalSlots - 
parallelism2);
 
             StateSourceBase.canFinishLatch.countDown();
@@ -441,7 +441,8 @@ public class AutoRescalingITCase extends TestLogger {
 
             restClusterClient.updateJobResourceRequirements(jobID, 
builder.build()).join();
 
-            waitForRunningTasks(restClusterClient, jobID, parallelism2);
+            // Source is parallelism, the flatMapper & Sink is parallelism2
+            waitForRunningTasks(restClusterClient, jobID, parallelism + 
parallelism2);
             waitForAvailableSlots(restClusterClient, totalSlots - 
parallelism2);
 
             SubtaskIndexSource.SOURCE_LATCH.trigger();