From 77dfa5ae7c8217f3b41c0134584c544df73e1ba7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dominik=20D=C4=99bowczyk?= <dominikd@spotify.com>
Date: Thu, 31 Aug 2023 16:43:34 +0200
Subject: [PATCH] [runners-flink] #28258 Wait infinitely if the duration value
 is less than 1ms in FlinkDetachedRunnerResult

---
 .../apache/beam/runners/flink/FlinkDetachedRunnerResult.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/runners/flink/src/main/java/org/apache/beam/runners/flink/FlinkDetachedRunnerResult.java b/runners/flink/src/main/java/org/apache/beam/runners/flink/FlinkDetachedRunnerResult.java
index ddf8a36a4f4ab..77d0e7d3434ca 100644
--- a/runners/flink/src/main/java/org/apache/beam/runners/flink/FlinkDetachedRunnerResult.java
+++ b/runners/flink/src/main/java/org/apache/beam/runners/flink/FlinkDetachedRunnerResult.java
@@ -92,7 +92,7 @@ public State waitUntilFinish(Duration duration) {
     long start = System.currentTimeMillis();
     long durationInMillis = duration.getMillis();
     State state = State.UNKNOWN;
-    while ((System.currentTimeMillis() - start) < durationInMillis) {
+    while (durationInMillis < 1 || (System.currentTimeMillis() - start) < durationInMillis) {
       state = getState();
       if (state.isTerminal()) {
         return state;