最近我希望为golang编写单元测试。该函数如下:
func (s *containerStats) Display(w io.Writer) error {
fmt.Fprintf(w, "%s %s\n", "hello", "world")
return nil
}
我该如何测试 "func Display" 函数的结果是否为 "hello world"?
最近我希望为golang编写单元测试。该函数如下:
func (s *containerStats) Display(w io.Writer) error {
fmt.Fprintf(w, "%s %s\n", "hello", "world")
return nil
}
我该如何测试 "func Display" 函数的结果是否为 "hello world"?
io.Writer
,并测试写入的内容是否与预期相符。 bytes.Buffer
是这种io.Writer
的很好选择,因为它只是将输出存储在其缓冲区中。func TestDisplay(t *testing.T) {
s := newContainerStats() // Replace this the appropriate constructor
var b bytes.Buffer
if err := s.Display(&b); err != nil {
t.Fatalf("s.Display() gave error: %s", err)
}
got := b.String()
want := "hello world\n"
if got != want {
t.Errorf("s.Display() = %q, want %q", got, want)
}
}
&b
或者b := new(bytes.Buffer)
,因为Write
方法有一个指针接收器。请注意,在Display调用中只需要使用一个选项。 - Dave C