go 1.4+ 中使用 testmain 实现测试前/后全局初始化与清理
在 go 测试中，若需在所有测试开始前执行一次初始化（如启动数据库、加载配置），并在全部测试结束后执行一次清理（如关闭连接、清空临时文件），应使用 func TestMain(m *testing.M) —— 这是 Go 1.4 引入的关键机制。它替代了默认的测试执行流程，使你获得对测试生命周期的完全控制。
⚠️ 注意：testing.M 仅在 Go 1.4 及以上版本中定义。若运行 go test 时提示 undefined: testing.M，请先执行 go version 确认版本。低于 1.4 的环境无法使用该特性（旧版需依赖 init() 函数或手动封装 TestXxx 函数中的重复逻辑）。
✅ 正确用法如下：
package main import ( "os" "testing" ) func TestSomeTest(t *testing.T) { t.Log("Running individual test") } func TestMain(m *testing.M) { // ✅ 测试前：全局初始化（如连接数据库、准备测试环境） println("Setting up test environment...") // ✅ 执行所有测试函数（必须调用 m.Run()） exitCode := m.Run() // ✅ 测试后：全局清理（如关闭资源、重置状态） println("Tearing down test environment...") // ⚠️ 必须使用 os.Exit() 传递退出码，否则测试框架无法正确返回结果 os.Exit(exitCode) }? 关键要点：
- TestMain 函数签名必须严格为 func TestMain(m *testing.M)，且位于 *_test.go 文件中；
- m.Run() 是核心调用，它按顺序执行所有 TestXxx 函数，并返回整型退出码（0 表示全部通过）；
- defer 在 TestMain 中不会自动生效于 os.Exit() 之后，因此清理逻辑必须显式写在 m.Run() 之后、os.Exit() 之前；
- 若需“每个测试后清理”，TestMain 并不适用（它是整个测试包级别的钩子）；此时应改用 t.Cleanup()（Go 1.14+）或在每个测试末尾手动 defer 清理逻辑。
例如，针对单个测试的后置清理（推荐现代写法）：
func TestWithCleanup(t *testing.T) { // 创建临时资源 tmpFile, _ := os.CreateTemp("", "test-*.txt") defer os.Remove(tmpFile.Name()) // 自动清理 // 注册测试结束时执行的清理（即使测试 panic 也保证执行） t.Cleanup(func() { tmpFile.Close() println("Cleaned up temp file:", tmpFile.Name()) }) // 实际测试逻辑... t.Log("Testing with cleanup") }总结：TestMain 是跨测试生命周期的“全局守门人”，适用于昂贵的一次性初始化/销毁场景；而 t.Cleanup() 更适合细粒度、测试级别的资源管理。两者结合可构建健壮、高效、易维护的 Go 测试套件。