From ae85c7bfe328fa2b76d87cbee2a4a48f60c87ae6 Mon Sep 17 00:00:00 2001
From: Egor Kulikov <egor.kulikov@jetbrains.com>
Date: Thu, 17 Aug 2023 10:54:39 +0000
Subject: [PATCH] [FIR] Add missing tests to resolve

Merge-request: KT-MR-11666
Merged-by: Egor Kulikov <Egor.Kulikov@jetbrains.com>
---
 ...rceModuleReferenceResolveTestGenerated.java | 18 ++++++++++++++++++
 ...rceModuleReferenceResolveTestGenerated.java | 18 ++++++++++++++++++
 ...rceModuleReferenceResolveTestGenerated.java | 18 ++++++++++++++++++
 ...rceModuleReferenceResolveTestGenerated.java | 18 ++++++++++++++++++
 ...rceModuleReferenceResolveTestGenerated.java | 18 ++++++++++++++++++
 .../testData/referenceResolve/InOperator.kt    |  4 ++++
 .../testData/referenceResolve/InOperator.txt   |  2 ++
 .../testData/referenceResolve/LabelsReturn.kt  |  3 +++
 .../testData/referenceResolve/LabelsReturn.txt |  2 ++
 .../testData/referenceResolve/NotInOperator.kt |  4 ++++
 .../referenceResolve/NotInOperator.txt         |  2 ++
 11 files changed, 107 insertions(+)
 create mode 100644 analysis/analysis-api/testData/referenceResolve/InOperator.kt
 create mode 100644 analysis/analysis-api/testData/referenceResolve/InOperator.txt
 create mode 100644 analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt
 create mode 100644 analysis/analysis-api/testData/referenceResolve/LabelsReturn.txt
 create mode 100644 analysis/analysis-api/testData/referenceResolve/NotInOperator.kt
 create mode 100644 analysis/analysis-api/testData/referenceResolve/NotInOperator.txt

diff --git a/analysis/analysis-api-fe10/tests-gen/org/jetbrains/kotlin/analysis/api/fe10/test/cases/generated/cases/references/Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java b/analysis/analysis-api-fe10/tests-gen/org/jetbrains/kotlin/analysis/api/fe10/test/cases/generated/cases/references/Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
index 15803754c7d..22087a78d17 100644
--- a/analysis/analysis-api-fe10/tests-gen/org/jetbrains/kotlin/analysis/api/fe10/test/cases/generated/cases/references/Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
+++ b/analysis/analysis-api-fe10/tests-gen/org/jetbrains/kotlin/analysis/api/fe10/test/cases/generated/cases/references/Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
@@ -310,6 +310,12 @@ public class Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exte
         runTest("analysis/analysis-api/testData/referenceResolve/InMethodVarargParameter.kt");
     }
 
+    @Test
+    @TestMetadata("InOperator.kt")
+    public void testInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/InOperator.kt");
+    }
+
     @Test
     @TestMetadata("InVaragReferenceInFunctionBody.kt")
     public void testInVaragReferenceInFunctionBody() throws Exception {
@@ -370,6 +376,12 @@ public class Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exte
         runTest("analysis/analysis-api/testData/referenceResolve/KotlinPropertyWithGetterAndSetterAssignment.kt");
     }
 
+    @Test
+    @TestMetadata("LabelsReturn.kt")
+    public void testLabelsReturn() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt");
+    }
+
     @Test
     @TestMetadata("MultiDeclarationExtension.kt")
     public void testMultiDeclarationExtension() throws Exception {
@@ -394,6 +406,12 @@ public class Fe10IdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exte
         runTest("analysis/analysis-api/testData/referenceResolve/NotEqualsOperator.kt");
     }
 
+    @Test
+    @TestMetadata("NotInOperator.kt")
+    public void testNotInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/NotInOperator.kt");
+    }
+
     @Test
     @TestMetadata("PackageFromAnnotationOnFile.kt")
     public void testPackageFromAnnotationOnFile() throws Exception {
diff --git a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated.java b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated.java
index a8dd5517d29..34598c0a8a1 100644
--- a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated.java
+++ b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated.java
@@ -310,6 +310,12 @@ public class FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated ex
         runTest("analysis/analysis-api/testData/referenceResolve/InMethodVarargParameter.kt");
     }
 
