最直接的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 检查。
立即学习“go语言免费学习笔记(深入)”;
- 用
http.HandlerFunc包裹逻辑,在函数体内读取req.Method、req.Header、req.URL.Query()等 - 对 JSON 响应,用
json.Marshal构造确定内容,避免时间戳、随机 ID 等不可预测字段干扰断言 - 若需模拟错误路径,可直接写入
http.StatusUnauthorized或http.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.Client 的 Transport 拦截请求(适合无服务场景)
有些情况不适合启 HTTP 服务:比如测试超时逻辑、重试机制、或想完全跳过网络层。这时可替换 http.Client.Transport 为自定义的 RoundTripper,直接返回预设响应,不走 socket。
标准库 net/http/httptest 提供了 httptest.NewUnstartedServer,但更轻量的做法是实现一个只返回固定 *http.Response 的 RoundTripper。
- 实现
RoundTrip(*http.Request) (*http.Response, error)方法,内部用io.NopCloser包装响应体 - 注意设置
Response.StatusCode、Response.Header和Response.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,就该有对应测试用例——哪怕只是验证错误类型是否匹配。










