Analysis API: fix call resolve inside plusAssign target

^KT-50864 fixed

analysis/analysis-api-fe10/tests/org/jetbrains/kotlin/analysis/api/fe10/fir/Fe10ResolveCallTestGenerated.java

@@ -771,4 +771,20 @@ public class Fe10ResolveCallTestGenerated extends AbstractResolveCallTest {
     public void testVariableWithMemberInvoke() throws Exception {
         runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/variableWithMemberInvoke.kt");
     }
+
+    @Nested
+    @TestMetadata("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Assignments {
+        @Test
+        public void testAllFilesPresentInAssignments() throws Exception {
+            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments"), Pattern.compile("^(.+)\\.kt$"), null, true);
+        }
+
+        @Test
+        @TestMetadata("insidePlusAssignTarget.kt")
+        public void testInsidePlusAssignTarget() throws Exception {
+            runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments/insidePlusAssignTarget.kt");
+        }
+    }
 }

analysis/analysis-api-fir/tests/org/jetbrains/kotlin/analysis/api/fir/fir/FirResolveCallTestGenerated.java

@@ -771,4 +771,20 @@ public class FirResolveCallTestGenerated extends AbstractResolveCallTest {
     public void testVariableWithMemberInvoke() throws Exception {
         runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/variableWithMemberInvoke.kt");
     }
+
+    @Nested
+    @TestMetadata("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Assignments {
+        @Test
+        public void testAllFilesPresentInAssignments() throws Exception {
+            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments"), Pattern.compile("^(.+)\\.kt$"), null, true);
+        }
+
+        @Test
+        @TestMetadata("insidePlusAssignTarget.kt")
+        public void testInsidePlusAssignTarget() throws Exception {
+            runTest("analysis/analysis-api/testData/components/callResolver/resolveCall/assignments/insidePlusAssignTarget.kt");
+        }
+    }
 }

analysis/analysis-api/testData/components/callResolver/resolveCall/assignments/insidePlusAssignTarget.kt

@@ -0,0 +1,5 @@
+// WITH_STDLIB
+
+fun foo(x: MutableMap<Int, MutableList<String>>) {
+    x.getOrPut(1) { <expr>mutableListOf</expr><String>() } += "str"
+}

analysis/analysis-api/testData/components/callResolver/resolveCall/assignments/insidePlusAssignTarget.txt

@@ -0,0 +1,12 @@
+KtSuccessCallInfo:
+  call = KtSimpleFunctionCall:
+    isImplicitInvoke = false
+    partiallyAppliedSymbol = KtPartiallyAppliedSymbol:
+      dispatchReceiver = null
+      extensionReceiver = null
+      signature = KtFunctionLikeSignature:
+        receiverType = null
+        returnType = kotlin.collections.MutableList<kotlin.String>
+        symbol = kotlin/collections/mutableListOf(): kotlin.collections.MutableList<T>
+        valueParameters = []
+    argumentMapping = {}

analysis/low-level-api-fir/testdata/getOrBuildFir/calls/compoundAssignOnVal_lhs.txt

@@ -1,6 +1,6 @@
 KT element: KtNameReferenceExpression
 FIR element: FirPropertyAccessExpressionImpl
-FIR source kind: DesugaredCompoundAssignment
+FIR source kind: KtRealSourceElementKind
 
 FIR element rendered:
 R|<local>/l|

analysis/low-level-api-fir/testdata/getOrBuildFir/expressions/insidePlusAssignTarget.kt

@@ -0,0 +1,6 @@
+// WITH_STDLIB
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: org.jetbrains.kotlin.psi.KtCallExpression
+
+fun foo(x: MutableMap<Int, MutableList<String>>) {
+    x.getOrPut(1) { <expr>mutableListOf<String>()</expr> } += "str"
+}

analysis/low-level-api-fir/testdata/getOrBuildFir/expressions/insidePlusAssignTarget.txt

@@ -0,0 +1,6 @@
+KT element: KtCallExpression
+FIR element: FirFunctionCallImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+R|kotlin/collections/mutableListOf|<R|kotlin/String|>()

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/GetOrBuildFirTestGenerated.java

@@ -279,6 +279,12 @@ public class GetOrBuildFirTestGenerated extends AbstractGetOrBuildFirTest {
             runTest("analysis/low-level-api-fir/testdata/getOrBuildFir/expressions/ifExpression.kt");
         }
 
+        @Test
+        @TestMetadata("insidePlusAssignTarget.kt")
+        public void testInsidePlusAssignTarget() throws Exception {
+            runTest("analysis/low-level-api-fir/testdata/getOrBuildFir/expressions/insidePlusAssignTarget.kt");
+        }
+
         @Test
         @TestMetadata("intLiteral.kt")
         public void testIntLiteral() throws Exception {

compiler/fir/raw-fir/raw-fir.common/src/org/jetbrains/kotlin/fir/builder/BaseFirBuilder.kt

@@ -8,7 +8,6 @@ package org.jetbrains.kotlin.fir.builder
 import com.intellij.psi.tree.IElementType
 import org.jetbrains.kotlin.*
 import org.jetbrains.kotlin.KtNodeTypes.*
-import org.jetbrains.kotlin.config.LanguageFeature
 import org.jetbrains.kotlin.descriptors.Modality
 import org.jetbrains.kotlin.descriptors.Visibilities
 import org.jetbrains.kotlin.fir.*
@@ -919,14 +918,13 @@ abstract class BaseFirBuilder<T>(val baseSession: FirSession, val context: Conte
             return buildAssignmentOperatorStatement {
                 source = baseSource
                 this.operation = operation
-                leftArgument = withDefaultSourceElementKind(KtFakeSourceElementKind.DesugaredCompoundAssignment) {
+                leftArgument = this@generateAssignment?.convert()
-                    this@generateAssignment?.convert()
+                    ?: buildErrorExpression {
-                } ?: buildErrorExpression {
+                        source = null
-                    source = null
+                        diagnostic = ConeSimpleDiagnostic(
-                    diagnostic = ConeSimpleDiagnostic(
+                            "Unsupported left value of assignment: ${baseSource?.psi?.text}", DiagnosticKind.ExpressionExpected
-                        "Unsupported left value of assignment: ${baseSource?.psi?.text}", DiagnosticKind.ExpressionExpected
+                        )
-                    )
+                    }
-                }
                 rightArgument = value
                 this.annotations += annotations
             }