Calculate and specify a wait-for-it timeout

arquillian · May 16, 2020 · f36c8cf · f36c8cf
1 parent 476094e
commit f36c8cf
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/docker/docker/src/main/java/org/arquillian/cube/docker/impl/await/PollingAwaitStrategy.java b/docker/docker/src/main/java/org/arquillian/cube/docker/impl/await/PollingAwaitStrategy.java
@@ -13,6 +13,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
+import java.util.concurrent.TimeUnit;
 import java.util.logging.Logger;
 import org.arquillian.cube.docker.impl.client.config.Await;
 import org.arquillian.cube.docker.impl.docker.DockerClientExecutor;
@@ -214,7 +215,12 @@ private Set<PosixFilePermission> getScriptPermissions() {
     }
 
     private String resolveWaitForItCommand(String containerIp, int port) {
-        return String.format("%s/%s %s:%s -s -- echo %s", CONTAINER_DIRECTORY, WAIT_FOR_IT_SCRIPT, containerIp, port,
+        final long waitForItSleepSeconds = TimeUnit.SECONDS.convert(
+                pollIterations * getSleepTime(),
+                getTimeUnit()
+        );
+        return String.format("%s/%s %s:%s -s -t %s -- echo %s", CONTAINER_DIRECTORY, WAIT_FOR_IT_SCRIPT, containerIp, port,
+            waitForItSleepSeconds,
             MESSAGE);
     }