如何在 golang 中测试 io.writer？答案

【问题标题】：How to test io.writer in golang?如何在 golang 中测试 io.writer？
【发布时间】：2015-04-08 02:27:28
【问题描述】：

最近希望为golang写一个单元测试。函数如下。

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

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

【问题讨论】：

标签： unit-testing go

【解决方案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)
    }
}

【讨论】：

非常感谢您的帮助！这正是我想要的！
这里的错字：您需要在 Display 调用中使用&b 或 b := new(bytes.Buffer)，因为Write 方法有一个指针接收器。
谢谢戴夫！我不知道它可以这么容易修复。 :)
当我这样做时，String() 返回“”。我可以说发生了写入是因为 Len() 发生了变化。当我将对象作为 Writer 传递时，我不知道它在读什么。有什么办法仍然可以读取缓冲区的内容吗？