17370845950

推荐用 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，尤其当方法不多、行为固定时：

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