Loop over entrySet

openrewrite · Jan 23, 2025 · 0de9e43 · 0de9e43
1 parent 607c079
commit 0de9e43
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/main/java/org/openrewrite/java/testing/search/FindUnitTests.java b/src/main/java/org/openrewrite/java/testing/search/FindUnitTests.java
@@ -88,19 +88,18 @@ public TreeVisitor<?, ExecutionContext> getVisitor(Accumulator acc) {
         JavaVisitor<ExecutionContext> tableRowVisitor = new JavaVisitor<ExecutionContext>() {
             @Override
             public J visitMethodDeclaration(J.MethodDeclaration methodDeclaration, ExecutionContext ctx) {
-                for (UnitTest unitTest : acc.unitTestAndTheirMethods.keySet()) {
-                    for (J.MethodInvocation method : acc.unitTestAndTheirMethods.get(unitTest)) {
+                for (Map.Entry<UnitTest, Set<J.MethodInvocation>> entry : acc.unitTestAndTheirMethods.entrySet()) {
+                    for (J.MethodInvocation method : entry.getValue()) {
                         if (method.getSimpleName().equals(methodDeclaration.getSimpleName())) {
                             unitTestTable.insertRow(ctx, new FindUnitTestTable.Row(
                                     methodDeclaration.getName().toString(),
                                     methodDeclaration.getSimpleName(),
                                     method.printTrimmed(getCursor()),
-                                    unitTest.getClazz(),
-                                    unitTest.getUnitTestName()
+                                    entry.getKey().getClazz(),
+                                    entry.getKey().getUnitTestName()
                             ));
                         }
                     }
-                    return super.visitMethodDeclaration(methodDeclaration, ctx);
                 }
                 return super.visitMethodDeclaration(methodDeclaration, ctx);
             }