Add http client test suite for long-running request patterns (#6924)

* http test suite added

* checkstyle fix

* fix test name

* fix check style

Author: Fred1155

http-clients/apache-client/src/test/java/software/amazon/awssdk/http/apache/ApacheHttpClientLongRunningRequestTest.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.apache;
+
+import software.amazon.awssdk.http.SdkHttpClient;
+import software.amazon.awssdk.http.SdkHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class ApacheHttpClientLongRunningRequestTest extends SdkHttpClientLongRunningRequestTestSuite {
+
+    @Override
+    protected SdkHttpClient createSdkHttpClient(AttributeMap config) {
+        return ApacheHttpClient.builder().buildWithDefaults(config);
+    }
+}

http-clients/apache5-client/src/test/java/software/amazon/awssdk/http/apache5/Apache5HttpClientLongRunningRequestTest.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.apache5;
+
+import software.amazon.awssdk.http.SdkHttpClient;
+import software.amazon.awssdk.http.SdkHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class Apache5HttpClientLongRunningRequestTest extends SdkHttpClientLongRunningRequestTestSuite {
+
+    @Override
+    protected SdkHttpClient createSdkHttpClient(AttributeMap config) {
+        return Apache5HttpClient.builder().buildWithDefaults(config);
+    }
+}

http-clients/aws-crt-client/src/test/java/software/amazon/awssdk/http/crt/AwsCrtAsyncHttpClientLongRunningRequestTest.java

@@ -0,0 +1,48 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.crt;
+
+import org.junit.jupiter.api.BeforeAll;
+import software.amazon.awssdk.crt.Log;
+import software.amazon.awssdk.http.SdkAsyncHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.http.async.SdkAsyncHttpClient;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class AwsCrtAsyncHttpClientLongRunningRequestTest extends SdkAsyncHttpClientLongRunningRequestTestSuite {
+
+    @BeforeAll
+    public static void beforeAll() {
+        System.setProperty("aws.crt.debugnative", "true");
+        Log.initLoggingToStdout(Log.LogLevel.Warn);
+    }
+
+    @Override
+    protected SdkAsyncHttpClient createSdkAsyncHttpClient(AttributeMap config) {
+        return AwsCrtAsyncHttpClient.builder().buildWithDefaults(config);
+    }
+
+    // Empty test; the CRT async client does not currently enforce READ_TIMEOUT. Delete this
+    // override when connection health monitoring is re-added.
+    @Override
+    public void executeWhenReadTimeoutAndServerDelaysResponseFailsWithinTimeoutBound() {
+    }
+
+    // Empty test; the CRT async client does not currently enforce READ_TIMEOUT. Delete this
+    // override when connection health monitoring is re-added.
+    @Override
+    public void executeWhenReadTimeoutAndStreamingResponsePausesFailsWithinTimeoutBound() {
+    }
+}

http-clients/aws-crt-client/src/test/java/software/amazon/awssdk/http/crt/AwsCrtHttpClientLongRunningRequestTest.java

@@ -0,0 +1,48 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.crt;
+
+import org.junit.jupiter.api.BeforeAll;
+import software.amazon.awssdk.crt.Log;
+import software.amazon.awssdk.http.SdkHttpClient;
+import software.amazon.awssdk.http.SdkHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class AwsCrtHttpClientLongRunningRequestTest extends SdkHttpClientLongRunningRequestTestSuite {
+
+    @BeforeAll
+    public static void beforeAll() {
+        System.setProperty("aws.crt.debugnative", "true");
+        Log.initLoggingToStdout(Log.LogLevel.Warn);
+    }
+
+    @Override
+    protected SdkHttpClient createSdkHttpClient(AttributeMap config) {
+        return AwsCrtHttpClient.builder().buildWithDefaults(config);
+    }
+
+    // Empty test; the CRT sync client does not currently enforce READ_TIMEOUT. Delete this
+    // override when connection health monitoring is re-added.
+    @Override
+    public void executeWhenReadTimeoutAndServerDelaysResponseFailsWithinTimeoutBound() {
+    }
+
+    // Empty test; the CRT sync client does not currently enforce READ_TIMEOUT. Delete this
+    // override when connection health monitoring is re-added.
+    @Override
+    public void executeWhenReadTimeoutAndStreamingResponsePausesFailsWithinTimeoutBound() {
+    }
+}

http-clients/netty-nio-client/src/test/java/software/amazon/awssdk/http/nio/netty/NettyAsyncHttpClientLongRunningRequestTest.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.nio.netty;
+
+import software.amazon.awssdk.http.SdkAsyncHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.http.async.SdkAsyncHttpClient;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class NettyAsyncHttpClientLongRunningRequestTest extends SdkAsyncHttpClientLongRunningRequestTestSuite {
+
+    @Override
+    protected SdkAsyncHttpClient createSdkAsyncHttpClient(AttributeMap config) {
+        return NettyNioAsyncHttpClient.builder().buildWithDefaults(config);
+    }
+}

http-clients/url-connection-client/src/test/java/software/amazon/awssdk/http/urlconnection/UrlConnectionHttpClientLongRunningRequestTest.java

@@ -0,0 +1,33 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http.urlconnection;
+
+import software.amazon.awssdk.http.SdkHttpClient;
+import software.amazon.awssdk.http.SdkHttpClientLongRunningRequestTestSuite;
+import software.amazon.awssdk.utils.AttributeMap;
+
+public class UrlConnectionHttpClientLongRunningRequestTest extends SdkHttpClientLongRunningRequestTestSuite {
+
+    @Override
+    protected SdkHttpClient createSdkHttpClient(AttributeMap config) {
+        return UrlConnectionHttpClient.builder().buildWithDefaults(config);
+    }
+
+    // Empty test; UrlConnectionHttpClient does not maintain a connection pool.
+    @Override
+    public void executeWhenConnectionAcquireTimeoutAndPoolExhaustedFailsWithinTimeoutBound() {
+    }
+}

test/http-client-tests/src/main/java/software/amazon/awssdk/http/LongRunningRequestTestSupport.java

@@ -0,0 +1,93 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.http;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
+import static com.github.tomakehurst.wiremock.client.WireMock.any;
+import static com.github.tomakehurst.wiremock.client.WireMock.urlPathEqualTo;
+
+import com.github.tomakehurst.wiremock.junit5.WireMockExtension;
+import java.time.Duration;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+/**
+ * Shared helpers for the long-running request test suites.
+ */
+public final class LongRunningRequestTestSupport {
+
+    public static final Duration CONFIGURED_TIMEOUT = Duration.ofSeconds(2);
+    public static final Duration SERVER_DELAY = Duration.ofSeconds(10);
+    public static final Duration TIME_BOUND_SAFETY_MARGIN = Duration.ofSeconds(10);
+    public static final Duration HANG_DELAY = Duration.ofMinutes(1);
+
+    private LongRunningRequestTestSupport() {
+    }
+
+    /**
+     * Simulates long-polling and synchronous-invoke-with-extended-wait (SFN getActivityTask, SQS receiveMessage,
+     * Lambda invoke): server delays the entire response.
+     */
+    public static void stubLongPolling(WireMockExtension mockServer) {
+        mockServer.stubFor(any(urlPathEqualTo("/"))
+                               .willReturn(aResponse().withStatus(200)
+                                                      .withBody("hello")
+                                                      .withFixedDelay((int) SERVER_DELAY.toMillis())));
+    }
+
+    /**
+     * Simulates streaming APIs (Transcribe Streaming, Kinesis subscribeToShard): server dribbles body chunks with
+     * gaps between them.
+     */
+    public static void stubStreamingWithPauses(WireMockExtension mockServer) {
+        mockServer.stubFor(any(urlPathEqualTo("/"))
+                               .willReturn(aResponse().withStatus(200)
+                                                      .withBody("hello")
+                                                      .withChunkedDribbleDelay(2, (int) SERVER_DELAY.toMillis())));
+    }
+
+    /**
+     * Hangs indefinitely so a holding request can exhaust a single-slot pool while another racing request
+     * exercises the acquire timeout.
+     */
+    public static void stubHanging(WireMockExtension mockServer) {
+        mockServer.stubFor(any(urlPathEqualTo("/"))
+                               .willReturn(aResponse().withStatus(200)
+                                                      .withBody("hello")
+                                                      .withFixedDelay((int) HANG_DELAY.toMillis())));
+    }
+
+    public static void assertFailsWithinTimeBound(CompletableFuture<?> future, Duration expectedTimeout) {
+        Duration maxWait = expectedTimeout.plus(TIME_BOUND_SAFETY_MARGIN);
+
+        try {
+            future.get(maxWait.toMillis(), TimeUnit.MILLISECONDS);
+            throw new AssertionError("Expected request to throw an exception but it completed successfully");
+        } catch (TimeoutException e) {
+            future.cancel(true);
+            throw new AssertionError(
+                "Expected request to fail within " + maxWait + " but it was still running - client appears to hang",
+                e);
+        } catch (InterruptedException e) {
+            Thread.currentThread().interrupt();
+            throw new AssertionError("Unexpected interruption while waiting for request to fail", e);
+        } catch (ExecutionException e) {
+            // expected
+        }
+    }
+}