最稳方案是用testmain统一初始化测试数据库。它在所有测试前执行一次,可安全清表、插seed数据,并支持跳过;需定义在main_test.go中,调用m.run(),db应隔离新建,seed失败须panic中断。

测试前用 TestMain 跑 seed 最稳
Go 测试里没类似 Rails 的 rake db:seed 机制,硬塞在每个 TestXxx 里会重复执行、污染状态。最靠谱的是统一入口——TestMain。
它在所有测试开始前执行一次,能控制 DB 连接、清空表、插入种子数据,还能跳过 seed(比如本地调试时加 -test.run=^$)。
-
TestMain必须定义在main_test.go文件中,且包名是main - 记得调用
m.Run(),否则测试根本不会跑 - DB 清空建议用
TRUNCATE TABLE xxx RESTART IDENTITY(PostgreSQL)或DELETE FROM xxx; ALTER TABLE xxx AUTO_INCREMENT = 1(MySQL),别只用DELETE留下自增 ID 断层 - seed 数据尽量用结构体初始化,别拼 SQL 字符串,防注入也易维护
func TestMain(m *testing.M) {
db := setupTestDB()
defer db.Close()
// 清表
db.Exec("TRUNCATE TABLE users RESTART IDENTITY")
// 插入种子
db.Create(&User{Name: "alice", Email: "a@example.com"})
db.Create(&User{Name: "bob", Email: "b@example.com"})
os.Exit(m.Run())
}
用 go:embed 管理 JSON/YAML 种子文件更灵活
硬编码在 TestMain 里,seed 一多就难读难改。把数据抽成 seed/users.json,用 go:embed 加载,既保持测试纯净,又支持按环境/场景分组。
注意:嵌入的文件路径是相对于 go:embed 所在文件的,不是 go test 当前目录;且 embed.FS 不能直接读子目录通配,得列清楚。
Go 配置库,使用 spf13/viper — 分层优先级(flag > env >file > KV > default),提供 BindPFlag/BindPFlags、SetEnvPrefix + SetEnvKeyReplace 等功能。
- JSON 文件字段要和 struct tag 对齐,比如
json:"email",否则json.Unmarshal失败静默丢数据 - 嵌入后用
fs.ReadFile拿内容,别用os.ReadFile,后者在测试二进制里会报 “no such file” - 如果 seed 数据含时间戳或 UUID,建议在加载后动态生成,避免测试因时间漂移失败
//go:embed seed/users.json
var seedFS embed.FS
func loadUsers(db *gorm.DB) {
data, _ := seedFS.ReadFile("seed/users.json")
var users []User
json.Unmarshal(data, &users)
db.CreateInBatches(users, 100)
}
TestDB 不该复用生产连接池
很多人图省事,在测试里直接复用 db 全局变量,结果出现连接泄漏、事务未提交、甚至测试间数据残留。测试 DB 必须隔离。
核心原则:每个测试函数(或每个 TestMain)应新建独立 *gorm.DB 实例,用完关闭;连接池大小设小点(比如 MaxOpenConns: 5),避免压垮本地 PostgreSQL。
- 别用
gorm.Open(...)后全局赋值,而是封装成newTestDB()函数,每次调用都建新实例 - PostgreSQL 测试库建议用
CREATE DATABASE testdb WITH TEMPLATE template0初始化,避免 template1 带脏数据 - SQLite 更简单:
file::memory:?cache=shared可实现内存 DB 隔离,但不支持外键级联,慎用于复杂关系测试
seed 失败时错误必须显式 panic 或 os.Exit(1)
seed 出错但没中断测试,后续断言全在错数据上跑,失败信息完全误导人。Golang 测试框架不会自动捕获 seed 阶段的 error 并 fail 整个 suite。
常见陷阱是忽略 db.Error 或用 log.Fatal —— 前者让测试继续跑,后者不触发 go test 的失败计数。
- 所有 DB 操作后加
if err != nil { panic(err) }或if db.Error != nil { panic(db.Error) } - 不要用
t.Fatal在TestMain里,它只对当前测试函数有效,而TestMain不是测试函数 - panic 消息里带上操作上下文,比如
"failed to seed users: " + err.Error(),方便定位哪条数据挂了
go test 都干净、可重现、不依赖外部状态。特别是并发测试时,哪怕漏关一个连接、少 truncate 一张关联表,问题都会藏很久才爆出来。golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!










