(flink) branch release-2.0 updated: [FLINK-35810][test] Fix AsyncWaitOperatorTest missing endInput bug

[gu...@apache.org]

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

guoweijie pushed a commit to branch release-2.0
in repository https://gitbox.apache.org/repos/asf/flink.git


The following commit(s) were added to refs/heads/release-2.0 by this push:
     new 56a1dcb75a2 [FLINK-35810][test] Fix AsyncWaitOperatorTest missing endInput bug
56a1dcb75a2 is described below

commit 56a1dcb75a276b2752d11865f87cde03b785b9ab
Author: yunfengzhou-hub <yu...@outlook.com>
AuthorDate: Wed Mar 5 10:03:31 2025 +0800

    [FLINK-35810][test] Fix AsyncWaitOperatorTest missing endInput bug
---
 .../flink/streaming/api/operators/async/AsyncWaitOperatorTest.java     | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/operators/async/AsyncWaitOperatorTest.java b/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/operators/async/AsyncWaitOperatorTest.java
index 2f1cc06d77a..25e51ef203f 100644
--- a/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/operators/async/AsyncWaitOperatorTest.java
+++ b/flink-streaming-java/src/test/java/org/apache/flink/streaming/api/operators/async/AsyncWaitOperatorTest.java
@@ -907,6 +907,7 @@ class AsyncWaitOperatorTest {
             harness.processAll();
             assertThat(harness.getOutput()).containsOnly(new StreamRecord<>(1));
             assertThat(TimeoutAfterCompletionTestFunction.TIMED_OUT).isFalse();
+            harness.waitForTaskCompletion();
         }
     }
 
@@ -1246,6 +1247,7 @@ class AsyncWaitOperatorTest {
                         testHarness.getOutput(),
                         new StreamRecordComparator());
             }
+            testHarness.waitForTaskCompletion();
         }
     }
 
@@ -1318,6 +1320,7 @@ class AsyncWaitOperatorTest {
             // case when test machine under high load)
             assertThat(asyncFunction.getTryCount(1)).isLessThanOrEqualTo(2);
             assertThat(asyncFunction.getTryCount(2)).isLessThanOrEqualTo(2);
+            testHarness.waitForTaskCompletion();
         }
     }