TestMain函数可用于实现测试前后钩子，先执行Setup初始化，再通过m.Run()运行所有测试，最后调用Teardown清理资源并返回退出码。
在Golang中，测试前后钩子可以通过 TestMain 函数来实现。它允许你在所有测试开始前和结束后执行自定义逻辑，比如初始化数据库、设置环境变量、清理资源等。
使用 TestMain 实现全局前后钩子
Go 的 testing 包提供了 TestMain 函数，它是测试的入口点。通过手动调用 m.Run()，你可以控制测试执行前后的行为。
示例代码：
package main
import (
"log"
"os"
"testing"
)
func Setup() {
log.Println("执行测试前准备...")
// 例如：连接数据库、创建临时文件
}
func Teardown() {
log.Println("执行测试后清理...")
// 例如：关闭数据库连接、删除临时文件
}
func TestMain(m *testing.M) {
Setup()
exitCode := m.Run()
Teardown()
os.Exit(exitCode)
}
在这个例子中：
- Setup() 在所有测试运行前执行一次
- m.Run() 执行所有测试函数
- Teardown() 在所有测试完成后执行
- 最后通过 os.Exit() 返回正确的退出码
为单个测试函数添加前后逻辑
如果只想对某个测试函数加钩子，可以直接在测试函数内部手动调用。
立即学习“go语言免费学习笔记（深入）”；
func TestWithHooks(t *testing.T) {
log.Println("测试开始前")
// 测试逻辑
t.Run("子测试1", func(t *testing.T) {
if 1+1 != 2 {
t.Fail()
}
})
log.Println("测试结束后")
}
这种方式适合局部资源管理，比如打开/关闭文件、mock 配置切换等。
结合子测试使用 defer 实现后置清理
利用 defer 可以确保清理逻辑在测试或子测试结束时执行。
func TestWithDefer(t *testing.T) {
t.Log("设置测试环境")
defer func() {
t.Log("清理测试环境")
}()
t.Run("case 1", func(t *testing.T) {
t.Log("运行测试用例 1")
})
}
这种模式常见于需要释放锁、关闭连接或恢复全局状态的场景。
基本上就这些。TestMain 用于包级初始化和销毁，函数内 defer 和手动调用适合更细粒度控制。不复杂但容易忽略的是记得调用 m.Run() 并传递退出码。
