Go语言中Mock结构体未导出字段失败的测试问题

问题场景 #

要测试的函数逻辑为调用sdk.Request的Send()方法，包装错误后返回结果：

func sendRequest(request *sdk.Request) (*sdk.Response, error) {
    response, err := request.Send()
    if err != nil {
        return nil, fmt.Errorf("cannot send request: %q", err)
    }
    return response, nil
}

但sdk.Request和sdk.Response包含未导出字段，手动初始化和Mock时遇到两个问题：

1. 直接声明sdk.Request结构体时，编译报错：cannot refer to unexported field transport in struct literal of type sdk.Request
2. 尝试用monkey补丁MockSend()方法失败，实际执行原方法后因transport字段为nil触发空指针panic：

panic: runtime error: invalid memory address or nil pointer dereference

解决方案 #

方案1：修复Monkey补丁，正确MockSend方法 #

问题核心是补丁时机错误，且手动初始化Request的方式不当。正确步骤：

1. 先补丁Send方法，再创建Request实例
2. 用反射创建Request实例，避免直接初始化未导出字段
3. 用反射设置Mock Response的未导出字段

import (
    "testing"
    "reflect"
    "github.com/stretchr/testify/assert"
    "github.com/bouk/monkey"
    sdk "github.com/openshift-online/ocm-sdk-go"
)

// 用反射创建Mock Response
func createMockResponse(status int) *sdk.Response {
    resp := &sdk.Response{}
    respVal := reflect.ValueOf(resp).Elem()
    if statusField := respVal.FieldByName("status"); statusField.IsValid() && statusField.CanSet() {
        statusField.SetInt(int64(status))
    }
    return resp
}

func TestSendRequest_Success(t *testing.T) {
    defer monkey.UnpatchAll()

    // 先补丁Send方法
    requestType := reflect.TypeOf(&sdk.Request{})
    monkey.PatchInstanceMethod(requestType, "Send", func(r *sdk.Request) (*sdk.Response, error) {
        return createMockResponse(200), nil
    })

    // 用反射创建Request实例，避免手动初始化的问题
    request := reflect.New(requestType.Elem()).Interface().(*sdk.Request)

    // 调用被测函数并断言
    response, err := sendRequest(request)
    assert.NoError(t, err)
    assert.Equal(t, 200, response.Status())
}

方案2：Mock底层http.RoundTripper（更推荐） #

既然Send()最终依赖transport.RoundTrip发送请求，直接Mock这个依赖更符合Go的测试设计，无需 monkey 补丁：

import (
    "testing"
    "net/http"
    "io"
    "strings"
    "reflect"
    "github.com/stretchr/testify/assert"
    sdk "github.com/openshift-online/ocm-sdk-go"
)

// 自定义Mock Transport
type mockTransport struct {
    mockResp *http.Response
    mockErr  error
}

func (m *mockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    return m.mockResp, m.mockErr
}

func TestSendRequest_Success(t *testing.T) {
    // 创建Mock HTTP响应
    mockHTTPResp := &http.Response{
        StatusCode: 200,
        Header:     make(http.Header),
        Body:       io.NopCloser(strings.NewReader("success")),
    }

    // 创建Request实例，用反射设置transport字段
    request := &sdk.Request{}
    reqVal := reflect.ValueOf(request).Elem()
    if transportField := reqVal.FieldByName("transport"); transportField.IsValid() && transportField.CanSet() {
        transportField.Set(reflect.ValueOf(&mockTransport{mockResp: mockHTTPResp}))
    }

    // 调用被测函数并断言
    response, err := sendRequest(request)
    assert.NoError(t, err)
    assert.Equal(t, 200, response.Status())
}

关键注意事项 #

• 优先使用SDK提供的公开构造函数创建Request/Response，避免直接操作未导出字段
• 必须操作未导出字段时，反射是Go中唯一合法的方式
• Monkey补丁要注意时机，务必在创建实例前完成补丁，否则无法生效
• 尽量避免使用Monkey补丁，优先通过依赖注入（比如替换transport）实现Mock

内容的提问来源于stack exchange，提问作者Aupa