added read timeout to http wait strategy

core/src/main/java/org/testcontainers/containers/wait/strategy/HttpWaitStrategy.java

@@ -2,7 +2,6 @@
 
 import com.google.common.base.Strings;
 import com.google.common.io.BaseEncoding;
-import java.util.Optional;
 import lombok.extern.slf4j.Slf4j;
 import org.rnorth.ducttape.TimeoutException;
 import org.testcontainers.containers.ContainerLaunchException;
@@ -13,7 +12,10 @@
 import java.net.HttpURLConnection;
 import java.net.URI;
 import java.net.URL;
+import java.time.Duration;
+import java.time.temporal.ChronoUnit;
 import java.util.HashSet;
+import java.util.Optional;
 import java.util.Set;
 import java.util.concurrent.TimeUnit;
 import java.util.function.Predicate;
@@ -41,6 +43,7 @@ public class HttpWaitStrategy extends AbstractWaitStrategy {
     private Predicate<String> responsePredicate;
     private Predicate<Integer> statusCodePredicate = null;
     private Optional<Integer> livenessPort = Optional.empty();
+    private Duration readTimeout = Duration.ofSeconds(1);
 
     /**
      * Waits for the given status code.
@@ -108,6 +111,17 @@ public HttpWaitStrategy withBasicCredentials(String username, String password) {
         return this;
     }
 
+    /**
+     * Set the HTTP connections read timeout.
+     *
+     * @param timeout the timeout in millis
+     * @return this
+     */
+    public HttpWaitStrategy withReadTimeout(Duration timeout) {
+        this.readTimeout = timeout;
+        return this;
+    }
+
     /**
      * Waits for the response to pass the given predicate
      * @param responsePredicate The predicate to test the response against
@@ -143,6 +157,7 @@ protected void waitUntilReady() {
                 getRateLimiter().doWhenReady(() -> {
                     try {
                         final HttpURLConnection connection = (HttpURLConnection) new URL(uri).openConnection();
+                        connection.setReadTimeout(Math.toIntExact(readTimeout.get(ChronoUnit.MILLIS)));
 
                         // authenticate
                         if (!Strings.isNullOrEmpty(username)) {

core/src/test/java/org/testcontainers/junit/wait/strategy/HttpWaitStrategyTest.java

@@ -5,6 +5,7 @@
 import org.rnorth.ducttape.RetryCountExceededException;
 import org.testcontainers.containers.wait.strategy.HttpWaitStrategy;
 
+import java.time.Duration;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.function.Predicate;
 
@@ -131,6 +132,15 @@ public void testWaitUntilReadyWithSpecificPort() {
         ));
     }
 
+    @Test
+    public void testWaitUntilReadyWithTimoutCausedByReadTimeout() {
+        waitUntilReadyAndTimeout(
+            startContainerWithCommand(createShellCommand("0 Connection Refused", GOOD_RESPONSE_BODY, 9090),
+                createHttpWaitStrategy(ready).forPort(9090).withReadTimeout(Duration.ofMillis(1)),
+                9090
+            ));
+    }
+
     /**
      * @param ready the AtomicBoolean on which to indicate success
      * @return the WaitStrategy under test
@@ -143,6 +153,7 @@ protected HttpWaitStrategy buildWaitStrategy(final AtomicBoolean ready) {
 
     /**
      * Create a HttpWaitStrategy instance with a waitUntilReady implementation
+     *
      * @param ready Indicates that the WaitStrategy has completed waiting successfully.
      * @return the HttpWaitStrategy instance
      */
@@ -163,9 +174,9 @@ private String createShellCommand(String header, String responseBody) {
 
     private String createShellCommand(String header, String responseBody, int port) {
         int length = responseBody.getBytes().length;
-        return "while true; do { echo -e \"HTTP/1.1 "+header+NEWLINE+
-                "Content-Type: text/html"+NEWLINE+
-                "Content-Length: "+length +NEWLINE+ "\";"
-                +" echo \""+responseBody+"\";} | nc -lp " + port + "; done";
+        return "while true; do { echo -e \"HTTP/1.1 " + header + NEWLINE +
+            "Content-Type: text/html" + NEWLINE +
+            "Content-Length: " + length + NEWLINE + "\";"
+            + " echo \"" + responseBody + "\";} | nc -lp " + port + "; done";
     }
 }