From cdf4b1705263922ed3c72088f4541074b9abc81d Mon Sep 17 00:00:00 2001
From: Ilya Chernikov <ilya.chernikov@jetbrains.com>
Date: Thu, 21 Dec 2023 17:28:52 +0100
Subject: [PATCH] K2 Scripting tests: adding tests on script decls visibility

fixes (makes it stable) behavior of the script top-level declarations
visibility in K2 scripts
---
 ...CompilerTestFE10TestdataTestGenerated.java |   6 +
 ...cCompilerTestFirTestdataTestGenerated.java |  10 +
 ...sticCompilerFE10TestDataTestGenerated.java |   6 +
 ...osticCompilerFirTestDataTestGenerated.java |  10 +
 ...clsShouldBeVisibleFromOtherSources.fir.txt |  57 +++++
 ...ptDeclsShouldBeVisibleFromOtherSources.kts |  28 +++
 .../scripts/scriptsDeclsMayConflict.fir.txt   |  20 ++
 .../scripts/scriptsDeclsMayConflict.kts       |  12 ++
 .../FirLightTreeDiagnosticsTestGenerated.java |  10 +
 .../FirPsiDiagnosticTestGenerated.java        | 204 ++++++++++--------
 ...siOldFrontendDiagnosticsTestGenerated.java |   6 +
 .../scriptDeclsShouldVisibleLocally.kts       |  25 +++
 .../test/runners/DiagnosticTestGenerated.java |   6 +
 .../generators/GenerateJUnit5CompilerTests.kt |   4 +-
 14 files changed, 311 insertions(+), 93 deletions(-)
 create mode 100644 compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.fir.txt
 create mode 100644 compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.kts
 create mode 100644 compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.fir.txt
 create mode 100644 compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.kts
 create mode 100644 compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts

diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
index 4f8c29bd29d..d0e2faf9f1c 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java
@@ -34176,6 +34176,12 @@ public class DiagnosticCompilerTestFE10TestdataTestGenerated extends AbstractDia
         runTest("compiler/testData/diagnostics/tests/script/resolveInitializerOfDestructuringDeclarationOnce.kts");
       }
 
+      @Test
+      @TestMetadata("scriptDeclsShouldVisibleLocally.kts")
+      public void testScriptDeclsShouldVisibleLocally() {
+        runTest("compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts");
+      }
+
       @Test
       @TestMetadata("scriptScopes.kts")
       public void testScriptScopes() {
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFirTestdataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFirTestdataTestGenerated.java
index 69018cef6ed..8fbc3e0bd8f 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFirTestdataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFirTestdataTestGenerated.java
@@ -4472,6 +4472,16 @@ public class DiagnosticCompilerTestFirTestdataTestGenerated extends AbstractDiag
       }
     }
 
+    @Nested
+    @TestMetadata("compiler/fir/analysis-tests/testData/resolve/scripts")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Scripts {
+      @Test
+      public void testAllFilesPresentInScripts() {
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scripts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+      }
+    }
+
     @Nested
     @TestMetadata("compiler/fir/analysis-tests/testData/resolve/smartcasts")
     @TestDataPath("$PROJECT_ROOT")
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
index aad6d76526e..9fa3d54c637 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java
@@ -34176,6 +34176,12 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated
         runTest("compiler/testData/diagnostics/tests/script/resolveInitializerOfDestructuringDeclarationOnce.kts");
       }
 
+      @Test
+      @TestMetadata("scriptDeclsShouldVisibleLocally.kts")
+      public void testScriptDeclsShouldVisibleLocally() {
+        runTest("compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts");
+      }
+
       @Test
       @TestMetadata("scriptScopes.kts")
       public void testScriptScopes() {
diff --git a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFirTestDataTestGenerated.java b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFirTestDataTestGenerated.java
index 391177e3a87..e85c8d1fb44 100644
--- a/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFirTestDataTestGenerated.java
+++ b/analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFirTestDataTestGenerated.java
@@ -4472,6 +4472,16 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFirTestDataTestGenerated
       }
     }
 
