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

package sample

import (
    "fmt"
    "net/http" // 假设会用到，但此处省略了实际的网络请求逻辑
)

// GetResponse 模拟一个从外部服务获取响应的函数
func GetResponse(path, employeeID string) string {
    url := fmt.Sprintf("http://example.com/%s/%s", path, employeeID)
    // 实际的网络请求和响应处理逻辑会在这里
    // 为了示例简化，直接返回一个硬编码值
    _ = url // 避免编译警告
    return "real_response_from_GetResponse"
}

// Process 业务逻辑函数，依赖 GetResponse
func Process(path, employeeID string) string {
    return GetResponse(path, employeeID)
}

package sample_test

import (
    "fmt"
    "testing"

    "github.com/golang/mock/gomock"
    "github.com/example/sample" // 假设这是你的项目路径
)

func TestProcess_WrongMockAttempt(t *testing.T) {
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()

    // 尝试直接在包上调用 MOCK() 和 EXPECT()
    // sample.MOCK()SetController(ctrl) // 错误：undefined: sample.MOCK
    // sample.EXPECT().GetResponse("path", "employeeID").Return("abc") // 错误：undefined: sample.EXPECT

    // v := sample.GetResponse("path", "employeeID")
    // fmt.Println(v)
    t.Errorf("此测试旨在演示错误，不会实际运行")
}

package sample

import (
    "fmt"
    // "net/http" // 如果GetResponseImpl需要，可以保留
)

// ResponseFetcher 定义了一个获取响应的接口
type ResponseFetcher interface {
    Get(path, employeeID string) string
}

// GetResponseImpl 是 ResponseFetcher 接口的一个具体实现
type GetResponseImpl struct{}

// Get 实现了 ResponseFetcher 接口的 Get 方法
func (g *GetResponseImpl) Get(path, employeeID string) string {
    url := fmt.Sprintf("http://example.com/%s/%s", path, employeeID)
    // 实际的网络请求和响应处理逻辑会在这里
    // 为了示例简化，直接返回一个硬编码值
    _ = url // 避免编译警告
    return "real_response_from_GetResponseImpl"
}

// Process 业务逻辑函数，现在依赖于 ResponseFetcher 接口
// 而不是具体的 GetResponseImpl 实现
func Process(fetcher ResponseFetcher, path, employeeID string) string {
    return fetcher.Get(path, employeeID)
}

// NewResponseFetcher 返回 ResponseFetcher 的默认实现
func NewResponseFetcher() ResponseFetcher {
    return &GetResponseImpl{}
}

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

# 假设你的模块路径是 github.com/example/sample
# 在项目根目录执行
mockgen -source=sample/sample.go -destination=sample/mock_sample.go -package=sample

package sample_test

import (
    "testing"

    "github.com/golang/mock/gomock"
    "github.com/stretchr/testify/assert" // 使用 testify 库进行断言
    "github.com/example/sample"          // 导入你的业务逻辑包
)

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

    // 2. 使用 mockgen 生成的函数创建 Mock 对象
    // 注意：NewMockResponseFetcher 是由 mockgen 在 mock_sample.go 中生成的
    mockFetcher := sample.NewMockResponseFetcher(ctrl)

    // 3. 设置预期行为
    // 当 mockFetcher 的 Get 方法被调用时，期望参数是 "path" 和 "employeeID"，
    // 并返回 "mocked_response"
    mockFetcher.EXPECT().Get("path", "employeeID").Return("mocked_response").Times(1)

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

    // 5. 断言结果
    assert.Equal(t, "mocked_response", result, "Process函数应该返回模拟的响应")
}

// 示例：测试 Process 函数的实际实现 (不使用 mock)
func TestProcess_RealImpl(t *testing.T) {
    fetcher := sample.NewResponseFetcher() // 使用真实的实现
    result := sample.Process(fetcher, "some_path", "some_id")
    assert.Equal(t, "real_response_from_GetResponseImpl", result, "Process函数应该返回真实实现的响应")
}