Files
kotlin-fork/compiler/testData/codegen/box/when/enumOptimization/manyWhensWithinClass.kt
T
Mikhail Glukhikh a4c7619c89 [FIR2IR] Introduce & use FirBuiltInsPackageFragment
Without this commit, JVM name mapping logic in BE does not work for FIR,
because FIR cannot use old BuiltInsPackageFragmentImpl descriptor.
In this commit we add our own implementation thus fixing
a pack of FIR black box tests.
2020-03-24 10:37:53 +03:00

57 lines
1.4 KiB
Kotlin
Vendored

// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar1_u51tkt$ count=3
// CHECK_IF_COUNT: function=bar1_u51tkt$ count=0
// CHECK_CASES_COUNT: function=A$bar2$lambda count=3
// CHECK_IF_COUNT: function=A$bar2$lambda count=0
import kotlin.test.assertEquals
enum class Season {
WINTER,
SPRING,
SUMMER,
AUTUMN
}
class A {
public fun bar1(x : Season) : String {
when (x) {
Season.WINTER, Season.SPRING -> return "winter_spring"
Season.SPRING -> return "spring"
Season.SUMMER -> return "summer"
}
return "autumn";
}
public fun bar2(y : Season) : String {
return bar3(y) { x ->
when (x) {
Season.WINTER, Season.SPRING -> "winter_spring"
Season.SPRING -> "spring"
Season.SUMMER -> "summer"
else -> "autumn"
}
}
}
private fun bar3(x : Season, block : (Season) -> String) = block(x)
}
fun box() : String {
val a = A()
assertEquals("winter_spring", a.bar1(Season.WINTER))
assertEquals("winter_spring", a.bar1(Season.SPRING))
assertEquals("summer", a.bar1(Season.SUMMER))
assertEquals("autumn", a.bar1(Season.AUTUMN))
assertEquals("winter_spring", a.bar2(Season.WINTER))
assertEquals("winter_spring", a.bar2(Season.SPRING))
assertEquals("summer", a.bar2(Season.SUMMER))
assertEquals("autumn", a.bar2(Season.AUTUMN))
return "OK"
}