贝利信息

Golang测试中如何构造HTTP请求与响应

日期:2026-01-06 00:00 / 作者:P粉602998670
httptest.NewRequest 构造请求对象无需网络调用,需正确设置 method、URL、body 和 Content-Type;配合 httptest.NewRecorder 捕获响应,可断言 status、header 和 body;路由参数需手动注入 context;避免启动真实 server,优先单元测试业务逻辑。

httptest.NewRequest 构造请求对象

测试 HTTP handler 时,不需要真实网络调用,httptest.NewRequest 可生成干净、可控的 *http.Request。它不依赖服务器启动,也不触发 DNS 或连接逻辑。

常见错误是忽略 body 类型或编码方式,导致 handler 解析失败。比如 POST JSON 时没设 Content-Type,或传了 nil body 却期望读取数据。

req := httptest.NewRequest("POST", "/api/users", strings.NewReader(`{"name":"alice"}`))
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer xyz")

httptest.NewRecorder 捕获响应

httptest.NewRecorder() 返回一个实现了 http.ResponseWriter 接口的内存 recorder,所有写入(status、header、body)都被缓存,可断言验证。

容易被忽略的是:它不会自动设置 Content-Length,也不会校验 header 合法性(比如重复的 Set-Cookie 不报错),但足够用于单元测试。

rec := httptest.NewRecorder()
handler.ServeHTTP(rec, req)

if rec.Code != http.StatusCreated {
    t.Fatalf("expected 201, got %d", rec.Code)
}
if !strings.Contains(rec.Body.String(), "alice") {
    t.Fatal("response body missing expected content")
}

测试带路由参数的 handler(如 gorilla/mux 或 chi)

第三方路由器(如 gorilla/mux)依赖 request.Context() 中的 URL 参数,而 httptest.NewRequest 默认不填充。必须手动把参数塞进 context,否则 mux.Vars(r) 返回空 map。

不同路由器注入方式不同:有的用 context.WithValue,有的要求调用特定方法(如 chiCtx.RoutePattern)。不匹配会导致测试通过但线上出错。

// gorilla/mux 示例
r := mux.NewRouter()
r.HandleFunc("/users/{id}", userHandler).Methods("GET")
req := httptest.NewRequest("GET", "/users/456", nil)
mux.SetURLVars(req, map[string]string{"id": "456"}) // 关键一步
rec := httptest.NewRecorder()
r.ServeHTTP(rec, req)

避免在测试中启动真实 HTTP server

http.ListenAndServehttptest.NewUnstartedServer 启动真实 server,会引入端口占用、竞态、超时、资源泄漏等问题。绝大多数 handler 单元测试完全不需要它。

只有两类场景才考虑真实 server:集成测试验证中间件链完整行为,或测试 client SDK 的重试/超时逻辑。即便如此,也应优先用 httptest.NewServer(它自动选空闲端口并管理生命周期)而非手写 ListenAndServe

真正难测的从来不是 HTTP 协议层,而是 handler 里混杂的业务逻辑、外部依赖、上下文传递。把它们拆出来单独测,比纠结怎么 mock 一个 response writer 更有效。