[FIR] Fix completion of synthetic call arguments

compiler/fir/analysis-tests/testData/resolve/extendedCheckers/RedundantExplicitTypeChecker.fir.txt

@@ -75,7 +75,7 @@ FILE: RedundantExplicitTypeChecker.kt
     public final fun test(boolean: R|kotlin/Boolean|): R|kotlin/Unit| {
         lval expectedLong: R|kotlin/Long| = when () {
             R|<local>/boolean| ->  {
-                Int(42)
+                Long(42)
             }
             else ->  {
                 ^test Unit

compiler/fir/analysis-tests/testData/resolveWithStdlib/problems/SpecialCallsWithLambdas.fir.txt

@@ -0,0 +1,17 @@
+FILE: SpecialCallsWithLambdas.kt
+    public final fun foo(): R|kotlin/Unit| {
+        lval inv: R|() -> kotlin/Function0<kotlin/Unit>| = fun <anonymous>(): R|() -> kotlin/Unit| {
+            ^ fun <anonymous>(): R|kotlin/Unit| {
+                ^ Unit
+            }
+
+        }
+        !!
+        lval bar: R|() -> kotlin/Function0<kotlin/Unit>| = fun <anonymous>(): R|() -> kotlin/Unit| {
+            ^ fun <anonymous>(): R|kotlin/Unit| {
+                ^ Unit
+            }
+
+        }
+
+    }

compiler/fir/analysis-tests/testData/resolveWithStdlib/problems/SpecialCallsWithLambdas.kt

@@ -0,0 +1,4 @@
+fun foo() {
+    val inv = {{}}!!
+    val bar = {{}}
+}

compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirDiagnosticTestGenerated.java

@@ -4889,6 +4889,12 @@ public class FirDiagnosticTestGenerated extends AbstractFirDiagnosticTest {
                 runTest("compiler/fir/analysis-tests/testData/resolveWithStdlib/problems/receiverResolutionInLambda.kt");
             }
 
+            @Test
+            @TestMetadata("SpecialCallsWithLambdas.kt")
+            public void testSpecialCallsWithLambdas() throws Exception {
+                runTest("compiler/fir/analysis-tests/testData/resolveWithStdlib/problems/SpecialCallsWithLambdas.kt");
+            }
+
             @Test
             @TestMetadata("TypesEligibleForSimpleVisit.kt")
             public void testTypesEligibleForSimpleVisit() throws Exception {

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirCallCompletionResultsWriterTransformer.kt

@@ -534,13 +534,16 @@ class FirCallCompletionResultsWriterTransformer(
         syntheticCall: D,
         data: ExpectedArgumentType?,
     ): CompositeTransformResult<FirStatement> where D : FirResolvable, D : FirExpression {
-        syntheticCall.transformChildren(this, data?.getExpectedType(syntheticCall)?.toExpectedType())
         val calleeReference = syntheticCall.calleeReference as? FirNamedReferenceWithCandidate ?: return syntheticCall.compose()
 
         val declaration = calleeReference.candidate.symbol.fir as? FirSimpleFunction ?: return syntheticCall.compose()
 
         val typeRef = typeCalculator.tryCalculateReturnType(declaration)
         syntheticCall.replaceTypeRefWithSubstituted(calleeReference, typeRef)
+        syntheticCall.transformChildren(
+            this,
+            data = data?.getExpectedType(syntheticCall)?.toExpectedType() ?: syntheticCall.typeRef.coneType.toExpectedType()
+        )
 
         return (syntheticCall.transformCalleeReference(
             StoreCalleeReference,

compiler/testData/codegen/box/inference/builderInference/specialCallsWithLambdas.kt

@@ -1,5 +1,4 @@
 // WITH_RUNTIME
-// IGNORE_BACKEND_FIR: JVM_IR
 // DONT_TARGET_EXACT_BACKEND: WASM
 
 import kotlin.experimental.ExperimentalTypeInference

compiler/testData/diagnostics/tests/inference/regressions/kt36342_2.fir.kt

@@ -5,7 +5,7 @@ fun <M> materialize(): M = TODO()
 
 fun test(b: Boolean) {
     <!INAPPLICABLE_CANDIDATE!>id<!>(if (b) {
-        <!INAPPLICABLE_CANDIDATE!>id<!>(<!UNRESOLVED_REFERENCE!>unresolved<!>)
+        id(<!UNRESOLVED_REFERENCE!>unresolved<!>)
     } else {
         id(
             materialize()