KT-51247 Fix resolution of functional types with context receivers

compiler/testData/diagnostics/tests/extensions/contextReceivers/fromKEEP/functionalType.fir.kt

@@ -1,21 +0,0 @@
-// !LANGUAGE: +ContextReceivers
-// !DIAGNOSTICS: -ASSIGNED_BUT_NEVER_ACCESSED_VARIABLE
-
-class Context
-class Receiver
-class Param
-
-fun foo(context: Context, receiver: Receiver, p: Param) {}
-
-context(Context)
-fun bar(receiver: Receiver, p: Param) {}
-
-context(Context)
-fun Receiver.baz(p: Param) {}
-
-fun main() {
-    var g: context(Context) Receiver.(Param) -> Unit
-    g = ::foo         // OK
-    g = ::<!UNRESOLVED_REFERENCE!>bar<!>         // OK
-    g = Receiver::<!UNRESOLVED_REFERENCE!>baz<!> // OK
-}

compiler/testData/diagnostics/tests/extensions/contextReceivers/fromKEEP/functionalType.kt

@@ -1,3 +1,4 @@
+// FIR_IDENTICAL
 // !LANGUAGE: +ContextReceivers
 // !DIAGNOSTICS: -ASSIGNED_BUT_NEVER_ACCESSED_VARIABLE