TestMain 是 Go 中唯一能在所有测试开始前/结束后执行一次全局初始化或清理的机制。它必须定义在 main_test.go 中，签名固定为 func TestMain(m *testing.M)，需显式调用 m.Run() 并用 os.Exit(m.Run()) 透传退出码，defer 清理逻辑须置于 m.Run() 前以确保执行。
TestMain 是什么，为什么不能只用 init()
Go 的 TestMain 不是可选的“高级技巧”，而是唯一能**在所有测试开始前/结束后执行一次全局初始化或清理**的机制。你如果在 init() 函数里初始化数据库连接、启动 mock HTTP server 或写临时文件，会遇到两个问题：一是 go test 并发运行测试时可能冲突；二是无法控制执行时机（比如想等 setup 完成后再跑任何测试，且确保 teardown 在所有测试之后）。TestMain 就是为此设计的入口函数。
如何定义 TestMain 函数
必须放在 main_test.go 文件中（注意后缀是 _test.go），且签名严格固定为：
func TestMain(m *testing.M)
它不是被自动调用的，你必须显式调用 m.Run() 来真正执行所有测试。不调用就等于“拦截了所有测试”，结果是 0 个测试运行，退出码为 0 —— 看似成功，实则没测。
常见错误写法包括：
- 函数名拼错，比如
TestMainn或TESTMAIN - 参数类型不是
*testing.M - 忘记调用
m.Run()，或把它放在defer里（会导致 teardown 提前执行） - 在
m.Run()后继续写逻辑但没 return，导致程序多走一遍 cleanup
典型初始化 + 清理结构
一个安全、可复用的 TestMain 模板要满足：setup 成功才跑测试、teardown 总被执行、失败时保留 exit code。
示例：
func TestMain(m *testing.M) {
// 初始化：比如启动本地 Redis mock
if err := startMockRedis(); err != nil {
fmt.Fprintf(os.Stderr, "failed to start mock redis: %v\n", err)
os.Exit(1)
}
defer func() {
// 清理：无论测试是否通过都执行
stopMockRedis()
}()
// 运行所有测试函数
os.Exit(m.Run())
}
注意几点：
-
os.Exit(m.Run())是关键，它把测试的退出码原样透传出去；直接写m.Run()会导致测试失败时返回 0 -
defer放在m.Run()前，才能保证 cleanup 在测试结束后执行 - 如果初始化失败，必须用
os.Exit(1)，不能只 return —— 否则go test会认为测试通过
什么时候不该用 TestMain
不是所有初始化都需要 TestMain。如果你只是设置几个包级变量、读配置文件、或初始化一个无状态的工具函数，用 init() 更轻量，也更符合 Go 的惯用法。
真正需要 TestMain 的场景很具体：
- 启动/关闭外部进程（如
mock-server、sqlite内存 DB） - 创建和销毁共享资源（如临时目录、监听端口）
- 设置全局环境变量并希望在所有测试中生效（如
os.Setenv("ENV", "test")） - 需要在测试前后打日志或统计耗时
容易被忽略的一点是：TestMain 中的 panic 不会被捕获，会直接终止整个测试流程 —— 所以涉及 I/O 或网络的操作，务必加 error 判断，别依赖 defer recover。
