IDEA-168296 Comments are duplicated when I replace Map.entrySet() loop with Map.forEach

java/java-impl/src/com/intellij/codeInspection/java18api/Java8MapForEachInspection.java

@@ -191,7 +191,7 @@ public class Java8MapForEachInspection extends BaseJavaBatchLocalInspectionTool
       LambdaRefactoringUtil.simplifyToExpressionLambda(newLambda);
       entrySetCall.getArgumentList().add(newLambda);
       ExpressionUtils.bindCallTo(entrySetCall, "forEach");
-      return entrySetCall.getText();
+      return ct.text(entrySetCall);
     }
 
     private static void fixInForeach(PsiForeachStatement loop) {

java/java-tests/testData/inspection/java8MapForEach/afterForEach.java

@@ -86,4 +86,10 @@ public class Test {
           System.out.println(str + ":" + num);
       });
   }
+
+  void forEach(Map<String, String> map) {
+      map.forEach((key, value) -> {
+          //long comment
+      });
+  }
 }

java/java-tests/testData/inspection/java8MapForEach/beforeForEach.java

@@ -108,4 +108,10 @@ public class Test {
       System.out.println(str + ":" + num);
     }
   }
+
+  void forEach(Map<String, String> map) {
+    for (Map.Entry<String, String> entry : map.entrySet()) {
+      //long comment
+    }
+  }
 }