Go数据库测试应优先用内存SQLite或testcontainers启动真实容器,确保隔离性与可重复性;避免mock驱动,通过真实SQL执行验证逻辑,辅以封装好的TestDB工具函数统一管理setup/teardown。

在 Go 中测试数据库操作,核心是隔离性和可重复性:不依赖真实生产库,每次测试运行前能快速准备干净数据,运行后不留痕迹。常用方法不是 mock 数据库驱动,而是用内存数据库(如 SQLite)或临时 PostgreSQL/MySQL 实例,配合 testcontainers 或本地轻量实例。
用 SQLite 做内存数据库测试
SQLite 支持 :memory: 模式,每次打开都是全新、独立的内存数据库,启动快、无副作用,特别适合单元测试。
- 将数据库初始化逻辑抽成函数,接受
*sql.DB参数,便于注入不同实例 - 测试时用
sql.Open("sqlite3", ":memory:")创建连接 - 立即执行建表语句(如
CREATE TABLE users(...)),确保结构就绪 - 注意:SQLite 默认不支持外键约束,需显式开启:
_, _ = db.Exec("PRAGMA foreign_keys = ON")
用 Testcontainers 启动真实数据库容器
当业务逻辑强依赖 PostgreSQL/MySQL 特有行为(如 JSONB、窗口函数、事务隔离级别),内存 SQLite 不够用,可用 testcontainers-go 在测试中自动拉起临时容器。
- 每个测试函数可启动独立容器,端口随机分配,避免冲突
- 用
container.GetConnectionURI()获取连接串,传给你的 DB 初始化函数 - 测试结束调用
container.Terminate(ctx)自动清理,不残留进程或数据 - 首次运行会下载镜像,后续复用,CI 环境也稳定(需 Docker 守护进程)
封装测试辅助函数,避免重复 setup/teardown
把“开库 → 建表 → 插入种子数据 → 关闭”逻辑收进工具函数,让测试用例聚焦业务逻辑本身。
立即学习“go语言免费学习笔记(深入)”;
- 定义
TestDB(t *testing.T) (*sql.DB, func()):返回 DB 和 cleanup 函数 - 内部自动处理 panic 捕获、错误 t.Fatal,cleanup 负责 Close + 删除临时文件(SQLite)或终止容器
- 测试函数中 defer 调用 cleanup,保证无论成功失败都释放资源
- 示例:
db, cleanup := TestDB(t); defer cleanup()
避免直接 mock sql.DB 或 driver
mock *sql.DB 或底层 driver 容易掩盖 SQL 错误、事务行为偏差、类型转换问题,导致测试通过但线上出错。
- 不要 mock
db.QueryRow返回固定值;应让真实 SQL 执行,验证结果是否符合预期 - 如果真要抽象数据访问层,可定义 interface(如
UserRepo),但实现仍基于真实 DB 连接 - 只对纯外部依赖(如 HTTP 调用、发邮件)做 mock,数据库不是“外部服务”,它是你代码的协作方
基本上就这些。关键不是“能不能测”,而是“测得像线上”。用内存 SQLite 快速验证逻辑,用 testcontainers 覆盖特性差异,再配上清晰的 setup/cleanup 封装——数据库测试就能既可靠又不拖慢开发节奏。










