Unwrap block expressions to report errors on them about not enough type information

^KT-44055 Fixed

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/DiagnosticReporterByTrackingStrategy.kt

@@ -461,7 +461,11 @@ class DiagnosticReporterByTrackingStrategy(
                         is TypeVariableForLambdaReturnType -> "return type of lambda"
                         else -> error("Unsupported type variable: $typeVariable")
                     }
-                    trace.reportDiagnosticOnce(NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER.on(expression, typeVariableName))
+                    val unwrappedExpression = if (expression is KtBlockExpression) {
+                        expression.statements.lastOrNull() ?: expression
+                    } else expression
+
+                    trace.reportDiagnosticOnce(NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER.on(unwrappedExpression, typeVariableName))
                 }
             }
 

compiler/testData/diagnostics/tests/inference/reportNotEnoughTypeInformationErrorsOnBlockExpressions.fir.kt

@@ -0,0 +1,29 @@
+// WITH_RUNTIME
+// SKIP_TXT
+// !DIAGNOSTICS: -CAST_NEVER_SUCCEEDS
+
+import kotlin.experimental.ExperimentalTypeInference
+
+class Foo7<T>
+
+fun foo7() = null as Foo7<Int>
+
+fun poll17(flag: Boolean): Any? {
+    val inv = if (flag) { foo7() } else { ::Foo7 }
+    return inv
+}
+
+fun poll26(flag: Boolean): Any? {
+    val inv = when (flag) { true -> ::Foo7 false -> foo7() else -> ::Foo7 }
+    return inv
+}
+
+fun poll36(flag: Boolean): Any? {
+    val inv = when (flag) { true -> ::Foo7 false -> foo7() }
+    return inv
+}
+
+fun poll56(): Any? {
+    val inv = try { ::Foo7 } catch (e: Exception) { foo7() } finally { foo7() }
+    return inv
+}

compiler/testData/diagnostics/tests/inference/reportNotEnoughTypeInformationErrorsOnBlockExpressions.kt

@@ -0,0 +1,29 @@
+// WITH_RUNTIME
+// SKIP_TXT
+// !DIAGNOSTICS: -CAST_NEVER_SUCCEEDS
+
+import kotlin.experimental.ExperimentalTypeInference
+
+class Foo7<T>
+
+fun foo7() = null as Foo7<Int>
+
+fun poll17(flag: Boolean): Any? {
+    val inv = if (flag) { <!IMPLICIT_CAST_TO_ANY!>foo7()<!> } else { <!IMPLICIT_CAST_TO_ANY!>::Foo7<!> }
+    return inv
+}
+
+fun poll26(flag: Boolean): Any? {
+    val inv = when (flag) { true -> <!IMPLICIT_CAST_TO_ANY, NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>::Foo7<!> false -> <!IMPLICIT_CAST_TO_ANY!>foo7()<!> <!REDUNDANT_ELSE_IN_WHEN!>else<!> -> <!IMPLICIT_CAST_TO_ANY, NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>::Foo7<!> }
+    return inv
+}
+
+fun poll36(flag: Boolean): Any? {
+    val inv = when (flag) { true -> <!IMPLICIT_CAST_TO_ANY, NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>::Foo7<!> false -> <!IMPLICIT_CAST_TO_ANY!>foo7()<!> }
+    return inv
+}
+
+fun poll56(): Any? {
+    val inv = try { ::Foo7 } catch (e: Exception) { foo7() } finally { foo7() }
+    return inv
+}