Golang使用mock实现单元测试示例-Golang-PHP中文网

Golang使用mock实现单元测试示例

P粉602998670

发布： 2025-09-26 22:33:02

原创

916人浏览过

首先定义接口并使用mockgen生成mock代码，再通过gomock在单元测试中模拟依赖行为。示例展示了UserService调用UserRepo获取用户信息时，如何用mock对象隔离数据库依赖。测试中设定期望输入输出，验证GetUserInfo方法的正确性。运行go test可快速执行测试，无需真实外部服务，提升测试效率与可靠性。关键在于接口设计、mock生成与预期设置。

golang使用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会在运行时验证是否按预期被调用。