0

0

如何使用Golang模拟数据库操作_Golang sql mock测试示例

P粉602998670

P粉602998670

发布时间:2026-01-08 14:46:58

|

621人浏览过

|

来源于php中文网

原创

sqlmock 只拦截 sql.DB 的公开方法,不拦截 sql.Tx,因其是独立类型且方法不经过 mock 的 *sql.DB 实例;事务测试需用 ExpectBegin/ExpectCommit 显式声明生命周期。

如何使用golang模拟数据库操作_golang sql mock测试示例

为什么 sqlmock 能拦截 *sql.DB 调用但不能拦截 *sql.Tx?

因为 sqlmock 只对 *sql.DB 的公开方法(如 QueryExecPrepare)做代理拦截,而 *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 无法覆盖

如何写一个能通过 sqlmock 的 INSERT 测试?

关键不是“怎么插”,而是“怎么让 mock 认出你要插哪条语句”。sqlmock 默认严格匹配 SQL 字符串(含空格、换行、参数占位符),所以拼接 SQL 或用 fmt.Sprintf 极易失败。

  • 使用 sqlmock.AnyArg() 替代具体值,避免因参数格式(如 time.Time 格式、float64 精度)导致匹配失败
  • SQL 中统一用 ? 占位符(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)
}

QueryRow 返回 nil error 但 scan 失败,mock 怎么模拟?

常见错误是只 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 对应 *intstring 对应 *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)
}

测试中复用 mock 实例导致 ExpectationsWereMet 失败?

每个 sqlmock.Sqlmock 实例维护自己的期望队列。如果多个测试函数共用同一个 mock(比如定义在包级变量里),前一个测试没清空期望,后一个测试的 ExpectationsWereMet() 就会失败。

百度文心百中
百度文心百中

百度大模型语义搜索体验中心

下载

立即学习go语言免费学习笔记(深入)”;

  • 每个测试函数内调用 sqlmock.New() 创建新 mock 和 *sql.DB
  • 不要在 TestMaininit() 里初始化 mock
  • 如果需要共享 setup 逻辑,封装成函数并返回新 *sql.DBsqlmock.Sqlmock,由每个测试自己接收和 cleanup
  • 务必在测试末尾调用 mock.ExpectationsWereMet(),它会检查是否所有 Expect* 都被触发且无多余调用

最容易被忽略的是:mock 不会自动清理未使用的 Expect —— 比如你写了 ExpectQuery 但实际代码根本没执行那条查询,ExpectationsWereMet 就会报错。这其实是好事,说明测试覆盖不全或逻辑分支没走到。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

177

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

336

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

208

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

194

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

189

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

191

2025.06.17

Golang 分布式缓存与高可用架构
Golang 分布式缓存与高可用架构

本专题系统讲解 Golang 在分布式缓存与高可用系统中的应用,涵盖缓存设计原理、Redis/Etcd集成、数据一致性与过期策略、分布式锁、缓存穿透/雪崩/击穿解决方案,以及高可用架构设计。通过实战案例,帮助开发者掌握 如何使用 Go 构建稳定、高性能的分布式缓存系统,提升大型系统的响应速度与可靠性。

53

2026.01.09

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
MySQL 教程
MySQL 教程

共48课时 | 1.7万人学习

MySQL 初学入门(mosh老师)
MySQL 初学入门(mosh老师)

共3课时 | 0.3万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 785人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号