FIR/UAST: regression test about exp type for range operator

^KT-59564

plugins/kotlin/uast/uast-kotlin-base/tests/test/org/jetbrains/uast/test/common/kotlin/UastApiFixtureTestBase.kt

@@ -352,6 +352,35 @@ interface UastApiFixtureTestBase : UastPluginSelection {
         TestCase.assertEquals(UastCallKind.CONSTRUCTOR_CALL, uCallExpression.kind)
     }
 
+    // Regression test from KT-59564
+    fun checkExpressionTypeOfForEach(myFixture: JavaCodeInsightTestFixture) {
+        myFixture.configureByText(
+            "main.kt", """
+                // !LANGUAGE: +RangeUntilOperator
+                @file:OptIn(ExperimentalStdlibApi::class)
+                fun test(a: Int, b: Int) {
+                  for (i in a..<b step 1) {
+                       println(i)
+                  }
+                }
+            """.trimIndent()
+        )
+
+        val uFile = myFixture.file.toUElement()!!
+        uFile.accept(object : AbstractUastVisitor() {
+            override fun visitForEachExpression(node: UForEachExpression): Boolean {
+                when (val exp = node.iteratedValue.skipParenthesizedExprDown()) {
+                    is UBinaryExpression -> {
+                        TestCase.assertEquals("kotlin.ranges.IntProgression", exp.getExpressionType()?.canonicalText)
+                        TestCase.assertEquals("kotlin.ranges.IntRange", exp.leftOperand.getExpressionType()?.canonicalText)
+                    }
+                }
+
+                return super.visitForEachExpression(node)
+            }
+        })
+    }
+
     // Regression test from KTIJ-23503
     fun checkExpressionTypeFromIncorrectObject(myFixture: JavaCodeInsightTestFixture) {
         myFixture.configureByText(

plugins/kotlin/uast/uast-kotlin-fir/tests/test/org/jetbrains/fir/uast/test/FirUastApiFixtureTest.kt

@@ -57,6 +57,10 @@ class FirUastApiFixtureTest : KotlinLightCodeInsightFixtureTestCase(), UastApiFi
         checkCallKindOfSamConstructor(myFixture)
     }
 
+    fun testExpressionTypeOfForEach() {
+        checkExpressionTypeOfForEach(myFixture)
+    }
+
     fun testExpressionTypeFromIncorrectObject() {
         checkExpressionTypeFromIncorrectObject(myFixture)
     }

plugins/kotlin/uast/uast-kotlin/tests/test/org/jetbrains/uast/test/kotlin/comparison/FE1UastApiFixtureTest.kt

@@ -55,6 +55,10 @@ class FE1UastApiFixtureTest : KotlinLightCodeInsightFixtureTestCase(), UastApiFi
         checkCallKindOfSamConstructor(myFixture)
     }
 
+    fun testExpressionTypeOfForEach() {
+        checkExpressionTypeOfForEach(myFixture)
+    }
+
     fun testExpressionTypeFromIncorrectObject() {
         checkExpressionTypeFromIncorrectObject(myFixture)
     }