ReplaceExtendWithAndContextConfiguration - only autoformat if other c…

…hanges were made
openrewrite · Jan 3, 2024 · eead6cf · eead6cf
1 parent d95bfad
commit eead6cf
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/...main/java/org/openrewrite/java/spring/boot2/ReplaceExtendWithAndContextConfiguration.java b/...main/java/org/openrewrite/java/spring/boot2/ReplaceExtendWithAndContextConfiguration.java
@@ -90,10 +90,11 @@ public J.Annotation visitAnnotation(J.Annotation annotation, ExecutionContext ct
                             maybeRemoveImport(FQN_CONTEXT_CONFIGURATION);
                             maybeAddImport(FQN_SPRING_JUNIT_CONFIG);
                             a = (J.Annotation) new ChangeType(FQN_CONTEXT_CONFIGURATION, FQN_SPRING_JUNIT_CONFIG, false)
-                                    .getVisitor().visit(a, ctx, getCursor());
+                                    .getVisitor().visitNonNull(a, ctx, getCursor());
+                            a = autoFormat(a, ctx);
                         }
 
-                        return a != null ? autoFormat(a, ctx) : annotation;
+                        return a;
                     }
 
                     private void replaceValueArgumentWithLocations(J.Annotation a, List<Expression> newArgs) {