Fix build and add end to end tests

Author: zoewangg

test/v2-migration-tests/src/test/resources/maven/after/src/main/java/foo/bar/Application.java

@@ -15,9 +15,10 @@
 
 package foo.bar;
 
-import software.amazon.awssdk.core.exception.SdkException;
 import software.amazon.awssdk.services.s3.S3Client;
 import software.amazon.awssdk.services.s3.model.GetObjectRequest;
+import software.amazon.awssdk.services.s3.model.PutObjectRequest;
+import software.amazon.awssdk.services.s3.model.PutObjectResponse;
 import software.amazon.awssdk.services.s3.model.S3Object;
 import software.amazon.awssdk.services.sqs.SqsClient;
 import software.amazon.awssdk.services.sqs.model.QueueDoesNotExistException;
@@ -32,6 +33,8 @@
 import java.nio.file.StandardCopyOption;
 import software.amazon.awssdk.awscore.exception.AwsServiceException;
 import software.amazon.awssdk.core.ResponseInputStream;
+import software.amazon.awssdk.core.exception.SdkException;
+import software.amazon.awssdk.core.sync.RequestBody;
 import software.amazon.awssdk.services.s3.model.GetObjectResponse;
 
 public class Application {
@@ -90,4 +93,11 @@ private static Path downloadFile(S3Client s3, String bucket, String key, Path ds
 
         return dst;
     }
+
+    private static PutObjectResponse uploadFile(S3Client s3, String bucket, String key, Path source) throws IOException {
+        PutObjectResponse result = s3.putObject(PutObjectRequest.builder()
+            .build(), RequestBody.fromFile(source.toFile()));
+
+        return result;
+    }
 }

test/v2-migration-tests/src/test/resources/maven/after/src/main/java/foo/bar/Enums.java

@@ -29,7 +29,8 @@ public static void main(String... args) {
         System.out.println(qan);
         System.out.println(qan2);
 
-        ReceiveMessageRequest v1Request = ReceiveMessageRequest.builder().build();
+        ReceiveMessageRequest v1Request = ReceiveMessageRequest.builder()
+            .build();
         List<String> attributes = v1Request.attributeNamesAsStrings();
         System.out.println(attributes);
 

test/v2-migration-tests/src/test/resources/maven/before/src/main/java/foo/bar/Application.java

@@ -19,6 +19,8 @@
 import com.amazonaws.AmazonServiceException;
 import com.amazonaws.services.s3.AmazonS3;
 import com.amazonaws.services.s3.model.GetObjectRequest;
+import com.amazonaws.services.s3.model.PutObjectRequest;
+import com.amazonaws.services.s3.model.PutObjectResult;
 import com.amazonaws.services.s3.model.S3Object;
 import com.amazonaws.services.sqs.AmazonSQS;
 import com.amazonaws.services.sqs.model.AmazonSQSException;
@@ -86,4 +88,10 @@ private static Path downloadFile(AmazonS3 s3, String bucket, String key, Path ds
 
         return dst;
     }
+
+    private static PutObjectResult uploadFile(AmazonS3 s3, String bucket, String key, Path source) throws IOException {
+        PutObjectResult result = s3.putObject(bucket, key, source.toFile());
+
+        return result;
+    }
 }

v2-migration/src/main/java/software/amazon/awssdk/v2migration/EnumCasingToV2.java

@@ -76,7 +76,12 @@ public J.Identifier visitIdentifier(J.Identifier identifier, ExecutionContext ct
             J.Identifier id = super.visitIdentifier(identifier, ctx);
 
             if (ENUMS.contains(id.getSimpleName())) {
-                return id.withFieldType(id.getFieldType().withName(id.getSimpleName()));
+                JavaType.Variable fieldType = id.getFieldType();
+                if (fieldType == null) {
+                    return id;
+                }
+                JavaType.Variable variable = fieldType.withName(id.getSimpleName());
+                return id.withFieldType(variable);
             }
 
             return id;

v2-migration/src/test/java/software/amazon/awssdk/v2migration/EnumGettersToV2Test.java

@@ -62,14 +62,15 @@ void shouldChangeSingleItemEnumGetter() {
                 "        String routeType = phoneNumberInformation.getRouteType();\n" +
                 "    }\n" +
                 "}\n",
-                "import software.amazon.awssdk.services.sns.model.PhoneNumberInformation;\n" +
-                "\n" +
-                "class Test {\n" +
-                "    static void method() {\n" +
-                "        PhoneNumberInformation phoneNumberInformation = PhoneNumberInformation.builder().build();\n" +
-                "        String routeType = phoneNumberInformation.routeTypeAsString();\n" +
-                "    }\n" +
-                "}\n"
+                "import software.amazon.awssdk.services.sns.model.PhoneNumberInformation;\n"
+                + "\n"
+                + "class Test {\n"
+                + "    static void method() {\n"
+                + "        PhoneNumberInformation phoneNumberInformation = PhoneNumberInformation.builder()\n"
+                + "                .build();\n"
+                + "        String routeType = phoneNumberInformation.routeTypeAsString();\n"
+                + "    }\n"
+                + "}"
             )
         );
     }
@@ -87,14 +88,15 @@ void shouldChangeCollectionItemEnumGetter() {
                 "        List<String> numberCapabilities = phoneNumberInformation.getNumberCapabilities();\n" +
                 "    }\n" +
                 "}\n",
-                "import software.amazon.awssdk.services.sns.model.PhoneNumberInformation;\n" +
-                "\n" +
-                "class Test {\n" +
-                "    static void method() {\n" +
-                "        PhoneNumberInformation phoneNumberInformation = PhoneNumberInformation.builder().build();\n" +
-                "        List<String> numberCapabilities = phoneNumberInformation.numberCapabilitiesAsStrings();\n" +
-                "    }\n" +
-                "}\n"
+                "import software.amazon.awssdk.services.sns.model.PhoneNumberInformation;\n"
+                + "\n"
+                + "class Test {\n"
+                + "    static void method() {\n"
+                + "        PhoneNumberInformation phoneNumberInformation = PhoneNumberInformation.builder()\n"
+                + "                .build();\n"
+                + "        List<String> numberCapabilities = phoneNumberInformation.numberCapabilitiesAsStrings();\n"
+                + "    }\n"
+                + "}"
             )
         );
     }