首先定义接口并使用mockgen生成mock代码，再通过gomock在单元测试中模拟依赖行为。示例展示了UserService调用UserRepo获取用户信息时，如何用mock对象隔离数据库依赖。测试中设定期望输入输出，验证GetUserInfo方法的正确性。运行go test可快速执行测试，无需真实外部服务，提升测试效率与可靠性。关键在于接口设计、mock生成与预期设置。
在Go语言开发中，单元测试是保证代码质量的重要手段。当被测代码依赖外部服务（如数据库、HTTP接口）时，使用mock可以隔离这些依赖，让测试更快速、稳定。下面通过一个简单示例展示如何在Golang中使用mock进行单元测试。
定义接口与实现
假设我们有一个用户服务，需要从数据库获取用户信息。先定义数据访问层接口和业务逻辑层：
// user_repo.go
package main
type UserRepo interface {
GetUserByID(id int) (*User, error)
}
type UserService struct {
repo UserRepo
}
func NewUserService(repo UserRepo) *UserService {
return &UserService{repo: repo}
}
func (s *UserService) GetUserInfo(id int) (string, error) {
user, err := s.repo.GetUserByID(id)
if err != nil {
return "", err
}
return "Hello, " + user.Name, nil
}
type User struct {
ID int
Name string
}
生成Mock代码
使用golang/mock工具生成mock实现：
- 安装mockgen：go install github.com/golang/mock/mockgen@latest
- 生成mock代码：mockgen -source=user_repo.go -destination=mock_user_repo.go -package=main
会自动生成MockUserRepo结构体，包含可编程的方法用于模拟行为。
立即学习“go语言免费学习笔记（深入）”；
编写单元测试
使用生成的mock对象来测试UserService：
// user_service_test.go
package main
import (
"testing"
"github.com/golang/mock/gomock"
"github.com/stretchr/testify/assert"
)
func TestGetUserInfo_Success(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockRepo := NewMockUserRepo(ctrl)
service := NewUserService(mockRepo)
// 设定期望调用和返回值
mockRepo.EXPECT().
GetUserByID(1).
Return(&User{ID: 1, Name: "Alice"}, nil)
result, err := service.GetUserInfo(1)
assert.NoError(t, err)
assert.Equal(t, "Hello, Alice", result)
}
func TestGetUserInfo_NotFound(t *testing.T) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
mockRepo := NewMockUserRepo(ctrl)
service := NewUserService(mockRepo)
mockRepo.EXPECT().
GetUserByID(999).
Return(nil, fmt.Errorf("user not found"))
result, err := service.GetUserInfo(999)
assert.Error(t, err)
assert.Empty(t, result)
}
通过EXPECT()设定方法调用的预期输入和输出，gomock会在运行时验证是否按预期被调用。
运行测试
执行命令：
go test -v可以看到测试通过，并且没有真实依赖数据库或其他外部服务。
基本上就这些。使用mock能有效解耦测试与外部依赖，提升测试效率和可靠性。关键是设计好接口，再利用工具生成mock，最后在测试中控制行为并验证结果。
