From 4aa05d85a4317e6716486111f81f65d8444fe79d Mon Sep 17 00:00:00 2001
From: Anastasiya Shadrina <anastasia.shadrina@jetbrains.com>
Date: Fri, 22 Oct 2021 02:32:53 +0700
Subject: [PATCH] [FE] Handle context receivers in
 CallableReferencesCandidateFactory

---
 .../calls/model/CallableReferencesCandidateFactory.kt        | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/compiler/resolution/src/org/jetbrains/kotlin/resolve/calls/model/CallableReferencesCandidateFactory.kt b/compiler/resolution/src/org/jetbrains/kotlin/resolve/calls/model/CallableReferencesCandidateFactory.kt
index 0c5ba45392f..32c6182e871 100644
--- a/compiler/resolution/src/org/jetbrains/kotlin/resolve/calls/model/CallableReferencesCandidateFactory.kt
+++ b/compiler/resolution/src/org/jetbrains/kotlin/resolve/calls/model/CallableReferencesCandidateFactory.kt
@@ -333,7 +333,10 @@ class CallableReferencesCandidateFactory(
         builtins: KotlinBuiltIns,
         buildTypeWithConversions: Boolean = true
     ): Pair<UnwrappedType, CallableReferenceAdaptation?> {
-        val argumentsAndReceivers = ArrayList<KotlinType>(descriptor.valueParameters.size + 2)
+        val argumentsAndReceivers = ArrayList<KotlinType>(descriptor.valueParameters.size + 2 + descriptor.contextReceiverParameters.size)
+
+        val contextReceiversTypes = descriptor.contextReceiverParameters.map { it.type }
+        argumentsAndReceivers.addAll(contextReceiversTypes)
 
         if (dispatchReceiver is CallableReceiver.UnboundReference) {
             argumentsAndReceivers.add(dispatchReceiver.receiver.stableType)