Add test collectionLiteralExpression and fix

openrewrite · Nov 7, 2023 · 5e9ae7d · 5e9ae7d
1 parent da8f517
commit 5e9ae7d
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java
@@ -2294,7 +2294,7 @@ public J visitDotQualifiedExpression(KtDotQualifiedExpression expression, Execut
                         .withName(((J.MethodInvocation) methodInvocation).getName().withPrefix(prefix(callExpression)))
                         .withPrefix(prefix(expression));
             } else if (methodInvocation instanceof J.NewClass) {
-                if (receiver instanceof J.FieldAccess) {
+                if (receiver instanceof J.FieldAccess || receiver instanceof J.Identifier) {
                     J.NewClass cur = (J.NewClass) methodInvocation;
                     if (cur.getClazz() instanceof J.ParameterizedType) {
                         cur = cur.withPrefix(prefix(expression));

diff --git a/src/test/java/org/openrewrite/kotlin/tree/AnnotationTest.java b/src/test/java/org/openrewrite/kotlin/tree/AnnotationTest.java
@@ -466,7 +466,7 @@ annotation class B
     }
 
     @Test
-    void AnntationEntryTrailingComma() {
+    void AnnotationEntryTrailingComma() {
         rewriteRun(
           spec -> spec.parser(KotlinParser.builder().classpath("jackson-annotations")),
           kotlin(
@@ -492,4 +492,22 @@ public sealed interface Fruit {
           )
         );
     }
+
+    @Test
+    void collectionLiteralExpression() {
+        rewriteRun(
+          spec -> spec.parser(KotlinParser.builder().classpath("jackson-annotations")),
+          kotlin(
+            """
+              import com.fasterxml.jackson.`annotation`.JsonSubTypes
+
+              @JsonSubTypes(value = [
+                JsonSubTypes .  Type(value = Employee::class, name = "Employee")
+              ])
+              public sealed interface Person {
+              }
+              """
+          )
+        );
+    }
 }