

新闻资讯
技术学院sqlmock 只拦截 sql.DB 的公开方法,不拦截 sql.Tx,因其是独立类型且方法不经过 mock 的 *sql.DB 实例;事务测试需用 ExpectBegin/ExpectCommit 显式声明生命周期。
因为 sqlmock 只对 *sql.DB 的公开方法(如 Query、Exec、Prepare)做代理拦截,而 *sql.Tx 是独立类型,其方法调用不经过 sqlmock 注入的 *sql.DB 实例。如果你在事务里直接调用 tx.Query,mock 就不会生效,测试会 panic 或报错 sql: Tx.Query: driver does not support tx.Query。
mock.ExpectQuery() / mock.ExpectExec() 等预设期望,并确保所有 DB 操作都走你传入的 *sql.DB 实例(不是 tx)*sql.Tx 并用 mock.ExpectBegin() + mock.ExpectCommit() 显式声明事务生命周期db.Begin() 后返回的 *sql.Tx 本身不被 mock,但后续对 tx.Stmt().Query/Exec 的调用仍会回退到 underlying *sql.DB —— 这是 Go 标准库行为,sqlmock 无法覆盖关键不是“怎么插”,而是“怎么让 mock 认出你要插哪条语句”。sqlmock 默认严格匹配 SQL 字符串(含空格、换行、参数占位符),所以拼接 SQL 或用 fmt.Sprintf 极易失败。
sqlmock.AnyArg() 替代具体值,避免因参数格式(如 time.Time 格式、float64 精度)导致匹配失败? 占位符(MySQL/SQLite 风格),不要混用 $1(PostgreSQL 风格),除非你显式配置了 sqlmock.WithDSN("postgres://...")
mock.ExpectExec("INSERT INTO users").WithArgs(...) 时,WithArgs 的参数顺序必须和 db.Exec 实际传入顺序完全一致mock.ExpectExec("INSERT INTO users\\(name, email\\)").WithArgs("alice", "a@example.com").WillReturnResult(sqlmock.NewResult(1, 1))
_, err := db.Exec("INSERT INTO users(name, email) VALUES (?, ?)", "alice", "a@example.com")
if err != nil {
t.
Fatal(err)
}
if err := mock.ExpectationsWereMet(); err != nil {
t.Error(err)
}
常见错误是只 mock ExpectQuery,却忘了设置返回的 Rows 行为。如果 rows.Scan() 报 sql: no rows in result set,说明 mock 没提供任何数据行;如果报 sql: Scan error on column index 0,说明列数或类型不匹配。
sqlmock.NewRows([]string{"id", "name"}).AddRow(123, "bob") 明确声明列名和一行数据QueryRow 中 SQL 的 SELECT 字段顺序、大小写完全一致(mock 不做字段映射)Scan(&id, &name),那么 AddRow() 的参数类型必须可赋值给对应变量(如 int64 对应 *int,string 对应 *string)rows := sqlmock.NewRows([]string{"id", "name"}).AddRow(456, "carol")
mock.ExpectQuery("SELECT id, name FROM users WHERE id = ?").WithArgs(456).WillReturnRows(rows)
var id int
var name string
err := db.QueryRow("SELECT id, name FROM users WHERE id = ?", 456).Scan(&id, &name)
if err != nil {
t.Fatal(err)
}
每个 sqlmock.Sqlmock 实例维护自己的期望队列。如果多个测试函数共用同一个 mock(比如定义在包级变量里),前一个测试没清空期望,后一个测试的 ExpectationsWereMet() 就会失败。
sqlmock.New() 创建新 mock 和 *sql.DB
TestMain 或 init() 里初始化 mock*sql.DB 和 sqlmock.Sqlmock,由每个测试自己接收和 cleanupmock.ExpectationsWereMet(),它会检查是否所有 Expect* 都被触发且无多余调用最容易被忽略的是:mock 不会自动清理未使用的 Expect —— 比如你写了 ExpectQuery 但实际代码根本没执行那条查询,ExpectationsWereMet 就会报错。这其实是好事,说明测试覆盖不全或逻辑分支没走到。