推荐用 mock 数据访问层接口而非数据库驱动,定义如 UserRepo 接口,手动编写轻量 mock 或用 testify/mock、gomock 自动生成,结合 test helper 封装常用场景,专注验证业务逻辑对数据层的调用正确性。

在 Go 中做数据库测试,推荐用 mock 方式隔离外部依赖,避免真实数据库连接带来的慢、不稳定和环境依赖问题。核心思路是:不 mock 数据库驱动本身(比如 database/sql),而是 mock 你的数据访问层(如 repository 接口),让测试只关注逻辑,不碰真实 DB。
定义清晰的 repository 接口
先把你操作数据库的代码抽象成接口,这是 mock 的前提。例如:
type UserRepo interface {
CreateUser(ctx context.Context, u *User) error
GetUserByID(ctx context.Context, id int64) (*User, error)
DeleteUser(ctx context.Context, id int64) error
}
真实实现(如基于 sql.DB)和 mock 实现都实现这个接口,测试时注入 mock 版本即可。
手动编写轻量 mock(适合简单场景)
不用第三方库也能快速 mock,尤其当方法不多、行为固定时:
立即学习“go语言免费学习笔记(深入)”;
拍客竞拍系统是一款免费竞拍网站建设软件,任何个人可以下载使用,但未经商业授权不能进行商业活动,程序源代码开源,任何个人和企业可以进行二次开发,但不能以出售和盈利为目的。安装方法,将www文件夹里面的所有文件上传至虚拟主机,在浏览器执行http://你的域名/install.php或者直接导入数据库文件执行。本次升级优化了一下内容1,程序和模板完美分离。2,优化了安装文件。3,后台增加模板切换功能。
- 为每个方法返回预设值或简单逻辑(如 ID 匹配就返回对应用户)
- 用字段记录调用次数或参数,便于断言是否被正确调用
- 示例:
type MockUserRepo struct {
Users map[int64]*User
CalledWithID int64 // 记录最后一次传入的 ID,用于验证
}
func (m *MockUserRepo) GetUserByID(ctx context.Context, id int64) (*User, error) {
m.CalledWithID = id
if u, ok := m.Users[id]; ok {
return u, nil
}
return nil, sql.ErrNoRows
}
用 testify/mock 或 gomock 自动生成 mock(适合中大型项目)
当接口方法多、需要复杂行为(如按调用顺序返回不同结果、校验参数结构)时,推荐工具辅助:
- testify/mock:轻量,基于结构体 + 手动注册期望,学习成本低
- gomock:官方推荐,生成类型安全 mock,支持精确参数匹配、多次调用控制、延迟返回等
- 使用 gomock 示例流程:
1. 安装:go install github.com/golang/mock/mockgen@latest
2. 生成 mock:mockgen -source=user_repo.go -destination=mocks/mock_user_repo.go
3. 在测试中设置期望:mockRepo.EXPECT().GetUserByID(gomock.Any(), int64(123)).Return(&User{ID: 123}, nil)
结合 test helper 封装常用 mock 场景
把高频 mock 模式(如“查不到用户”、“插入失败”、“并发调用”)封装成函数,提升测试可读性与复用性:
- 写一个
newMockRepoWithUser(id int64, u *User)快速构造带数据的 mock - 写一个
newMockRepoWithError(err error)统一返回错误,用于异常路径测试 - 避免每个测试里重复写相同 mock 配置
不复杂但容易忽略:mock 的目标不是模拟数据库语法或事务细节,而是控制输入输出,验证业务逻辑对数据层的调用是否正确。只要接口契约守住了,上层逻辑就可信。









