[IR] Add more complex nonlocal break/continue tests

^KT-56466

Author: vsukharev

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirBlackBoxCodegenBasedTestGenerated.java

@@ -12,8 +12,7 @@ inline fun <T> Iterable<T>.myForEach(action: (T) -> Unit): Unit {
     for (element in this) action(element)
 }
 
-
-fun box(): String {
+private fun testMyForEach() {
     val visited = mutableListOf<Pair<Int, Int>>()
 
     for (i in 1..3) {
@@ -26,5 +25,57 @@ fun box(): String {
     }
 
     assertEquals(listOf(1 to 1, 1 to 2), visited)
+}
+
+inline fun <T> Iterable<T>.myForEachWithBreak(dealBreaker: T, action: (T) -> Unit): Unit {
+    for (element in this) {
+        if (element == dealBreaker)
+            break
+        action(element)
+    }
+}
+
+private fun testMyForEachWithBreak() {
+    val visited = mutableListOf<Pair<Int, Int>>()
+
+    for (i in 1..3) {
+        (1..3).myForEachWithBreak(3) { j ->
+            if (i == 3) {
+                break
+            }
+            visited += i to j
+        }
+    }
+
+    assertEquals(listOf(1 to 1, 1 to 2, 2 to 1, 2 to 2), visited)
+}
+
+inline fun <T> Iterable<T>.myForEachWithContinue(skipper: T, action: (T) -> Unit): Unit {
+    for (element in this) {
+        if (element == skipper)
+            continue
+        action(element)
+    }
+}
+
+private fun testMyForEachWithContinue() {
+    val visited = mutableListOf<Pair<Int, Int>>()
+
+    for (i in 1..3) {
+        (1..3).myForEachWithContinue(2) { j ->
+            if (j == 3) {
+                break
+            }
+            visited += i to j
+        }
+    }
+
+    assertEquals(listOf(1 to 1), visited)
+}
+
+fun box(): String {
+    testMyForEach()
+    testMyForEachWithBreak()
+    testMyForEachWithContinue()
     return "OK"
 }

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirReversedBlackBoxCodegenBasedTestGenerated.java

@@ -0,0 +1,75 @@
+// LANGUAGE: +BreakContinueInInlineLambdas
+// IGNORE_BACKEND: JVM
+// WITH_STDLIB
+
+inline fun foo(block: () -> Int): Int  = block()
+
+fun box(): String {
+    val resultNonLabeled = testNonLabeledBreaks()
+    if (resultNonLabeled != (1)+(1+2)+(1+2)+(1+2)) return "FAIL testNonLabeledBreaks: $resultNonLabeled"
+
+    val resultLabeled = testLabeledBreaks()
+    if (resultLabeled != (1)+(1+2)) return "FAIL testLabeledBreaks: $resultLabeled"
+
+    val resultForEach = testForEach()
+    if (resultForEach != (1)+(1+2)) return "FAIL testForEach: $resultForEach"
+    return "OK"
+}
+
+private fun testNonLabeledBreaks(): Int {
+    var sum = 0
+
+    for (i in 1..10) {
+        sum += foo {
+            if (i == 5) break else {
+                var innerSum = 0
+                for (j in 1..i) {
+                    innerSum += foo {
+                        if (j == 3) break else j
+                    }
+                }
+                innerSum
+            }
+        }
+    }
+    return sum
+}
+
+private fun testLabeledBreaks(): Int {
+    var sum = 0
+
+    outer@
+    for (i in 1..10) {
+        sum += foo {
+            if (i == 5) break else {
+                var innerSum = 0
+                for (j in 1..i) {
+                    innerSum += foo {
+                        if (j == 3) break@outer else j
+                    }
+                }
+                innerSum
+            }
+        }
+    }
+    return sum
+}
+
+private fun testForEach(): Int {
+    var sum = 0
+
+    for (i in 1..10) {
+        sum += foo {
+            if (i == 5) break else {
+                var innerSum = 0
+                (1..i).forEach { j: Int ->
+                    innerSum += foo {
+                        if (j == 3) break else j // breaks not `forEach`, but outer `for`, so similar to `break@outer`
+                    }
+                }
+                innerSum
+            }
+        }
+    }
+    return sum
+}