Expect to add import as needed

Author: timtebeek

src/main/java/org/openrewrite/java/recipes/UseRewriteTestDefaults.java

@@ -69,6 +69,7 @@ public J.ClassDeclaration visitClassDeclaration(J.ClassDeclaration classDecl, Ex
                 if (!allSpecsAreIdentical(specInfos)) {
                     return cd;
                 }
+                maybeAddImport("org.openrewrite.test.RecipeSpec");
                 cd = newlineBeforeFirstStatement(cd);
                 cd = addDefaultsMethod(cd, specInfos.get(0));
                 return removeSpecsFromRewriteRuns(cd, ctx);

src/test/java/org/openrewrite/java/recipes/UseRewriteTestDefaultsTest.java

@@ -38,7 +38,6 @@ void shouldRefactorWhenAllTestsUseSameRecipe() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {
@@ -96,7 +95,6 @@ void shouldNotRefactorWhenOnlyOneTest() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {
@@ -119,7 +117,6 @@ void shouldNotRefactorWhenTestsUseDifferentRecipes() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {
@@ -150,7 +147,6 @@ void shouldHandleComplexRecipeSpec() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
               import org.openrewrite.java.JavaParser;
 
@@ -216,7 +212,6 @@ void shouldNotRefactorWhenSomeTestsHaveDifferentSpecs() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {
@@ -311,9 +306,8 @@ void shouldHandleNestedClasses() {
         rewriteRun(
           java(
             """
-              import org.junit.jupiter.api.Test;
               import org.junit.jupiter.api.Nested;
-              import org.openrewrite.test.RecipeSpec;
+              import org.junit.jupiter.api.Test;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {
@@ -354,8 +348,8 @@ void test4() {
               }
               """,
             """
-              import org.junit.jupiter.api.Test;
               import org.junit.jupiter.api.Nested;
+              import org.junit.jupiter.api.Test;
               import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
@@ -410,7 +404,6 @@ void shouldPreserveOtherRewriteRunOverloads() {
           java(
             """
               import org.junit.jupiter.api.Test;
-              import org.openrewrite.test.RecipeSpec;
               import org.openrewrite.test.RewriteTest;
 
               class MyTest implements RewriteTest {