From 4f4b30ee285f78c802ed51c9792009f50a4a2368 Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Thu, 10 Aug 2023 15:15:28 +0200
Subject: [PATCH] [LL FIR] dependent analyze: add test on different statements
 in script

It contains a wrong result. Will be fixed in the next commit

^KT-60987
---
 .../moreStatementsInCopy.context.txt          | 40 +++++++++++++++++++
 .../moreStatementsInCopy.copy.txt             | 18 +++++++++
 .../moreStatementsInCopy.fir.txt              | 22 ++++++++++
 .../dependentCopy/moreStatementsInCopy.kts    | 14 +++++++
 ...riptDependentCopyContextTestGenerated.java |  6 +++
 .../ScriptDependentCopyFirTestGenerated.java  |  6 +++
 6 files changed, 106 insertions(+)
 create mode 100644 analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.context.txt
 create mode 100644 analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.copy.txt
 create mode 100644 analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.fir.txt
 create mode 100644 analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts

diff --git a/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.context.txt b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.context.txt
new file mode 100644
index 00000000000..f9b1591c42f
--- /dev/null
+++ b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.context.txt
@@ -0,0 +1,40 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirDefaultSimpleImportingScope
+    Element 2
+        Scope: FirExplicitStarImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirDefaultSimpleImportingScope
+    Element 5
+        Scope: FirPackageMemberScope
+    Element 6
+        Scope: FirExplicitSimpleImportingScope
+    Element 7
+        Scope: FirScriptDeclarationsScope
+            Functions
+                FirNamedFunctionSymbol public? final? fun foo(i: Int, action: ( (Int) -> Unit )): R|kotlin/Unit|
+                FirNamedFunctionSymbol public? final? fun foo(i: Int, action: ( (Int) -> Unit )): R|kotlin/Unit|
+                FirNamedFunctionSymbol public? final? fun scriptFunction(): <implicit>
+                FirNamedFunctionSymbol public? final? fun unusedScriptFunction(p: String): <implicit>
+                FirNamedFunctionSymbol public? final? fun unusedScriptFunction(p: String): <implicit>
+    Element 8
+        Scope: FirLocalScope
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 9
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT:
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public? final? fun scriptFunction(): <implicit>
+                public? final? fun unusedScriptFunction(p: String): <implicit>
+                public? final? fun unusedScriptFunction(p: String): <implicit>
+                public? final? fun foo(i: Int, action: ( (Int) -> Unit )): R|kotlin/Unit|
+                public? final? fun foo(i: Int, action: ( (Int) -> Unit )): R|kotlin/Unit|
+                Type: kotlin/script/templates/standard/ScriptTemplateWithArgs
+                Label: <script>
diff --git a/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.copy.txt b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.copy.txt
new file mode 100644
index 00000000000..03ce3fb1838
--- /dev/null
+++ b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.copy.txt
@@ -0,0 +1,18 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtScript
+
+fun scriptFunction() = 42
+scriptFunction()
+
+fun unusedScriptFunction(p: String) = 22
+
+scriptFunction()
+
+fun foo(i: Int, action: (Int) -> Unit) {
+    action(i)
+}
+
+foo(scriptFunction()) {
+    scriptFunction()
+}
+
+unusedScriptFunction("str")
diff --git a/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.fir.txt b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.fir.txt
new file mode 100644
index 00000000000..7cb15ac0548
--- /dev/null
+++ b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.fir.txt
@@ -0,0 +1,22 @@
+KT element: KtScript
+FIR element: FirScriptImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+SCRIPT: [ResolvedTo(BODY_RESOLVE)]
+    [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+    public? final? [ResolvedTo(RAW_FIR)] fun scriptFunction(): <implicit> { LAZY_BLOCK }
+
+    public? final? [ResolvedTo(RAW_FIR)] fun unusedScriptFunction([ResolvedTo(RAW_FIR)] p: String): <implicit> { LAZY_BLOCK }
+
+    public? final? [ResolvedTo(RAW_FIR)] fun unusedScriptFunction([ResolvedTo(RAW_FIR)] p: String): <implicit> {
+        ^unusedScriptFunction IntegerLiteral(22)
+    }
+
+    public? final? [ResolvedTo(RAW_FIR)] fun foo([ResolvedTo(RAW_FIR)] i: Int, [ResolvedTo(RAW_FIR)] action: ( (Int) -> Unit )): R|kotlin/Unit| { LAZY_BLOCK }
+
+    public? final? [ResolvedTo(RAW_FIR)] fun foo([ResolvedTo(RAW_FIR)] i: Int, [ResolvedTo(RAW_FIR)] action: ( (Int) -> Unit )): R|kotlin/Unit| {
+        action#(i#)
+    }
diff --git a/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts
new file mode 100644
index 00000000000..b6803581195
--- /dev/null
+++ b/analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts
@@ -0,0 +1,14 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtScript
+<caret>
+fun scriptFunction() = 42
+fun unusedScriptFunction(p: String) = 22
+
+scriptFunction()
+
+fun foo(i: Int, action: (Int) -> Unit) {
+    action(i)
+}
+
+foo(scriptFunction()) {
+    scriptFunction()
+}
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyContextTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyContextTestGenerated.java
index 7a07f02c790..878a8dd0171 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyContextTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyContextTestGenerated.java
@@ -36,6 +36,12 @@ public class ScriptDependentCopyContextTestGenerated extends AbstractScriptDepen
         runTest("analysis/low-level-api-fir/testdata/dependentCopy/classScript.kts");
     }
 
+    @Test
+    @TestMetadata("moreStatementsInCopy.kts")
+    public void testMoreStatementsInCopy() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts");
+    }
+
     @Test
     @TestMetadata("script.kts")
     public void testScript() throws Exception {
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyFirTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyFirTestGenerated.java
index d09ee4e329e..a4d7739ff55 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyFirTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptDependentCopyFirTestGenerated.java
@@ -36,6 +36,12 @@ public class ScriptDependentCopyFirTestGenerated extends AbstractScriptDependent
         runTest("analysis/low-level-api-fir/testdata/dependentCopy/classScript.kts");
     }
 
+    @Test
+    @TestMetadata("moreStatementsInCopy.kts")
+    public void testMoreStatementsInCopy() throws Exception {
+        runTest("analysis/low-level-api-fir/testdata/dependentCopy/moreStatementsInCopy.kts");
+    }
+
     @Test
     @TestMetadata("script.kts")
     public void testScript() throws Exception {