Go 语言如何编写数据库 Mock 测试？-Golang-PHP中文网

因为sql.db是不可导出字段的高度封装结构体，无法直接赋值mock实例；正确做法是通过sqlmock.new()创建符合database/sql语义的fake sql.db并注入业务代码。

go 语言如何编写数据库 mock 测试？

为什么直接 mock `*sql.DB` 会失败？

Go 标准库的 *sql.DB 是一个高度封装的结构体，它不导出内部字段，也没有实现可被接口替换的“行为抽象”。你不能直接对 *sql.DB 做类型断言或赋值 mock 实例——编译会报错：cannot assign *mockDB to db (type *sql.DB)。真正该 mock 的是它的行为载体：数据库操作所依赖的接口，比如 sql.Queryer、sql.Execer，但更推荐的方式是定义自己的仓储接口（repository interface），然后只 mock 那个接口。

用接口隔离 + `sqlmock` 模拟查询和事务

sqlmock 是目前最主流的方案，但它不是 mock *sql.DB 本身，而是 mock 它背后的 driver.Conn 行为，并通过 sqlmock.New() 返回一个符合 database/sql 语义的 fake *sql.DB。关键点在于：你必须把业务逻辑从直接调用 db.Query 解耦到接口方法中，再让测试传入 sqlmock 创建的 DB 实例。

定义仓储接口，例如：type UserRepo interface { GetUserByID(id int) (*User, error) }
实现时接收 *sql.DB 作为依赖，内部调用 db.QueryRow 等
测试中用 sqlmock.New() 创建 mock DB，再构造 mock 实现（或直接注入到被测对象）
预设期望：mock.ExpectQuery("SELECT id FROM users").WithArgs(123).WillReturnRows(rows)
执行业务逻辑后，务必调用 mock.AssertExpectations(t)，否则未触发的期望不会报错

`sqlmock` 对事务的支持很脆弱

事务在 sqlmock 中需要显式开启并跟踪状态。如果你的代码里用了 tx, _ := db.Begin()，那么 mock 必须用 mock.ExpectBegin()；后续所有查询/执行都得在 tx 上调用，且需对应 ExpectQuery 或 ExpectExec。稍有不匹配（比如忘了 ExpectCommit() 或用了 db.Query 而非 tx.Query），就会 panic 或静默失败。

使用Go语言搭建家庭相册系统-相关课件

下载

避免在测试中混合使用 db 和 tx —— 统一用事务对象操作
mock.ExpectCommit().WillReturnError(nil) 是常见写法，但若业务中可能回滚，就得补上 ExpectRollback()
注意：事务内嵌套事务（如 Savepoint）不被 sqlmock 支持，遇到就只能绕过或改用内存 SQLite