使用TestMain配合*testing.M可在测试前后执行初始化和清理操作。1. 定义TestMain函数作为测试入口；2. 在setup中建立数据库连接、加载配置或启动服务；3. 调用m.Run()运行所有TestXXX函数；4. 在teardown中关闭资源；5. 必须通过os.Exit(exitCode)退出以确保正确返回状态码。适用于需共享资源的集成测试场景，注意全局状态并发安全与资源释放。
在Go语言中，testing.M 是 testing 包提供的一个结构体，用于控制测试的执行流程。通过它，我们可以在所有测试用例运行前后执行全局初始化和清理操作，比如连接数据库、加载配置、启动服务等。这在集成测试或需要共享资源的场景中非常有用。
何时使用 testing.M
默认情况下，Go的测试直接运行所有以 Test 开头的函数。但如果你需要：
这时就应该使用 TestMain 函数配合 *testing.M 来控制生命周期。
基本用法：定义 TestMain 函数
在一个测试包中，你可以定义一个名为 TestMain(m *testing.M) 的函数。它是测试的入口点，由 go test 调用。
立即学习“go语言免费学习笔记（深入）”；
// example_test.gopackage main
import (
"log"
"os"
"testing"
)
var db *MockDB // 模拟全局资源
// 模拟数据库
type MockDB struct {
connected bool
}
func (m *MockDB) Close() {
m.connected = false
}
func setup() {
db = &MockDB{connected: true}
log.Println("✅ 数据库连接已建立")
}
func teardown() {
if db != nil && db.connected {
db.Close()
log.Println("? 数据库连接已关闭")
}
}
func TestMain(m *testing.M) {
setup()
// 运行所有测试
exitCode := m.Run()
teardown()
// 使用 os.Exit 退出，确保返回正确的状态码
os.Exit(exitCode)
}
func TestSomething(t *testing.T) {
if !db.connected {
t.Fatal("数据库未连接")
}
t.Log("测试通过：数据库可用")
}
在这个例子中：
-
setup()在测试前执行 -
m.Run()启动所有 TestXXX 函数 -
teardown()在测试后清理资源 -
os.Exit(exitCode)必须调用，否则 TestMain 返回后仍会继续执行其他测试
常见注意事项
使用 TestMain 时有几个关键点要注意：
- 只能在一个包中定义一个 TestMain：多个文件中不能重复定义
- 必须调用 m.Run()：否则测试不会执行
- 必须用 os.Exit 结束：不能直接 return，否则可能忽略失败状态
- 并发测试需谨慎：全局状态可能被多个测试并发访问，注意同步或避免共享可变状态
- 子测试中慎用 flag.Parse()：TestMain 中不要手动解析 flag，go test 已处理
实际应用场景示例
假设你要测试一个依赖 Redis 和配置文件的服务：
func TestMain(m *testing.M) {
// 加载配置
config, err := LoadConfig("config.test.yaml")
if err != nil {
log.Fatalf("❌ 配置加载失败: %v", err)
}
// 初始化 Redis
redisClient = redis.NewClient(&redis.Options{
Addr: config.RedisAddr,
})
_, err = redisClient.Ping(context.Background()).Result()
if err != nil {
log.Fatalf("❌ Redis 连接失败: %v", err)
}
log.Println("? 测试环境准备就绪")
exitCode := m.Run()
// 清理
redisClient.Close()
log.Println("? Redis 连接已关闭")
os.Exit(exitCode)
}
这样，所有测试都能安全使用 redisClient，且资源会在最后统一释放。
基本上就这些。合理使用 testing.M 能让集成测试更简洁可靠，关键是记得收尾和正确退出。
