Go 测试如何只验证一次临时目录清理：t.Cleanup、子测试与失败现场保留

测试临时目录最容易出现两种相反的问题：清理逻辑注册了两遍，成功用例留下垃圾；清理得太早，失败时却找不到现场。更稳妥的做法是让每个测试只注册一次 t.Cleanup，目录由 t.TempDir 创建，清理函数在测试和全部子测试结束后再根据 t.Failed() 决定删除还是保留。

把“创建目录”和“清理目录”绑定在同一个测试层级；成功时删除，失败时保留，父测试不要替子测试重复接管目录。

先分清 t.TempDir 与 t.Cleanup 的责任

t.TempDir 解决的是“目录从哪里来以及默认何时回收”；t.Cleanup 解决的是“测试结束时还要做哪些动作”。官方 testing 文档说明，TempDir 返回的目录会在测试及其全部子测试完成后自动移除，而 Cleanup 注册的函数也会在这个测试层级收尾时运行。

用一次注册覆盖成功与失败两条路径

下面的辅助函数只做一件事：申请目录，并注册一次清理函数。成功时目录由 t.TempDir 的默认机制回收；失败时把 result.txt 留在目录中，便于 CI 收集。这里的 t.Failed() 必须放进清理函数，而不是在测试主体中提前判断。

package cleanupdemo

import (
    "os"
    "path/filepath"
    "testing"
)

func testDir(t *testing.T) string {
    t.Helper()
    dir := t.TempDir()
    t.Cleanup(func() {
        if t.Failed() {
            t.Logf("保留失败现场: %s", dir)
            return
        }
        _ = os.RemoveAll(dir)
    })
    return dir
}

func TestReport(t *testing.T) {
    dir := testDir(t)
    path := filepath.Join(dir, "result.txt")
    if err := os.WriteFile(path, []byte("ok"), 0o600); err != nil {
        t.Fatal(err)
    }
}

创建目录只发生一次；清理函数等测试结束后再走成功删除或失败保留分支。

子测试要自己拥有目录和清理边界

当一个测试用例需要覆盖多个输入，建议让每个 t.Run 单独调用 testDir。父测试只组织场景，不要把一个目录传给所有子测试。这样即使某个子测试失败，保留的 dir 也能对应到具体名称，其他子测试不会覆盖它的文件。

func TestImport(t *testing.T) {
    cases := []struct {
        name string
        data string
    }{
        {name: "empty", data: ""},
        {name: "header", data: "name,age\n"},
    }

    for _, tc := range cases {
        tc := tc
        t.Run(tc.name, func(t *testing.T) {
            dir := testDir(t)
            path := filepath.Join(dir, "input.txt")
            if err := os.WriteFile(path, []byte(tc.data), 0o600); err != nil {
                t.Fatal(err)
            }
            // 在这里执行只依赖当前 path 的导入检查。
        })
    }
}

父测试负责组织，empty 与 header 子测试各自拥有自己的 testDir。

失败现场保留时要检查什么

这个方案保留的是测试失败时的目录，不等于自动把目录上传到 CI。工程上还需要在流水线里收集目录，或者在日志中打印路径。若测试程序运行在容器里，容器退出后目录仍可能消失，所以“保留”只覆盖当前测试进程存活期间；持久化要交给 CI 的 artifact 机制。

另一个边界是清理函数本身不要承担断言。清理阶段再调用 t.Error 往往会让原始失败变得难以定位。目录清理失败通常应记录日志，并把真正的测试断言留在测试主体。

一组小实验验证执行顺序

可以用下面的命令观察父测试和子测试的输出顺序：

go test -run 'TestImport|TestReport' -v ./...

成功路径会显示测试完成，临时目录随后回收；人为加入一个错误断言后，清理函数会先打印“保留失败现场”，目录路径与子测试名一起出现在日志中。若看不到日志，先确认用例确实失败，以及 CI 是否收集了测试工作目录。

常见问题

t.TempDir 后还要调用 os.RemoveAll 吗？

通常不需要。只有在成功路径想提前释放大型目录，或清理的对象不是 t.TempDir 创建的目录时，才考虑额外删除；重复删除要避免和失败现场保留策略冲突。

t.Cleanup 会在子测试前执行吗？

不会。当前测试注册的清理函数会等它的全部子测试结束后执行；同一层级多个清理函数按后注册先执行。

并行子测试可以共用一个临时目录吗？

不建议。共享目录会引入文件名覆盖和清理时机竞争。让每个子测试调用自己的 t.TempDir，再把目录路径传给被测函数更容易排查。

把清理责任留在最小测试单元

最小且可靠的约定是：测试单元自己创建资源，自己注册一次收尾动作；成功由框架回收，失败由清理函数留下证据。父测试只组织 t.Run，不跨层级接管子测试目录。这样的边界既减少重复清理，也让失败日志能直接指向实际输入。