StreamToLoopInspection: handle labeled expression statement (found by property testing)

java/java-impl/src/com/intellij/codeInspection/streamToLoop/StreamToLoopInspection.java

@@ -563,7 +563,14 @@ public class StreamToLoopInspection extends AbstractBaseJavaLocalInspectionTool
     public PsiElement makeFinalReplacement() {
       LOG.assertTrue(myStreamExpression != null);
       if (myFinisher == null || myStreamExpression instanceof PsiStatement) {
-        myCommentTracker.delete(myStreamExpression);
+        PsiElement toDelete = myStreamExpression;
+        if (toDelete instanceof PsiExpression && toDelete.getParent() instanceof PsiExpressionStatement) {
+          toDelete = toDelete.getParent();
+          while (toDelete instanceof PsiExpressionStatement && toDelete.getParent() instanceof PsiLabeledStatement) {
+            toDelete = toDelete.getParent();
+          }
+        }
+        myCommentTracker.delete(toDelete);
         return null;
       }
       else {

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamToLoop/afterLabeled.java

@@ -0,0 +1,15 @@
+// "Fix all 'Stream API call chain can be replaced with loop' problems in file" "true"
+
+import java.util.Arrays;
+import java.util.stream.Stream;
+
+public class Main {
+  void test() {
+      for (String s : Arrays.asList("foo", "bar", "baz")) {
+          System.out.println(s);
+      }
+
+      String s = "xyz";
+    System.out.println(s);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamToLoop/beforeLabeled.java

@@ -0,0 +1,13 @@
+// "Fix all 'Stream API call chain can be replaced with loop' problems in file" "true"
+
+import java.util.stream.Stream;
+
+public class Main {
+  void test() {
+    test:
+    Stream<caret>.of("foo", "bar", "baz").forEach(System.out::println);
+
+    String s = "xyz";
+    System.out.println(s);
+  }
+}