[NI] Fix shouldRunCompletion for builder inference session

#KT-41308 Fixed
#KT-41363

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/callableReferenceAndCoercionToUnit.kt

@@ -0,0 +1,9 @@
+// FIR_IDENTICAL
+// !DIAGNOSTICS: -EXPERIMENTAL_API_USAGE_ERROR -UNUSED_EXPRESSION
+
+fun test(s: String?) {
+    val list = buildList {
+        s?.let(::add)
+    }
+    <!DEBUG_INFO_EXPRESSION_TYPE("kotlin.collections.List<kotlin.String>")!>list<!>
+}

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/callableReferenceAndCoercionToUnit.txt

@@ -0,0 +1,3 @@
+package
+
+public fun test(/*0*/ s: kotlin.String?): kotlin.Unit

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/kt35684.kt

@@ -21,7 +21,7 @@ fun test_2() {
 fun test_3() {
     sequence {
         yield(materialize<Int>())
-        <!TYPE_MISMATCH, TYPE_MISMATCH, TYPE_MISMATCH, TYPE_MISMATCH!>materialize()<!>
+        <!NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>materialize<!>()
     }
 }
 
@@ -32,4 +32,4 @@ interface Inv<T> {
     fun yield(element: T)
 }
 
-fun <K> materialize(): Inv<K> = TODO()
+fun <K> materialize(): Inv<K> = TODO()

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/kt41308.fir.fail

@@ -0,0 +1,7 @@
+Failures detected in FirBodyResolveTransformerAdapter, file: /kt41308.fir.kt
+Cause: java.lang.RuntimeException: While resolving call R?C|kotlin/sequences/sequence|(<L> = sequence@fun R|kotlin/sequences/SequenceScope<Stub: TypeVariable(T)>|.<anonymous>(): R|kotlin/Unit| {
+    lval list: R|kotlin/collections/List<kotlin/String>?| = Null(null)
+    lval outputList: R|TypeVariable(K)| = R|<local>/list| ?: R?C|kotlin/collections/listOf|()
+    yieldAll#(R|<local>/outputList|)
+}
+)

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/kt41308.fir.kt

@@ -0,0 +1,9 @@
+// ISSUE: KT-41308
+
+fun main() {
+    sequence {
+        val list: List<String>? = null
+        val outputList = <!DEBUG_INFO_EXPRESSION_TYPE("kotlin.collections.List<kotlin.String>")!>list ?: listOf()<!>
+        yieldAll(outputList)
+    }
+}

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/kt41308.kt

@@ -0,0 +1,9 @@
+// ISSUE: KT-41308
+
+fun main() {
+    sequence {
+        val list: List<String>? = null
+        val outputList = <!DEBUG_INFO_EXPRESSION_TYPE("kotlin.collections.List<kotlin.String>")!>list ?: listOf()<!>
+        yieldAll(outputList)
+    }
+}

compiler/testData/diagnostics/testsWithStdLib/coroutines/inference/kt41308.txt

@@ -0,0 +1,3 @@
+package
+
+public fun main(): kotlin.Unit

compiler/testData/diagnostics/testsWithStdLib/coroutines/kt37309.kt

@@ -21,4 +21,4 @@ fun main() {
     }
 }
 
-class Foo(val notNullProp: String)
+class Foo(val notNullProp: String)