17370845950

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

把“开库 → 建表 → 插入种子数据 → 关闭”逻辑收进工具函数，让测试用例聚焦业务逻辑本身。

• 定义 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 封装——数据库测试就能既可靠又不拖慢开发节奏。