使用TestMain可实现测试前setup和测试后teardown，通过m.Run()控制测试流程，并用defer或封装工具函数管理资源，确保初始化与清理操作正确执行。
在Go语言中进行测试时，有时需要在多个测试函数运行前执行一些初始化操作（setup），并在所有测试完成后进行清理（teardown）。虽然Go的testing包没有像其他语言那样直接提供setUp和tearDown方法，但可以通过一些模式来实现类似功能。
使用TestMain控制测试流程
最常用且推荐的方式是使用TestMain函数。它允许你自定义测试的入口点，从而在测试开始前和结束后执行特定逻辑。
例如，你想在测试前启动数据库连接、加载配置，测试结束后关闭资源：
func TestMain(m *testing.M) {// Setup: 测试前准备
setup()
// 运行所有测试用例
code := m.Run()
// Teardown: 测试后清理
teardown()
// 退出并返回测试结果状态码
os.Exit(code)
}
func setup() {
fmt.Println("执行 setup...")
// 初始化数据库、配置等
}
func teardown() {
fmt.Println("执行 teardown...")
// 关闭连接、删除临时文件等
}
只要在同一个包下定义TestMain，它就会接管整个测试流程。注意要调用m.Run()来触发实际的测试函数，并将返回值传给os.Exit。
立即学习“go语言免费学习笔记（深入）”；
为子测试使用defer实现局部teardown
如果只是想对某个测试函数内的多个子测试进行setup/teardown，可以在主测试函数中先做setup，然后用defer执行清理。
比如测试一个服务的多个接口场景：
func TestService(t *testing.T) {
// Setup
svc := NewTestService()
svc.Start()
// Teardown 使用 defer
defer func() {
svc.Stop()
cleanupTestData()
}()
t.Run("case 1", func(t *testing.T) {
// 测试逻辑
})
t.Run("case 2", func(t *testing.T) {
// 另一个测试逻辑
})
}
这种方式适合单个测试文件或功能模块内部的资源管理，简洁明了。
共享setup逻辑的封装建议
当多个测试包都需要相同的初始化流程时，可以将setup和teardown逻辑封装成公共函数，甚至放在一个专门的测试工具包中。
例如创建testutil包：
func SetupTestDB() *sql.DB {
// 创建测试数据库连接
}
func ResetTestDB(db *sql.DB) {
// 清空表数据
}
在具体测试中引入并使用：
func TestUserRepository(t *testing.T) {db := testutil.SetupTestDB()
defer db.Close()
repo := NewUserRepo(db)
defer testutil.ResetTestDB(db) // 或其他清理
// 执行测试
}
这样既保持了代码复用性，又避免重复编写setup逻辑。
基本上就这些。Go的设计哲学倾向于简单直接，通过TestMain加defer已经能覆盖大多数测试生命周期管理需求。关键是确保资源正确释放，不影响其他测试或本地环境。
