Fix package name typo which was leading to imports not being added by…

… OutputCaptureExtension. Should fix #46 for real this time.
openrewrite · Apr 24, 2021 · 7db05b4 · 7db05b4
1 parent 1de360c
commit 7db05b4
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/main/java/org/openrewrite/java/spring/boot2/OutputCaptureExtension.java b/src/main/java/org/openrewrite/java/spring/boot2/OutputCaptureExtension.java
@@ -40,8 +40,9 @@ public class OutputCaptureExtension extends Recipe {
                             Parser.Input.fromString("package org.springframework.boot.test.system;\n" +
                                     "public interface CapturedOutput {\n" +
                                     "  String getAll();\n" +
-                                    "}"),
-                            Parser.Input.fromString("package org.junit.jupiter.api.extension.ExtendWith;\n" +
+                                    "}"
+                            ),
+                            Parser.Input.fromString("package org.junit.jupiter.api.extension;\n" +
                                     "public @interface ExtendWith {\n" +
                                     "  Class[] value();\n" +
                                     "}")
@@ -115,10 +116,7 @@ public J.ClassDeclaration visitClassDeclaration(J.ClassDeclaration classDecl, Ex
                             ))
                     );
 
-                    // We got a bug report that this import wasn't being added
-                    // Couldn't reproduce why, assumed that type attribution failed for reasons that aren't obvious.
-                    // Setting onlyIfReferenced to "false" just to be sure.
-                    doAfterVisit(new AddImport<>("org.springframework.boot.test.system.OutputCaptureExtension", null, false));
+                    maybeAddImport("org.springframework.boot.test.system.OutputCaptureExtension");
                     maybeAddImport("org.junit.jupiter.api.extension.ExtendWith");
                 }