[FLINK-18020] Increase timeout in SQLClientKafkaITCase

is · Jun 5, 2020 · 971e7b2 · 971e7b2
1 parent f6bbff4
commit 971e7b2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/...ts-common-kafka/src/test/java/org/apache/flink/tests/util/kafka/SQLClientKafkaITCase.java b/...ts-common-kafka/src/test/java/org/apache/flink/tests/util/kafka/SQLClientKafkaITCase.java
@@ -18,6 +18,7 @@
 
 package org.apache.flink.tests.util.kafka;
 
+import org.apache.flink.api.common.time.Deadline;
 import org.apache.flink.tests.util.TestUtils;
 import org.apache.flink.tests.util.cache.DownloadCache;
 import org.apache.flink.tests.util.categories.TravisGroup1;
@@ -52,6 +53,7 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.StandardOpenOption;
+import java.time.Duration;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collection;
@@ -234,8 +236,8 @@ private String initializeSessionYaml(Map<String, String> vars) throws IOExceptio
 
  private void checkCsvResultFile() throws Exception {
  boolean success = false;
- long maxRetries = 10, duration = 5000L;
- for (int i = 0; i < maxRetries; i++) {
+ final Deadline deadline = Deadline.fromNow(Duration.ofSeconds(120));
+ while (!success && deadline.hasTimeLeft()) {
  if (Files.exists(result)) {
  byte[] bytes = Files.readAllBytes(result);
  String[] lines = new String(bytes, Charsets.UTF_8).split("\n");
@@ -255,8 +257,8 @@ private void checkCsvResultFile() throws Exception {
  } else {
  LOG.info("The target CSV {} does not exist now", result);
  }
- Thread.sleep(duration);
+ Thread.sleep(500);
  }
- Assert.assertTrue("Timeout(" + (maxRetries * duration) + " sec) to read the correct CSV results.", success);
+ Assert.assertTrue("Did not get expected results before timeout.", success);
  }
 }