Fix flaky StreamingDataflowWorkerTest which wasn't waiting for enough…

… commits. (apache#31781)
hjtran · Jul 4, 2024 · a5eee58 · a5eee58
1 parent 0c89a0e
commit a5eee58
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...er/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java b/...er/src/test/java/org/apache/beam/runners/dataflow/worker/StreamingDataflowWorkerTest.java
@@ -1227,7 +1227,7 @@ public void testKeyCommitTooLargeException() throws Exception {
         .thenReturn(makeInput(2, 0, "key", DEFAULT_SHARDING_KEY));
     server.waitForEmptyWorkQueue();
 
-    Map<Long, Windmill.WorkItemCommitRequest> result = server.waitForAndGetCommits(1);
+    Map<Long, Windmill.WorkItemCommitRequest> result = server.waitForAndGetCommits(2);
 
     assertEquals(2, result.size());
     assertEquals(