如何在Golang架构中使用Sqlmock模拟SQL数据库执行链路断言-Golang-PHP中文网

最常见原因是sql字符串字面量不匹配，包括占位符差异（如?与$1）、空格换行、大小写及驱动重写；需用sqlmock.anyarg()或正则宽松匹配，并确保expectquery在实际查询前注册。

如何在golang架构中使用sqlmock模拟sql数据库执行链路断言

为什么 sqlmock.ExpectQuery 不匹配你的 SQL 语句

最常见的情况是：你写的 ExpectQuery 正则表达式太严格，或没处理 Go 的 SQL 驱动自动重写。比如 db.Query("SELECT * FROM users WHERE id = ?") 在 MySQL 驱动里可能被转成 SELECT * FROM users WHERE id = ?，但 PostgreSQL 驱动会变成 SELECT * FROM users WHERE id = $1 —— 而你却用 ExpectQuery("SELECT.*users") 硬匹配原始字符串，结果断言直接失败。

实操建议：

用 sqlmock.AnyArg() 替代具体占位符，例如 mock.ExpectQuery("SELECT.*users").WithArgs(sqlmock.AnyArg())
若需校验参数值，改用 WithArgs(123) 或 WithArgs(sqlmock.AnyArg(), "admin")，不要依赖 SQL 字符串含参
避免在正则中写死换行、空格或括号；用 .* 宽松匹配，如 "INSERT\s+INTO\s+orders"（注意双反斜杠）
调用 mock.ExpectationsWereMet() 前，确保所有 query/exec 已被触发，否则报 there is a remaining expectation which was not matched

如何让 sqlmock 捕获事务中的多条语句

Sqlmock 默认不跟踪事务内部的嵌套执行，Begin() 后的 Query 或 Exec 如果没显式绑定到 mock 对象，就会报 there is a query pending。

实操建议：

必须对事务内每条语句单独设期望，例如：

tx, _ := db.Begin()
mock.ExpectQuery("SELECT.*").WillReturnRows(rows)
mock.ExpectExec("UPDATE.*").WillReturnResult(sqlmock.NewResult(1, 1))
tx.Query(...)
tx.Exec(...)
tx.Commit()

不要复用同一个 sqlmock.Sqlmock 实例跨多个测试函数；每次测试用 sqlmock.New() 重建
如果业务代码用了 db.WithContext(ctx).Query(...)，而你 mock 的是普通 db.Query，也会失配——确认被测代码实际调用的是哪个方法签名

怎样断言错误路径：比如数据库连接失败或查询超时

Sqlmock 本身不模拟网络层错误，它只控制 SQL 执行阶段的返回。想测 “查不到数据”“主键冲突”“context canceled”，得靠 WillReturnError 和预设结果集行为。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

实操建议：

模拟空结果：用 WillReturnRows(sqlmock.NewRows([]string{"id"}))（零行）
模拟查询错误：用 WillReturnError(fmt.Errorf("timeout"))，然后检查业务函数是否返回该 error
模拟约束失败（如唯一键冲突）：PostgreSQL 返回 pq.Error{Code: "23505"}，可用 WillReturnError(&pq.Error{Code: "23505"})，前提是你的代码真正在 catch pq.Error
注意：若业务函数里用了 errors.Is(err, sql.ErrNoRows)，就别传自定义 error，改用 WillReturnRows().RowError(0, sql.ErrNoRows)

sqlmock 和真实 driver 兼容性有哪些坑

Sqlmock 是基于 database/sql/driver 接口实现的 fake driver，但它不解析 SQL、不维护 schema、不支持 prepare statement 缓存——这些都会在迁移真实环境时暴露。

实操建议：

避免在测试中调用 db.Prepare()；sqlmock 对 Prepare 的支持有限，容易 panic。改用 db.Query/db.Exec 直接执行
不要依赖列名大小写；sqlmock 返回的 Rows.Columns() 默认全小写，而某些 PG 驱动返回原样字段名，会导致 struct scan 失败
时间类型慎用：mock 返回 time.Time 时，务必用 UTC 时间构造，否则和本地时区不一致引发比较失败
如果你用 GORM 或 sqlx，它们可能绕过 sqlmock 的 hook 点（比如直接调用 driver.Stmt.Exec），此时要换用其内置 mock 方式（如 gorm.io/gorm/testutils）