+    @Nested
+    @TestMetadata("compiler/fir/analysis-tests/testData/resolve/scripts")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Scripts {
+      @Test
+      public void testAllFilesPresentInScripts() {
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scripts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+      }
+    }
+
     @Nested
     @TestMetadata("compiler/fir/analysis-tests/testData/resolve/smartcasts")
     @TestDataPath("$PROJECT_ROOT")
diff --git a/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.fir.txt b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.fir.txt
new file mode 100644
index 00000000000..ed15df1a433
--- /dev/null
+++ b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.fir.txt
@@ -0,0 +1,57 @@
+FILE: script.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: <script-script.kts>
+        lval args: R|kotlin/Array<kotlin/String>|
+
+        public final val a: R|kotlin/Int| = Int(42)
+            public get(): R|kotlin/Int|
+
+        public final class A : R|kotlin/Any| {
+            public constructor(): R|A| {
+                super<R|kotlin/Any|>()
+            }
+
+        }
+
+        public final enum class E : R|kotlin/Enum<E>| {
+            private constructor(): R|E| {
+                super<R|kotlin/Enum<E>|>()
+            }
+
+            public final static enum entry V: R|E|
+            public final static fun values(): R|kotlin/Array<E>| {
+            }
+
+            public final static fun valueOf(value: R|kotlin/String|): R|E| {
+            }
+
+            public final static val entries: R|kotlin/enums/EnumEntries<E>|
+                public get(): R|kotlin/enums/EnumEntries<E>|
+
+        }
+
+        public final object O : R|kotlin/Any| {
+            private constructor(): R|O| {
+                super<R|kotlin/Any|>()
+            }
+
+            public final val v: R|kotlin/Int| = Int(42)
+                public get(): R|kotlin/Int|
+
+        }
+
+        public final fun foo(): R|kotlin/Int| {
+            ^foo Int(42)
+        }
+
+FILE: main.kt
+    public final val b: R|kotlin/Int| = R|/a|
+        public get(): R|kotlin/Int|
+    public final val ca: R|A| = R|/A.A|()
+        public get(): R|A|
+    public final val ev: R|E| = Q|E|.R|/E.V|
+        public get(): R|E|
+    public final val ov: R|kotlin/Int| = Q|O|.R|/O.v|
+        public get(): R|kotlin/Int|
+    public final val rfoo: R|kotlin/Int| = R|/foo|()
+        public get(): R|kotlin/Int|
diff --git a/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.kts b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.kts
new file mode 100644
index 00000000000..d1fa9ba02e2
--- /dev/null
+++ b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.kts
@@ -0,0 +1,28 @@
+
+// FILE: script.kts
+
+val a = 42
+
+class A
+
+enum class E {
+    V
+}
+
+object O {
+    val v = 42
+}
+
+fun foo() = 42
+
+// FILE: main.kt
+
+val b =  a
+
+val ca = A()
+
+val ev = E.V
+
+val ov = O.v
+
+val rfoo = foo()
diff --git a/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.fir.txt b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.fir.txt
new file mode 100644
index 00000000000..b5eff4368ec
--- /dev/null
+++ b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.fir.txt
@@ -0,0 +1,20 @@
+FILE: script1.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: <script-script1.kts>
+        lval args: R|kotlin/Array<kotlin/String>|
+
+        public final val a: R|kotlin/Int| = Int(42)
+            public get(): R|kotlin/Int|
+
+FILE: script2.kts
+    context(<script>@R|kotlin/script/templates/standard/ScriptTemplateWithArgs|)
+    SCRIPT: <script-script2.kts>
+        lval args: R|kotlin/Array<kotlin/String>|
+
+        public final val a: R|kotlin/String| = String(42)
+            public get(): R|kotlin/String|
+
+FILE: main.kt
+    public final fun foo(): <ERROR TYPE REF: Ambiguity: a, [/a, /a]> {
+        ^foo <Ambiguity: a, [/a, /a]>#
+    }
diff --git a/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.kts b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.kts
new file mode 100644
index 00000000000..e65c34a9cc8
--- /dev/null
+++ b/compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.kts
@@ -0,0 +1,12 @@
+
+// FILE: script1.kts
+
+val a = 42
+
+// FILE: script2.kts
+
+val a = "42"
+
+// FILE: main.kt
+
+fun foo() =  <!OVERLOAD_RESOLUTION_AMBIGUITY!>a<!>
diff --git a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeDiagnosticsTestGenerated.java b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeDiagnosticsTestGenerated.java
index b044ead70bf..da6b7392e1f 100644
--- a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeDiagnosticsTestGenerated.java
+++ b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeDiagnosticsTestGenerated.java
@@ -4472,6 +4472,16 @@ public class FirLightTreeDiagnosticsTestGenerated extends AbstractFirLightTreeDi
       }
     }
 
