Go单元测试通过接口隔离外部依赖，使用模拟对象替代数据库、网络等服务，结合依赖注入和testify/mock工具实现快速、稳定的可重复测试。
在Go语言中，单元测试的关键是隔离被测代码与外部依赖，比如数据库、网络请求或第三方服务。通过模拟这些依赖，可以确保测试快速、稳定且可重复。以下是几种常见的模拟依赖的方法和实践。
使用接口定义依赖
Go的接口机制是实现依赖模拟的基础。将外部依赖抽象为接口，便于在测试时替换为模拟实现。
例如，有一个服务需要调用数据库：
type UserRepository interface {
GetUser(id int) (*User, error)
}
type UserService struct {
repo UserRepository
}
func (s *UserService) GetUserInfo(id int) (string, error) {
user, err := s.repo.GetUser(id)
if err != nil {
return "", err
}
return "Hello " + user.Name, nil
}
测试时，可以实现一个模拟的 UserRepository：
立即学习“go语言免费学习笔记（深入）”；
type MockUserRepo struct {
users map[int]*User
}
func (m *MockUserRepo) GetUser(id int) (*User, error) {
if user, exists := m.users[id]; exists {
return user, nil
}
return nil, fmt.Errorf("user not found")
}
然后在测试中注入模拟对象：
func TestGetUserInfo(t *testing.T) {
mockRepo := &MockUserRepo{
users: map[int]*User{
1: {ID: 1, Name: "Alice"},
},
}
service := &UserService{repo: mockRepo}
result, err := service.GetUserInfo(1)
if err != nil {
t.Fatal(err)
}
if result != "Hello Alice" {
t.Errorf("expected Hello Alice, got %s", result)
}
}
使用 testify/mock 简化模拟
手动编写模拟结构体在复杂接口下会变得繁琐。testify/mock 提供了更简洁的方式来生成和管理模拟对象。
安装 testify：
go get github.com/stretchr/testify/mock
定义模拟类：
type MockUserRepository struct {
mock.Mock
}
func (m *MockUserRepository) GetUser(id int) (*User, error) {
args := m.Called(id)
return args.Get(0).(*User), args.Error(1)
}
测试中设置期望行为：
func TestGetUserInfoWithTestify(t *testing.T) {
mockRepo := new(MockUserRepository)
service := &UserService{repo: mockRepo}
expectedUser := &User{ID: 1, Name: "Bob"}
mockRepo.On("GetUser", 1).Return(expectedUser, nil)
result, err := service.GetUserInfo(1)
assert.NoError(t, err)
assert.Equal(t, "Hello Bob", result)
mockRepo.AssertExpectations(t)
}
这种方式能验证方法是否被调用、参数是否正确，适合复杂的交互场景。
依赖注入提升可测试性
为了方便替换依赖，建议使用依赖注入（DI），而不是在代码内部直接实例化具体类型。
错误做法：
func NewUserService() *UserService {
return &UserService{
repo: &RealUserRepo{}, // 硬编码依赖
}
}
正确做法：
func NewUserService(repo UserRepository) *UserService {
return &UserService{repo: repo}
}
这样在测试中可以自由传入模拟对象，生产代码则传入真实实现。
模拟HTTP客户端或其他外部服务
当依赖外部API时，可以使用 httptest 包启动一个临时HTTP服务器来模拟响应。
func TestExternalAPICall(t *testing.T) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
fmt.Fprintln(w, `{"name": "mocked user"}`)
}))
defer ts.Close()
client := &http.Client{}
resp, err := client.Get(ts.URL)
// 解析响应并断言结果
}
也可以封装HTTP调用为接口，便于模拟。
基本上就这些。核心思路是：用接口解耦、用模拟实现替代真实依赖、通过依赖注入传递。这样写的测试不依赖环境，运行快，也更容易维护。
