如何在 Go 中测试 io.Writer 接口的输出行为-Golang

本文介绍使用 `bytes.buffer` 作为可捕获输出的 `io.writer` 实现，对依赖 `io.writer` 的函数进行单元测试，验证其写入内容是否符合预期。

在 Go 单元测试中，当函数接受 io.Writer 作为参数（如用于格式化输出），我们无需依赖真实文件、标准输出或网络连接——只需提供一个内存中的、可读取的 io.Writer 实现即可。bytes.Buffer 是标准库中最常用且最合适的选项：它同时实现了 io.Writer 和 io.Reader 接口，能无副作用地累积写入内容，并通过 .String() 或 .Bytes() 暴露结果。

以下是一个典型测试示例，对应被测方法：

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

对应的单元测试如下：

func TestDisplay(t *testing.T) {
    s := newContainerStats() // 替换为实际的构造逻辑（如 &containerStats{} 或工厂函数）
    var buf bytes.Buffer

    if err := s.Display(&buf); err != nil {
        t.Fatalf("s.Display() returned unexpected error: %v", err)
    }

    got := buf.String()
    want := "hello world\n"

    if got != want {
        t.Errorf("s.Display() output = %q, want %q", got, want)
    }
}

✅ 关键要点说明：