如何在Golang中测试io.writer？

Question

最近，我希望为golang编写单元测试。功能如下。

func (s *containerStats) Display(w io.Writer) error {
    fmt.Fprintf(w, "%s %s\n", "hello", "world")
    return nil
}

那么，如何测试“ func Display”的结果是“ hello world”呢？

Answer 1

您只需输入自己的值，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)
    }
}