diff --git a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java index 025aab232..6c51f4476 100644 --- a/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java +++ b/src/main/java/org/openrewrite/kotlin/internal/KotlinTreeParserVisitor.java @@ -2571,7 +2571,7 @@ public J visitObjectLiteralExpression(KtObjectLiteralExpression expression, Exec args = JContainer.empty(); args = args.withMarkers(Markers.build(singletonList(new OmitParentheses(randomId())))); } else { - if (declaration.getSuperTypeList().getEntries().size() > 1) { + if (declaration.getSuperTypeList().getEntries().size() != 1) { throw new UnsupportedOperationException("TODO"); } @@ -2583,7 +2583,7 @@ public J visitObjectLiteralExpression(KtObjectLiteralExpression expression, Exec args = args.withMarkers(Markers.EMPTY.addIfAbsent(new OmitParentheses(randomId()))); } - clazz = declaration.getSuperTypeList().accept(this, data).withPrefix(Space.EMPTY); + clazz = (TypeTree) declaration.getSuperTypeList().getEntries().get(0).accept(this, data); } // TODO: fix NPE. diff --git a/src/test/java/org/openrewrite/kotlin/tree/MethodInvocationTest.java b/src/test/java/org/openrewrite/kotlin/tree/MethodInvocationTest.java index 228195b92..6cb5c93bb 100644 --- a/src/test/java/org/openrewrite/kotlin/tree/MethodInvocationTest.java +++ b/src/test/java/org/openrewrite/kotlin/tree/MethodInvocationTest.java @@ -248,7 +248,7 @@ open class Test fun test ( a : Test ) { } fun method ( ) { - test ( object : Test ( ) { + test ( object : Test ( ) { } ) } """