scala - 我如何在 Scala 中对控制台输入进行单元测试？

Question

我如何使用 scalaTest 在 scala 中对控制台输入进行单元测试。被测代码:

object ConsoleAction {
  def readInput(in: InputStream): List[String] = {
    val bs = new BufferedSource(in)(Codec.default)
    val l = bs.getLines()
    l.takeWhile(_!="").toList
  }

  def main(args: Array[String]) {
    val l = ConsoleAction.readInput(System.in)
    println("--> "+l)
  }
}

我想测试 readInput 方法。一行输入可以这样测试:

"Result list" should "has 1 element" in {
  val input = "Hello\\n"
  val is = new ByteArrayInputStream(input.getBytes(StandardCharsets.UTF_8))
  assert(ConsoleAction.readInput(is).size===1)
}

...但是多行输入的方式是什么？

input line 1
input line 2

谢谢

Answer 1

您的问题在于您如何转义换行符。你在做“\\n”而不是“\n”。这个测试应该通过。

"Result list" should "has 2 elements" in {
  val input = "Hello\nWorld\n"
  val is = new ByteArrayInputStream(input.getBytes(StandardCharsets.UTF_8))
  assert(ConsoleAction.readInput(is).size===2)
}