17370845950

Golang测试中如何构造HTTP请求与响应
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 却期望读取数据。

  • methodurl 必须非空,否则 Request.URLnil,多数 handler 会 panic
  • 若需 body,用 strings.NewReaderbytes.NewBuffer 包装,别直接传字符串字面量
  • 设置 header 要在创建 request 后调用 .Header.Set(),不能传入构造函数
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 不报错),但足够用于单元测试。

  • 调用 handler 后,从 recorder.Code 读 status code,recorder.Body.String() 读响应体
  • recorder.Header() 返回的是 http.Header 的副本,修改它不影响已写入的响应
  • 如果 handler 调用了 http.Redirectrecorder.Code 是 302,且 Location 在 header 中
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,用 mux.SetURLVars(req, map[string]string{"id": "123"})
  • chi,需构造 chi.Context 并赋给 req = req.WithContext(context.WithValue(req.Context(), chi.RouteCtxKey, rctx))
  • 若 handler 直接解析 r.URL.Path,则无需额外处理——但这是耦合度高的写法,不推荐
// 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

  • httptest.NewServer 返回的 *httptest.ServerURLClose() 方法,适合 client 测试
  • 若 handler 依赖全局状态(如数据库连接池、配置单例),应在测试前显式初始化和清理,而不是靠 server 启停来“模拟”环境
  • 并发测试多个 handler 时,每个测试用独立 httptest.NewRequest + NewRecorder,无共享状态,天然线程安全

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