本文介绍通过替换 http.DefaultClient 并自定义 RoundTripper 实现对 http.Head() 的可控模拟，从而覆盖网络错误分支的单元测试场景。
本文介绍通过替换 `http.defaultclient` 并自定义 `roundtripper` 实现对 `http.head()` 的可控模拟，从而覆盖网络错误分支的单元测试场景。
在 Go 标准库中，http.Head(url) 并非独立实现，而是内部委托给 http.DefaultClient.Head() 方法执行。这意味着——它本质上是 http.DefaultClient 的一次方法调用。因此，Mock http.Head() 的核心思路不是“打桩函数”，而是在测试期间临时替换 http.DefaultClient 为一个可控的客户端，使其底层 Transport 可按需返回指定响应或错误。
✅ 推荐方案：自定义 RoundTripper
http.Client 的行为由其 Transport 字段决定（类型为 http.RoundTripper 接口）。我们只需实现该接口，在 RoundTrip 方法中精确控制返回值即可：
type mockTransport struct {
response *http.Response
err error
}
func (m *mockTransport) RoundTrip(req *http.Request) (*http.Response, error) {
// 可选：校验请求 URL、Method 等，确保被测逻辑调用了预期的 Head 请求
if req.Method != "HEAD" {
return nil, fmt.Errorf("unexpected method: %s", req.Method)
}
return m.response, m.err
}在测试中使用：
func TestHandleRelease_ErrorOnHead(t *testing.T) {
// 1. 保存原始 DefaultClient
originalClient := http.DefaultClient
defer func() { http.DefaultClient = originalClient }()
// 2. 构造一个会返回错误的 Transport
transport := &mockTransport{
err: errors.New("network timeout"),
}
http.DefaultClient = &http.Client{Transport: transport}
// 3. 执行被测逻辑（例如调用 main.handleRelease）
// 此时 http.Head() 将触发 mockTransport.RoundTrip 并返回 err
// 预期日志输出或状态码应体现错误处理分支
// 测试断言（略）...
}⚠️ 注意事项与最佳实践
- 务必恢复 http.DefaultClient：使用 defer 或 t.Cleanup() 确保测试结束后还原，避免污染其他测试。
- 避免全局副作用：若项目中存在并发测试，建议改用依赖注入（如将 *http.Client 作为参数传入业务函数），而非依赖 http.DefaultClient —— 这更易测试且符合 Go 最佳实践。
- 不要 Mock http.Head 函数本身：Go 不支持直接 Monkey Patch 函数；试图用 //go:linkname 或反射修改函数指针属于未定义行为，极易引发 panic 或竞态。
- 推荐补充 HTTP 状态码模拟：除错误外，还可构造不同 *http.Response（如 StatusCode: 404, 500）验证业务层对各类 HTTP 状态的容错逻辑。
✅ 总结
Mock http.Head() 的本质是 Mock http.DefaultClient 的传输行为。通过实现轻量级 RoundTripper，你既能精准控制网络响应，又能保持测试隔离性与可读性。这一模式同样适用于 http.Get、http.Post 等所有基于 DefaultClient 的便捷函数，是 Go HTTP 客户端测试的基石技术。
