如何在 Go 中构建和使用 Mock 类进行单元测试-Golang-PHP中文网

如何在 Go 中构建和使用 Mock 类进行单元测试

本文介绍在 go 语言中通过组合接口与第三方 mock 框架（如 testify/mock）构建可测试的 mock 类，替代硬编码继承式伪实现，提升单元测试的可靠性与可维护性。

本文介绍在 go 语言中通过组合接口与第三方 mock 框架（如 testify/mock）构建可测试的 mock 类，替代硬编码继承式伪实现，提升单元测试的可靠性与可维护性。

在 Go 中，“Mock 类”并非像面向对象语言（如 Java 或 C#）那样通过继承或动态代理生成，而是基于接口抽象 + 依赖注入 + 第三方 mock 工具的组合实践。直接嵌入结构体（如 FakeMyClass 匿名嵌入 MyClass）看似能覆盖方法，但存在严重缺陷：PrintInfo() 调用的是 FakeMyClass.Object()，而该方法若未显式重写 *MyClass 的行为，实际仍可能调用原始逻辑（尤其当方法被内联或接收者类型不匹配时），且无法验证调用次数、参数或设置条件返回值。

✅ 正确做法是：先定义接口，再为接口实现 Mock。这是 Go 测试的最佳实践。

1. 提取接口，解耦依赖

首先将 MyClass 的行为抽象为接口：

type Objecter interface {
    Object() int
}

type MyClass struct{}

func (*MyClass) Object() int {
    return 0 // 真实实现
}

func (mc *MyClass) PrintInfo() {
    fmt.Printf("%v\n", mc.Object())
}

注意：PrintInfo 方法应依赖接口而非具体类型，以支持注入 Mock：

func PrintInfo(o Objecter) {
    fmt.Printf("%v\n", o.Object())
}

2. 使用 testify/mock 构建 Mock 实现

安装依赖：

go get github.com/stretchr/testify/mock

定义 Mock 结构体并实现接口：

type MockObjecter struct {
    mock.Mock
}

func (m *MockObjecter) Object() int {
    args := m.Called()
    return args.Int(0)
}

mock.Mock 提供了 Called() 和 Return() 等核心能力，允许你在测试中精确控制行为：

func TestPrintInfo(t *testing.T) {
    mockObj := new(MockObjecter)
    mockObj.On("Object").Return(42) // 声明：调用 Object() 时返回 42

    PrintInfo(mockObj) // 输出: 42

    mockObj.AssertExpectations(t) // 验证 Object() 被调用一次
}

3. 关键注意事项

❌ 不要试图“伪造类继承”（如 FakeMyClass 嵌入 MyClass）——Go 不支持虚函数多态，该方式无法保证方法调用链可控，且违背依赖倒置原则；
✅ 必须通过接口隔离依赖，Mock 才有意义；
✅ 在测试中始终调用 AssertExpectations(t) 或 AssertNumberOfCalls(t, "Method", n)，确保行为契约被满足；
⚠️ testify/mock 生成的 Mock 是运行时反射驱动，适用于中小型项目；超大型项目可考虑 gomock（需代码生成）或 mockery（自动生成 Mock 文件）进一步提升类型安全与 IDE 支持。