(beam) branch master updated: Fix flaky AsyncBatchWriteHandlerTest.correctlyLimitConcurrency (#36975)

[yh...@apache.org]

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

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


The following commit(s) were added to refs/heads/master by this push:
     new e24a2a2eeb9 Fix flaky AsyncBatchWriteHandlerTest.correctlyLimitConcurrency (#36975)
e24a2a2eeb9 is described below

commit e24a2a2eeb9643b22a929ee2727b432a891611d5
Author: Sam Whittle <sc...@users.noreply.github.com>
AuthorDate: Thu Dec 4 00:27:17 2025 +0100

    Fix flaky AsyncBatchWriteHandlerTest.correctlyLimitConcurrency (#36975)
---
 .../org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java b/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
index cd7aca3c9f9..056d856e442 100644
--- a/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
+++ b/sdks/java/io/amazon-web-services2/src/test/java/org/apache/beam/sdk/io/aws2/common/AsyncBatchWriteHandlerTest.java
@@ -230,7 +230,7 @@ public class AsyncBatchWriteHandlerTest {
     nextResults.complete(emptyList());
 
     handler.waitForCompletion();
-    assertThat(future).isDone();
+    eventually(5, () -> assertThat(future).isDone());
   }
 
   static class SubmitFn<T, V> implements BiFunction<String, List<T>, CompletableFuture<List<V>>> {