17370845950

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，得把整条链“组装”起来再测。

• 中间件本质是 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 里，别用全局变量存配置。越早把依赖抽象成参数或接口，后续加测试就越省力。