在 Go 单元测试中，可通过 bytes.Buffer（实现 io.Reader 和 io.Writer）模拟文件读写，无需真实磁盘 I/O，提升测试速度与可靠性。
在 go 单元测试中，可通过 `bytes.buffer`（实现 `io.reader` 和 `io.writer`）模拟文件读写，无需真实磁盘 i/o，提升测试速度与可靠性。
Go 语言本身不提供类似 Python 的 io.StringIO 或 io.BytesIO 独立类型，但标准库中的 bytes.Buffer 完美承担了这一角色——它同时实现了 io.Reader、io.Writer、io.Seeker 和 io.Closer 等核心接口，天然适配任何接受这些接口的函数，是构建可测试 I/O 逻辑的基石。
✅ 核心原则：面向接口编程
为确保可测试性，业务函数应依赖抽象接口而非具体类型。例如：
func ParseLines(r io.Reader) ([]string, error) {
scanner := bufio.NewScanner(r)
var lines []string
for scanner.Scan() {
lines = append(lines, scanner.Text())
}
return lines, scanner.Err()
}
func WriteLines(w io.Writer, lines []string) error {
for _, line := range lines {
if _, err := fmt.Fprintln(w, line); err != nil {
return err
}
}
return nil
}以上函数不关心输入来自文件、网络还是内存——只要传入符合 io.Reader 或 io.Writer 的实例即可。
? 测试示例：用 bytes.Buffer 模拟文件流
测试读取逻辑（模拟 stdin 或配置文件）：
func TestParseLines(t *testing.T) {
input := "apple\nbanana\ncherry"
buf := bytes.NewBufferString(input) // ← 内存输入流
result, err := ParseLines(buf)
if err != nil {
t.Fatal(err)
}
expected := []string{"apple", "banana", "cherry"}
if !reflect.DeepEqual(result, expected) {
t.Errorf("expected %v, got %v", expected, result)
}
}测试写入逻辑（模拟 stdout 或日志文件）：
func TestWriteLines(t *testing.T) {
var buf bytes.Buffer // ← 内存输出流
lines := []string{"first", "second", "third"}
err := WriteLines(&buf, lines)
if err != nil {
t.Fatal(err)
}
expected := "first\nsecond\nthird\n"
if buf.String() != expected {
t.Errorf("expected %q, got %q", expected, buf.String())
}
}⚠️ 注意事项与最佳实践
- *避免 `bytes.Buffer
vsbytes.Buffer混用**：WriteLines接收io.Writer，而*bytes.Buffer才实现该接口（值类型bytes.Buffer不实现io.Writer），因此调用时需传&buf`（如上例）。 -
初始化选择：
bytes.NewBufferString(s)适用于已知字符串输入；bytes.NewBuffer(b)适用于字节切片；空缓冲区直接new(bytes.Buffer)或&bytes.Buffer{}。 -
无需手动
Seek(0)：与 Python 的StringIO不同，bytes.Buffer读写位置自动管理；若需重读，可用buf.Reset()清空或buf.Bytes()获取底层数据后重新构造。 -
结合
io.MultiReader/io.MultiWriter：复杂场景下可组合多个内存流（如模拟多文件输入）。
通过将 I/O 依赖抽象为接口，并用 bytes.Buffer 在测试中替代真实文件，你不仅能写出零副作用、快速稳定的单元测试，还能让生产代码天然支持灵活的输入源（本地文件、HTTP 响应、管道等），真正实现“一次编写，多处运行”。
