Moves functional test from integration test suite to unit tests (aws#2745)

Author: cenedhryn

services/s3/src/test/java/software/amazon/awssdk/services/s3/checksums/ChecksumResetsOnRetryTest.java

@@ -53,8 +53,7 @@
  */
 public class ChecksumResetsOnRetryTest {
     @Rule
-    public WireMockRule mockServer = new WireMockRule(new WireMockConfiguration().port(0)
-                                                                                 .notifier(new ConsoleNotifier(true)));
+    public WireMockRule mockServer = new WireMockRule(new WireMockConfiguration().port(0));
 
     private S3Client s3Client;
 

services/s3/src/it/java/software/amazon/awssdk/services/s3/AsyncResponseTransformerIntegrationTest.java renamed to services/s3/src/test/java/software/amazon/awssdk/services/s3/functionaltests/AsyncResponseTransformerTest.java

@@ -13,29 +13,31 @@
  * permissions and limitations under the License.
  */
 
-package software.amazon.awssdk.services.s3;
+package software.amazon.awssdk.services.s3.functionaltests;
 
 import static org.assertj.core.api.Assertions.fail;
 import static org.mockito.Matchers.any;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.verify;
 
+import java.util.concurrent.CompletionException;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.runners.MockitoJUnitRunner;
 import software.amazon.awssdk.auth.credentials.AwsCredentialsProviderChain;
 import software.amazon.awssdk.auth.credentials.ProfileCredentialsProvider;
 import software.amazon.awssdk.core.async.AsyncResponseTransformer;
+import software.amazon.awssdk.services.s3.S3AsyncClient;
 
 @RunWith(MockitoJUnitRunner.class)
-public class AsyncResponseTransformerIntegrationTest {
+public class AsyncResponseTransformerTest {
 
     @Mock
     private AsyncResponseTransformer asyncResponseTransformer;
 
     @Test
-    public void AsyncResponseTransformerPrepareCalled_BeforeCredentailsResolution() {
+    public void AsyncResponseTransformerPrepareCalled_BeforeCredentialsResolution() {
         S3AsyncClient client = S3AsyncClient.builder()
                                             .credentialsProvider(AwsCredentialsProviderChain.of(
                                                 ProfileCredentialsProvider.create("dummyprofile")))
@@ -44,8 +46,7 @@ public void AsyncResponseTransformerPrepareCalled_BeforeCredentailsResolution()
         try {
             client.getObject(b -> b.bucket("dummy").key("key"), asyncResponseTransformer).join();
             fail("Expected an exception during credential resolution");
-        } catch (Throwable t) {
-
+        } catch (CompletionException e) {
         }
 
         verify(asyncResponseTransformer, times(1)).prepare();

test/protocol-tests/src/test/java/software/amazon/awssdk/protocol/tests/AsyncResponseTransformerTest.java

@@ -19,8 +19,12 @@
 import static com.github.tomakehurst.wiremock.client.WireMock.anyUrl;
 import static com.github.tomakehurst.wiremock.client.WireMock.post;
 import static com.github.tomakehurst.wiremock.client.WireMock.stubFor;
+import static org.assertj.core.api.Assertions.fail;
 import static org.assertj.core.api.AssertionsForClassTypes.assertThatThrownBy;
 import static org.assertj.core.api.AssertionsForInterfaceTypes.assertThat;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
 
 import com.github.tomakehurst.wiremock.junit.WireMockRule;
 import java.net.URI;
@@ -29,14 +33,18 @@
 import org.junit.Rule;
 import org.junit.Test;
 import software.amazon.awssdk.auth.credentials.AwsBasicCredentials;
+import software.amazon.awssdk.auth.credentials.AwsCredentialsProviderChain;
+import software.amazon.awssdk.auth.credentials.ProfileCredentialsProvider;
 import software.amazon.awssdk.auth.credentials.StaticCredentialsProvider;
 import software.amazon.awssdk.awscore.AwsResponse;
 import software.amazon.awssdk.core.async.AsyncResponseTransformer;
 import software.amazon.awssdk.core.async.SdkPublisher;
+import software.amazon.awssdk.core.exception.SdkClientException;
 import software.amazon.awssdk.regions.Region;
 import software.amazon.awssdk.services.protocolquery.ProtocolQueryAsyncClient;
 import software.amazon.awssdk.services.protocolquery.model.ProtocolQueryException;
 import software.amazon.awssdk.services.protocolrestjson.ProtocolRestJsonAsyncClient;
+import software.amazon.awssdk.services.protocolrestjson.model.GetOperationWithBodyResponse;
 import software.amazon.awssdk.services.protocolrestjson.model.ProtocolRestJsonException;
 import software.amazon.awssdk.services.protocolrestjson.model.StreamingOutputOperationResponse;
 import software.amazon.awssdk.utils.builder.SdkBuilder;
@@ -86,6 +94,26 @@ public void xmlClient_nonRetriableError_shouldNotifyAsyncResponseTransformer() {
         assertThat(responseTransformer.exceptionOccurred).isEqualTo(true);
     }
 
+    @Test
+    public void jsonClient_incorrectCredentials_shouldNotifyAsyncResponseTransformer() {
+        ProtocolRestJsonAsyncClient clientWithIncorrectCredentials = ProtocolRestJsonAsyncClient
+            .builder()
+            .credentialsProvider(AwsCredentialsProviderChain.of(ProfileCredentialsProvider.create("dummyprofile")))
+            .region(Region.US_EAST_1)
+            .endpointOverride(URI.create("http://localhost:" + wireMock.port()))
+            .build();
+
+        stubFor(post(anyUrl())
+                    .willReturn(aResponse()
+                                    .withStatus(400)));
+
+        TestAsyncResponseTransformer<StreamingOutputOperationResponse> responseTransformer = new TestAsyncResponseTransformer<>();
+        assertThatThrownBy(() -> clientWithIncorrectCredentials.streamingOutputOperation(SdkBuilder::build, responseTransformer).join())
+            .hasCauseExactlyInstanceOf(SdkClientException.class)
+            .hasMessageContaining("Unable to load credentials");
+        assertThat(responseTransformer.exceptionOccurred).isEqualTo(true);
+    }
+
     private class TestAsyncResponseTransformer<T extends AwsResponse> implements AsyncResponseTransformer<T, Void> {
         private boolean exceptionOccurred = false;