Golang如何测试HTTP接口？

使用httptest可无需启动服务器测试Golang的HTTP接口，通过NewRequest和NewRecorder模拟请求与响应。示例涵盖GET请求参数处理、路由注册、POST JSON数据解析及状态码校验。推荐采用表格驱动测试提升可维护性，并结合testify等断言库优化断言逻辑。核心是构造请求、验证状态码与响应体，确保测试独立可重复。

测试 HTTP 接口在 Golang 中非常常见，尤其是构建 RESTful 服务时。我们可以使用标准库中的 net/http/httptest 和 testing 包来完成单元测试，无需启动真实服务器。下面介绍如何编写可维护、清晰的 HTTP 接口测试。

使用 httptest 模拟 HTTP 请求

Go 的 httptest 包提供了一种无需绑定端口即可测试 HTTP 处理器的方式。你可以创建一个模拟的请求并捕获响应。

假设你有一个简单的处理函数：

func HelloHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello, %s!", r.URL.Query().Get("name"))
}

对应的测试可以这样写：

func TestHelloHandler(t *testing.T) {
    req := httptest.NewRequest("GET", "/hello?name=Gopher", nil)
    w := httptest.NewRecorder()

    HelloHandler(w, req)

    resp := w.Result()
    body, _ := io.ReadAll(resp.Body)

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

    if string(body) != "Hello, Gopher!" {
        t.Errorf("expected body %q, got %q", "Hello, Gopher!", string(body))
    }
}

测试路由和多方法（使用 net/http）

如果你使用的是 net/http 的路由（比如基于 http.ServeMux），可以将处理器注册到 Mux 上再进行测试。

示例代码：

func setupRouter() *http.ServeMux {
    mux := http.NewServeMux()
    mux.HandleFunc("/api/v1/hello", HelloHandler)
    return mux
}

func TestHelloRoute(t *testing.T) {
    req := httptest.NewRequest("GET", "/api/v1/hello?name=World", nil)
    w := httptest.NewRecorder()

    setupRouter().ServeHTTP(w, req)

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

    if w.Body.String() != "Hello, World!" {
        t.Errorf("expected body %q, got %q", "Hello, World!", w.Body.String())
    }
}

测试 JSON 接口（POST 请求）

大多数现代 API 使用 JSON 数据。你需要构造 JSON 请求体并验证返回的 JSON 结构。

处理函数示例：

type User struct {
    Name string `json:"name"`
}

func CreateUser(w http.ResponseWriter, r *http.Request) {
    var user User
    if err := json.NewDecoder(r.Body).Decode(&user); err != nil {
        http.Error(w, "invalid json", http.StatusBadRequest)
        return
    }

    w.Header().Set("Content-Type", "application/json")
    w.WriteHeader(http.StatusCreated)
    json.NewEncoder(w).Encode(map[string]string{
        "message": "User created",
        "name":    user.Name,
    })
}

测试代码：

func TestCreateUser(t *testing.T) {
    payload := strings.NewReader(`{"name": "Alice"}`)

    req := httptest.NewRequest("POST", "/api/v1/users", payload)
    req.Header.Set("Content-Type", "application/json")
    w := httptest.NewRecorder()

    CreateUser(w, req)

    if w.Code != http.StatusCreated {
        t.Errorf("expected status %d, got %d", http.StatusCreated, w.Code)
    }

    var resp map[string]string
    if err := json.Unmarshal(w.Body.Bytes(), &resp); err != nil {
        t.Fatalf("can't decode json: %v", err)
    }

    if resp["name"] != "Alice" {
        t.Errorf("expected name %q, got %q", "Alice", resp["name"])
    }
}

组织测试与断言优化

为了提升可读性和维护性，建议使用表格驱动测试，并引入断言工具（如 testify/assert）。

表格驱动示例：

func TestHelloHandler_TableDriven(t *testing.T) {
    tests := []struct {
        name     string
        query    string
        expected string
    }{
        {"with name", "?name=Bob", "Hello, Bob!"},
        {"without name", "", "Hello, !"},
    }

    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            req := httptest.NewRequest("GET", "/hello"+tt.query, nil)
            w := httptest.NewRecorder()

            HelloHandler(w, req)

            if w.Body.String() != tt.expected {
                t.Errorf("got %q, want %q", w.Body.String(), tt.expected)
            }
        })
    }
}

基本上就这些。Golang 的测试机制简洁高效，配合 httptest 能轻松覆盖大部分 HTTP 接口场景。关键是构造好请求、检查状态码、解析响应内容，保持测试独立且可重复。