Add native image integ test

Author: zoewangg

test/sdk-native-image-test/src/main/java/software/amazon/awssdk/nativeimagetest/S3TestRunner.java

@@ -15,7 +15,10 @@
 
 package software.amazon.awssdk.nativeimagetest;
 
+import java.io.IOException;
 import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.UUID;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -24,11 +27,14 @@
 import software.amazon.awssdk.services.s3.S3AsyncClient;
 import software.amazon.awssdk.services.s3.S3Client;
 import software.amazon.awssdk.services.s3.model.CreateBucketResponse;
+import software.amazon.awssdk.services.s3.model.HeadObjectResponse;
 
 public class S3TestRunner implements TestRunner {
     private static final String BUCKET_NAME = "v2-native-image-tests-" + UUID.randomUUID();
     private static final Logger logger = LoggerFactory.getLogger(S3TestRunner.class);
     private static final String KEY = "key";
+    private static final String MIMETYPE_KEY = "mimetype-key.txt";
+    private static final String EXPECTED_TXT_CONTENT_TYPE = "text/plain";
     private final S3Client s3ApacheHttpClient;
     private final S3Client s3UrlConnectionHttpClient;
     private final S3AsyncClient s3NettyClient;
@@ -43,6 +49,8 @@ public S3TestRunner() {
     public void runTests() {
         logger.info("starting to run S3 tests");
         CreateBucketResponse bucketResponse = null;
+        Path tempFile = null;
+        boolean mimetypeObjectCreated = false;
         try {
             bucketResponse = s3UrlConnectionHttpClient.createBucket(b -> b.bucket(BUCKET_NAME));
 
@@ -56,12 +64,42 @@ public void runTests() {
             s3NettyClient.getObject(b -> b.bucket(BUCKET_NAME).key(KEY),
                                AsyncResponseTransformer.toBytes()).join();
 
+            tempFile = createTempTextFile();
+            s3ApacheHttpClient.putObject(b -> b.bucket(BUCKET_NAME).key(MIMETYPE_KEY),
+                                         RequestBody.fromFile(tempFile));
+
+            HeadObjectResponse head = s3ApacheHttpClient.headObject(b -> b.bucket(BUCKET_NAME).key(MIMETYPE_KEY));
+            String contentType = head.contentType();
+            if (contentType == null || !contentType.startsWith(EXPECTED_TXT_CONTENT_TYPE)) {
+                throw new RuntimeException("Expected Content-Type to start with '" + EXPECTED_TXT_CONTENT_TYPE
+                                           + "' but was '" + contentType + "'. The mime.types resource may be missing"
+                                           + " from the native image classpath.");
+            }
+
         } finally {
             if (bucketResponse != null) {
                 s3NettyClient.deleteObject(b -> b.bucket(BUCKET_NAME).key(KEY)).join();
+                s3NettyClient.deleteObject(b -> b.bucket(BUCKET_NAME).key(MIMETYPE_KEY)).join();
 
                 s3NettyClient.deleteBucket(b -> b.bucket(BUCKET_NAME)).join();
             }
+            if (tempFile != null) {
+                try {
+                    Files.deleteIfExists(tempFile);
+                } catch (IOException e) {
+                    logger.warn("Failed to delete temp file {}", tempFile, e);
+                }
+            }
+        }
+    }
+
+    private static Path createTempTextFile() {
+        try {
+            Path file = Files.createTempFile("native-image-mimetype-", ".txt");
+            Files.write(file, "helloworld".getBytes(StandardCharsets.UTF_8));
+            return file;
+        } catch (IOException e) {
+            throw new RuntimeException("Failed to create temp .txt file for mimetype test", e);
         }
     }
 }