+    @Nested
+    @TestMetadata("compiler/fir/analysis-tests/testData/resolve/scripts")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Scripts {
+      @Test
+      public void testAllFilesPresentInScripts() {
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scripts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+      }
+    }
+
     @Nested
     @TestMetadata("compiler/fir/analysis-tests/testData/resolve/smartcasts")
     @TestDataPath("$PROJECT_ROOT")
diff --git a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiDiagnosticTestGenerated.java b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiDiagnosticTestGenerated.java
index fb78cd87b02..116f72b60bf 100644
--- a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiDiagnosticTestGenerated.java
+++ b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiDiagnosticTestGenerated.java
@@ -53,7 +53,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
     @Test
     public void testAllFilesPresentInResolve() {
-      KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+      KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
     }
 
     @Test
@@ -776,7 +776,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Annotations {
       @Test
       public void testAllFilesPresentInAnnotations() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/annotations"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/annotations"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -810,7 +810,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Arguments {
       @Test
       public void testAllFilesPresentInArguments() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/arguments"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/arguments"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1048,7 +1048,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Arrays {
       @Test
       public void testAllFilesPresentInArrays() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/arrays"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/arrays"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1070,7 +1070,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Builtins {
       @Test
       public void testAllFilesPresentInBuiltins() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/builtins"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/builtins"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1086,7 +1086,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class CallResolution {
       @Test
       public void testAllFilesPresentInCallResolution() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/callResolution"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/callResolution"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1246,7 +1246,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Cfa {
       @Test
       public void testAllFilesPresentInCfa() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/cfa"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/cfa"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1286,7 +1286,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Cfg {
       @Test
       public void testAllFilesPresentInCfg() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/cfg"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/cfg"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1500,7 +1500,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Checkers {
       @Test
       public void testAllFilesPresentInCheckers() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/checkers"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/checkers"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1588,7 +1588,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class ConstVal {
       @Test
       public void testAllFilesPresentInConstVal() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/constVal"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/constVal"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1622,7 +1622,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Constructors {
       @Test
       public void testAllFilesPresentInConstructors() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/constructors"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/constructors"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1644,7 +1644,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Contracts {
       @Test
       public void testAllFilesPresentInContracts() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/contracts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/contracts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1690,7 +1690,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Delegates {
       @Test
       public void testAllFilesPresentInDelegates() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/delegates"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/delegates"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -1766,7 +1766,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
       @Test
       public void testAllFilesPresentInDiagnostics() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/diagnostics"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/diagnostics"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -2087,7 +2087,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class FunctionAsExpression {
         @Test
         public void testAllFilesPresentInFunctionAsExpression() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/diagnostics/functionAsExpression"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/diagnostics/functionAsExpression"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2104,7 +2104,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Exhaustiveness {
       @Test
       public void testAllFilesPresentInExhaustiveness() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Nested
@@ -2113,7 +2113,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Negative {
         @Test
         public void testAllFilesPresentInNegative() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness/negative"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness/negative"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2153,7 +2153,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Positive {
         @Test
         public void testAllFilesPresentInPositive() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness/positive"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/exhaustiveness/positive"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2254,7 +2254,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
       @Test
       public void testAllFilesPresentInExpresssions() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -2635,7 +2635,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Inference {
         @Test
         public void testAllFilesPresentInInference() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/inference"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/inference"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2663,7 +2663,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Invoke {
         @Test
         public void testAllFilesPresentInInvoke() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/invoke"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/invoke"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2763,7 +2763,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Operators {
         @Test
         public void testAllFilesPresentInOperators() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/operators"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/expresssions/operators"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2792,7 +2792,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class ExtendedCheckers {
       @Test
       public void testAllFilesPresentInExtendedCheckers() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -2867,7 +2867,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class CanBeReplacedWithOperatorAssignment {
         @Test
         public void testAllFilesPresentInCanBeReplacedWithOperatorAssignment() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/canBeReplacedWithOperatorAssignment"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/canBeReplacedWithOperatorAssignment"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -2991,7 +2991,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class EmptyRangeChecker {
         @Test
         public void testAllFilesPresentInEmptyRangeChecker() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/emptyRangeChecker"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/emptyRangeChecker"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -3013,7 +3013,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class RedundantCallOfConversionMethod {
         @Test
         public void testAllFilesPresentInRedundantCallOfConversionMethod() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/RedundantCallOfConversionMethod"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/RedundantCallOfConversionMethod"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -3143,7 +3143,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Unused {
         @Test
         public void testAllFilesPresentInUnused() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/unused"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/unused"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -3255,7 +3255,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class UselessCallOnNotNullChecker {
         @Test
         public void testAllFilesPresentInUselessCallOnNotNullChecker() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/UselessCallOnNotNullChecker"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/extendedCheckers/UselessCallOnNotNullChecker"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -3332,7 +3332,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class FromBuilder {
       @Test
       public void testAllFilesPresentInFromBuilder() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/fromBuilder"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/fromBuilder"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3372,7 +3372,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Inference {
       @Test
       public void testAllFilesPresentInInference() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inference"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inference"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3585,7 +3585,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class BuilderInference {
         @Test
         public void testAllFilesPresentInBuilderInference() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inference/builderInference"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inference/builderInference"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -3608,7 +3608,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class InlineClasses {
       @Test
       public void testAllFilesPresentInInlineClasses() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inlineClasses"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/inlineClasses"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3636,7 +3636,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class InnerClasses {
       @Test
       public void testAllFilesPresentInInnerClasses() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/innerClasses"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/innerClasses"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3670,7 +3670,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class LocalClasses {
       @Test
       public void testAllFilesPresentInLocalClasses() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/localClasses"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/localClasses"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3722,7 +3722,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Multifile {
       @Test
       public void testAllFilesPresentInMultifile() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/multifile"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/multifile"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3798,7 +3798,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Overrides {
       @Test
       public void testAllFilesPresentInOverrides() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/overrides"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/overrides"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -3862,7 +3862,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Problems {
       @Test
       public void testAllFilesPresentInProblems() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/problems"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/problems"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4064,7 +4064,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Properties {
       @Test
       public void testAllFilesPresentInProperties() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/properties"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/properties"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4128,7 +4128,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class PropertyVsField {
       @Test
       public void testAllFilesPresentInPropertyVsField() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/propertyVsField"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/propertyVsField"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4162,7 +4162,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class References {
       @Test
       public void testAllFilesPresentInReferences() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/references"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/references"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4244,7 +4244,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class SamConstructors {
       @Test
       public void testAllFilesPresentInSamConstructors() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/samConstructors"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/samConstructors"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4296,7 +4296,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class SamConversions {
       @Test
       public void testAllFilesPresentInSamConversions() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/samConversions"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/samConversions"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4372,7 +4372,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Scopes {
       @Test
       public void testAllFilesPresentInScopes() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scopes"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scopes"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4472,6 +4472,28 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       }
     }
 
+    @Nested
+    @TestMetadata("compiler/fir/analysis-tests/testData/resolve/scripts")
+    @TestDataPath("$PROJECT_ROOT")
+    public class Scripts {
+      @Test
+      public void testAllFilesPresentInScripts() {
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/scripts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
+      }
+
+      @Test
+      @TestMetadata("scriptDeclsShouldBeVisibleFromOtherSources.kts")
+      public void testScriptDeclsShouldBeVisibleFromOtherSources() {
+        runTest("compiler/fir/analysis-tests/testData/resolve/scripts/scriptDeclsShouldBeVisibleFromOtherSources.kts");
+      }
+
+      @Test
+      @TestMetadata("scriptsDeclsMayConflict.kts")
+      public void testScriptsDeclsMayConflict() {
+        runTest("compiler/fir/analysis-tests/testData/resolve/scripts/scriptsDeclsMayConflict.kts");
+      }
+    }
+
     @Nested
     @TestMetadata("compiler/fir/analysis-tests/testData/resolve/smartcasts")
     @TestDataPath("$PROJECT_ROOT")
@@ -4484,7 +4506,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
       @Test
       public void testAllFilesPresentInSmartcasts() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -4661,7 +4683,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Booleans {
         @Test
         public void testAllFilesPresentInBooleans() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/booleans"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/booleans"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4695,7 +4717,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class BoundSmartcasts {
         @Test
         public void testAllFilesPresentInBoundSmartcasts() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/boundSmartcasts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/boundSmartcasts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4741,7 +4763,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class ControlStructures {
         @Test
         public void testAllFilesPresentInControlStructures() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/controlStructures"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/controlStructures"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4799,7 +4821,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Lambdas {
         @Test
         public void testAllFilesPresentInLambdas() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/lambdas"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/lambdas"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4827,7 +4849,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Loops {
         @Test
         public void testAllFilesPresentInLoops() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/loops"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/loops"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4855,7 +4877,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Problems {
         @Test
         public void testAllFilesPresentInProblems() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/problems"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/problems"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4883,7 +4905,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Receivers {
         @Test
         public void testAllFilesPresentInReceivers() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/receivers"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/receivers"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4917,7 +4939,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class SafeCalls {
         @Test
         public void testAllFilesPresentInSafeCalls() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/safeCalls"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/safeCalls"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4963,7 +4985,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Stability {
         @Test
         public void testAllFilesPresentInStability() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/stability"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/stability"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -4979,7 +5001,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Variables {
         @Test
         public void testAllFilesPresentInVariables() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/variables"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/smartcasts/variables"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -5008,7 +5030,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Stdlib {
       @Test
       public void testAllFilesPresentInStdlib() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/stdlib"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/stdlib"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Nested
@@ -5017,7 +5039,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class J_k {
         @Test
         public void testAllFilesPresentInJ_k() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/stdlib/j+k"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/stdlib/j+k"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -5040,7 +5062,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Suppress {
       @Test
       public void testAllFilesPresentInSuppress() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/suppress"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/suppress"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5086,7 +5108,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class TypeParameters {
       @Test
       public void testAllFilesPresentInTypeParameters() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/typeParameters"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/typeParameters"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5108,7 +5130,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Types {
       @Test
       public void testAllFilesPresentInTypes() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/types"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/types"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5154,7 +5176,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class UnqualifiedEnum {
       @Test
       public void testAllFilesPresentInUnqualifiedEnum() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/unqualifiedEnum"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/unqualifiedEnum"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5212,7 +5234,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Visibility {
       @Test
       public void testAllFilesPresentInVisibility() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/visibility"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/visibility"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5306,7 +5328,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class WithAllowedKotlinPackage {
       @Test
       public void testAllFilesPresentInWithAllowedKotlinPackage() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/withAllowedKotlinPackage"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolve/withAllowedKotlinPackage"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5329,7 +5351,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
     @Test
     public void testAllFilesPresentInResolveWithStdlib() {
-      KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+      KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
     }
 
     @Test
@@ -5716,7 +5738,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class CallableReferences {
       @Test
       public void testAllFilesPresentInCallableReferences() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/callableReferences"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/callableReferences"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -5869,7 +5891,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class FromBasicDiagnosticTests {
         @Test
         public void testAllFilesPresentInFromBasicDiagnosticTests() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/callableReferences/fromBasicDiagnosticTests"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/callableReferences/fromBasicDiagnosticTests"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -5958,7 +5980,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Contracts {
       @Test
       public void testAllFilesPresentInContracts() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Nested
@@ -5967,7 +5989,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class FromLibrary {
         @Test
         public void testAllFilesPresentInFromLibrary() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromLibrary"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromLibrary"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -5995,7 +6017,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class FromSource {
         @Test
         public void testAllFilesPresentInFromSource() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Nested
@@ -6004,7 +6026,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
         public class Bad {
           @Test
           public void testAllFilesPresentInBad() {
-            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
           }
 
           @Test
@@ -6019,7 +6041,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
           public class CallsInPlace {
             @Test
             public void testAllFilesPresentInCallsInPlace() {
-              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad/callsInPlace"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad/callsInPlace"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
             }
 
             @Test
@@ -6053,7 +6075,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
           public class ReturnsImplies {
             @Test
             public void testAllFilesPresentInReturnsImplies() {
-              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad/returnsImplies"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/bad/returnsImplies"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
             }
 
             @Test
@@ -6082,7 +6104,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
         public class Good {
           @Test
           public void testAllFilesPresentInGood() {
-            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+            KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
           }
 
           @Nested
@@ -6091,7 +6113,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
           public class CallsInPlace {
             @Test
             public void testAllFilesPresentInCallsInPlace() {
-              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/callsInPlace"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/callsInPlace"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
             }
 
             @Test
@@ -6167,7 +6189,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
           public class ReturnsImplies {
             @Test
             public void testAllFilesPresentInReturnsImplies() {
-              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/returnsImplies"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/returnsImplies"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
             }
 
             @Test
@@ -6243,7 +6265,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
           public class VariousContracts {
             @Test
             public void testAllFilesPresentInVariousContracts() {
-              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/variousContracts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+              KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/variousContracts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
             }
 
             @Nested
@@ -6252,7 +6274,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
             public class NewSyntax {
               @Test
               public void testAllFilesPresentInNewSyntax() {
-                KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/variousContracts/newSyntax"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+                KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/contracts/fromSource/good/variousContracts/newSyntax"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
               }
 
               @Test
@@ -6272,7 +6294,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Delegates {
       @Test
       public void testAllFilesPresentInDelegates() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/delegates"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/delegates"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6354,7 +6376,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Diagnostics {
       @Test
       public void testAllFilesPresentInDiagnostics() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6369,7 +6391,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class FunctionReturnTypeMismatchChecker {
         @Test
         public void testAllFilesPresentInFunctionReturnTypeMismatchChecker() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics/functionReturnTypeMismatchChecker"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics/functionReturnTypeMismatchChecker"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -6385,7 +6407,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class InitializerTypeMismatchChecker {
         @Test
         public void testAllFilesPresentInInitializerTypeMismatchChecker() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics/initializerTypeMismatchChecker"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/diagnostics/initializerTypeMismatchChecker"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -6402,7 +6424,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Inference {
       @Test
       public void testAllFilesPresentInInference() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/inference"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/inference"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6453,7 +6475,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class Problems {
         @Test
         public void testAllFilesPresentInProblems() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/inference/problems"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/inference/problems"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -6488,7 +6510,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Initialization {
       @Test
       public void testAllFilesPresentInInitialization() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/initialization"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/initialization"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6510,7 +6532,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
       @Test
       public void testAllFilesPresentInIntellij() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/intellij"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/intellij"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6610,7 +6632,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class J_k {
       @Test
       public void testAllFilesPresentInJ_k() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/j+k"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/j+k"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6932,7 +6954,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class MultiModule {
       @Test
       public void testAllFilesPresentInMultiModule() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/multiModule"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/multiModule"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -6978,7 +7000,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
 
       @Test
       public void testAllFilesPresentInProblems() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/problems"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/problems"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -7186,7 +7208,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Properties {
       @Test
       public void testAllFilesPresentInProperties() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/properties"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/properties"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -7201,7 +7223,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
       public class BackingField {
         @Test
         public void testAllFilesPresentInBackingField() {
-          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/properties/backingField"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+          KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/properties/backingField"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
         }
 
         @Test
@@ -7296,7 +7318,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Reinitializations {
       @Test
       public void testAllFilesPresentInReinitializations() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/reinitializations"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/reinitializations"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
@@ -7312,7 +7334,7 @@ public class FirPsiDiagnosticTestGenerated extends AbstractFirPsiDiagnosticTest
     public class Smartcasts {
       @Test
       public void testAllFilesPresentInSmartcasts() {
-        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/smartcasts"), Pattern.compile("^([^.]+)\\.kt$"), null, true);
+        KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("compiler/fir/analysis-tests/testData/resolveWithStdlib/smartcasts"), Pattern.compile("^([^.]+)\\.(kt|kts)$"), null, true);
       }
 
       @Test
diff --git a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
index 49a17793af7..326a1c66abb 100644
--- a/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
+++ b/compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java
@@ -31928,6 +31928,12 @@ public class FirPsiOldFrontendDiagnosticsTestGenerated extends AbstractFirPsiDia
         runTest("compiler/testData/diagnostics/tests/script/resolveInitializerOfDestructuringDeclarationOnce.kts");
       }
 
+      @Test
+      @TestMetadata("scriptDeclsShouldVisibleLocally.kts")
+      public void testScriptDeclsShouldVisibleLocally() {
+        runTest("compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts");
+      }
+
       @Test
       @TestMetadata("scriptScopes.kts")
       public void testScriptScopes() {
diff --git a/compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts b/compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts
new file mode 100644
index 00000000000..bd9f8efc76f
--- /dev/null
+++ b/compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts
@@ -0,0 +1,25 @@
+// FIR_IDENTICAL
+
+val a = 42
+
+class A
+
+enum class E {
+    V
+}
+
+object O {
+    val v = 42
+}
+
+fun foo() = 42
+
+val b =  a
+
+val ca = A()
+
+val ev = E.V
+
+val ov = O.v
+
+val rfoo = foo()
diff --git a/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java b/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
index 0a5822a2c77..d604f6e2ca0 100644
--- a/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
+++ b/compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java
@@ -34176,6 +34176,12 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
         runTest("compiler/testData/diagnostics/tests/script/resolveInitializerOfDestructuringDeclarationOnce.kts");
       }
 
+      @Test
+      @TestMetadata("scriptDeclsShouldVisibleLocally.kts")
+      public void testScriptDeclsShouldVisibleLocally() {
+        runTest("compiler/testData/diagnostics/tests/script/scriptDeclsShouldVisibleLocally.kts");
+      }
+
       @Test
       @TestMetadata("scriptScopes.kts")
       public void testScriptScopes() {
diff --git a/compiler/tests-for-compiler-generator/tests/org/jetbrains/kotlin/test/generators/GenerateJUnit5CompilerTests.kt b/compiler/tests-for-compiler-generator/tests/org/jetbrains/kotlin/test/generators/GenerateJUnit5CompilerTests.kt
index 8a1164a6ba8..ed9edf88007 100644
--- a/compiler/tests-for-compiler-generator/tests/org/jetbrains/kotlin/test/generators/GenerateJUnit5CompilerTests.kt
+++ b/compiler/tests-for-compiler-generator/tests/org/jetbrains/kotlin/test/generators/GenerateJUnit5CompilerTests.kt
@@ -366,8 +366,8 @@ fun generateJUnit5CompilerTests(args: Array<String>, mainClassName: String?) {
 
         testGroup("compiler/fir/analysis-tests/tests-gen", "compiler/fir/analysis-tests/testData") {
             testClass<AbstractFirPsiDiagnosticTest> {
-                model("resolve", pattern = TestGeneratorUtil.KT_WITHOUT_DOTS_IN_NAME)
-                model("resolveWithStdlib", pattern = TestGeneratorUtil.KT_WITHOUT_DOTS_IN_NAME)
+                model("resolve", pattern = TestGeneratorUtil.KT_OR_KTS_WITHOUT_DOTS_IN_NAME)
+                model("resolveWithStdlib", pattern = TestGeneratorUtil.KT_OR_KTS_WITHOUT_DOTS_IN_NAME)
             }
 
             testClass<AbstractFirLightTreeDiagnosticsTest> {