Support annotations on assignments

openrewrite · Sep 11, 2023 · 0ba5ec8 · 0ba5ec8
1 parent 8a5a0d5
commit 0ba5ec8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/main/kotlin/org/openrewrite/kotlin/internal/KotlinParserVisitor.kt b/src/main/kotlin/org/openrewrite/kotlin/internal/KotlinParserVisitor.kt
@@ -4492,7 +4492,7 @@ class KotlinParserVisitor(
             }
         }
 
-        if (firElement is FirExpression && firElement.annotations.isNotEmpty()) {
+        if ((firElement is FirExpression || firElement is FirVariableAssignment) && (firElement as FirStatement).annotations.isNotEmpty()) {
             val annotations = ArrayList<J.Annotation>(firElement.annotations.size)
             for (annotation in firElement.annotations) {
                 annotations.add(visitElement(annotation, data) as J.Annotation)

diff --git a/src/test/java/org/openrewrite/kotlin/tree/AssignmentTest.java b/src/test/java/org/openrewrite/kotlin/tree/AssignmentTest.java
@@ -135,4 +135,19 @@ void not() {
           )
         );
     }
+
+    @Test
+    void annotation() {
+        rewriteRun(
+          kotlin(
+            """
+              fun test() {
+                  var a: Boolean
+                  @Suppress
+                  a = true
+              }
+              """
+          )
+        );
+    }
 }