Go语言测试之模拟（mock）依赖（手把手教你用Mock进行Go单元测试）

为什么需要 Mock？

假设你有一个函数，它会调用数据库来获取用户信息：

// user.gopackage maintype User struct {    ID   int    Name string}type UserRepository interface {    GetUserByID(id int) (*User, error)}func GetUserDetails(repo UserRepository, id int) (string, error) {    user, err := repo.GetUserByID(id)    if err != nil {        return "", err    }    return fmt.Sprintf("User: %s (ID: %d)", user.Name, user.ID), nil}

如果你直接在测试中使用真实的数据库，会导致测试变慢、不可靠（比如网络问题），甚至污染生产数据。这时候，我们就需要通过 Go mock依赖 的方式，创建一个假的 UserRepository 实现。

方法一：手动编写 Mock

最简单的方式是自己写一个 mock 结构体：

// user_test.gopackage mainimport (    "errors"    "testing")// MockUserRepository 是 UserRepository 的 mock 实现type MockUserRepository struct {    ReturnUser *User    ReturnErr  error}func (m *MockUserRepository) GetUserByID(id int) (*User, error) {    return m.ReturnUser, m.ReturnErr}func TestGetUserDetails(t *testing.T) {    // 测试正常情况    mockRepo := &MockUserRepository{        ReturnUser: &User{ID: 1, Name: "Alice"},        ReturnErr:  nil,    }    result, err := GetUserDetails(mockRepo, 1)    if err != nil {        t.Fatalf("Expected no error, got %v", err)    }    expected := "User: Alice (ID: 1)"    if result != expected {        t.Errorf("Expected %s, got %s", expected, result)    }    // 测试错误情况    mockRepoError := &MockUserRepository{        ReturnUser: nil,        ReturnErr:  errors.New("user not found"),    }    _, err = GetUserDetails(mockRepoError, 999)    if err == nil {        t.Fatal("Expected an error, got none")    }}

这种方式简单直观，适合小型项目或初学者理解 Go依赖注入测试 的核心思想。

方法二：使用 testify/mock 库（推荐）

对于复杂项目，我们可以使用第三方库如 github.com/stretchr/testify/mock 来自动生成 mock。

首先安装：

go get github.com/stretchr/testify/mock

然后定义 mock 类型：

// user_mock.gopackage mainimport (    "github.com/stretchr/testify/mock")type MockUserRepository struct {    mock.Mock}func (m *MockUserRepository) GetUserByID(id int) (*User, error) {    args := m.Called(id)    return args.Get(0).(*User), args.Error(1)}

编写测试：

func TestGetUserDetailsWithTestify(t *testing.T) {    mockRepo := new(MockUserRepository)    // 设置期望调用    mockRepo.On("GetUserByID", 1).Return(&User{ID: 1, Name: "Bob"}, nil)    result, err := GetUserDetails(mockRepo, 1)    if err != nil {        t.Fatalf("Unexpected error: %v", err)    }    expected := "User: Bob (ID: 1)"    if result != expected {        t.Errorf("Expected %s, got %s", expected, result)    }    // 验证所有期望都被调用    mockRepo.AssertExpectations(t)}

总结

通过本文，你已经学会了在 Go语言测试 中如何使用 mock 来隔离外部依赖。无论是手动 mock 还是使用 testify，核心思想都是：用可控的假对象替代真实依赖，使测试更快速、稳定、可维护。

记住，良好的 Go单元测试模拟 实践不仅能提升代码质量，还能让你在重构时更有信心。赶快在你的项目中试试吧！

关键词回顾：Go语言测试、Go mock依赖、Go单元测试模拟、Go依赖注入测试