Also parse other class members in enum classes

Fixes: #307
openrewrite · Sep 11, 2023 · 7612f24 · 7612f24
1 parent f8fb467
commit 7612f24
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 6 deletions.
diff --git a/src/main/kotlin/org/openrewrite/kotlin/internal/KotlinParserVisitor.kt b/src/main/kotlin/org/openrewrite/kotlin/internal/KotlinParserVisitor.kt
@@ -3989,10 +3989,6 @@ class KotlinParserVisitor(
                 val range = TextRange(declaration.source!!.startOffset, declaration.source!!.endOffset)
                 generatedFirProperties[range] = declaration
             } else {
-                // We aren't interested in the generated values.
-                if (ClassKind.ENUM_CLASS == classKind && declaration.source != null) {
-                    continue
-                }
                 membersMultiVariablesSeparated.add(declaration)
             }
         }

diff --git a/src/test/java/org/openrewrite/kotlin/tree/EnumTest.java b/src/test/java/org/openrewrite/kotlin/tree/EnumTest.java
@@ -16,7 +16,6 @@
 package org.openrewrite.kotlin.tree;
 
 import org.junit.jupiter.api.Test;
-import org.junitpioneer.jupiter.ExpectedToFail;
 import org.openrewrite.Issue;
 import org.openrewrite.test.RewriteTest;
 
@@ -141,7 +140,6 @@ fun foo() = print("foo",)
 
     @Test
     @Issue("https://github.com/openrewrite/rewrite-kotlin/issues/307")
-    @ExpectedToFail
     void enumWithFunction() {
         rewriteRun(
           kotlin(