如何使用Golang测试HTTP接口_结合httptest创建模拟请求

Go 的 httptest 包支持内存中 HTTP 接口测试：NewRecorder 用于单元测试单个 handler，NewServer 用于集成测试完整服务；需配合依赖注入与 mock 实现隔离。

使用 Go 的 httptest 包可以轻松测试 HTTP 接口，无需启动真实服务器，也不依赖网络。核心思路是把你的 HTTP 处理函数（如 http.HandlerFunc）直接传给 httptest.NewServer 或 httptest.NewRecorder，让测试在内存中完成请求-响应流程。

用 httptest.NewRecorder 测试单个 handler

适合单元测试单个路由处理逻辑，不涉及中间件或完整服务启动。

• 创建 *httptest.ResponseRecorder 模拟响应体、状态码、Header 等
• 构造 *http.Request（可用 http.NewRequest，注意设置 URL、Method、Body）
• 直接调用你的 handler 函数：handler.ServeHTTP(recorder, req)
• 断言 recorder.Code、recorder.Body.String()、recorder.Header()

示例：

func TestLoginHandler(t *testing.T) {
    req, _ := http.NewRequest("POST", "/login", strings.NewReader(`{"user":"a","pass":"123"}`))
    req.Header.Set("Content-Type", "application/json")
    rr := httptest.NewRecorder()
    handler := http.HandlerFunc(LoginHandler)
    handler.ServeHTTP(rr, req)

    if rr.Code != http.StatusOK {
        t.Errorf("expected status OK, got %d", rr.Code)
    }
    if !strings.Contains(rr.Body.String(), "token") {
        t.Error("response missing token")
    }
}

用 httptest.NewServer 测试完整服务行为

适合集成测试多个路由、中间件、重定向、客户端行为等，模拟真实 HTTP 服务端点。

• 传入一个 http.Handler（比如你整个 http.ServeMux 或 Gin/Echo 的 engine）
• httptest.NewServer 返回一个带随机端口的临时服务器，提供 .URL 字段
• 用标准 http.Client 向该 URL 发起请求（就像调用真实 API）
• 服务器会在测试结束后自动关闭

示例：

func TestAPIWithMiddleware(t *testing.T) {
    mux := http.NewServeMux()
    mux.HandleFunc("/api/user", UserHandler)
    server := httptest.NewServer(mux)
    defer server.Close() // 自动关闭

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

    if resp.StatusCode != http.StatusOK {
        t.Errorf("expected 200, got %d", resp.StatusCode)
    }
}

测试带 JSON 请求/响应的常见技巧

多数 REST 接口使用 JSON，测试时需注意序列化与内容类型。

• 发 JSON 请求：用 strings.NewReader(jsonBytes) 构造 body，并设 Content-Type: application/json
• 解析响应 JSON：用 json.NewDecoder(resp.Body).Decode(&v)，避免手动字符串匹配
• 验证结构体字段：定义预期结构体，解码后逐字段比对（或用 reflect.DeepEqual，但注意时间、指针等细节）
• 错误响应也要覆盖：比如空 body、非法 JSON、400 错误码，检查 error message 是否合理

处理依赖（如数据库、外部服务）的测试隔离

真实 handler 常依赖 DB、缓存、第三方 API。测试时应替换为可控的 mock 实现。

• 将依赖（如 *sql.DB 或 client 接口）作为 handler 的闭包变量或结构体字段注入
• 测试时传入 mock 实现（例如用 github.com/DATA-DOG/go-sqlmock 或自定义 interface）
• 避免在测试中启动真实数据库或调用外网服务；httptest 本身不解决依赖问题，但和依赖注入配合效果最佳

例如：

type UserService struct{ db *sql.DB }
func (s *UserService) GetUser(w http.ResponseWriter, r *http.Request) {
    // ...
}

// 测试时：
mockDB, mock, _ := sqlmock.New()
service := &UserService{db: mockDB}
handler := http.HandlerFunc(service.GetUser)