blob: 9867cb4ecf2033df7145a99a9bc35ea874ab4e6c (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
package test
import dotty.tools.dotc.repl._
import dotty.tools.dotc.core.Contexts.Context
import collection.mutable
import java.io.StringWriter
/** A subclass of REPL used for testing.
* It takes a transcript of a REPL session in `script`. The transcript
* starts with the first input prompt `scala> ` and ends with `scala> :quit` and a newline.
* Invoking `process()` on the `TestREPL` runs all input lines and
* collects then interleaved with REPL output in a string writer `out`.
* Invoking `check()` checks that the collected output matches the original
* `script`.
*/
class TestREPL(script: String) extends REPL {
private val out = new StringWriter()
override lazy val config = new REPL.Config {
override val output = new NewLinePrintWriter(out)
override def context(ctx: Context) =
ctx.fresh.setSetting(ctx.settings.color, "never")
override def input(in: Interpreter)(implicit ctx: Context) = new InteractiveReader {
val lines = script.lines.buffered
def readLine(prompt: String): String = {
val line = lines.next
val buf = new StringBuilder
if (line.startsWith(prompt)) {
output.println(line)
buf append line.drop(prompt.length)
while (lines.hasNext && lines.head.startsWith(continuationPrompt)) {
val continued = lines.next
output.println(continued)
buf append System.lineSeparator()
buf append continued.drop(continuationPrompt.length)
}
buf.toString
}
else readLine(prompt)
}
val interactive = false
}
}
def check() = {
out.close()
val printed = out.toString
val transcript = printed.drop(printed.indexOf(config.prompt))
if (transcript.toString.lines.toList != script.lines.toList) {
println("input differs from transcript:")
println(transcript)
assert(false)
}
}
}
|