+    @Test
+    @TestMetadata("InOperator.kt")
+    public void testInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/InOperator.kt");
+    }
+
     @Test
     @TestMetadata("InVaragReferenceInFunctionBody.kt")
     public void testInVaragReferenceInFunctionBody() throws Exception {
@@ -370,6 +376,12 @@ public class FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated ex
         runTest("analysis/analysis-api/testData/referenceResolve/KotlinPropertyWithGetterAndSetterAssignment.kt");
     }
 
+    @Test
+    @TestMetadata("LabelsReturn.kt")
+    public void testLabelsReturn() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt");
+    }
+
     @Test
     @TestMetadata("MultiDeclarationExtension.kt")
     public void testMultiDeclarationExtension() throws Exception {
@@ -394,6 +406,12 @@ public class FirIdeDependentAnalysisSourceModuleReferenceResolveTestGenerated ex
         runTest("analysis/analysis-api/testData/referenceResolve/NotEqualsOperator.kt");
     }
 
+    @Test
+    @TestMetadata("NotInOperator.kt")
+    public void testNotInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/NotInOperator.kt");
+    }
+
     @Test
     @TestMetadata("PackageFromAnnotationOnFile.kt")
     public void testPackageFromAnnotationOnFile() throws Exception {
diff --git a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerated.java b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerated.java
index ee53637c8c5..960faa571d8 100644
--- a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerated.java
+++ b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerated.java
@@ -310,6 +310,12 @@ public class FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/InMethodVarargParameter.kt");
     }
 
+    @Test
+    @TestMetadata("InOperator.kt")
+    public void testInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/InOperator.kt");
+    }
+
     @Test
     @TestMetadata("InVaragReferenceInFunctionBody.kt")
     public void testInVaragReferenceInFunctionBody() throws Exception {
@@ -370,6 +376,12 @@ public class FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/KotlinPropertyWithGetterAndSetterAssignment.kt");
     }
 
+    @Test
+    @TestMetadata("LabelsReturn.kt")
+    public void testLabelsReturn() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt");
+    }
+
     @Test
     @TestMetadata("MultiDeclarationExtension.kt")
     public void testMultiDeclarationExtension() throws Exception {
@@ -394,6 +406,12 @@ public class FirIdeNormalAnalysisLibrarySourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/NotEqualsOperator.kt");
     }
 
+    @Test
+    @TestMetadata("NotInOperator.kt")
+    public void testNotInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/NotInOperator.kt");
+    }
+
     @Test
     @TestMetadata("PackageFromAnnotationOnFile.kt")
     public void testPackageFromAnnotationOnFile() throws Exception {
diff --git a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
index 3a3af033363..a4295b40840 100644
--- a/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
+++ b/analysis/analysis-api-fir/tests-gen/org/jetbrains/kotlin/analysis/api/fir/test/cases/generated/cases/references/FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
@@ -310,6 +310,12 @@ public class FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exten
         runTest("analysis/analysis-api/testData/referenceResolve/InMethodVarargParameter.kt");
     }
 
+    @Test
+    @TestMetadata("InOperator.kt")
+    public void testInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/InOperator.kt");
+    }
+
     @Test
     @TestMetadata("InVaragReferenceInFunctionBody.kt")
     public void testInVaragReferenceInFunctionBody() throws Exception {
@@ -370,6 +376,12 @@ public class FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exten
         runTest("analysis/analysis-api/testData/referenceResolve/KotlinPropertyWithGetterAndSetterAssignment.kt");
     }
 
+    @Test
+    @TestMetadata("LabelsReturn.kt")
+    public void testLabelsReturn() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt");
+    }
+
     @Test
     @TestMetadata("MultiDeclarationExtension.kt")
     public void testMultiDeclarationExtension() throws Exception {
@@ -394,6 +406,12 @@ public class FirIdeNormalAnalysisSourceModuleReferenceResolveTestGenerated exten
         runTest("analysis/analysis-api/testData/referenceResolve/NotEqualsOperator.kt");
     }
 
