Golang如何测试HTTP路由_Golang HTTP路由单元测试方法_技术教程

答案：使用Go的httptest包可无需启动服务器测试HTTP路由。1. 将路由注册封装为函数返回Handler便于测试；2. 用httptest.NewRequest创建请求，httptest.NewRecorder捕获响应；3. 验证状态码、响应体，如测试/hello返回200和"Hello, World!"；4. 测试方法限制如POST/user/返回405；5. 推荐表驱动测试覆盖多场景。

在Go语言中，测试HTTP路由是确保Web应用正确处理请求的重要环节。使用标准库中的 net/http/httptest 包，可以轻松实现对HTTP路由的单元测试，无需启动真实服务器。

1. 编写可测试的HTTP路由

为了让路由可测试，建议将路由注册逻辑封装成一个可复用的函数，例如返回一个 http.ServeMux 或 http.Handler。这样可以在测试中直接传入模拟请求，而不依赖端口绑定。

示例：定义简单路由

假设你有一个简单的HTTP服务：

package main

import "net/http"

func NewRouter() *http.ServeMux {
    mux := http.NewServeMux()
    mux.HandleFunc("/hello", helloHandler)
    mux.HandleFunc("/user/", userHandler)
    return mux
}

func helloHandler(w http.ResponseWriter, r *http.Request) {
    w.WriteHeader(http.StatusOK)
    w.Write([]byte("Hello, World!"))
}

func userHandler(w http.ResponseWriter, r *http.Request) {
    if r.Method != "GET" {
        w.WriteHeader(http.StatusMethodNotAllowed)
        return
    }
    w.WriteHeader(http.StatusOK)
    w.Write([]byte("User profile"))
}

2. 使用 httptest 进行单元测试

Go 的 net/http/httptest 提供了 httptest.NewRecorder() 来捕获响应，配合 http.NewRequest() 构造请求，即可完成完整流程测试。

示例：测试 /hello 路由

package main

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

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

    handler := NewRouter()
    handler.ServeHTTP(rec, req)

    if rec.Code != http.StatusOK {
        t.Errorf("期望状态码 %d，实际得到 %d", http.StatusOK, rec.Code)
    }

    expected := "Hello, World!"
    if rec.Body.String() != expected {
        t.Errorf("期望响应体 %q，实际得到 %q", expected, rec.Body.String())
    }
}

3. 测试路径参数与方法限制

虽然 ServeMux 支持前缀匹配（如 /user/），但不支持通配符参数（如 /user/:id）。如果你使用的是 Gin、Echo 等框架，测试方式类似，仍可通过 httptest 模拟请求。

示例：测试不允许的方法

func TestUserHandler_MethodNotAllowed(t *testing.T) {
    req := httptest.NewRequest("POST", "/user/123", nil)
    rec := httptest.NewRecorder()

    NewRouter().ServeHTTP(rec, req)

    if rec.Code != http.StatusMethodNotAllowed {
        t.Errorf("期望 %d，实际 %d", http.StatusMethodNotAllowed, rec.Code)
    }
}