Go TestMain 里 os.Exit 后为什么没有执行清理代码

在 TestMain 里写了 defer closeResource()，最后却用 os.Exit(m.Run()) 结束测试，清理函数不会执行。原因不是 defer 失效，而是 os.Exit 直接结束进程，绕过了当前函数的正常返回路径。

TestMain 返回和 os.Exit 不是同一条退出路径

go test 会为测试包生成一个包装入口。存在 TestMain(m *testing.M) 时，包装器调用它；如果 TestMain 返回，包装器会把 m.Run() 得到的结果交给 os.Exit。这给了 TestMain 一个完整的收尾机会。

下面两种写法看起来都在“用测试结果退出”，实际行为不同：

把资源清理放在 m.Run 前，并让 TestMain 正常结束

如果资源由测试包级别的初始化创建，可以在调用 m.Run 前注册清理函数。重点是不要把 m.Run 嵌入 os.Exit 参数中：

package example

import (
    "log"
    "testing"
)

func TestMain(m *testing.M) {
    resource, err := openTestResource()
    if err != nil {
        log.Printf("open test resource: %v", err) // 初始化失败时不要继续跑测试
        return // 未创建资源，无需关闭；测试包装器会完成最终退出
    }
    defer resource.Close() // 正常返回时关闭资源，失败测试也会走到这里

    code := m.Run() // 保存测试结果，不要直接作为 os.Exit 的参数
    log.Printf("test exit code: %d", code) // 只记录结果，不改变退出路径
}

这里即使某个测试失败，m.Run 也只是返回非零值；函数继续执行日志和 defer，随后正常返回。生成的包装器仍能拿到同一次 m.Run 的结果。初始化失败时如果还没有资源句柄，直接返回即可；如果初始化过程已经创建了部分资源，应在返回前显式清理，或把已经创建的对象交给一个可安全重复调用的 cleanup。

必须显式 os.Exit 时，先完成所有清理

有些自定义测试入口会要求在 TestMain 内显式结束进程。这时可以使用 os.Exit，但要接受一个事实：它不会替你执行 defer。需要清理的动作必须在调用它之前写完。

func TestMain(m *testing.M) {
    resource, err := openTestResource()
    if err != nil {
        log.Printf("open test resource: %v", err) // 此处没有可用资源可关闭
        os.Exit(2) // 直接退出，不能依赖 defer
    }

    code := m.Run()
    if err := resource.Close(); err != nil {
        log.Printf("close test resource: %v", err) // 清理错误不能静默吞掉
        if code == 0 {
            code = 1 // 只有原测试成功时，清理失败才提升为失败
        }
    }
    os.Exit(code) // 所有显式清理已经完成
}

这种写法的代价是 TestMain 中其他 defer 仍然不会执行，所以通常不如“清理后正常返回”简单。只有确实需要自定义退出策略时才选它，并把每一项清理放到明确的顺序里。

测试失败、初始化失败和清理失败要分开处理

m.Run() 的非零结果表示测试或基准流程失败；它不等于初始化失败，也不自动替清理失败分配新的退出码。排查时可以按下面的清单区分：

相关问题

TestMain 里的 defer 为什么不执行？

只有在 TestMain 正常返回时，函数内的 defer 才会按后进先出执行。直接调用 os.Exit 会终止进程，不会展开 defer 栈。

测试失败时 cleanup 会不会执行？

会。只要 m.Run 返回后仍回到 TestMain 的清理代码，测试失败不会阻止清理；真正会切断它的是提前退出进程。

TestMain 能不能直接 return m.Run()？

不能把整数返回给没有返回值的 TestMain。应保存 m.Run() 的结果，完成清理后让函数自然返回，由测试包装器传递退出码。