Golang测试中临时目录使用技巧

Go 1.16+ 推荐使用 t.TempDir() 创建测试临时目录，它自动注册清理逻辑、保证隔离性、无需手动删除；支持子测试并行、可构建嵌套结构，但不可在 init() 或全局变量中调用。

Go 1.16+ 提供了 t.TempDir()，是测试中创建临时目录最安全、最推荐的方式——它自动注册清理逻辑，无需手动 os.RemoveAll，且每个测试用例独享隔离目录，彻底避免竞态和残留问题。

✅ 基本用法：一行创建，自动清理

在测试函数中直接调用 t.TempDir()，它返回一个绝对路径字符串。该目录会在测试结束（无论成功或失败）时被自动删除：

• 不需要 defer，不需显式调用 os.RemoveAll
• 即使测试 panic 或提前 return，目录仍会被清理
• 路径唯一，多个子测试（t.Run）也互不干扰

func TestProcessConfig(t *testing.T) {
    dir := t.TempDir() // 自动注册 cleanup
    cfgPath := filepath.Join(dir, "config.json")
    os.WriteFile(cfgPath, []byte(`{"port":8080}`), 0600)

    result := LoadConfig(cfgPath)
    if result.Port != 8080 {
        t.Fatal("expected port 8080")
    }
    // 测试结束，dir 自动删除 ✅
}

? 配合子测试（t.Run）安全复用

在表驱动测试中，每个子测试调用 t.TempDir() 会生成独立目录，天然支持并行（t.Parallel()）：

• 每个子测试拥有自己的 dir，无共享状态风险
• 可放心在子测试内创建文件、修改内容、甚至递归写入
• 父测试结束时，所有子测试的 TempDir 已全部清理完毕

func TestReadFiles(t *testing.T) {
    tests := []struct{
        name, content string
    }{
        {"empty", ""},
        {"json", `{"ok":true}`},
        {"text", "hello\nworld"},
    }
    for _, tt := range tests {
        tt := tt // capture range var
        t.Run(tt.name, func(t *testing.T) {
            t.Parallel()
            dir := t.TempDir()
            fpath := filepath.Join(dir, "input.txt")
            os.WriteFile(fpath, []byte(tt.content), 0644)

            data, _ := os.ReadFile(fpath)
            if string(data) != tt.content {
                t.Errorf("got %q, want %q", string(data), tt.content)
            }
            // ✅ 自动清理 dir 及其全部内容
        })
    }
}

? 进阶技巧：预置结构 + 模拟嵌套目录

临时目录本身是空的，但你可以快速构建任意层级结构，适合模拟真实项目布局：

• 用 os.MkdirAll 创建子目录（如 dir/logs、dir/data/cache）
• 用 os.WriteFile 或 ioutil.WriteFile（Go <1.16）写入多文件
• 用 os.Symlink 模拟符号链接（注意跨平台兼容性）

func TestAppWithLayout(t *testing.T) {
    root := t.TempDir()
    
    // 创建嵌套目录
    os.MkdirAll(filepath.Join(root, "config"), 0755)
    os.MkdirAll(filepath.Join(root, "data", "cache"), 0755)
    
    // 写入配置
    os.WriteFile(filepath.Join(root, "config", "app.yaml"), 
        []byte("mode: production\nlog_level: info"), 0644)
    
    // 写入缓存文件
    os.WriteFile(filepath.Join(root, "data", "cache", "token.bin"), 
        []byte("secret-token-123"), 0600)
    
    app := NewApp(root) // 传入 root 作为 base dir
    if !app.IsProduction() {
        t.Fatal("should be production mode")
    }
}

⚠️ 注意事项与常见误区

t.TempDir() 很好用，但要注意几个边界点：

• 不能在 init() 或包级变量中调用 —— 它只在 *testing.T 上下文有效
• 不要把返回路径保存到全局变量或长期引用，目录生命周期仅限当前测试
• 如果需要“跨测试保留”，说明设计有问题 —— 应该重构为可注入路径的接口，或改用内存 FS（如 github.com/spf13/afero）
• Windows 下路径含盘符（如 C:\Users\...\TestXXX\... ），避免硬编码路径分隔符，始终用 filepath.Join

基本上就这些。t.TempDir 是 Go 测试基建里低调但关键的一环——写得少，错得少，清理得干净。