IDEA-161260 convert to findFirst() if assignment is followed by break

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/afterFindFirstAssignment.java

@@ -0,0 +1,12 @@
+// "Replace with findFirst()" "true"
+
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    int firstSize = map.values().stream().filter(Objects::nonNull).findFirst().map(List::size).orElse(0);
+      System.out.println(firstSize);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/afterFindFirstAssignmentIntermediate.java

@@ -0,0 +1,15 @@
+// "Replace with findFirst()" "true"
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    List<String> firstList = new ArrayList<>();
+    firstList.add("none");
+      firstList = map.values().stream().filter(Objects::nonNull).findFirst().orElse(firstList);
+    System.out.println(firstList);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/afterFindFirstAssignmentWithLabel.java

@@ -0,0 +1,13 @@
+// "Replace with findFirst()" "true"
+
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+import java.util.Objects;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    String firstStr = map.values().stream().filter(Objects::nonNull).flatMap(Collection::stream).filter(str -> !str.isEmpty()).findFirst().orElse("");
+      System.out.println(firstStr);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/beforeFindFirstAssignment.java

@@ -0,0 +1,17 @@
+// "Replace with findFirst()" "true"
+
+import java.util.List;
+import java.util.Map;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    int firstSize = 0;
+    for(List<String> list : map.valu<caret>es()) {
+      if(list != null) {
+        firstSize = list.size();
+        break;
+      }
+    }
+    System.out.println(firstSize);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/beforeFindFirstAssignmentIntermediate.java

@@ -0,0 +1,19 @@
+// "Replace with findFirst()" "true"
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    List<String> firstList = new ArrayList<>();
+    firstList.add("none");
+    for(List<String> list : ma<caret>p.values()) {
+      if(list != null) {
+        firstList = list;
+        break;
+      }
+    }
+    System.out.println(firstList);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/beforeFindFirstAssignmentNested.java

@@ -0,0 +1,22 @@
+// "Replace with findFirst()" "false"
+
+import java.util.List;
+import java.util.Map;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    String firstStr = "";
+    OUTER:
+    for(List<String> list : map.valu<caret>es()) {
+      if(list != null) {
+        for(String str : list) {
+          if(!str.isEmpty()) {
+            firstStr = str;
+            break;
+          }
+        }
+      }
+    }
+    System.out.println(firstStr);
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/beforeFindFirstAssignmentWithLabel.java

@@ -0,0 +1,22 @@
+// "Replace with findFirst()" "true"
+
+import java.util.List;
+import java.util.Map;
+
+public class Main {
+  public void testMap(Map<String, List<String>> map) throws Exception {
+    String firstStr = "";
+    OUTER:
+    for(List<String> list : map.valu<caret>es()) {
+      if(list != null) {
+        for(String str : list) {
+          if(!str.isEmpty()) {
+            firstStr = str;
+            break OUTER;
+          }
+        }
+      }
+    }
+    System.out.println(firstStr);
+  }
+}