(flink) 02/02: [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 release-1.19
in repository https://gitbox.apache.org/repos/asf/flink.git

commit c91029b0456ddd7635fe0f04ad9ecb4813c5d5a3
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 9b95c8aa41c..98df5533fd2 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
@@ -343,7 +343,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();
@@ -433,7 +433,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();