From eb11901d432cb4564f89fe9b60dee3fa609800dc Mon Sep 17 00:00:00 2001
From: Nikolay Lunyak <nikolay.lunyak@jetbrains.com>
Date: Mon, 15 Jan 2024 14:21:52 +0200
Subject: [PATCH] [FIR] Add some corner case tests for KT-64891

Note that there are some that K2
correctly supports, while K1 fails.

^KT-64891
---
 ...CompilerTestFE10TestdataTestGenerated.java | 18 ++++++++++
 ...sticCompilerFE10TestDataTestGenerated.java | 18 ++++++++++
 ...eeOldFrontendDiagnosticsTestGenerated.java | 18 ++++++++++
 ...siOldFrontendDiagnosticsTestGenerated.java | 18 ++++++++++
 ...expressionsInQaSelectorsCornerCases.fir.kt | 34 +++++++++++++++++++
 .../expressionsInQaSelectorsCornerCases.kt    | 34 +++++++++++++++++++
 ...ionsInQaSelectorsReceiverConversion.fir.kt |  5 +++
 ...ressionsInQaSelectorsReceiverConversion.kt |  5 +++
 .../expressionsInQaSelectorsWithElvis.kt      |  6 ++++
 .../test/runners/DiagnosticTestGenerated.java | 18 ++++++++++
 10 files changed, 174 insertions(+)
 create mode 100644 compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.fir.kt
 create mode 100644 compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt
 create mode 100644 compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.fir.kt
 create mode 100644 compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt
 create mode 100644 compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt

diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
index 2914410d91c..6432cb9e107 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
@@ -375,6 +375,24 @@ public class DiagnosticCompilerTestFE10TestdataTestGenerated extends AbstractDia
             runTest("compiler/testData/diagnostics/tests/equalityWithSmartCastInIfBlock.kt");
         }
 
+        @Test
+        @TestMetadata("expressionsInQaSelectorsCornerCases.kt")
+        public void testExpressionsInQaSelectorsCornerCases() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsReceiverConversion.kt")
+        public void testExpressionsInQaSelectorsReceiverConversion() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsWithElvis.kt")
+        public void testExpressionsInQaSelectorsWithElvis() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt");
+        }
+
         @Test
         @TestMetadata("ExtensionCallInvoke.kt")
         public void testExtensionCallInvoke() throws Exception {
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
index 1d29fcf58f7..c795ae06f25 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
@@ -375,6 +375,24 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated
             runTest("compiler/testData/diagnostics/tests/equalityWithSmartCastInIfBlock.kt");
         }
 
+        @Test
+        @TestMetadata("expressionsInQaSelectorsCornerCases.kt")
+        public void testExpressionsInQaSelectorsCornerCases() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsReceiverConversion.kt")
+        public void testExpressionsInQaSelectorsReceiverConversion() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsWithElvis.kt")
+        public void testExpressionsInQaSelectorsWithElvis() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt");
+        }
+
         @Test
         @TestMetadata("ExtensionCallInvoke.kt")
         public void testExtensionCallInvoke() throws Exception {
diff --git a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java
index 71a3023c889..dbe579eba1e 100644
--- a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java
+++ b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java
@@ -375,6 +375,24 @@ public class FirLightTreeOldFrontendDiagnosticsTestGenerated extends AbstractFir
             runTest("compiler/testData/diagnostics/tests/equalityWithSmartCastInIfBlock.kt");
         }
 
+        @Test
+        @TestMetadata("expressionsInQaSelectorsCornerCases.kt")
+        public void testExpressionsInQaSelectorsCornerCases() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsReceiverConversion.kt")
+        public void testExpressionsInQaSelectorsReceiverConversion() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsWithElvis.kt")
+        public void testExpressionsInQaSelectorsWithElvis() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt");
+        }
+
         @Test
         @TestMetadata("ExtensionCallInvoke.kt")
         public void testExtensionCallInvoke() throws Exception {
diff --git a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
index 109c1f20fbd..1821a6fdc0f 100644
--- a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
+++ b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
@@ -375,6 +375,24 @@ public class FirPsiOldFrontendDiagnosticsTestGenerated extends AbstractFirPsiDia
             runTest("compiler/testData/diagnostics/tests/equalityWithSmartCastInIfBlock.kt");
         }
 
