TImeout StreamingDataflowWorkerTest when test stuck (#29065)

apache · Oct 19, 2023 · 7632d48 · 7632d48
1 parent 67a39a5
commit 7632d48
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -178,6 +178,7 @@
 import org.junit.Test;
 import org.junit.rules.ErrorCollector;
 import org.junit.rules.TestRule;
+import org.junit.rules.Timeout;
 import org.junit.runner.Description;
 import org.junit.runner.RunWith;
 import org.junit.runners.Parameterized;
@@ -254,6 +255,7 @@ public Long get() {
           return idGenerator.getAndIncrement();
         }
       };
+  @Rule public transient Timeout globalTimeout = Timeout.seconds(600);
   @Rule public BlockingFn blockingFn = new BlockingFn();
   @Rule public TestRule restoreMDC = new RestoreDataflowLoggingMDC();
   @Rule public ErrorCollector errorCollector = new ErrorCollector();