Go测试无内置setup/teardown,可用TestMain实现包级初始化与清理,子测试+闭包或辅助函数+defer实现函数级隔离,需注意并发安全与资源唯一性。

在 Go 的测试中,没有内置的 setup/teardown 机制(如 JUnit 的 @Before/@After),但可以通过标准测试框架和合理结构模拟出类似行为,确保每次测试前资源就绪、测试后干净释放,从而维持环境一致性。
用 TestMain 实现包级 setup/teardown
适用于需要在整组测试开始前初始化一次、结束后清理一次的场景(例如启动本地数据库、创建临时目录、设置全局 mock)。
- 定义
func TestMain(m *testing.M),调用setup()→m.Run()→teardown() - 必须显式调用
os.Exit(m.Run()),否则测试会提前退出 - 注意:
TestMain是包级的,不适用于单个测试函数的隔离需求
示例:
func TestMain(m *testing.M) {
setupDB()
defer teardownDB()
os.Exit(m.Run())
}
用子测试(subtest)+ 闭包模拟函数级 setup/teardown
对单个测试函数需要独立资源时最常用。把 setup/teardown 逻辑封装进闭包,配合 t.Run 使用,天然支持并行(需手动加锁或使用独立资源实例)。
立即学习“go语言免费学习笔记(深入)”;
- 每个 subtest 自己调用 setup(如创建临时文件、初始化内存 DB 实例)
- 用
defer在函数末尾触发 cleanup(如os.RemoveAll、关闭连接) - 避免跨 subtest 共享可变状态,防止干扰
示例:
func TestUserService(t *testing.T) {
t.Run("create user", func(t *testing.T) {
db := setupInMemoryDB() // 每次新建
defer db.Close() // 自动清理
svc := NewUserService(db)
// ... 测试逻辑
})
}
用测试辅助函数 + defer 组织可复用逻辑
将重复的初始化/清理抽象成函数,提高可读性和复用性。关键点是让 setup 返回 cleanup 函数,由调用方 defer 执行。
- setup 函数返回资源句柄 + cleanup 函数(类型为
func()) - 调用方统一用
defer cleanup(),语义清晰且不易遗漏 - 适合数据库连接、HTTP server、临时目录等常见资源
示例:
func setupTempDir(t *testing.T) (string, func()) {
dir, err := os.MkdirTemp("", "test-*")
if err != nil {
t.Fatal(err)
}
return dir, func() { os.RemoveAll(dir) }
}
func TestReadConfig(t *testing.T) {
dir, cleanup := setupTempDir(t)
defer cleanup()
// ... 写配置、调用被测函数
}
注意并发与资源隔离
Go 测试默认并发执行(go test -race 可检测竞争),若多个测试共用同一资源(如端口、文件路径、全局变量),易导致失败或非确定行为。
- 避免硬编码端口、文件名;改用
net.Listen("tcp", ":0")或os.MkdirTemp获取唯一值 - 不要在多个测试间共享可变全局状态(如
var db *sql.DB),应每个测试新建或重置 - 若必须共享(如慢启动的外部服务),加互斥锁 + once 控制,但要注明“非并行安全”