+        @Test
+        @TestMetadata("expressionsInQaSelectorsCornerCases.kt")
+        public void testExpressionsInQaSelectorsCornerCases() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsReceiverConversion.kt")
+        public void testExpressionsInQaSelectorsReceiverConversion() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsWithElvis.kt")
+        public void testExpressionsInQaSelectorsWithElvis() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt");
+        }
+
         @Test
         @TestMetadata("ExtensionCallInvoke.kt")
         public void testExtensionCallInvoke() throws Exception {
diff --git a/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.fir.kt b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.fir.kt
new file mode 100644
index 00000000000..e0a3353309a
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.fir.kt
@@ -0,0 +1,34 @@
+// ISSUE: KT-64891
+
+fun <T> T.b(): Int = 10
+
+fun <T> Int.a(it: T): Int = this
+
+fun main() {
+    5.(<!UNRESOLVED_REFERENCE!>::<!UNRESOLVED_REFERENCE!>b<!><!>)().inv()
+    5.(<!UNRESOLVED_REFERENCE!>::<!UNRESOLVED_REFERENCE!>a<!><!>)('=').inv()
+
+    val c = Int::b
+    5.(c)().inv()
+
+    val f: Int.() -> Int = Int::b
+    5.(f)().inv()
+
+    5.(Int::b)().inv()
+
+    val d = Int::<!NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>a<!>
+    5.(d)('=').inv()
+
+    val e: Int.(Char) -> Int = Int::a
+    5.(e)('=').inv()
+}
+
+val <T> T.x get(): Int = 10
+
+fun rain() {
+    5.(<!UNRESOLVED_REFERENCE!>x<!>)().inv()
+
+    5.(<!UNRESOLVED_REFERENCE!>::<!UNRESOLVED_REFERENCE!>x<!><!>)().inv()
+
+    5.(Int::x)().inv()
+}
diff --git a/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt
new file mode 100644
index 00000000000..a8cbc4e9f52
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt
@@ -0,0 +1,34 @@
+// ISSUE: KT-64891
+
+fun <T> T.b(): Int = 10
+
+fun <T> Int.a(it: T): Int = this
+
+fun main() {
+    5.<!FUNCTION_EXPECTED!>(::<!UNRESOLVED_REFERENCE!>b<!>)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+    5.<!FUNCTION_EXPECTED!>(::<!UNRESOLVED_REFERENCE!>a<!>)<!>('=').<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    val c = Int::b
+    5.<!FUNCTION_EXPECTED!>(c)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    val f: Int.() -> Int = Int::b
+    5.(f)().inv()
+
+    5.<!FUNCTION_EXPECTED!>(Int::b)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    val d = Int::<!NEW_INFERENCE_NO_INFORMATION_FOR_PARAMETER!>a<!>
+    5.<!FUNCTION_EXPECTED!>(<!DEBUG_INFO_ELEMENT_WITH_ERROR_TYPE!>d<!>)<!>('=').<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    val e: Int.(Char) -> Int = Int::a
+    5.(e)('=').inv()
+}
+
+val <T> T.x get(): Int = 10
+
+fun rain() {
+    5.<!FUNCTION_EXPECTED!>(<!UNRESOLVED_REFERENCE!>x<!>)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    5.<!FUNCTION_EXPECTED!>(::<!UNRESOLVED_REFERENCE!>x<!>)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+
+    5.<!FUNCTION_EXPECTED!>(Int::x)<!>().<!DEBUG_INFO_MISSING_UNRESOLVED!>inv<!>()
+}
diff --git a/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.fir.kt b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.fir.kt
new file mode 100644
index 00000000000..ec751e6aef8
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.fir.kt
@@ -0,0 +1,5 @@
+// ISSUE: KT-64891
+
+fun test(f: (Int) -> Int) {
+    2.(f)()
+}
\ No newline at end of file
diff --git a/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt
new file mode 100644
index 00000000000..9b88d7adc1b
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt
@@ -0,0 +1,5 @@
+// ISSUE: KT-64891
+
+fun test(f: (Int) -> Int) {
+    2.<!NO_RECEIVER_ALLOWED!>(f)<!>()
+}
diff --git a/compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt
new file mode 100644
index 00000000000..d4ccafc3964
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt
@@ -0,0 +1,6 @@
+// FIR_IDENTICAL
+// ISSUE: KT-64891
+
+fun test(a: (Int.() -> Int)?, b: Int.() -> Int) {
+    2.(a ?: b)()
+}
diff --git a/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java b/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
index ed6f553592e..3e86d289508 100644
--- a/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
+++ b/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
@@ -375,6 +375,24 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
             runTest("compiler/testData/diagnostics/tests/equalityWithSmartCastInIfBlock.kt");
         }
 
+        @Test
+        @TestMetadata("expressionsInQaSelectorsCornerCases.kt")
+        public void testExpressionsInQaSelectorsCornerCases() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsCornerCases.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsReceiverConversion.kt")
+        public void testExpressionsInQaSelectorsReceiverConversion() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsReceiverConversion.kt");
+        }
+
+        @Test
+        @TestMetadata("expressionsInQaSelectorsWithElvis.kt")
+        public void testExpressionsInQaSelectorsWithElvis() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/expressionsInQaSelectorsWithElvis.kt");
+        }
+
         @Test
         @TestMetadata("ExtensionCallInvoke.kt")
         public void testExtensionCallInvoke() throws Exception {