[NI] Fix initial data flow info for analyzing lambdas

#KT-30734 Fixed

compiler/fir/resolve/tests/org/jetbrains/kotlin/fir/FirDiagnosticsSmokeTestGenerated.java

@@ -13944,6 +13944,11 @@ public class FirDiagnosticsSmokeTestGenerated extends AbstractFirDiagnosticsSmok
             runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt244.kt");
         }
 
+        @TestMetadata("kt30734.kt")
+        public void testKt30734() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt30734.kt");
+        }
+
         @TestMetadata("kt362.kt")
         public void testKt362() throws Exception {
             runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt362.kt");

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/tower/KotlinResolutionCallbacksImpl.kt

@@ -155,7 +155,7 @@ class KotlinResolutionCallbacksImpl(
             .replaceBindingTrace(trace)
             .replaceContextDependency(lambdaInfo.contextDependency)
             .replaceExpectedType(approximatesExpectedType)
-            .replaceDataFlowInfo(psiCallArgument.lambdaInitialDataFlowInfo).let {
+            .replaceDataFlowInfo(psiCallArgument.dataFlowInfoBeforeThisArgument).let {
                 if (coroutineSession != null) it.replaceInferenceSession(coroutineSession) else it
             }
 

compiler/testData/diagnostics/tests/infos/SmartCasts.kt

@@ -183,7 +183,7 @@ fun returnFunctionLiteral(a: Any?): Function0<Int> {
 }
 
 fun returnFunctionLiteralExpressionBody(a: Any?): Function0<Int> =
-        if (a is Int) { -> <!NI;TYPE_MISMATCH, NI;TYPE_MISMATCH, NI;TYPE_MISMATCH, NI;TYPE_MISMATCH, OI;DEBUG_INFO_SMARTCAST!>a<!> }
+        if (a is Int) { -> <!DEBUG_INFO_SMARTCAST!>a<!> }
         else { -> 1 }
 
 

compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt30734.kt

@@ -0,0 +1,16 @@
+// !DIAGNOSTICS: -UNUSED_VARIABLE
+// !WITH_NEW_INFERENCE
+// !LANGUAGE: +NewInference
+// Issue: KT-30734
+
+class Sample {
+    fun foo(): Boolean = true
+}
+
+fun test(ls: Sample?) {
+    val filter: () -> Boolean = if (ls == null) {
+        { false }
+    } else {
+        { <!DEBUG_INFO_SMARTCAST!>ls<!>.foo() } // OK in OI, error in NI
+    }
+}

compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt30734.txt

@@ -0,0 +1,11 @@
+package
+
+public fun test(/*0*/ ls: Sample?): kotlin.Unit
+
+public final class Sample {
+    public constructor Sample()
+    public open override /*1*/ /*fake_override*/ fun equals(/*0*/ other: kotlin.Any?): kotlin.Boolean
+    public final fun foo(): kotlin.Boolean
+    public open override /*1*/ /*fake_override*/ fun hashCode(): kotlin.Int
+    public open override /*1*/ /*fake_override*/ fun toString(): kotlin.String
+}

compiler/tests/org/jetbrains/kotlin/checkers/DiagnosticsTestGenerated.java

@@ -13951,6 +13951,11 @@ public class DiagnosticsTestGenerated extends AbstractDiagnosticsTest {
                 runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt244.kt");
             }
 
+            @TestMetadata("kt30734.kt")
+            public void testKt30734() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt30734.kt");
+            }
+
             @TestMetadata("kt362.kt")
             public void testKt362() throws Exception {
                 runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt362.kt");

compiler/tests/org/jetbrains/kotlin/checkers/javac/DiagnosticsUsingJavacTestGenerated.java

@@ -13946,6 +13946,11 @@ public class DiagnosticsUsingJavacTestGenerated extends AbstractDiagnosticsUsing
                 runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt244.kt");
             }
 
+            @TestMetadata("kt30734.kt")
+            public void testKt30734() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt30734.kt");
+            }
+
             @TestMetadata("kt362.kt")
             public void testKt362() throws Exception {
                 runTest("compiler/testData/diagnostics/tests/nullabilityAndSmartCasts/kt362.kt");