Add timeout for multinode tests

Signed-off-by: bowenlan-amzn <[email protected]>
bowenlan-amzn · Oct 4, 2023 · dd10cf2 · dd10cf2
1 parent 8b0cd6d
commit dd10cf2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/test/kotlin/org/opensearch/indexmanagement/TestHelpers.kt b/src/test/kotlin/org/opensearch/indexmanagement/TestHelpers.kt
@@ -114,17 +114,18 @@ fun <T> waitFor(
     timeout: Instant = Instant.ofEpochSecond(20),
     block: () -> T
 ): T {
+    var to = timeout
     if (isMultiNode) {
         // job scheduling could be skipped in multi-node tests
         // https://github.com/opensearch-project/job-scheduler/issues/173
-        timeout.plusSeconds(60)
+        to = timeout.plusSeconds(60)
     }
     val startTime = Instant.now().toEpochMilli()
     do {
         try {
             return block()
         } catch (e: Throwable) {
-            if ((Instant.now().toEpochMilli() - startTime) > timeout.toEpochMilli()) {
+            if ((Instant.now().toEpochMilli() - startTime) > to.toEpochMilli()) {
                 throw e
             } else {
                 Thread.sleep(100L)