TestMain 是 Go 中用于包级测试初始化和清理的钩子函数，仅在需跨所有测试共享 setup/cleanup 时使用；必须定义在 _test.go 文件中，签名固定为 func TestMain(m testing.M) int，且必须调用 m.Run() 并返回其结果。
TestMain 是什么，什么时候必须用它
Go 的 TestMain 不是替代 main 的测试入口，而是测试包级初始化/清理的钩子。只有当你需要在所有测试开始前做一次 setup（比如启动 mock 服务、初始化全局状态），或在所有测试结束后做 cleanup（比如关闭数据库连接、释放临时文件），才需要它。
常见错误现象：go test 报错 cannot define TestMain in package main —— 因为 TestMain 只能定义在测试包（即 _test.go 文件）中，且包名必须是 xxx_test，不能是 main。
- 它只对当前包的测试生效，不影响其他包
- 如果你只是测单个函数，完全不需要
TestMain；TestXxx函数各自独立运行更安全 - Go 1.14+ 要求
TestMain必须调用m.Run()，否则测试会卡住或 panic
怎么写一个合法的 TestMain 函数
TestMain 签名固定，参数类型和返回值都不能改：必须是 func TestMain(m *testing.M)，返回 int。它的核心逻辑就是“前置操作 → m.Run() → 后置操作 → 返回 exit code”。
典型误用：defer cleanup() 放在 m.Run() 前面 —— 这会导致 cleanup 在所有测试执行前就被调用，起不到收尾作用。
立即学习“go语言免费学习笔记（深入）”；
-
m.Run()会阻塞直到所有TestXxx执行完毕，并返回进程退出码（0 表示全通过） - 必须显式 return 该返回值，否则测试框架无法判断成功与否
- 如果前置失败（比如端口被占），应直接 return 非零值，避免继续跑测试
func TestMain(m *testing.M) {
// setup
if err := startMockServer(); err != nil {
fmt.Fprintln(os.Stderr, "failed to start mock:", err)
os.Exit(1)
}
defer stopMockServer() // ✅ 正确：defer 在 m.Run 后触发
// run tests
code := m.Run()
os.Exit(code)
}
TestMain 和 init() 的区别在哪
init() 在包加载时自动执行，每次 go test 都会触发一次，但无法控制执行时机（比如不能等 flag 解析完），也不方便做带 error 的初始化。而 TestMain 是唯一能拿到 *testing.M 实例的地方，可以读取 -test.* 参数、控制是否跳过测试、甚至重定向 stdout。
容易踩的坑：init() 中调用 log.Fatal 或 os.Exit 会让整个测试提前终止，且不输出任何失败信息；TestMain 则可以优雅地报错并 exit。
-
init()无参数、无返回值，适合无副作用的静态初始化（如注册 encoder） -
TestMain可以解析os.Args，比如根据-test.short决定是否跳过耗时 setup - 多个
init()函数按源码顺序执行；TestMain只允许一个，重复定义会编译失败
为什么 TestMain 里不能用 t.Log 或 t.Fatal
TestMain 不在测试函数上下文中，没有 *testing.T 实例，所以所有以 t. 开头的方法都不可用。试图传入或伪造 T 会导致 panic 或行为未定义。
真实场景：你想在 setup 失败时像测试失败一样输出堆栈和文件行号 —— 但做不到。只能靠 fmt.Fprintln(os.Stderr, ...) + os.Exit(1)。
-
testing.T对象由测试框架为每个TestXxx单独构造，生命周期仅限于该函数 -
TestMain属于测试框架的“宿主层”，比单个测试更早、更底层 - 如果真需要结构化日志，建议封装自己的 logger，输出到 stderr 并带上时间戳
真正难的不是写对 TestMain 的签名，而是判断“这个初始化真的需要跨所有测试共享吗”。多数时候，把 setup/cleanup 拆进每个 TestXxx 里，反而更清晰、可复现、易调试。
