From f972c8d239331bcabb02ee7e689e1cc9760a414b Mon Sep 17 00:00:00 2001
From: Dmitrii Gridin <dmitry.gridin@jetbrains.com>
Date: Fri, 2 Feb 2024 20:04:18 +0100
Subject: [PATCH] [LL FIR] add more tests for script initializer

^KT-65344
---
 .../syntheticResultDeclarationInitializer.kts |  3 ++
 .../syntheticResultDeclarationInitializer.txt | 28 +++++++++++++++++
 .../expressions/firstStatement.kts            |  9 ++++++
 .../expressions/firstStatement.txt            | 29 ++++++++++++++++++
 .../expressions/firstStatementCall.kts        |  9 ++++++
 .../expressions/firstStatementCall.txt        | 27 +++++++++++++++++
 .../expressions/secondStatement.kts           |  9 ++++++
 .../expressions/secondStatement.txt           | 29 ++++++++++++++++++
 .../expressions/secondStatementCall.kts       |  9 ++++++
 .../expressions/secondStatementCall.txt       | 27 +++++++++++++++++
 .../fir/ScriptGetOrBuildFirTestGenerated.java | 30 +++++++++++++++++++
 11 files changed, 209 insertions(+)
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.kts
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.txt
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.kts
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.txt
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.kts
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.txt
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.kts
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.txt
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.kts
 create mode 100644 analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.txt

diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.kts b/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.kts
new file mode 100644
index 00000000000..2e0b28e847e
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.kts
@@ -0,0 +1,3 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtScriptInitializer
+fun foo() = 24
+<expr>foo()</expr>
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.txt b/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.txt
new file mode 100644
index 00000000000..dfcd4ff9626
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.txt
@@ -0,0 +1,28 @@
+KT element: KtScriptInitializer
+FIR element: FirScriptImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-syntheticResultDeclarationInitializer.kts>
+    [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+    public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+        ^foo Int(24)
+    }
+
+    public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Int| = R|/foo|()
+        public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+FIR FILE:
+FILE: [ResolvedTo(IMPORTS)] syntheticResultDeclarationInitializer.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-syntheticResultDeclarationInitializer.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+            ^foo Int(24)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Int| = R|/foo|()
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.kts b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.kts
new file mode 100644
index 00000000000..5603edcb124
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.kts
@@ -0,0 +1,9 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtScriptInitializer
+fun foo() = 4
+
+<expr>foo()</expr>
+bar()
+
+fun bar() {
+
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.txt b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.txt
new file mode 100644
index 00000000000..f5f595bfc88
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.txt
@@ -0,0 +1,29 @@
+KT element: KtScriptInitializer
+FIR element: FirAnonymousInitializerImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+[ResolvedTo(BODY_RESOLVE)] init {
+    R|/foo|()
+}
+
+FIR FILE:
+FILE: [ResolvedTo(IMPORTS)] firstStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-firstStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+            ^foo Int(4)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/foo|()
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/bar|()
+        }
+
+        public final [ResolvedTo(CONTRACTS)] fun bar(): R|kotlin/Unit| {
+        }
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.kts b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.kts
new file mode 100644
index 00000000000..13a8d74536f
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.kts
@@ -0,0 +1,9 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtCallExpression
+fun foo() = 4
+
+<expr>foo()</expr>
+bar()
+
+fun bar() {
+
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.txt b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.txt
new file mode 100644
index 00000000000..0d362218bb4
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.txt
@@ -0,0 +1,27 @@
+KT element: KtCallExpression
+FIR element: FirFunctionCallImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+R|/foo|()
+
+FIR FILE:
+FILE: [ResolvedTo(IMPORTS)] firstStatementCall.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-firstStatementCall.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+            ^foo Int(4)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/foo|()
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/bar|()
+        }
+
+        public final [ResolvedTo(CONTRACTS)] fun bar(): R|kotlin/Unit| {
+        }
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.kts b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.kts
new file mode 100644
index 00000000000..3c302c608ef
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.kts
@@ -0,0 +1,9 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtScriptInitializer
+fun foo() = 4
+
+foo()
+<expr>bar()</expr>
+
+fun bar() {
+
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.txt b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.txt
new file mode 100644
index 00000000000..ce1301e932c
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.txt
@@ -0,0 +1,29 @@
+KT element: KtScriptInitializer
+FIR element: FirAnonymousInitializerImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+[ResolvedTo(BODY_RESOLVE)] init {
+    R|/bar|()
+}
+
+FIR FILE:
+FILE: [ResolvedTo(IMPORTS)] secondStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-secondStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+            ^foo Int(4)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/foo|()
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/bar|()
+        }
+
+        public final [ResolvedTo(CONTRACTS)] fun bar(): R|kotlin/Unit| {
+        }
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.kts b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.kts
new file mode 100644
index 00000000000..2c1aff13432
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.kts
@@ -0,0 +1,9 @@
+// LOOK_UP_FOR_ELEMENT_OF_TYPE: KtCallExpression
+fun foo() = 4
+
+foo()
+<expr>bar()</expr>
+
+fun bar() {
+
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.txt b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.txt
new file mode 100644
index 00000000000..9a30057face
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.txt
@@ -0,0 +1,27 @@
+KT element: KtCallExpression
+FIR element: FirFunctionCallImpl
+FIR source kind: KtRealSourceElementKind
+
+FIR element rendered:
+R|/bar|()
+
+FIR FILE:
+FILE: [ResolvedTo(IMPORTS)] secondStatementCall.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-secondStatementCall.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] fun foo(): R|kotlin/Int| {
+            ^foo Int(4)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/foo|()
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/bar|()
+        }
+
+        public final [ResolvedTo(CONTRACTS)] fun bar(): R|kotlin/Unit| {
+        }
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptGetOrBuildFirTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptGetOrBuildFirTestGenerated.java
index a11e6389c23..a57b00c584c 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptGetOrBuildFirTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ScriptGetOrBuildFirTestGenerated.java
@@ -340,6 +340,12 @@ public class ScriptGetOrBuildFirTestGenerated extends AbstractScriptGetOrBuildFi
         public void testSyntheticResultDeclaration() throws Exception {
             runTest("analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclaration.kts");
         }
