Polish. Fixed a few tests to pass type validation.

openrewrite · Nov 30, 2023 · 033a1bc · 033a1bc
1 parent 20aac80
commit 033a1bc
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 13 deletions.
diff --git a/src/test/java/org/openrewrite/kotlin/AssertionsTest.java b/src/test/java/org/openrewrite/kotlin/AssertionsTest.java
@@ -30,25 +30,22 @@
 import static org.openrewrite.kotlin.Assertions.kotlin;
 import static org.openrewrite.test.RewriteTest.toRecipe;
 
-// This class may be removed after recipes with Assertions are added.
 public class AssertionsTest implements RewriteTest {
-    @Override
-    public void defaults(RecipeSpec spec) {
-        spec.recipe(toRecipe(() -> new KotlinIsoVisitor<>() {
-            @Override
-            public J.VariableDeclarations.NamedVariable visitVariable(J.VariableDeclarations.NamedVariable variable, ExecutionContext executionContext) {
-                if ("a".equals(variable.getSimpleName())) {
-                    return variable.withName(variable.getName().withSimpleName("b"));
-                }
-                return variable;
-            }
-        }));
-    }
 
     @Issue("https://github.com/openrewrite/rewrite-kotlin/issues/30")
     @Test
     void isChanged() {
         rewriteRun(
+          spec -> spec.recipe(toRecipe(() -> new KotlinIsoVisitor<>() {
+              @Override
+              public J.VariableDeclarations.NamedVariable visitVariable(J.VariableDeclarations.NamedVariable variable, ExecutionContext executionContext) {
+                  J.VariableDeclarations.NamedVariable n = super.visitVariable(variable, executionContext);
+                  if ("a".equals(n.getSimpleName())) {
+                      return n.withName(n.getName().withSimpleName("b").withType(null).withFieldType(null)).withType(null).withVariableType(null);
+                  }
+                  return n;
+              }
+          })),
           kotlin(
             """
             class A {

diff --git a/src/test/java/org/openrewrite/kotlin/format/SpacesTest.java b/src/test/java/org/openrewrite/kotlin/format/SpacesTest.java
@@ -159,12 +159,14 @@ fun foo() {
                       foo ()
                       val test = Test ()
                   }
+                  class Test
                   """,
                 """
                   fun foo() {
                       foo()
                       val test = Test()
                   }
+                  class Test
                   """
               )
             );