如何在 Go 中构建可测试的 Mock 类-Golang-PHP中文网

如何在 Go 中构建可测试的 Mock 类

本文介绍 go 语言中构建 mock 类的两种主流方式：基于接口抽象的手动 mock 和使用 testify/mock 的自动化 mock，重点讲解如何解耦依赖、定义契约、控制返回值，并给出完整可运行示例。

本文介绍 go 语言中构建 mock 类的两种主流方式：基于接口抽象的手动 mock 和使用 testify/mock 的自动化 mock，重点讲解如何解耦依赖、定义契约、控制返回值，并给出完整可运行示例。

在 Go 中，“Mock 类”并非像 Java 或 Python 那样通过继承或动态代理直接模拟具体类型，而是遵循 Go 的设计哲学——面向接口编程。真正的可测试性来源于提前定义清晰的接口契约，再为该接口提供真实实现与测试专用的 Mock 实现。下面我们将从原理到实践，系统说明如何构建高效、可维护的 Mock。

✅ 正确姿势：先定义接口，再实现与 Mock

原始代码的问题在于 FakeMyClass 嵌入了 MyClass 并试图覆盖方法，但 PrintInfo 在接收者 *MyClass 上调用的是 *MyClass.Object()，而非 *FakeMyClass.Object() —— 因为 PrintInfo 是 MyClass 的方法，它内部硬编码调用自身类型的方法，无法被子类（Go 中无真正继承）重写。这是典型的“紧耦合导致不可测”。

正确做法是将行为抽象为接口：

package main

import "fmt"

// Step 1: 定义接口（契约）
type Objecter interface {
    Object() int
}

// Step 2: 实现真实逻辑（依赖该接口）
type MyClass struct{}

func (*MyClass) Object() int {
    return 0
}

func (mc *MyClass) PrintInfo() {
    fmt.Printf("Real: %v\n", mc.Object()) // 调用接口方法
}

// Step 3: 编写 Mock 实现（满足同一接口）
type FakeMyClass struct{}

func (*FakeMyClass) Object() int {
    return 1 // 可控返回值
}

func main() {
    // 通过接口注入，运行时可替换实现
    var obj Objecter = &FakeMyClass{}
    fmt.Printf("Mock: %v\n", obj.Object()) // 输出 1

    // 若需复用 PrintInfo，应将其改为接受接口参数
    printInfo(obj) // 而非绑定到 *MyClass
}

func printInfo(o Objecter) {
    fmt.Printf("From interface: %v\n", o.Object())
}

✅ 这种方式轻量、无依赖、符合 Go idioms，适用于大多数单元测试场景。

? 进阶方案：使用 testify/mock 自动生成 Mock 行为

当接口方法较多、需要灵活配置调用次数、参数校验或返回动态值时，可借助 github.com/stretchr/testify/mock 自动生成 Mock 类（需配合 mockgen 工具或手写，此处以手写简化版为例）：

package main

import (
    "fmt"
    "github.com/stretchr/testify/mock"
)

type Objecter interface {
    Object() int
}

type MyClass struct{}

func (*MyClass) Object() int { return 0 }

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

// Mock 实现（嵌入 mock.Mock，按约定实现接口方法）
type MockObjecter struct {
    mock.Mock
}

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

func main() {
    mockObj := new(MockObjecter)
    mockObj.On("Object").Return(42) // 声明期望：调用 Object() 返回 42

    // 使用 Mock 实例
    printInfo(mockObj) // 输出 "From interface: 42"

    // ✅ 可验证是否按预期调用（需在测试中使用 t）
    // mockObj.AssertExpectations(t)
}

⚠️ 注意事项：

testify/mock 不替代接口抽象，它只是对接口实现的辅助工具；

mock.Mock 本身不实现任何业务接口，必须由你手动编写方法体并调用 Called()/Return()；

生产代码中不应引入 testify/mock，仅用于 _test.go 文件；

更推荐结合 mockgen（来自 gomock 或 testify）生成强类型 Mock，避免手写错误。