Compute type for callable references and lambdas inside not null assertion if couldn't get already recorded type

2020-12-23 14:36:34 +03:00
parent 39e579db91
commit 954c9cecca
6 changed files with 293 additions and 6 deletions
@@ -0,0 +1,113 @@
+// WITH_RUNTIME
+
+import kotlin.experimental.ExperimentalTypeInference
+
+fun <K> FlowCollector<K>.bar(): K = null as K
+fun <K> FlowCollector<K>.foo(): K = null as K
+
+fun bar2(): Int = 1
+fun foo2(): Float = 1f
+
+fun <T> materialize() = null as T
+
+interface FlowCollector<in T> {}
+
+@Suppress("EXPERIMENTAL_API_USAGE_ERROR")
+fun <L> flow(@BuilderInference block: suspend FlowCollector<L>.() -> Unit) = Flow(block)
+
+class Flow<out R>(private val block: suspend FlowCollector<R>.() -> Unit)
+
+fun <R> select(vararg x: R) = x[0]
+
+fun poll01(): Flow<String> {
+    return flow {
+        val inv = select(::bar2, ::foo2)
+        inv()
+    }
+}
+
+fun poll1(flag: Boolean): Flow<String> {
+    return flow {
+        val inv = if (flag) { ::bar2 } else { ::foo2 }
+        inv()
+    }
+}
+
+fun poll11(flag: Boolean): Flow<String> {
+    return flow {
+        val inv = if (flag) { ::bar2 } else { ::foo2 }
+        inv()
+    }
+}
+
+
+fun poll21(flag: Boolean): Flow<String> {
+    return flow {
+        val inv = when (flag) { true -> ::bar2 else -> ::foo2 }
+        inv()
+    }
+}
+
+fun poll31(flag: Boolean): Flow<String> {
+    return flow {
+        val inv = when (flag) { true -> ::bar2 false -> ::foo2 }
+        inv()
+    }
+}
+
+fun poll41(): Flow<String> {
+    return flow {
+        val inv = try { ::bar2 } finally { ::foo2 }
+        inv()
+    }
+}
+
+fun poll51(): Flow<String> {
+    return flow {
+        val inv = try { ::bar2 } catch (e: Exception) { ::foo2 } finally { ::foo2 }
+        inv()
+    }
+}
+
+fun poll61(): Flow<String> {
+    return flow {
+        val inv = ::bar2
+        inv
+    }
+}
+
+fun poll71(): Flow<String> {
+    return flow {
+        val inv = ::bar2!!
+        inv()
+    }
+}
+
+fun poll81(): Flow<String> {
+    return flow {
+        val inv = ::bar2 in setOf(::foo2)
+        inv
+    }
+}
+
+fun poll91(): Flow<String> {
+    return flow {
+        val inv = ::foo2 in setOf(::foo2)
+        inv
+    }
+}
+
+fun box(): String {
+    poll01()
+    poll1(true)
+    poll11(true)
+    poll21(true)
+    poll31(true)
+    poll41()
+    poll51()
+    poll61()
+    poll71()
+    poll81()
+    poll91()
+    return "OK"
+}