httptest 包提供两种核心测试方式:NewServer 用于集成测试完整 HTTP 流程(启动真实本地服务器),NewRequest+NewRecorder 用于单元测试单个 handler(内存中无网络调用)。

Go 的 net/http/httptest 包不是“模拟 HTTP 客户端”,而是专门用来**测试 HTTP 服务端逻辑**的——它帮你启动一个内存中的假服务器,绕过网络层,直接把请求打进去,再检查响应。如果你在测自己写的 http.HandlerFunc 或 http.ServeMux,这才是正路。
用 httptest.NewServer 测试完整 HTTP 服务流程
适用于已封装好 http.Handler(比如 Gin、Echo 或自定义 mux),需要走一遍真实 TCP 连接、HTTP 解析、中间件链路的场景。它会起一个监听随机端口的真实本地服务器,适合集成测试。
- 每次调用
httptest.NewServer都会绑定新端口,记得调用srv.Close()释放资源,否则可能端口耗尽 - 返回的
srv.URL是带协议和端口的完整地址(如"http://127.0.0.1:34212"),客户端代码必须用它,不能硬写"localhost:8080" - 不支持 HTTPS,默认只跑 HTTP;若需 TLS 测试,得用
httptest.NewUnstartedServer+ 手动配置srv.TLS
func TestFullFlow(t *testing.T) {
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
if r.URL.Path == "/api/v1/users" && r.Method == "GET" {
w.WriteHeader(200)
json.NewEncoder(w).Encode([]map[string]string{{"id": "1", "name": "alice"}})
}
}))
defer srv.Close() // 必须
resp, err := http.Get(srv.URL + "/api/v1/users")
if err != nil {
t.Fatal(err)
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
t.Errorf("expected 200, got %d", resp.StatusCode)
}
}
用 httptest.NewRequest + httptest.NewRecorder 测试单个 handler 函数
这是最轻量、最常用的方式,适合单元测试单个 http.HandlerFunc。不走网络、不占端口、无并发干扰,执行快且可断言内部状态(比如是否调用了某个依赖函数)。
-
httptest.NewRequest构造请求时,body参数为nil表示空体;若要传 JSON,得用strings.NewReader(`{"name":"bob"}`)并设Content-Type: application/json -
httptest.NewRecorder返回的*httptest.ResponseRecorder不是http.ResponseWriter接口实现体,但兼容;它的Body.Bytes()、Code、Header()都可直接读取 - 注意:handler 内部若调用
http.Redirect,recorder.Code会是 302,recorder.Header().Get("Location")可取跳转地址
func TestUserCreateHandler(t *testing.T) {
req := httptest.NewRequest("POST", "/users", strings.NewReader(`{"name":"bob"}`))
req.Header.Set("Content-Type", "application/json")
rr := httptest.NewRecorder()
handler := http.HandlerFunc(userCreateHandler)
handler.ServeHTTP(rr, req)
if status := rr.Code; status != http.StatusCreated {
t.Errorf("handler returned wrong status code: got %v want %v",
status, http.StatusCreated)
}
var result map[string]string
if err := json.Unmarshal(rr.Body.Bytes(), &result); err != nil {
t.Fatal(err)
}
if result["name"] != "bob" {
t.Error("expected name 'bob'")
}
}
测试带中间件或依赖注入的 handler 链
真实项目中 handler 往往嵌套中间件(如日志、鉴权)或接收外部依赖(DB、cache)。这时不能只测裸 handler,得把整条链“组装”起来再测。
立即学习“go语言免费学习笔记(深入)”;
- 中间件本质是
func(http.Handler) http.Handler,测试时用middleware(next)包一层再传给ServeHTTP即可 - 若 handler 依赖结构体字段(如
type UserHandler struct { DB *sql.DB }),先构造实例,再用其方法作为 handler:h := &UserHandler{DB: mockDB}; h.ServeHTTP(rr, req) - 避免在测试里 new 出真实 DB 或 Redis 客户端;用 interface + mock(如
mockDB := &MockQuerier{...})控制返回值
func TestAuthMiddleware(t *testing.T) {
mockDB := &MockUserDB{}
handler := AuthMiddleware(UserCreateHandler(mockDB))
req := httptest.NewRequest("POST", "/users", nil)
rr := httptest.NewRecorder()
// 无 token → 应该 401
handler.ServeHTTP(rr, req)
if rr.Code != http.StatusUnauthorized {
t.Error("expected 401 without Authorization header")
}
// 带有效 token → 应该 201
req.Header.Set("Authorization", "Bearer valid-token")
rr = httptest.NewRecorder()
handler.ServeHTTP(rr, req)
if rr.Code != http.StatusCreated {
t.Error("expected 201 with valid token")
}
}
真正难的不是怎么写测试,而是怎么让 handler 足够“可测”:别在函数里直接调 log.Println 或 os.Exit,别把数据库连接写死在 handler 里,别用全局变量存配置。越早把依赖抽象成参数或接口,后续加测试就越省力。










