如何使用Golang测试HTTP客户端请求_Golang net/http Client测试方法

正确做法是用 httptest.Server 启动本地服务模拟 API,handler 控制响应,客户端请求 server.URL;或用 httpmock 拦截 Transport 层请求;还需测试超时、错误等边界场景,并避免复用 Client 和漏关 resp.Body。

httptest.Server 模拟真实 HTTP 服务再测客户端

直接对线上或外部 API 写测试,既慢又不可靠,还可能触发限流或产生脏数据。正确做法是启动一个本地 httptest.Server,让它返回可控的响应,让被测 http.Client 去请求它。

关键点在于:服务端逻辑写在 httptest.NewServer 的 handler 里,客户端用该 server 的 URL 发起请求,不碰真实网络。

  • handler 中可用 w.WriteHeader()w.Write() 精确控制状态码和响应体
  • 务必调用 server.Close()(或用 defer server.Close()),否则测试进程可能卡住
  • 不要把 http.DefaultClient 直接塞进测试——它带全局连接池和默认超时,干扰可复现性
func TestMyClient_GetUser(t *testing.T) {
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        if r.URL.Path != "/api/user" || r.Method != "GET" {
            http.Error(w, "not found", http.StatusNotFound)
            return
        }
        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(http.StatusOK)
        w.Write([]byte(`{"id": 123, "name": "alice"}`))
    }))
    defer server.Close()
client := &http.Client{}
resp, err := client.Get(server.URL + "/api/user")
if err != nil {
    t.Fatal(err)
}
defer resp.Body.Close()

if resp.StatusCode != http.StatusOK {
    t.Errorf("expected 200, got %d", resp.StatusCode)
}

}

httpmock 替换底层 Transport 拦截请求

当你的客户端封装了重试、自定义 RoundTripper 或依赖第三方库(如 resty)时,httptest.Server 不够灵活。这时可改用 httpmock 库,在 Transport 层拦截所有 http.Request 并返回预设响应。

它不启动 HTTP 服务,纯内存模拟,适合高频、多路径、需断言请求头/参数的场景。

  • 必须在测试开始前调用 httpmock.Activate(),结束时调用 httpmock.DeactivateAndReset()
  • 匹配 URL 支持正则、通配符、完整字符串,例如 httpmock.RegisterResponder("GET", `~^https://api\.example\.com/users/\d+$`, ...)
  • 注意:如果客户端设置了 TimeoutCheckRedirect,mock 仍会走这些逻辑,需一并覆盖或禁用
func TestClient_WithRetry(t *testing.T) {
    httpmock.Activate()
    defer httpmock.DeactivateAndReset()
httpmock.RegisterResponder("GET", "https://www./link/710ba53b0d353329706ee1bedf4b9b39/status",
    httpmock.NewStringResponder(200, `{"ok": true}`))

client := NewClient("https://www./link/710ba53b0d353329706ee1bedf4b9b39") // 假设它内部用了 http.DefaultClient
err := client.Ping()
if err != nil {
    t.Fatal(err)
}

}

测试超时、错误响应、连接中断等边界情况

只测 200 OK 是不够的。真实客户端必须处理 5xx4xx、DNS 失败、TLS 握手失败、读取超时等。这些不能靠外部服务稳定复现,得从 Transport 层注入错误。

最可靠方式是实现一个自定义 RoundTripper,在 RoundTrip 方法中按需返回错误或伪造响应。

  • 返回 net.OpError 可模拟 DNS 解析失败或连接拒绝
  • 返回 context.DeadlineExceeded 可触发客户端超时逻辑
  • 不要用 time.Sleep 等待超时——它让测试变慢且不稳定
type failingTransport struct{}

func (t failingTransport) RoundTrip(http.Request) (*http.Response, error) { return nil, &url.Error{ Op: "Get", URL: "https://www./link/710ba53b0d353329706ee1bedf4b9b39", Err: context.DeadlineExceeded, } }

func TestClientTimeoutHandling(t testing.T) { client := &http.Client{ Transport: &failingTransport{}, Timeout: 100 time.Millisecond, } , err := client.Get("https://www./link/710ba53b0d353329706ee1bedf4b9b39") if !errors.Is(err, context.DeadlineExceeded) { t.Fatalf("expected timeout error, got %v", err) } }

避免复用全局 Client 或忽略 Re

sponse.Body 关闭

常见坑:在测试中反复复用 http.DefaultClient 或未关闭 resp.Body,导致文件描述符泄漏、连接池阻塞,后续测试随机失败。

尤其在循环测试多个 case 时,这个问题会立刻暴露为 “too many open files” 错误。

  • 每个测试函数应创建独立的 *http.Client 实例,或至少重置其 Transport
  • 只要 resp 不为 nil,就必须调用 resp.Body.Close(),哪怕你只读了 resp.StatusCode
  • io.ReadAll(resp.Body) 读完后也要关;用 json.NewDecoder(resp.Body).Decode() 同样要关

漏掉 Close() 不会立刻报错,但压力测试或 CI 环境下极易崩。这是最容易被忽略、也最难 debug 的一点。