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 函数中的重复逻辑）。
✅ 正确用法如下：
AI舌诊面诊问诊医疗系统+诊疗商城系统源码
下载
AI 舌诊面诊问诊医疗系统 + 诊疗商城系统源码是一套结合 AI 辅助诊断与商品转化的健康类程序，基于浩创云、ThinkPHP（PHP）开发，支持微信服务号与 H5 端，适合做中医舌诊面诊问答、健康建议推送以及配套诊疗商品的运营。
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 测试套件。
