如何使用Golang测试HTTP客户端_Golang net/http客户端请求测试示例

最直接的HTTP客户端测试方式是用httptest.Server启动本地可控服务,自动分配端口并支持自定义Handler,需defer srv.Close()、使用srv.URL,并精确断言状态码、JSON响应及请求细节。

httptest.Server 替换真实后端

测试 HTTP 客户端最直接的方式,是让客户端请求发到一个本地可控的测试服务器,而不是真实 API。Go 标准库的 httptest.Server 就是为此设计的——它启动一个临时 HTTP 服务,返回你指定的响应。

常见错误是手动起 goroutine 启动 http.ListenAndServe,既难控制生命周期,又容易端口冲突或泄漏。而 httptest.Server 自动分配空闲端口、自动关闭,且支持自定义 http.Handler

  • 每次测试前调用 srv := httptest.NewServer(handler)handler 可以是 http.HandlerFunc 或任意实现 ServeHTTP 的对象
  • 务必在测试结束时调用 srv.Close()(通常放在 defer 中),否则资源不释放,多次运行会 panic
  • 客户端应使用 srv.URL(如 "http://127.0.0.1:34212")作为基础地址,而非硬编码 "http://localhost:8080"

构造可预测的 JSON 响应并验证请求头/参数

真实场景中,客户端往往依赖特定状态码、JSON 字段或响应头。测试时不能只检查“有没有返回”,而要精确断言结构和行为。

比如客户端设置了 Authorization 头、发送了 query 参数、用了 POST 方法,这些都该被测试覆盖。你可以通过闭包捕获请求信息,再用 assert 类库或原生 if 检查。

  • http.HandlerFunc 包裹逻辑,在函数体内读取 req.Methodreq.Headerreq.URL.Query()
  • 对 JSON 响应,用 json.Marshal 构造确定内容,避免时间戳、随机 ID 等不可预测字段干扰断言
  • 若需模拟错误路径,可直接写入 http.StatusUnauthorizedhttp.StatusInternalServerError,再检查客户端是否正确处理
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    if r.Method != "POST" {
        w.WriteHeader(http.StatusBadRequest)
        return
    }
    if r.Header.Get("Authorization") == "" {
        w.WriteHeader(http.StatusUnauthorized)
        return
    }
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(map[string]string{"status": "ok"})
}))
defer srv.Close()

client := &http.Client{} resp, _ := client.Post(srv.URL+"/api/v1/users", "application/json", strings.NewReader({"name":"test"}))

http.ClientTransport 拦截请求(适合无服务场景)

有些情况不适合启 HTTP 服务:比如测试超时逻辑、重试机制、或想完全跳过网络层。这时可替换 http.Client.Transport 为自定义的 RoundTripper,直接返回预设响应,不走 socket。

标准库 net/http/httptest 提供了 httptest.NewUnstartedServer,但更轻量的做法是实现一个只返回固定 *http.ResponseRoundTripper

  • 实现 RoundTrip(*http.Request) (*http.Response, error) 方法,内部用 io.NopCloser 包装响应体
  • 注意设置 Response.StatusCodeResponse.HeaderResponse.Body,否则客户端可能 panic
  • 此方式绕过 DNS、连接、TLS 等所有网络环节,适合单元测试;但无法测重定向、代理、超时触发等依赖真实网络栈的行为

测试超时、取消和上下文传播

生产客户端几乎都带超时或 context 控制。测试这些行为不能靠 time.Sleep,而要用可控的延迟或立即取消。

典型错误是写 time.Sleep(3 * time.Second) 等待超时,导致测试慢、不稳定。正确做法是用 context.WithTimeout 配合一个极短时间(如 1 * time.Millisecond),或用 context.WithCancel 立即 cancel。

  • 测试超时:创建 ctx, cancel := context.WithTimeout(context.Background(), 1*time.Millisecond),然后在 client.Do(req.WithContext(ctx)) 后检查是否返回 context.DeadlineExceeded
  • 测试取消:调用 cancel() 后立刻发起请求,预期得到 context.Canceled
  • 注意:http.DefaultClient 不受 context 影响,必须显式传入 req.WithContext(ctx) 或使用自定义 http.Client

超时和取消逻辑容易漏测,尤其当客户端封装了多层调用。只要用了 context,就该有对应测试用例——哪怕只是验证错误类型是否匹配。