StreamApiMigrationInspection: improve stream source recognition

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/afterJoiningInfiniteSource.java

@@ -0,0 +1,11 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+      String sb = IntStream.iterate(0, i -> i + 1).filter(i -> i % 100 == 0).mapToObj(String::valueOf).collect(Collectors.joining());
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/afterJoiningInfiniteSourceComplexUpdate.java

@@ -0,0 +1,11 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+      String sb = IntStream.iterate(0, i -> i + 23 * 11).filter(i -> i % 100 == 0).mapToObj(String::valueOf).collect(Collectors.joining());
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/afterJoiningInfiniteSourceComplexUpdateOp.java

@@ -0,0 +1,11 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+      String sb = IntStream.iterate(0, i -> i * (i + 23)).filter(i -> i % 100 == 0).mapToObj(String::valueOf).collect(Collectors.joining());
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/afterJoiningInfiniteSourceLong.java

@@ -0,0 +1,11 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.IntStream;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+      String sb = IntStream.iterate(0, i -> (int) (i + 10L)).filter(i -> i % 100 == 0).mapToObj(String::valueOf).collect(Collectors.joining());
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/afterJoiningInfiniteSourceReference.java

@@ -0,0 +1,19 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+
+class A {
+  B next() {return null;}
+  int x;
+}
+
+class B extends A {}
+
+
+public class Main {
+  public static int find(List<List<String>> list) {
+      String sb = Stream.iterate(new A(), A::next).filter(a -> a.x % 100 == 0).map(a -> String.valueOf(a.x)).collect(Collectors.joining());
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/beforeJoiningInfiniteSource.java

@@ -0,0 +1,14 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+    StringBuilder sb = new StringBuilder();
+    for <caret> (int i = 0;; i++) {
+      if(i % 100 == 0) {
+        sb.append(i);
+      }
+    }
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/beforeJoiningInfiniteSourceComplexUpdate.java

@@ -0,0 +1,14 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+    StringBuilder sb = new StringBuilder();
+    for <caret> (int i = 0;; i = i + 23 * 11) {
+      if(i % 100 == 0) {
+        sb.append(i);
+      }
+    }
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/beforeJoiningInfiniteSourceComplexUpdateOp.java

@@ -0,0 +1,14 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+    StringBuilder sb = new StringBuilder();
+    for <caret> (int i = 0;; i *= i + 23) {
+      if(i % 100 == 0) {
+        sb.append(i);
+      }
+    }
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/beforeJoiningInfiniteSourceLong.java

@@ -0,0 +1,14 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+
+public class Main {
+  public static int find(List<List<String>> list) {
+    StringBuilder sb = new StringBuilder();
+    for <caret> (int i = 0;; i += 10L) {
+      if(i % 100 == 0) {
+        sb.append(i);
+      }
+    }
+  }
+}

java/java-tests/testData/codeInsight/daemonCodeAnalyzer/quickFix/streamApiMigration/joining/beforeJoiningInfiniteSourceReference.java

@@ -0,0 +1,22 @@
+// "Replace with collect" "true"
+
+import java.util.List;
+
+class A {
+  B next() {return null;}
+  int x;
+}
+
+class B extends A {}
+
+
+public class Main {
+  public static int find(List<List<String>> list) {
+    StringBuilder sb = new StringBuilder();
+    for <caret> (A a = new A();; a = a.next()) {
+      if(a.x % 100 == 0) {
+        sb.append(a.x);
+      }
+    }
+  }
+}