+    @Test
+    @TestMetadata("NotInOperator.kt")
+    public void testNotInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/NotInOperator.kt");
+    }
+
     @Test
     @TestMetadata("PackageFromAnnotationOnFile.kt")
     public void testPackageFromAnnotationOnFile() throws Exception {
diff --git a/analysis/analysis-api-standalone/tests-gen/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/generated/cases/references/FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerated.java b/analysis/analysis-api-standalone/tests-gen/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/generated/cases/references/FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
index ec4fc8725e6..0a19f826766 100644
--- a/analysis/analysis-api-standalone/tests-gen/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/generated/cases/references/FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
+++ b/analysis/analysis-api-standalone/tests-gen/org/jetbrains/kotlin/analysis/api/standalone/fir/test/cases/generated/cases/references/FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerated.java
@@ -310,6 +310,12 @@ public class FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/InMethodVarargParameter.kt");
     }
 
+    @Test
+    @TestMetadata("InOperator.kt")
+    public void testInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/InOperator.kt");
+    }
+
     @Test
     @TestMetadata("InVaragReferenceInFunctionBody.kt")
     public void testInVaragReferenceInFunctionBody() throws Exception {
@@ -370,6 +376,12 @@ public class FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/KotlinPropertyWithGetterAndSetterAssignment.kt");
     }
 
+    @Test
+    @TestMetadata("LabelsReturn.kt")
+    public void testLabelsReturn() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt");
+    }
+
     @Test
     @TestMetadata("MultiDeclarationExtension.kt")
     public void testMultiDeclarationExtension() throws Exception {
@@ -394,6 +406,12 @@ public class FirStandaloneNormalAnalysisSourceModuleReferenceResolveTestGenerate
         runTest("analysis/analysis-api/testData/referenceResolve/NotEqualsOperator.kt");
     }
 
+    @Test
+    @TestMetadata("NotInOperator.kt")
+    public void testNotInOperator() throws Exception {
+        runTest("analysis/analysis-api/testData/referenceResolve/NotInOperator.kt");
+    }
+
     @Test
     @TestMetadata("PackageFromAnnotationOnFile.kt")
     public void testPackageFromAnnotationOnFile() throws Exception {
diff --git a/analysis/analysis-api/testData/referenceResolve/InOperator.kt b/analysis/analysis-api/testData/referenceResolve/InOperator.kt
new file mode 100644
index 00000000000..dbb0f844522
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/InOperator.kt
@@ -0,0 +1,4 @@
+class Foo {
+    operator fun contains(v: Int) = true
+}
+fun foo() = 0 <caret>in Foo()
diff --git a/analysis/analysis-api/testData/referenceResolve/InOperator.txt b/analysis/analysis-api/testData/referenceResolve/InOperator.txt
new file mode 100644
index 00000000000..4ccaafffd25
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/InOperator.txt
@@ -0,0 +1,2 @@
+Resolved to:
+0: (in Foo) operator fun contains(v: kotlin.Int): kotlin.Boolean
\ No newline at end of file
diff --git a/analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt b/analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt
new file mode 100644
index 00000000000..d2f52054ab3
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/LabelsReturn.kt
@@ -0,0 +1,3 @@
+fun foo() {
+    return@<caret>foo
+}
diff --git a/analysis/analysis-api/testData/referenceResolve/LabelsReturn.txt b/analysis/analysis-api/testData/referenceResolve/LabelsReturn.txt
new file mode 100644
index 00000000000..e4bdf71735e
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/LabelsReturn.txt
@@ -0,0 +1,2 @@
+Resolved to:
+0: (in ROOT) fun foo()
diff --git a/analysis/analysis-api/testData/referenceResolve/NotInOperator.kt b/analysis/analysis-api/testData/referenceResolve/NotInOperator.kt
new file mode 100644
index 00000000000..d132e0a38c7
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/NotInOperator.kt
@@ -0,0 +1,4 @@
+class Foo {
+    operator fun contains(v: Int) = true
+}
+fun foo() = 0 <caret>!in Foo()
diff --git a/analysis/analysis-api/testData/referenceResolve/NotInOperator.txt b/analysis/analysis-api/testData/referenceResolve/NotInOperator.txt
new file mode 100644
index 00000000000..4ccaafffd25
--- /dev/null
+++ b/analysis/analysis-api/testData/referenceResolve/NotInOperator.txt
@@ -0,0 +1,2 @@
+Resolved to:
+0: (in Foo) operator fun contains(v: kotlin.Int): kotlin.Boolean
\ No newline at end of file