KT-51243 Fix parameterized contextual lambda

^KT-51243 Fixed

compiler/testData/codegen/box/extensionFunctions/contextReceivers/parameterizedContextualLambda.kt

@@ -0,0 +1,13 @@
+// !LANGUAGE: +ContextReceivers
+// TARGET_BACKEND: JVM_IR
+// FIR status: context receivers aren't yet supported
+
+class A {
+    val result = "OK"
+}
+
+fun <T> test(receiver: T, action: context(T) () -> String) = action(receiver)
+
+fun box(): String = with(A()) {
+    result
+}

compiler/testData/diagnostics/tests/extensions/contextReceivers/parameterizedContextualLambda.kt

@@ -0,0 +1,20 @@
+// FIR_IDENTICAL
+// SKIP_TXT
+// !LANGUAGE: +ContextReceivers
+
+fun <T> test(action: context(T) () -> Unit) {}
+
+fun <T> test2(actionWithArg: context(T) (T) -> Unit) {}
+
+fun main() {
+    test<String> {
+        length
+    }
+    test<Int> {
+        toDouble()
+    }
+    test2<String> { a ->
+        length
+        a.length
+    }
+}