当前位置：首页 > 文章列表 > Golang > Go教程 > Golang接口测试与mock实战教程

Golang接口测试与mock实战教程

2025-10-03 13:16:20 0浏览收藏

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《Golang接口测试与mock对象实战示例》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

答案：通过接口抽象依赖并使用mock对象可实现Go语言中可测试的代码设计。定义UserRepository接口及InMemoryUserRepo实现，UserService通过接口解耦业务逻辑与数据访问；测试时创建MockUserRepository模拟 GetUserByID 行为，验证 GetUserInfo 正确性，从而隔离外部依赖、覆盖正常与错误路径，提升代码可维护性和测试完整性。

Golang接口实现测试与mock对象示例

在Go语言开发中，接口和mock对象是编写可测试代码的关键。通过接口抽象依赖，我们可以轻松替换真实实现为模拟对象，从而隔离外部依赖进行单元测试。下面用一个简单示例说明如何设计接口、实现以及使用mock进行测试。

定义接口与实现

假设我们要实现一个用户服务，它依赖于数据库操作。我们先定义一个数据访问接口：

type UserRepository interface {
    GetUserByID(id int) (*User, error)
    SaveUser(user *User) error
}

type User struct {
    ID   int
    Name string
}

接着实现一个基于内存的版本用于真实运行时：

type InMemoryUserRepo struct {
    users map[int]*User
}

func NewInMemoryUserRepo() *InMemoryUserRepo {
    return &InMemoryUserRepo{users: make(map[int]*User)}
}

func (r *InMemoryUserRepo) GetUserByID(id int) (*User, error) {
    user, exists := r.users[id]
    if !exists {
        return nil, fmt.Errorf("user not found")
    }
    return user, nil
}

func (r *InMemoryUserRepo) SaveUser(user *User) error {
    r.users[user.ID] = user
    return nil
}

业务逻辑依赖接口

UserService 不关心具体的数据存储方式，只依赖 UserRepository 接口：

type UserService struct {
    repo UserRepository
}

func NewUserService(repo UserRepository) *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
}

使用Mock对象进行测试

在测试中，我们创建一个mock的 UserRepository 实现，用来控制行为并验证调用：

type MockUserRepository struct {
    mockGetUser func(id int) (*User, error)
    mockSaveUser func(user *User) error
}

func (m *MockUserRepository) GetUserByID(id int) (*User, error) {
    if m.mockGetUser != nil {
        return m.mockGetUser(id)
    }
    return nil, fmt.Errorf("not implemented")
}

func (m *MockUserRepository) SaveUser(user *User) error {
    if m.mockSaveUser != nil {
        return m.mockSaveUser(user)
    }
    return nil
}

现在可以编写测试，验证 UserService 的行为：

func TestUserService_GetUserInfo(t *testing.T) {
    mockRepo := &MockUserRepository{
        mockGetUser: func(id int) (*User, error) {
            if id == 1 {
                return &User{ID: 1, Name: "Alice"}, nil
            }
            return nil, fmt.Errorf("user not found")
        },
    }

    service := NewUserService(mockRepo)

    result, err := service.GetUserInfo(1)
    if err != nil {
        t.Fatalf("expected no error, got %v", err)
    }
    if result != "Hello, Alice" {
        t.Errorf("expected 'Hello, Alice', got '%s'", result)
    }

    _, err = service.GetUserInfo(999)
    if err == nil {
        t.Error("expected error for unknown user, got none")
    }
}