From 08727bf0be9a9e21eb6e21f53fcce0604b9a5210 Mon Sep 17 00:00:00 2001
From: Yan Zhulanow <yan.zhulanow@jetbrains.com>
Date: Thu, 11 Jan 2024 17:34:34 +0900
Subject: [PATCH] [Analysis API] Add more ContextCollector tests inspired by
 dependentCopy

As the on-air analysis is obsolete, related tests are going to be
deleted. Here we add cases not yet covered by 'ContextCollector' tests.
---
 .../superTypeDelegatedToCompanion.copy.txt    | 42 +++++++++
 .../superTypeDelegatedToCompanion.kt          |  7 ++
 .../superTypeDelegatedToCompanion.txt         | 42 +++++++++
 ...elegatedToPrimaryConstructorParam.copy.txt | 29 ++++++
 ...rTypeDelegatedToPrimaryConstructorParam.kt |  5 +
 ...TypeDelegatedToPrimaryConstructorParam.txt | 29 ++++++
 .../scripts/insideClassBody.copy.txt          | 56 +++++++++++
 .../scripts/insideClassBody.kts               |  5 +
 .../scripts/insideClassBody.txt               | 56 +++++++++++
 .../scripts/lastStatement.copy.txt            | 86 +++++++++++++++++
 .../scripts/lastStatement.kts                 | 16 ++++
 .../scripts/lastStatement.txt                 | 86 +++++++++++++++++
 .../scriptInsideLastStatement.copy.txt        | 92 +++++++++++++++++++
 .../scripts/scriptInsideLastStatement.kts     | 16 ++++
 .../scripts/scriptInsideLastStatement.txt     | 92 +++++++++++++++++++
 .../scripts/statement.copy.txt                | 89 ++++++++++++++++++
 .../contextCollector/scripts/statement.kts    | 16 ++++
 .../contextCollector/scripts/statement.txt    | 89 ++++++++++++++++++
 ...daryConstructor_superCallArgument.copy.txt | 31 +++++++
 .../secondaryConstructor_superCallArgument.kt |  9 ++
 ...secondaryConstructor_superCallArgument.txt | 31 +++++++
 .../ContextCollectorScriptTestGenerated.java  | 24 +++++
 .../ContextCollectorSourceTestGenerated.java  | 18 ++++
 23 files changed, 966 insertions(+)
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.kt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.kt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.kts
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.kts
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.kts
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/statement.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/statement.kts
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/scripts/statement.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.copy.txt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.kt
 create mode 100644 analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.txt

diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.copy.txt
new file mode 100644
index 00000000000..07e0853f0a3
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.copy.txt
@@ -0,0 +1,42 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Implicit receiver:
+            FirRegularClassSymbol public final companion object Companion : R|test/MyInterface|
+                Type: test.MyClass.Companion
+    Element 7
+        Scope: FirNestedClassifierScopeImpl
+            Classifiers:
+                FirRegularClassSymbol public final companion object Companion : R|test/MyInterface|
+    Element 8
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] superTypeDelegatedToCompanion.kt
+    public final? [ResolvedTo(RAW_FIR)] interface MyInterface : R|kotlin/Any| {
+    }
+    public final [ResolvedTo(STATUS)] class MyClass : R|test/MyInterface| {
+        public [ResolvedTo(BODY_RESOLVE)] constructor(): R|test/MyClass| {
+            super<R|kotlin/Any|>()
+        }
+
+        private final [ResolvedTo(BODY_RESOLVE)] field $$delegate_0: R|test/MyInterface| = Q|test/MyClass.Companion|
+
+        public final companion [ResolvedTo(STATUS)] object Companion : R|test/MyInterface| {
+            private [ResolvedTo(RAW_FIR)] constructor(): R|test/MyClass.Companion| {
+                LAZY_super<<implicit>>
+            }
+
+        }
+
+    }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.kt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.kt
new file mode 100644
index 00000000000..b52e70145f8
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.kt
@@ -0,0 +1,7 @@
+package test
+
+interface MyInterface
+
+class MyClass: MyInterface by <expr>Companion</expr> {
+    companion object : MyInterface
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.txt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.txt
new file mode 100644
index 00000000000..bf16beaabf6
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.txt
@@ -0,0 +1,42 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Implicit receiver:
+            FirRegularClassSymbol public final companion object Companion : R|test/MyInterface|
+                Type: test.MyClass.Companion
+    Element 7
+        Scope: FirNestedClassifierScopeImpl
+            Classifiers:
+                FirRegularClassSymbol public final companion object Companion : R|test/MyInterface|
+    Element 8
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] superTypeDelegatedToCompanion.kt
+    public abstract [ResolvedTo(STATUS)] interface MyInterface : R|kotlin/Any| {
+    }
+    public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class MyClass : R|test/MyInterface| {
+        public [ResolvedTo(BODY_RESOLVE)] constructor(): R|test/MyClass| {
+            super<R|kotlin/Any|>()
+        }
+
+        private final [ResolvedTo(BODY_RESOLVE)] field $$delegate_0: R|test/MyInterface| = Q|test/MyClass.Companion|
+
+        public final companion [ResolvedTo(STATUS)] object Companion : R|test/MyInterface| {
+            private [ResolvedTo(STATUS)] constructor(): R|test/MyClass.Companion| {
+                LAZY_super<<implicit>>
+            }
+
+        }
+
+    }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.copy.txt
new file mode 100644
index 00000000000..0796a2cd968
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.copy.txt
@@ -0,0 +1,29 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirLocalScope
+            Properties:
+                FirValueParameterSymbol param: R|test/MyInterface|
+
+FILE: [ResolvedTo(IMPORTS)] superTypeDelegatedToPrimaryConstructorParam.kt
+    public final? [ResolvedTo(RAW_FIR)] interface MyInterface : R|kotlin/Any| {
+    }
+    public final [ResolvedTo(STATUS)] class MyClass : R|test/MyInterface| {
+        public [ResolvedTo(BODY_RESOLVE)] constructor([ResolvedTo(BODY_RESOLVE)] param: R|test/MyInterface|): R|test/MyClass| {
+            super<R|kotlin/Any|>()
+        }
+
+        private final [ResolvedTo(BODY_RESOLVE)] field $$delegate_0: R|test/MyInterface| = R|<local>/param|
+
+    }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.kt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.kt
new file mode 100644
index 00000000000..8f52823e710
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.kt
@@ -0,0 +1,5 @@
+package test
+
+interface MyInterface
+
+class MyClass(param: MyInterface): MyInterface by <expr>param</expr>
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.txt b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.txt
new file mode 100644
index 00000000000..7bc4fc8099e
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.txt
@@ -0,0 +1,29 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirLocalScope
+            Properties:
+                FirValueParameterSymbol param: R|test/MyInterface|
+
+FILE: [ResolvedTo(IMPORTS)] superTypeDelegatedToPrimaryConstructorParam.kt
+    public abstract [ResolvedTo(STATUS)] interface MyInterface : R|kotlin/Any| {
+    }
+    public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class MyClass : R|test/MyInterface| {
+        public [ResolvedTo(BODY_RESOLVE)] constructor([ResolvedTo(BODY_RESOLVE)] param: R|test/MyInterface|): R|test/MyClass| {
+            super<R|kotlin/Any|>()
+        }
+
+        private final [ResolvedTo(BODY_RESOLVE)] field $$delegate_0: R|test/MyInterface| = R|<local>/param|
+
+    }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.copy.txt
new file mode 100644
index 00000000000..3c7a01b6b70
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.copy.txt
@@ -0,0 +1,56 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final? class Foo : R|kotlin/Any|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class Foo : R|kotlin/Any|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-insideClassBody.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final? class Foo : R|kotlin/Any|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Implicit receiver:
+            FirRegularClassSymbol public final class Foo : R|kotlin/Any|
+                Type: Foo
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] insideClassBody.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(TYPES)] <script-insideClassBody.kts>
+        [ResolvedTo(RAW_FIR)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final? [ResolvedTo(RAW_FIR)] class Foo : R|kotlin/Any| {
+            public? [ResolvedTo(RAW_FIR)] constructor(): R|Foo| {
+                LAZY_super<R|kotlin/Any|>
+            }
+
+            public? final? [ResolvedTo(RAW_FIR)] val x: <implicit> = LAZY_EXPRESSION
+                public? [ResolvedTo(RAW_FIR)] get(): <implicit>
+
+            public final [ResolvedTo(BODY_RESOLVE)] fun foo(): R|kotlin/Unit| {
+            }
+
+        }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.kts b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.kts
new file mode 100644
index 00000000000..598240f5c60
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.kts
@@ -0,0 +1,5 @@
+class Foo {
+    val x = 0
+
+    <expr>fun foo() {}</expr>
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.txt
new file mode 100644
index 00000000000..1869345c4db
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.txt
@@ -0,0 +1,56 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class Foo : R|kotlin/Any|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class Foo : R|kotlin/Any|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-insideClassBody.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class Foo : R|kotlin/Any|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Implicit receiver:
+            FirRegularClassSymbol public final class Foo : R|kotlin/Any|
+                Type: Foo
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] insideClassBody.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(TYPES)] <script-insideClassBody.kts>
+        [ResolvedTo(RAW_FIR)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(STATUS)] class Foo : R|kotlin/Any| {
+            public [ResolvedTo(STATUS)] constructor(): R|Foo| {
+                LAZY_super<R|kotlin/Any|>
+            }
+
+            public final [ResolvedTo(STATUS)] val x: <implicit> = LAZY_EXPRESSION
+                public [ResolvedTo(STATUS)] get(): <implicit>
+
+            public final [ResolvedTo(BODY_RESOLVE)] fun foo(): R|kotlin/Unit| {
+            }
+
+        }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.copy.txt
new file mode 100644
index 00000000000..3f11f1abef9
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.copy.txt
@@ -0,0 +1,86 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-lastStatement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+
+FILE: [ResolvedTo(IMPORTS)] lastStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-lastStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(STATUS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.kts b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.kts
new file mode 100644
index 00000000000..2e0add991ae
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.kts
@@ -0,0 +1,16 @@
+class ScriptClass {
+    val memberProperty = 4
+}
+
+fun scriptFunction() = 42
+fun unusedScriptFunction(p: String) = 22
+
+scriptFunction()
+
+fun foo(i: Int, action: (Int) -> Unit) {
+    action(i)
+}
+
+<expr>foo(scriptFunction()) {
+    scriptFunction()
+}</expr>
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.txt
new file mode 100644
index 00000000000..ffd0e8059bd
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.txt
@@ -0,0 +1,86 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-lastStatement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+
+FILE: [ResolvedTo(IMPORTS)] lastStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-lastStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.copy.txt
new file mode 100644
index 00000000000..410919b6b09
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.copy.txt
@@ -0,0 +1,92 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-scriptInsideLastStatement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Scope: FirLocalScope
+            Properties:
+                FirValueParameterSymbol it: R|kotlin/Int|
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] scriptInsideLastStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-scriptInsideLastStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(STATUS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.kts b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.kts
new file mode 100644
index 00000000000..f2fdc484de2
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.kts
@@ -0,0 +1,16 @@
+class ScriptClass {
+    val memberProperty = 4
+}
+
+fun scriptFunction() = 42
+fun unusedScriptFunction(p: String) = 22
+
+scriptFunction()
+
+fun foo(i: Int, action: (Int) -> Unit) {
+    action(i)
+}
+
+foo(scriptFunction()) {
+    <expr>scriptFunction()</expr>
+}
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.txt
new file mode 100644
index 00000000000..e888786f132
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.txt
@@ -0,0 +1,92 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-scriptInsideLastStatement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Scope: FirLocalScope
+            Properties:
+                FirValueParameterSymbol it: R|kotlin/Int|
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] scriptInsideLastStatement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-scriptInsideLastStatement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.copy.txt
new file mode 100644
index 00000000000..5b88a295615
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.copy.txt
@@ -0,0 +1,89 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-statement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Scope: FirLocalScope
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] statement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-statement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(STATUS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.kts b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.kts
new file mode 100644
index 00000000000..55e07e63858
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.kts
@@ -0,0 +1,16 @@
+class ScriptClass {
+    val memberProperty = 4
+}
+
+fun scriptFunction() = 42
+fun unusedScriptFunction(p: String) = 22
+
+<expr>scriptFunction()</expr>
+
+fun foo(i: Int, action: (Int) -> Unit) {
+    action(i)
+}
+
+foo(scriptFunction()) {
+    scriptFunction()
+}
diff --git a/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.txt b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.txt
new file mode 100644
index 00000000000..da3e1cc2257
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/scripts/statement.txt
@@ -0,0 +1,89 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirScriptDeclarationsScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+    Element 7
+        Scope: FirLocalScope
+            Classifiers:
+                FirRegularClassSymbol public final class ScriptClass : R|kotlin/Any|
+            Functions
+                FirNamedFunctionSymbol public final fun scriptFunction(): R|kotlin/Int|
+                FirNamedFunctionSymbol public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+            Properties:
+                FirPropertySymbol lval args: R|kotlin/Array<kotlin/String>|
+    Element 8
+        Context receivers:
+            FirScriptSymbol context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+            SCRIPT: <script-statement.kts>
+                lval args: R|kotlin/Array<kotlin/String>|
+
+                public final class ScriptClass : R|kotlin/Any|
+                public final fun scriptFunction(): R|kotlin/Int|
+                public final fun unusedScriptFunction(p: R|kotlin/String|): R|kotlin/Int|
+                init
+                public final fun foo(i: R|kotlin/Int|, action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit|
+                public final val $$result: R|kotlin/Unit|
+                    public get(): R|kotlin/Unit|
+                Type: kotlin.script.templates.standard.ScriptTemplateWithArgs
+                Label: <script>
+    Element 9
+        Scope: FirLocalScope
+    Element 10
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] statement.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: [ResolvedTo(BODY_RESOLVE)] <script-statement.kts>
+        [ResolvedTo(BODY_RESOLVE)] lval args: R|kotlin/Array<kotlin/String>|
+
+        public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class ScriptClass : R|kotlin/Any| {
+            public [ResolvedTo(BODY_RESOLVE)] constructor(): R|ScriptClass| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final [ResolvedTo(BODY_RESOLVE)] val memberProperty: R|kotlin/Int| = Int(4)
+                public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Int|
+
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun scriptFunction(): R|kotlin/Int| {
+            ^scriptFunction Int(42)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun unusedScriptFunction([ResolvedTo(BODY_RESOLVE)] p: R|kotlin/String|): R|kotlin/Int| {
+            ^unusedScriptFunction Int(22)
+        }
+
+        [ResolvedTo(BODY_RESOLVE)] init {
+            R|/scriptFunction|()
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] fun foo([ResolvedTo(BODY_RESOLVE)] i: R|kotlin/Int|, [ResolvedTo(BODY_RESOLVE)] action: R|(kotlin/Int) -> kotlin/Unit|): R|kotlin/Unit| {
+            R|<local>/action|.R|SubstitutionOverride<kotlin/Function1.invoke: R|kotlin/Unit|>|(R|<local>/i|)
+        }
+
+        public final [ResolvedTo(BODY_RESOLVE)] val $$result: R|kotlin/Unit| = R|/foo|(R|/scriptFunction|(), <L> = [ResolvedTo(BODY_RESOLVE)] foo@fun <anonymous>([ResolvedTo(BODY_RESOLVE)] it: R|kotlin/Int|): R|kotlin/Unit| <inline=NoInline>  {
+            R|/scriptFunction|()
+        }
+        )
+            public [ResolvedTo(BODY_RESOLVE)] get(): R|kotlin/Unit|
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.copy.txt b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.copy.txt
new file mode 100644
index 00000000000..b8fa83282b8
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.copy.txt
@@ -0,0 +1,31 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] secondaryConstructor_superCallArgument.kt
+    public open [ResolvedTo(RAW_FIR)] class Base : R|kotlin/Any| {
+        public? [ResolvedTo(RAW_FIR)] constructor([ResolvedTo(RAW_FIR)] i: Int): R|test/Base| {
+            LAZY_super<R|kotlin/Any|>
+        }
+
+    }
+    public? final? [ResolvedTo(RAW_FIR)] val property: <implicit> = LAZY_EXPRESSION
+        public? [ResolvedTo(RAW_FIR)] get(): <implicit>
+    public final? [ResolvedTo(RAW_FIR)] class Child : Base {
+        public [ResolvedTo(BODY_RESOLVE)] constructor(): R|test/Child| {
+            super<R|test/Base|>(R|test/property|)
+        }
+
+    }
diff --git a/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.kt b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.kt
new file mode 100644
index 00000000000..b5fa5a92633
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.kt
@@ -0,0 +1,9 @@
+package test
+
+open class Base(i: Int)
+
+val property = 10
+
+class Child : Base {
+    constructor(): super(<expr>property</expr>)
+}
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.txt b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.txt
new file mode 100644
index 00000000000..3b5d7b7bd53
--- /dev/null
+++ b/analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.txt
@@ -0,0 +1,31 @@
+Tower Data Context:
+    Element 0
+        Scope: FirDefaultStarImportingScope
+    Element 1
+        Scope: FirExplicitStarImportingScope
+    Element 2
+        Scope: FirDefaultSimpleImportingScope
+    Element 3
+        Scope: FirDefaultSimpleImportingScope
+    Element 4
+        Scope: FirPackageMemberScope
+    Element 5
+        Scope: FirExplicitSimpleImportingScope
+    Element 6
+        Scope: FirLocalScope
+
+FILE: [ResolvedTo(IMPORTS)] secondaryConstructor_superCallArgument.kt
+    public open [ResolvedTo(STATUS)] class Base : R|kotlin/Any| {
+        public [ResolvedTo(STATUS)] constructor([ResolvedTo(STATUS)] i: R|kotlin/Int|): R|test/Base| {
+            LAZY_super<R|kotlin/Any|>
+        }
+
+    }
+    public final [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] val property: R|kotlin/Int| = Int(10)
+        public [ResolvedTo(IMPLICIT_TYPES_BODY_RESOLVE)] get(): R|kotlin/Int|
+    public final [ResolvedTo(ANNOTATION_ARGUMENTS)] class Child : R|test/Base| {
+        public [ResolvedTo(BODY_RESOLVE)] constructor(): R|test/Child| {
+            super<R|test/Base|>(R|test/property|)
+        }
+
+    }
\ No newline at end of file
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorScriptTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorScriptTestGenerated.java
index e40564d8521..c7e34ed6a13 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorScriptTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorScriptTestGenerated.java
@@ -53,6 +53,12 @@ public class ContextCollectorScriptTestGenerated extends AbstractContextCollecto
             KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/low-level-api-fir/testData/contextCollector/scripts"), Pattern.compile("^(.+)\\.(kts)$"), null, true);
         }
 
+        @Test
+        @TestMetadata("insideClassBody.kts")
+        public void testInsideClassBody() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/scripts/insideClassBody.kts");
+        }
+
         @Test
         @TestMetadata("insideClassMember.kts")
         public void testInsideClassMember() throws Exception {
@@ -65,6 +71,24 @@ public class ContextCollectorScriptTestGenerated extends AbstractContextCollecto
             runTest("analysis/low-level-api-fir/testData/contextCollector/scripts/insideTopLevelFunction.kts");
         }
 
