Format and suggestion

openrewrite · Sep 12, 2024 · a99e5d6 · a99e5d6
1 parent b865ce4
commit a99e5d6
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/main/java/org/openrewrite/java/testing/mockito/MockitoWhenOnStaticToMockStatic.java b/src/main/java/org/openrewrite/java/testing/mockito/MockitoWhenOnStaticToMockStatic.java
@@ -56,13 +56,13 @@ public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method, Ex
                         List<Statement> statementsAfterWhen = new ArrayList<>();
                         for (Statement stmt : m.getBody().getStatements()) {
                             if (stmt instanceof J.MethodInvocation &&
-                                MOCKITO_WHEN.matches(((J.MethodInvocation)stmt).getSelect())) {
-                                J.MethodInvocation when = (J.MethodInvocation)((J.MethodInvocation) stmt).getSelect();
-                                if (when.getArguments().get(0) instanceof J.MethodInvocation && ((J.MethodInvocation)when.getArguments().get(0)).getMethodType().getFlags().contains(Flag.Static)){
+                                MOCKITO_WHEN.matches(((J.MethodInvocation) stmt).getSelect())) {
+                                J.MethodInvocation when = (J.MethodInvocation) ((J.MethodInvocation) stmt).getSelect();
+                                if (when.getArguments().get(0) instanceof J.MethodInvocation && ((J.MethodInvocation) when.getArguments().get(0)).getMethodType().getFlags().contains(Flag.Static)) {
                                     JavaType.FullyQualified arg_fq = TypeUtils.asFullyQualified(when.getArguments().get(0).getType());
                                     String template = String.format("try(MockedStatic<%s> mock%s = mockStatic(%s.class)){\n" +
                                                                     "mock%s.when(%s::%s).thenReturn(%s);\n" +
-                                                                    "}", arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), ((J.MethodInvocation)when.getArguments().get(0)).getSimpleName(), ((J.MethodInvocation) stmt).getArguments().get(0));
+                                                                    "}", arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), arg_fq.getClassName(), ((J.MethodInvocation) when.getArguments().get(0)).getSimpleName(), ((J.MethodInvocation) stmt).getArguments().get(0));
                                     m = JavaTemplate.builder(template)
                                             .contextSensitive()
                                             .javaParser(JavaParser.fromJavaVersion())

diff --git a/src/test/java/org/openrewrite/java/testing/mockito/MockitoWhenOnStaticToMockStaticTest.java b/src/test/java/org/openrewrite/java/testing/mockito/MockitoWhenOnStaticToMockStaticTest.java
@@ -24,7 +24,7 @@
 
 import static org.openrewrite.java.Assertions.java;
 
-public class MockitoWhenOnStaticToMockStaticTest implements RewriteTest {
+class MockitoWhenOnStaticToMockStaticTest implements RewriteTest {
 
     @Override
     public void defaults(RecipeSpec spec) {