Golang接口实现测试与mock对象示例-Golang-PHP中文网

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

P粉602998670

发布： 2025-10-13 22:58:01

原创

129人浏览过

答案：通过接口抽象依赖并使用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 接口：

立即学习“go语言免费学习笔记（深入）”；

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 实现，用来控制行为并验证调用：

白瓜面试

白瓜面试 - AI面试助手,辅助笔试面试神器

查看详情

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)
    <strong>if err != nil</strong> {
        t.Fatalf("expected no error, got %v", err)
    }
    <strong>if result != "Hello, Alice"</strong> {
        t.Errorf("expected 'Hello, Alice', got '%s'", result)
    }

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

登录后复制