使用Go语言gomock进行接口模拟测试指南

// sample/client.go
package sample

import (
    "fmt"
    "net/http" // 假设会用到
)

// DataFetcher 定义了获取数据的方法
type DataFetcher interface {
    FetchResponse(path, employeeID string) string
}

// HTTPDataFetcher 是 DataFetcher 接口的一个具体实现
type HTTPDataFetcher struct{}

// FetchResponse 实现了 DataFetcher 接口
func (h *HTTPDataFetcher) FetchResponse(path, employeeID string) string {
    url := fmt.Sprintf("http://example.com/%s/%s", path, employeeID)
    // 实际的网络请求逻辑，这里简化
    // resp, err := http.Get(url)
    // defer resp.Body.Close()
    // bodyBytes, _ := io.ReadAll(resp.Body)
    // return string(bodyBytes)
    fmt.Printf("Making actual HTTP request to: %s\n", url) // 模拟实际请求
    return "Actual Response from " + url
}

// Process 依赖于 DataFetcher 接口
func Process(fetcher DataFetcher, path, employeeID string) string {
    return fetcher.FetchResponse(path, employeeID)
}

go install github.com/golang/mock/mockgen@latest

# 从接口定义所在的go文件生成mock
# -source 指定包含接口定义的源文件
# -destination 指定生成的mock文件路径
# -package 指定生成的mock文件的包名
mockgen -source=sample/client.go -destination=sample/mock_datafetcher_test.go -package=sample_test

// sample/client_test.go
package sample_test

import (
    "testing"

    "github.com/golang/mock/gomock"
    "yourproject/sample" // 替换为你的实际项目路径
)

func TestProcessWithMock(t *testing.T) {
    // 1. 创建一个gomock控制器
    ctrl := gomock.NewController(t)
    defer ctrl.Finish() // 确保在测试结束时调用ctrl.Finish()来检查所有期望是否满足

    // 2. 使用mockgen生成的构造函数创建模拟对象
    mockFetcher := sample.NewMockDataFetcher(ctrl)

    // 3. 设置期望：当FetchResponse方法被调用时，返回指定的值
    // EXPECT().FetchResponse("path", "employeeID") 表示期望以这两个参数调用
    // .Return("mocked_response") 表示当调用发生时返回此值
    mockFetcher.EXPECT().FetchResponse("path", "employeeID").Return("mocked_abc_response").Times(1)

    // 4. 调用被测试的函数，传入模拟对象
    // 注意：Process函数现在接收一个DataFetcher接口
    result := sample.Process(mockFetcher, "path", "employeeID")

    // 5. 断言结果是否符合预期
    expected := "mocked_abc_response"
    if result != expected {
        t.Errorf("Expected %q, got %q", expected, result)
    }

    // 再次调用，如果之前设置了Times(1)，这里会报错
    // result2 := sample.Process(mockFetcher, "path", "employeeID")
    // fmt.Println("Second call result:", result2)
}

// 原始的TestProcess，现在需要调整以使用真实的HTTPDataFetcher或模拟
func TestProcessActual(t *testing.T) {
    // 实例化真实的DataFetcher实现
    realFetcher := &sample.HTTPDataFetcher{}
    result := sample.Process(realFetcher, "actual_path", "actual_employee")
    // 验证结果，这里只是打印，实际应有断言
    t.Logf("Actual process result: %s", result)
    // 假设我们知道实际请求会返回什么
    if result == "" { // 简单检查非空
        t.Errorf("Expected non-empty result, got empty")
    }
}