FE: Fix null assertion error when left is CollectionLiteral

#KT-49961 fixed

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosisCompilerTestFE10TestdataTestGenerated.java

@@ -13870,6 +13870,12 @@ public class DiagnosisCompilerTestFE10TestdataTestGenerated extends AbstractDiag
                 runTest("compiler/testData/diagnostics/tests/inference/kt49658Strict.kt");
             }
 
+            @Test
+            @TestMetadata("kt49961.kt")
+            public void testKt49961() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/inference/kt49961.kt");
+            }
+
             @Test
             @TestMetadata("kt51844.kt")
             public void testKt51844() throws Exception {

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

@@ -13870,6 +13870,12 @@ public class FirOldFrontendDiagnosticsTestGenerated extends AbstractFirDiagnosti
                 runTest("compiler/testData/diagnostics/tests/inference/kt49658Strict.kt");
             }
 
+            @Test
+            @TestMetadata("kt49961.kt")
+            public void testKt49961() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/inference/kt49961.kt");
+            }
+
             @Test
             @TestMetadata("kt51844.kt")
             public void testKt51844() throws Exception {

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

@@ -13870,6 +13870,12 @@ public class FirOldFrontendDiagnosticsWithLightTreeTestGenerated extends Abstrac
                 runTest("compiler/testData/diagnostics/tests/inference/kt49658Strict.kt");
             }
 
+            @Test
+            @TestMetadata("kt49961.kt")
+            public void testKt49961() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/inference/kt49961.kt");
+            }
+
             @Test
             @TestMetadata("kt51844.kt")
             public void testKt51844() throws Exception {

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/ArgumentTypeResolver.java

@@ -224,6 +224,10 @@ public class ArgumentTypeResolver {
         return null;
     }
 
+    public static boolean isCollectionLiteralArgument(@NotNull KtExpression expression) {
+        return expression instanceof KtCollectionLiteralExpression;
+    }
+
     @NotNull
     public KotlinTypeInfo getArgumentTypeInfo(
             @Nullable KtExpression expression,

compiler/frontend/src/org/jetbrains/kotlin/types/expressions/BasicExpressionTypingVisitor.java

@@ -1303,7 +1303,8 @@ public class BasicExpressionTypingVisitor extends ExpressionTypingVisitor {
         KotlinTypeInfo leftTypeInfo = BindingContextUtils.getRecordedTypeInfo(left, context.trace.getBindingContext());
         boolean isLeftFunctionLiteral = ArgumentTypeResolver.isFunctionLiteralArgument(left, context);
         boolean isLeftCallableReference = ArgumentTypeResolver.isCallableReferenceArgument(left, context);
-        if (leftTypeInfo == null && (isLeftFunctionLiteral || isLeftCallableReference)) {
+        boolean isLeftCollectionLiteral = ArgumentTypeResolver.isCollectionLiteralArgument(left);
+        if (leftTypeInfo == null && (isLeftFunctionLiteral || isLeftCallableReference || isLeftCollectionLiteral)) {
             return TypeInfoFactoryKt.noTypeInfo(context);
         }
         assert leftTypeInfo != null : "Left expression was not processed: " + expression;

compiler/testData/diagnostics/tests/inference/kt49961.fir.kt

@@ -0,0 +1,10 @@
+class Xyz {
+    fun x(): String? {
+        return try {
+            <!UNSUPPORTED!>[<!UNRESOLVED_REFERENCE!>a<!>]<!> <!USELESS_ELVIS!>?: <!UNRESOLVED_REFERENCE!>XYZ<!><!>
+        }
+        catch (e: Exception) {
+            null
+        }
+    }
+}

compiler/testData/diagnostics/tests/inference/kt49961.kt

@@ -0,0 +1,10 @@
+class Xyz {
+    fun x(): String? {
+        return try {
+            <!TYPE_MISMATCH, UNSUPPORTED!>[<!UNRESOLVED_REFERENCE!>a<!>]<!> <!USELESS_ELVIS!>?: <!UNRESOLVED_REFERENCE!>XYZ<!><!>
+        }
+        catch (e: Exception) {
+            null
+        }
+    }
+}

compiler/testData/diagnostics/tests/inference/kt49961.txt

@@ -0,0 +1,9 @@
+package
+
+public final class Xyz {
+    public constructor Xyz()
+    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
+    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
+    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
+    public final fun x(): kotlin.String?
+}

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java

@@ -13876,6 +13876,12 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
                 runTest("compiler/testData/diagnostics/tests/inference/kt49658Strict.kt");
             }
 
+            @Test
+            @TestMetadata("kt49961.kt")
+            public void testKt49961() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/inference/kt49961.kt");
+            }
+
             @Test
             @TestMetadata("kt51844.kt")
             public void testKt51844() throws Exception {