如何在Go语言中测试文件写入功能-Golang-PHP中文网

应优先用 os.createtemp 创建临时文件并显式清理，或用 bytes.buffer 避免磁盘 i/o；注意 os.file 需 close 刷新缓冲，测试失败时可暂存临时文件辅助调试。

如何在go语言中测试文件写入功能

用 `os.CreateTemp` 替代真实路径写入测试

直接在测试中往 /tmp 或当前目录写文件，会污染环境、依赖权限、难以清理。Go 标准库提供 os.CreateTemp，它自动创建唯一临时文件并返回 *os.File，适合模拟写入目标。

关键点：

传入空字符串 "" 作为目录参数，让系统选默认临时目录（如 /var/folders/... 或 C:\Users\...\AppData\Local\Temp\）
务必在测试末尾调用 os.Remove 删除临时文件，否则多次运行后残留文件会堆积
不要用 defer os.Remove(...) 在函数开头注册——如果 CreateTemp 失败，defer 仍会执行并报错 no such file or directory

func TestWriteToFile(t *testing.T) {
    f, err := os.CreateTemp("", "test-*.txt")
    if err != nil {
        t.Fatal(err)
    }
    defer os.Remove(f.Name()) // ✅ 在成功创建后才 defer

    err = WriteContent(f, "hello world")
    if err != nil {
        t.Fatal(err)
    }

    // 验证内容
    data, _ := os.ReadFile(f.Name())
    if string(data) != "hello world" {
        t.Error("expected 'hello world'")
    }
}

用 `bytes.Buffer` 替代文件句柄做单元测试

如果你的写入逻辑封装在某个函数里（比如 WriteContent(w io.Writer, s string)），根本不需要碰磁盘。把 bytes.Buffer 当作 io.Writer 传进去，测的是逻辑本身，速度快、无副作用、可重复。

适用场景：

函数签名接受 io.Writer（如 json.Encoder.Encode、自定义日志写入器）
想隔离“生成内容”和“落盘”两个关注点
避免因文件系统权限、只读挂载等外部因素导致测试失败

func TestWriteContent_Buffer(t *testing.T) {
    var buf bytes.Buffer
    err := WriteContent(&buf, "test data")
    if err != nil {
        t.Fatal(err)
    }
    if buf.String() != "test data" {
        t.Error("unexpected output")
    }
}

注意 `os.File` 的写入缓冲与 `Close` 行为

很多 bug 出在忘记 Close() 或误以为 Write() 立即落盘。实际上 os.File 内部有缓冲，且部分操作系统（尤其是 Windows）在文件关闭前可能不刷新元数据（如文件大小）。测试时若只写不关，后续 os.Stat 或 os.ReadFile 可能拿到过期结果。