Go语言中的测试替身实战指南（深入理解Stub与Mock在Go测试中的应用）

什么是测试替身？

测试替身是在测试过程中用来替代真实依赖的对象。常见的类型包括：

• Stub（桩）：提供预设的返回值，不关心调用细节，仅用于让被测代码正常运行。
• Mock（模拟对象）：不仅提供返回值，还会验证方法是否被正确调用（如调用次数、参数等）。

在 Go语言测试 中，由于 Go 没有内置 Mock 框架（不像 Java 或 Python），我们通常通过接口（interface）和自定义实现来构建 Stub 或 Mock。

实战：使用 Stub 替代真实 HTTP 客户端

假设我们有一个函数 GetUserInfo，它通过 HTTP 调用外部 API 获取用户信息：

// user.gopackage mainimport (    "encoding/json"    "net/http")type User struct {    ID   int    `json:"id"`    Name string `json:"name"`}type HTTPClient interface {    Get(url string) (*http.Response, error)}func GetUserInfo(client HTTPClient, userID int) (*User, error) {    resp, err := client.Get(fmt.Sprintf("https://api.example.com/users/%d", userID))    if err != nil {        return nil, err    }    defer resp.Body.Close()    var user User    if err := json.NewDecoder(resp.Body).Decode(&user); err != nil {        return nil, err    }    return &user, nil}

注意：我们将 HTTP 客户端抽象为 HTTPClient 接口，这是实现 测试替身 的关键！

编写 Stub 测试

现在我们创建一个 Stub 来模拟 HTTP 响应：

// user_test.gopackage mainimport (    "bytes"    "io"    "net/http"    "testing")// Stub 实现 HTTPClient 接口type StubHTTPClient struct {    ResponseBody string    StatusCode   int}func (s *StubHTTPClient) Get(url string) (*http.Response, error) {    return &http.Response{        StatusCode: s.StatusCode,        Body:       io.NopCloser(bytes.NewBufferString(s.ResponseBody)),    }, nil}func TestGetUserInfo(t *testing.T) {    stub := &StubHTTPClient{        ResponseBody: `{"id": 123, "name": "Alice"}`,        StatusCode:   200,    }    user, err := GetUserInfo(stub, 123)    if err != nil {        t.Fatalf("expected no error, got %v", err)    }    if user.ID != 123 || user.Name != "Alice" {        t.Errorf("expected user Alice with ID 123, got %+v", user)    }}

这个 Stub 返回固定的 JSON 字符串，完全绕过了网络请求，使测试快速且可重复。

进阶：使用 Mock 验证调用行为

如果我们还想验证 Get 方法是否被正确调用（比如 URL 是否包含正确的用户 ID），就需要 Mock。我们可以扩展 Stub 添加记录功能：

type MockHTTPClient struct {    CalledURL    string    ResponseBody string}func (m *MockHTTPClient) Get(url string) (*http.Response, error) {    m.CalledURL = url // 记录被调用的 URL    return &http.Response{        StatusCode: 200,        Body:       io.NopCloser(bytes.NewBufferString(m.ResponseBody)),    }, nil}func TestGetUserInfo_WithMock(t *testing.T) {    mock := &MockHTTPClient{        ResponseBody: `{"id": 456, "name": "Bob"}`,    }    _, err := GetUserInfo(mock, 456)    if err != nil {        t.Fatalf("unexpected error: %v", err)    }    expectedURL := "https://api.example.com/users/456"    if mock.CalledURL != expectedURL {        t.Errorf("expected URL %s, got %s", expectedURL, mock.CalledURL)    }}

总结

通过合理使用 Stub 和 Mock，我们可以写出高效、可靠、可维护的 Go语言测试。关键在于：

1. 将外部依赖抽象为接口；
2. 在测试中提供接口的简单实现（Stub）或带验证逻辑的实现（Mock）；
3. 避免在单元测试中调用真实网络、数据库等外部资源。

掌握 测试替身 技巧，是成为 Go 语言高级开发者的重要一步。赶快在你的项目中试试吧！