+
+        @Test
+        @TestMetadata("syntheticResultDeclarationInitializer.kts")
+        public void testSyntheticResultDeclarationInitializer() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/getOrBuildFir/calls/syntheticResultDeclarationInitializer.kts");
+        }
     }
 
     @Nested
@@ -495,6 +501,18 @@ public class ScriptGetOrBuildFirTestGenerated extends AbstractScriptGetOrBuildFi
             runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/callInsideStatement.kts");
         }
 
+        @Test
+        @TestMetadata("firstStatement.kts")
+        public void testFirstStatement() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatement.kts");
+        }
+
+        @Test
+        @TestMetadata("firstStatementCall.kts")
+        public void testFirstStatementCall() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/firstStatementCall.kts");
+        }
+
         @Test
         @TestMetadata("forExpressionRangeScript.kts")
         public void testForExpressionRangeScript() throws Exception {
@@ -555,6 +573,18 @@ public class ScriptGetOrBuildFirTestGenerated extends AbstractScriptGetOrBuildFi
             runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/parenthesizedExpressionScript.kts");
         }
 
+        @Test
+        @TestMetadata("secondStatement.kts")
+        public void testSecondStatement() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatement.kts");
+        }
+
+        @Test
+        @TestMetadata("secondStatementCall.kts")
+        public void testSecondStatementCall() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/getOrBuildFir/expressions/secondStatementCall.kts");
+        }
+
         @Test
         @TestMetadata("stringLiteralScript.kts")
         public void testStringLiteralScript() throws Exception {