如何使用Golang模拟HTTP请求_Golang httptest模拟客户端请求示例

应使用 httptest.NewServer 启动临时 HTTP 服务器并传入 handler,获取 server.URL 发起请求,且必须 defer server.Close();需精细控制请求时用 httptest.NewRequest 构造 *http.Request,配合 httptest.NewRecorder 测试 handler 行为。

httptest.NewServer 启动测试 HTTP 服务

别直接写 http.Get 去调真实接口——测试时应隔离外部依赖。Go 标准库的 httptest 包提供 NewServer,它会启动一个临时 HTTP 服务器,返回可访问的 URL(如 http://127.0.0.1:34212),你拿这个 URL 当目标发请求即可。

关键点:

  • NewServer 接收一个 http.Handler,通常传入你的业务路由(比如 http.HandlerFunchttp.ServeMux
  • 它自动监听随机空闲端口,无需手动选端口或处理 ListenAndServe 错误
  • 必须调用 server.Close() 结束,否则 goroutine 和端口会泄漏
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    if r.URL.Path == "/api/user" && r.Method == "GET" {
        w.WriteHeader(http.StatusOK)
        w.Write([]byte(`{"id":1,"name":"alice"}`))
    }
}))
defer server.Close() // 必须加

resp, err := http.Get(server.URL + "/api/user")
if err != nil {
    log.Fatal(err)
}
defer resp.Body.Close()

httptest.NewRequest 构造请求对象

当你需要控制请求头、Body、Method、URL 查询参数等细节(比如测试登录接口带 Authorization 头),就不能只靠 http.Get。此时用 httptest.NewRequest 创建原始 *http.Request,再交给你的 handler 处理。

常见使用场景:

  • 测试中间件(如鉴权、日志)是否正确读取 header
  • 验证 JSON 请求体是否被正确解析
  • 模拟不同 method(PUTDELETE)或 content-type
req := httptest.NewRequest("POST", "/login", strings.NewReader(`{"user":"bob","pass":"123"}`))
req.Header.Set("Content-Type", "application/json")
req.Header.Set("Authorization", "Bearer abc123")

rr := httptest.NewRecorder()
handler := http.HandlerFunc(yourLoginHandler)
handler.ServeHTTP(rr, req)

if rr.Code != http.StatusOK {
    t.Errorf("expected status OK, got %d", rr.Code)
}

为什么不用 net/http/httptest 模拟客户端?

名字容易误导:httptest 的核心定位是「测试服务端逻辑」,不是模拟客户端行为。它提供的 NewRequestNewRecorder 是为服务端 handler 测试服务的;NewServer 是为客户端代码提供可控后端——但它本身不封装 http.Client 行为。

也就是说:

  • 你仍要用标准 http.Client 发请求(如 client.Do(req)
  • httptest 不替代 http.Client,只帮你解决「往哪发」和「怎么构造请求对象」的问题
  • 若需 mock 客户端行为(如强制返回错误、延迟响应),得自己包装 http.RoundTripper,或用第三方库如 gock

容易漏掉的清理和边界情况

测试中端口泄漏和 Body 未关闭是最常导致 CI 失败或本地运行变慢的问题。

  • httptest.NewServer 启动后,必须配对 defer server.Close(),哪怕测试 panic 也要生效,建议用 defer func() { server.Close() }()
  • 所有 resp.Body 必须显式 Close(),否则连接不会释放,后续请求可能卡住
  • 如果 handler 内部调用了 http.Redirect,默认不会跟随跳转,需手动检查 resp.StatusCode 是否为 302,或用带 CheckRedirect 的 client
  • httptest.NewRequest 的 body 参数如果是 strings.NewReader,记得内容要合法(比如 JSON 字符串不能少引号)

复杂点往往不在逻辑,而在这些隐式资源管理上。漏掉一次 Close(),可能让整个测试套件在并发下间歇性失败。