go - 需要 *os.File 作为参数的测试函数

Question

我想为波纹管函数编写一个测试，但我不明白我可以将什么作为参数发送给 toCount，因为我不想打开/创建文件，我知道 os.Stdin 会起作用，但我认为你不能写入它。

func toCount(f *os.File) int {
  input := buffo.NewScanner(f)
  sum := 0;
  for input.Scan() {
    sum++
  }
  return sum
}

Answer 1

您的toCount 函数只需要一个io.Reader。如果将签名更改为

func toCount(f io.Reader) int

它可以接受 *os.File 和任何其他你想用来测试的阅读器。