os.MkdirTemp 目录清理遗漏的测试修复方法

测试里用 os.MkdirTemp 创建工作目录后，目录不会因为测试失败而自动消失。真正容易遗漏的地方不是创建调用，而是“谁拥有清理责任”：创建函数返回成功后，必须马上把 os.RemoveAll 绑定到当前测试生命周期。对普通单元测试，优先使用 t.TempDir()；只有需要自定义临时根目录、提前清理或复用一段已有目录逻辑时，才保留 os.MkdirTemp。

为什么 os.MkdirTemp 会留下测试目录

官方文档对 MkdirTemp 的约定很明确：它创建一个权限为 0700 的目录，返回路径，目录不再需要时由调用者负责移除。也就是说，下面这段代码即使断言失败，也不会替你执行删除。

dir, err := os.MkdirTemp("", "parser-test-")
if err != nil {
    t.Fatal(err) // 创建失败时没有可清理的路径，直接终止测试
}
// 这里只拿到了路径；如果没有登记 RemoveAll，测试结束后目录仍可能存在。

如果临时目录由辅助函数创建并返回，问题会更隐蔽：调用方可能只记得使用目录，却忘记它还要负责回收。另一个常见误区是在辅助函数里写 defer os.RemoveAll(dir)，这样清理发生在辅助函数返回时，而不是调用方期望的整个测试阶段结束时。

用 t.Cleanup 绑定目录的所有权

更稳妥的修复方式是让创建目录的辅助函数同时登记清理回调。这样调用方只拿到“可用目录”，不会再因为忘记配套删除而留下垃圾：

func makeTestDir(t *testing.T, parent string) string {
    t.Helper()

    dir, err := os.MkdirTemp(parent, "parser-test-")
    if err != nil {
        t.Fatalf("创建临时目录失败: %v", err) // 错误路径没有注册清理回调
    }
    t.Cleanup(func() {
        if err := os.RemoveAll(dir); err != nil {
            t.Errorf("清理临时目录 %q 失败: %v", dir, err) // 保留失败原因
        }
    })
    return dir
}

func TestParseFile(t *testing.T) {
    dir := makeTestDir(t, "")
    path := filepath.Join(dir, "input.txt")
    if err := os.WriteFile(path, []byte("hello"), 0o600); err != nil {
        t.Fatal(err) // 文件写入失败仍会走 t.Cleanup
    }
}

t.Cleanup 会在当前测试及其子测试完成后执行，并且即使测试通过、失败或调用 Fatal，都沿着 testing 的清理生命周期运行。清理回调里使用 RemoveAll，是因为目录下通常还会有测试输入文件；只调用 os.Remove 只能删除空目录。

检查子测试、并发和自定义临时根目录

修复后不要只看主流程。下面这张检查表能快速发现目录所有权是否仍然模糊：

特别是 t.Parallel 场景，不能把一个测试创建的目录交给另一个测试清理。父目录也不应由子测试随意删除，否则其他并发用例可能在写文件时收到 path does not exist。Go 官方的临时目录测试同样会在创建成功后登记删除，并单独覆盖不存在的父目录与非法 pattern。

什么时候直接换成 t.TempDir

如果测试只需要一个独立目录，不需要指定父目录，也不需要在测试中途提前删除，直接改成 dir := t.TempDir() 通常更简单。它由 testing 包负责创建，并在测试和全部子测试结束后自动移除。

需要自定义根目录、验证 MkdirTemp 本身的 pattern、控制提前清理时机，或兼容非 testing 的普通辅助函数时，再使用 os.MkdirTemp。无论选哪种方式，检查清单都只有三项：创建成功后是否立刻绑定清理；清理是否覆盖非空目录；并发测试之间是否没有共享删除责任。

常见问题与修复清单

为什么 defer os.RemoveAll 有时不够直观？

它只绑定当前函数作用域。若目录由辅助函数创建，清理时机可能早于调用方预期；把回调注册到 t.Cleanup 更能表达测试级所有权。

清理失败要不要让测试失败？

建议用 t.Errorf 记录路径和错误，让测试结果暴露环境问题；不要静默吞掉 RemoveAll 的失败。

目录里只有一个文件，可以只删文件吗？

不建议。测试可能在后续步骤新增文件，统一使用 RemoveAll 才不会让清理逻辑随内容变化而失效。

官方文档入口：https://pkg.go.dev/os