Add @ExtendWith(MockitoExtension.class) Test using Mockito Annotations

src/main/java/org/openrewrite/java/testing/mockito/AddMockitoExtensionIfAnnotationsUsed.java

@@ -0,0 +1,68 @@
+/*
+ * Copyright 2025 the original author or authors.
+ * <p>
+ * Licensed under the Moderne Source Available License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * <p>
+ * https://docs.moderne.io/licensing/moderne-source-available-license
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.openrewrite.java.testing.mockito;
+
+import org.openrewrite.ExecutionContext;
+import org.openrewrite.Recipe;
+import org.openrewrite.TreeVisitor;
+import org.openrewrite.internal.ListUtils;
+import org.openrewrite.java.JavaIsoVisitor;
+import org.openrewrite.java.JavaParser;
+import org.openrewrite.java.JavaTemplate;
+import org.openrewrite.java.search.FindAnnotations;
+import org.openrewrite.java.search.IsLikelyTest;
+import org.openrewrite.java.tree.J;
+
+import static org.openrewrite.Preconditions.*;
+
+public class AddMockitoExtensionIfAnnotationsUsed extends Recipe {
+    @Override
+    public String getDisplayName() {
+        return "Adds Mockito extensions to Mockito tests";
+    }
+
+    @Override
+    public String getDescription() {
+        return "Adds `@ExtendWith(MockitoExtension.class)` to tests using `@Mock` or `@Captor`.";
+    }
+
+    @Override
+    public TreeVisitor<?, ExecutionContext> getVisitor() {
+        return check(and(
+                new IsLikelyTest().getVisitor(),
+                not(new FindAnnotations("org.junit.jupiter.api.extension.ExtendWith(org.mockito.junit.jupiter.MockitoExtension.class)", false).getVisitor()),
+                or(new FindAnnotations("org.mockito.Captor", false).getVisitor(),
+                        new FindAnnotations("org.mockito.Mock", false).getVisitor())), new JavaIsoVisitor<ExecutionContext>() {
+            @Override
+            public J.ClassDeclaration visitClassDeclaration(J.ClassDeclaration classDecl, ExecutionContext ctx) {
+
+                maybeAddImport("org.mockito.junit.jupiter.MockitoExtension");
+                maybeAddImport("org.junit.jupiter.api.extension.ExtendWith");
+
+                J.Annotation extendsWith = ((J.ClassDeclaration) JavaTemplate.builder("@ExtendWith(MockitoExtension.class)")
+                        .imports("org.mockito.junit.jupiter.MockitoExtension")
+                        .imports("org.junit.jupiter.api.extension.ExtendWith")
+                        .javaParser(JavaParser.fromJavaVersion().classpathFromResources(ctx, "junit-jupiter-api", "mockito-junit-jupiter"))
+                        .build()
+                        .apply(getCursor(), classDecl.getCoordinates().replaceAnnotations()))
+                        .getLeadingAnnotations().get(0);
+
+                return autoFormat(classDecl.withLeadingAnnotations(ListUtils.concat(classDecl.getLeadingAnnotations(), extendsWith)), ctx);
+            }
+        });
+    }
+}

src/main/resources/META-INF/rewrite/mockito.yml

@@ -168,6 +168,7 @@ recipeList:
   - org.openrewrite.java.testing.mockito.CleanupMockitoImports
   - org.openrewrite.java.testing.mockito.MockUtilsToStatic
   - org.openrewrite.java.testing.junit5.MockitoJUnitToMockitoExtension
+  - org.openrewrite.java.testing.mockito.AddMockitoExtensionIfAnnotationsUsed
   - org.openrewrite.java.testing.mockito.RemoveInitMocksIfRunnersSpecified
   - org.openrewrite.java.testing.mockito.ReplacePowerMockito
   - org.openrewrite.java.dependencies.ChangeDependency:

src/test/java/org/openrewrite/java/testing/mockito/AddMockitoExtensionIfAnnotationsUsedTest.java

@@ -0,0 +1,117 @@
+/*
+ * Copyright 2025 the original author or authors.
+ * <p>
+ * Licensed under the Moderne Source Available License (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * <p>
+ * https://docs.moderne.io/licensing/moderne-source-available-license
+ * <p>
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.openrewrite.java.testing.mockito;
+
+import org.junit.jupiter.api.Test;
+import org.openrewrite.DocumentExample;
+import org.openrewrite.InMemoryExecutionContext;
+import org.openrewrite.java.JavaParser;
+import org.openrewrite.test.RecipeSpec;
+import org.openrewrite.test.RewriteTest;
+
+import static org.openrewrite.java.Assertions.java;
+
+class AddMockitoExtensionIfAnnotationsUsedTest implements RewriteTest {
+
+    @Override
+    public void defaults(RecipeSpec spec) {
+        spec.recipe(new AddMockitoExtensionIfAnnotationsUsed())
+          .parser(JavaParser.fromJavaVersion()
+            .classpathFromResources( new InMemoryExecutionContext(),"junit-jupiter-api", "mockito-junit-jupiter", "mockito-core")
+            .dependsOn("public class Service {}"));
+    }
+
+    @DocumentExample
+    @Test
+    void addForMock(){
+        rewriteRun(
+          //language=java
+          java(
+            """
+              import org.mockito.Mock;
+
+              class Test {
+                  @Mock
+                  Service service;
+              }
+              """,
+            """
+              import org.junit.jupiter.api.extension.ExtendWith;
+              import org.mockito.Mock;
+              import org.mockito.junit.jupiter.MockitoExtension;
+
+              @ExtendWith(MockitoExtension.class)
+              class Test {
+                  @Mock
+                  Service service;
+              }
+              """
+          )
+        );
+    }
+
+    @Test
+    void addForCaptor(){
+        rewriteRun(
+          //language=java
+          java(
+            """
+              import org.mockito.Captor;
+
+              class Test {
+                  @Captor
+                  Service service;
+              }
+              """,
+            """
+              import org.junit.jupiter.api.extension.ExtendWith;
+              import org.mockito.Captor;
+              import org.mockito.junit.jupiter.MockitoExtension;
+
+              @ExtendWith(MockitoExtension.class)
+              class Test {
+                  @Captor
+                  Service service;
+              }
+              """
+          )
+        );
+    }
+
+    @Test
+    void dontAddIfPresent(){
+        rewriteRun(
+          //language=java
+          java(
+            """
+              import org.junit.jupiter.api.extension.ExtendWith;
+              import org.mockito.Captor;
+              import org.mockito.Mock;
+              import org.mockito.junit.jupiter.MockitoExtension;
+
+              @ExtendWith(MockitoExtension.class)
+              class Test {
+                  @Captor
+                  Service service;
+                  @Mock
+                  Service service;
+              }
+              """
+          )
+        );
+    }
+}