推荐用 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语言免费学习笔记（深入）”；
- 为每个方法返回预设值或简单逻辑（如 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 的目标不是模拟数据库语法或事务细节，而是控制输入输出，验证业务逻辑对数据层的调用是否正确。只要接口契约守住了，上层逻辑就可信。
