Test framework for the scripting engine implemented

Author: anatoliykmetyuk

compiler/test-resources/scripting/basic-math.scala

@@ -0,0 +1 @@
+@main def Test = assert(2 + 2 == 4)

compiler/test-resources/scripting/say-hi.args

@@ -0,0 +1 @@
+World

compiler/test-resources/scripting/say-hi.scala

@@ -0,0 +1,2 @@
+@main def Test(name: String) =
+  assert(name == "World")

compiler/test/dotty/tools/repl/ReplTest.scala

@@ -50,12 +50,6 @@ class ReplTest(withStaging: Boolean = false, out: ByteArrayOutputStream = new By
   extension [A](state: State)
     def andThen(op: State => A): A = op(state)
 
-  def scripts(path: String): Array[JFile] = {
-    val dir = new JFile(getClass.getResource(path).getPath)
-    assert(dir.exists && dir.isDirectory, "Couldn't load scripts dir")
-    dir.listFiles
-  }
-
   def testFile(f: JFile): Unit = {
     val prompt = "scala>"
 
@@ -77,12 +71,11 @@ class ReplTest(withStaging: Boolean = false, out: ByteArrayOutputStream = new By
         case nonEmptyLine => nonEmptyLine :: Nil
       }
 
-    val expectedOutput =
-      Using(Source.fromFile(f, StandardCharsets.UTF_8.name))(_.getLines().flatMap(filterEmpties).toList).get
+    val expectedOutput = readLines(f).flatMap(filterEmpties)
     val actualOutput = {
       resetToInitial()
 
-      val lines = Using(Source.fromFile(f, StandardCharsets.UTF_8.name))(_.getLines.toList).get
+      val lines = readLines(f)
       assert(lines.head.startsWith(prompt),
         s"""Each file has to start with the prompt: "$prompt"""")
       val inputRes = lines.filter(_.startsWith(prompt))

compiler/test/dotty/tools/scripting/ScriptingTests.scala

@@ -0,0 +1,42 @@
+package dotty
+package tools
+package scripting
+
+import java.io.File
+
+import org.junit.Test
+
+import vulpix.TestConfiguration
+
+
+/** Runs all tests contained in `compiler/test-resources/repl/` */
+class ScriptingTests:
+  extension (str: String) def dropExtension =
+    str.reverse.dropWhile(_ != '.').drop(1).reverse
+
+  @Test def scriptingTests =
+    val testFiles = scripts("/scripting")
+
+    val argss: Map[String, Array[String]] = (
+      for
+        argFile <- testFiles
+        if argFile.getName.endsWith(".args")
+        name = argFile.getName.dropExtension
+        scriptArgs = readLines(argFile).toArray
+      yield name -> scriptArgs).toMap
+
+    for
+      scriptFile <- testFiles
+      if scriptFile.getName.endsWith(".scala")
+      name = scriptFile.getName.dropExtension
+      scriptArgs = argss.getOrElse(name, Array.empty[String])
+    do
+      ScriptingDriver(
+        compilerArgs = Array(
+            "-classpath", TestConfiguration.basicClasspath,
+            "-color:never",
+            "-Yerased-terms",
+          ),
+        scriptFile = scriptFile,
+        scriptArgs = scriptArgs
+      ).compileAndRun()

compiler/test/dotty/tools/utils.scala

@@ -0,0 +1,19 @@
+package dotty.tools
+
+import java.io.File
+import java.nio.charset.StandardCharsets.UTF_8
+
+import scala.io.Source
+import scala.util.Using.resource
+
+def scripts(path: String): Array[File] = {
+  val dir = new File(getClass.getResource(path).getPath)
+  assert(dir.exists && dir.isDirectory, "Couldn't load scripts dir")
+  dir.listFiles
+}
+
+private def withFile[T](file: File)(action: Source => T): T =
+  resource(Source.fromFile(file, UTF_8.name))(action)
+
+def readLines(f: File): List[String] = withFile(f)(_.getLines.toList)
+def readFile(f: File): String = withFile(f)(_.mkString)

staging/test/scala/quoted/staging/repl/StagingScriptedReplTests.scala

@@ -1,6 +1,7 @@
 package scala.quoted.staging.repl
 
 import dotty.BootstrappedOnlyTests
+import dotty.tools.scripts
 import dotty.tools.repl.ReplTest
 import dotty.tools.vulpix.TestConfiguration
 import org.junit.Test