如何在 Go 中正确测试 HTTP 请求处理器-Golang-PHP中文网

如何在 Go 中正确测试 HTTP 请求处理器

本文介绍使用 net/http/httptest 包对 Go 的 HTTP 处理器进行高效、隔离的单元测试，涵盖模拟请求/响应、解耦数据库依赖及结构化 handler 设计的最佳实践。

本文介绍使用 net/http/httptest 包对 go 的 http 处理器进行高效、隔离的单元测试，涵盖模拟请求/响应、解耦数据库依赖及结构化 handler 设计的最佳实践。

在 Go Web 开发中，HTTP 处理器（如 func(w http.ResponseWriter, r *http.Request)）是核心逻辑入口，但直接启动真实 HTTP 服务器进行测试既低效又难隔离。幸运的是，Go 标准库提供了轻量、可靠的测试工具——net/http/httptest，它允许你在不启动网络服务的前提下，完全控制请求输入与响应输出。

✅ 使用 `httptest.NewRequest` 和 `httptest.NewRecorder`

httptest.NewRequest 可构造任意方法、URL、Header 和 Body 的模拟请求；httptest.NewRecorder 则实现 http.ResponseWriter 接口，捕获状态码、Header 和响应体内容，便于断言：

func TestGetProductsHandler(t *testing.T) {
    // 模拟请求
    req := httptest.NewRequest("GET", "/products", nil)
    w := httptest.NewRecorder()

    // 调用待测处理器（注意：需确保 db 已初始化或可被替换）
    GetProductsHandler(w, req)

    // 断言响应
    if w.Code != http.StatusOK {
        t.Errorf("expected status OK, got %d", w.Code)
    }
    if !strings.Contains(w.Header().Get("Content-Type"), "application/json") {
        t.Error("expected JSON Content-Type")
    }
    if !strings.HasPrefix(w.Body.String(), "[") {
        t.Error("response body should be a JSON array")
    }
}

⚠️ 注意：上述测试仍隐式依赖全局 db 变量，这会破坏测试的可重复性与隔离性——多个测试并发运行时可能相互干扰，且无法精确控制数据库返回值（如空列表、错误等边界场景）。

? 推荐方案：结构化 Handler + 依赖注入

将处理器封装为结构体，并通过字段注入依赖（如数据库接口），是实现真正可测试、可维护代码的关键模式：

Miller CSV TSV JSON 数据处理器

Miller (mlr) 是一个命令行工具，用于查询、整形和重新格式化名称索引数据，如 CSV、TSV、JSON 和 JSON Lines。它将 awk、sed、cut、join 和 sort 的功能整合到一个专为结构化数据处理而构建的单一工具中。

下载

type ProductDB interface {
    GetProducts() []Product
}

type Handlers struct {
    db ProductDB
}

func (h *Handlers) GetProductsHandler(w http.ResponseWriter, r *http.Request) {
    defer r.Body.Close()
    products := h.db.GetProducts()

    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(products)
}

测试时，可轻松传入内存 Mock 或桩实现（stub）：

type mockDB struct {
    data []Product
}

func (m *mockDB) GetProducts() []Product {
    return m.data
}

func TestGetProductsHandler_WithMockDB(t *testing.T) {
    // 构造测试数据
    mockDB := &mockDB{data: []Product{{ID: 1, Name: "Laptop"}}}
    handlers := &Handlers{db: mockDB}

    req := httptest.NewRequest("GET", "/products", nil)
    w := httptest.NewRecorder()

    handlers.GetProductsHandler(w, req)

    assert.Equal(t, http.StatusOK, w.Code)
    assert.JSONEq(t, `[{"ID":1,"Name":"Laptop"}]`, w.Body.String())
}