需要 *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您想用来测试的任何其他类型的阅读器。