Remove unused imports after remove unused private methods

Fixes #347
openrewrite · Oct 6, 2024 · 34b5938 · 34b5938
1 parent d523fc1
commit 34b5938
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/staticanalysis/RemoveUnusedPrivateMethods.java b/src/main/java/org/openrewrite/staticanalysis/RemoveUnusedPrivateMethods.java
@@ -21,6 +21,7 @@
 import org.openrewrite.TreeVisitor;
 import org.openrewrite.java.JavaIsoVisitor;
 import org.openrewrite.java.NoMissingTypes;
+import org.openrewrite.java.RemoveUnusedImports;
 import org.openrewrite.java.search.FindAnnotations;
 import org.openrewrite.java.service.AnnotationService;
 import org.openrewrite.java.tree.*;
@@ -123,6 +124,7 @@ public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method,
                         }
                     }
 
+                    doAfterVisit(new RemoveUnusedImports().getVisitor());
                     //noinspection ConstantConditions
                     return null;
                 }

diff --git a/src/test/java/org/openrewrite/staticanalysis/RemoveUnusedPrivateMethodsTest.java b/src/test/java/org/openrewrite/staticanalysis/RemoveUnusedPrivateMethodsTest.java
@@ -153,8 +153,6 @@ private Stream<Object> unused() {
               }
               """,
             """
-              import java.util.stream.Stream;
-              
               class Test {
                   void test(String input) {
                   }
-Original file line number
+Diff line change
@@ Expand Up / @@ -153,8 +153,6 @@ private Stream<Object> unused() { @@
                   }
                   """,
                 """
-                  import java.util.stream.Stream;
                   class Test {
                       void test(String input) {
                       }
@@ Expand Down @@