+        @Test
+        @TestMetadata("lastStatement.kts")
+        public void testLastStatement() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/scripts/lastStatement.kts");
+        }
+
+        @Test
+        @TestMetadata("scriptInsideLastStatement.kts")
+        public void testScriptInsideLastStatement() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/scripts/scriptInsideLastStatement.kts");
+        }
+
+        @Test
+        @TestMetadata("statement.kts")
+        public void testStatement() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/scripts/statement.kts");
+        }
+
         @Test
         @TestMetadata("topLevel.kts")
         public void testTopLevel() throws Exception {
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorSourceTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorSourceTestGenerated.java
index d6fe025fd94..8f936db0066 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorSourceTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/ContextCollectorSourceTestGenerated.java
@@ -180,6 +180,12 @@ public class ContextCollectorSourceTestGenerated extends AbstractContextCollecto
         runTest("analysis/low-level-api-fir/testData/contextCollector/propertyDelegateInitializer.kt");
     }
 
+    @Test
+    @TestMetadata("secondaryConstructor_superCallArgument.kt")
+    public void testSecondaryConstructor_superCallArgument() throws Exception {
+        runTest("analysis/low-level-api-fir/testData/contextCollector/secondaryConstructor_superCallArgument.kt");
+    }
+
     @Test
     @TestMetadata("simple.kt")
     public void testSimple() throws Exception {
@@ -297,6 +303,18 @@ public class ContextCollectorSourceTestGenerated extends AbstractContextCollecto
             runTest("analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedExpression.kt");
         }
 
+        @Test
+        @TestMetadata("superTypeDelegatedToCompanion.kt")
+        public void testSuperTypeDelegatedToCompanion() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToCompanion.kt");
+        }
+
+        @Test
+        @TestMetadata("superTypeDelegatedToPrimaryConstructorParam.kt")
+        public void testSuperTypeDelegatedToPrimaryConstructorParam() throws Exception {
+            runTest("analysis/low-level-api-fir/testData/contextCollector/classHeaderPositions/superTypeDelegatedToPrimaryConstructorParam.kt");
+        }
+
         @Test
         @TestMetadata("superTypeDelegatedTypeRef.kt")
         public void testSuperTypeDelegatedTypeRef() throws Exception {