Go语言单元测试中Mock Hijack方法触发错误以实现全量代码覆盖

我们有一个处理HTTP请求的函数，其中会劫持入站连接，代码如下：

func(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("ACK:"))
        hijacker, ok := w.(http.Hijacker)
        if !ok {
            http.Error(w, "hijacking not supported", http.StatusInternalServerError)
            return
        }
        clientConn, _, err := hijacker.Hijack()

        //[TODO] 如何触发这个错误分支？
        if err != nil {
            http.Error(w, err.Error(), http.StatusServiceUnavailable)
            return
        }
        // 其他业务逻辑
    }

我们需要编写测试用例实现100%代码覆盖率，但不知道如何触发Hijack()方法返回错误的分支。

我尝试了以下代码来Mock Hijack方法，但出现编译错误：

oldHf := http.Hijacker.Hijack
http.Hijacker.Hijack = func(net.Conn, *bufio.ReadWriter, error) (net.Conn, *bufio.ReadWriter, error) {
    return &net.TCPConn{}, &bufio.ReadWriter{}, errors.New("dummy error")
}
defer func() { http.Hijacker.Hijack = oldHf }()

编译错误信息：

defer func() { http.Hijacker.Hijack = oldHf }()
http.Hijacker (type) is not an expressioncompilerNotAnExpr
cannot assign to http.Hijacker.Hijack (value of type func(http.Hijacker) (net.Conn, *bufio.ReadWriter, error))compilerUnassignableOperand

你不能直接修改http.Hijacker接口的方法，因为接口是类型定义，并非可赋值的变量。正确的做法是自定义一个实现了http.Hijacker接口的Mock结构体，在Hijack方法中返回预设的错误。

1. 定义Mock Hijack Writer结构体 #

import (
    "bufio"
    "errors"
    "net"
    "net/http"
    "net/http/httptest"
)

// MockHijackWriter 同时实现http.ResponseWriter和http.Hijacker接口
type MockHijackWriter struct {
    *httptest.ResponseRecorder // 嵌入基础的ResponseWriter实现
    hijackErr error            // 预设的Hijack方法返回错误
}

// Hijack 实现http.Hijacker接口，直接返回预设错误
func (m *MockHijackWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
    return nil, nil, m.hijackErr
}

2. 编写测试用例 #

使用上述Mock结构体，传入预设错误即可触发目标错误分支：

import (
    "strings"
    "testing"
    "net/http"
    "net/http/httptest"
    "errors"
)

func TestHandlerHijackError(t *testing.T) {
    // 创建测试请求
    req, err := http.NewRequest("GET", "/test", nil)
    if err != nil {
        t.Fatalf("创建请求失败: %v", err)
    }

    // 初始化Mock Writer，设置要返回的错误
    mockWriter := &MockHijackWriter{
        ResponseRecorder: httptest.NewRecorder(),
        hijackErr:        errors.New("模拟劫持失败错误"),
    }

    // 调用待测试的处理函数
    yourRequestHandler(mockWriter, req)

    // 验证结果：状态码应为503，响应体包含错误信息
    if mockWriter.Code != http.StatusServiceUnavailable {
        t.Errorf("期望状态码%d，实际得到%d", http.StatusServiceUnavailable, mockWriter.Code)
    }
    if !strings.Contains(mockWriter.Body.String(), "模拟劫持失败错误") {
        t.Error("响应体未包含预期的错误信息")
    }
}

额外说明 #

• 如果需要测试w.(http.Hijacker)断言失败的分支，可以再定义一个仅实现http.ResponseWriter但不实现http.Hijacker的Mock结构体，传入后即可触发该分支。
• 这种基于接口实现的Mock方式是Go测试中的标准实践，避免了修改全局类型的风险，更符合Go的面向接口设计思想。

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