support return in forEach replacement

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/continue/afterContinueInside.java

@@ -0,0 +1,14 @@
+// "Replace with forEach" "true"
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Objects;
+
+class Sample {
+  List<String> foo = new ArrayList<>();
+  {
+      foo.stream().filter(Objects::isNull).forEach(s -> {
+          return;
+      });
+
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/continue/afterForEachContinue.java

@@ -0,0 +1,14 @@
+// "Replace with forEach" "true"
+
+import java.util.Arrays;
+import java.util.Collection;
+
+public class Test {
+  void test(int[] arr) {
+      Arrays.stream(arr).forEach(x -> {
+          int y = x * 2;
+          if (x > y) return;
+          System.out.println(x);
+      });
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/continue/beforeContinueInside.java

@@ -1,4 +1,4 @@
-// "Replace with forEach" "false"
+// "Replace with forEach" "true"
 import java.util.ArrayList;
 import java.util.List;
 
@@ -12,4 +12,4 @@ class Sample {
     }
 
   }
-}
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/foreach/afterForEachContinue.java

@@ -1,13 +0,0 @@
-// "Replace with forEach" "true"
-
-import java.util.Collection;
-
-public class Test {
-  void test(int[] arr) {
-    for<caret>(int x : arr) {
-      int y = x*2;
-      if(x > y) continue;
-      System.out.println(x);
-    }
-  }
-}