Go中测试HTTP中间件需解耦中间件与路由，用httptest模拟请求响应；验证其修改请求头、拦截访问及上下文注入等行为，重点覆盖短路逻辑与上下文完整性。
在Golang中测试HTTP中间件逻辑并不复杂，关键在于将中间件与具体的路由处理函数解耦，并使用标准库中的 net/http/httptest 包来模拟请求和响应。通过构造一个被中间件包装的处理器，我们可以验证中间件是否按预期修改请求、添加头信息、拦截未授权访问等行为。
理解HTTP中间件的基本结构
Go中的HTTP中间件通常是一个函数，接收一个 http.HandlerFunc 并返回一个新的 http.HandlerFunc。它可以在请求到达最终处理器之前或之后执行逻辑。
例如，一个简单的日志中间件可能如下：func LoggingMiddleware(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
log.Printf("%s %s", r.Method, r.URL.Path)
next(w, r)
}
}
使用 httptest 测试中间件行为
要测试中间件，我们不需要启动真实服务器。使用 httptest.NewRecorder() 可以捕获响应，而 httptest.NewRequest() 用于构造模拟请求。
示例：测试一个身份验证中间件func AuthMiddleware(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
token := r.Header.Get("Authorization")
if token != "Bearer valid-token" {
http.Error(w, "Forbidden", http.StatusForbidden)
return
}
next(w, r)
}
}
对应的测试代码：
立即学习“go语言免费学习笔记（深入）”；
func TestAuthMiddleware_AllowsValidToken(t *testing.T) {
handler := AuthMiddleware(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
})
req := httptest.NewRequest("GET", "/protected", nil)
req.Header.Set("Authorization", "Bearer valid-token")
rec := httptest.NewRecorder()
handler(rec, req)
if rec.Code != http.StatusOK {
t.Errorf("Expected status OK; got %v", rec.Code)
}
}
func TestAuthMiddleware_RejectsInvalidToken(t *testing.T) {
handler := AuthMiddleware(func(w http.ResponseWriter, r *http.Request) {
t.Fatal("Handler should not be called")
})
req := httptest.NewRequest("GET", "/protected", nil)
req.Header.Set("Authorization", "Bearer invalid-token")
rec := httptest.NewRecorder()
handler(rec, req)
if rec.Code != http.StatusForbidden {
t.Errorf("Expected Forbidden; got %v", rec.Code)
}
}
测试中间件对请求或上下文的修改
有些中间件会向请求中注入数据（如用户信息），通常使用 context 来实现。测试时需要验证这些值是否正确设置。
示例中间件：func UserMiddleware(next http.HandlerFunc) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
ctx := context.WithValue(r.Context(), "user", "alice")
next(w, r.WithContext(ctx))
}
}
测试代码中可以检查上下文值：
func TestUserMiddleware_SetsUserInContext(t *testing.T) {
var receivedUser string
handler := UserMiddleware(func(w http.ResponseWriter, r *http.Request) {
receivedUser = r.Context().Value("user").(string)
w.WriteHeader(http.StatusOK)
})
req := httptest.NewRequest("GET", "/user", nil)
rec := httptest.NewRecorder()
handler(rec, req)
if receivedUser != "alice" {
t.Errorf("Expected user 'alice', got '%s'", receivedUser)
}
}
基本上就这些。只要把中间件当作可组合的函数来处理，用 httptest 构造请求和记录响应，就能全面覆盖各种场景。不复杂但容易忽略的是：确保测试了短路情况（如认证失败直接返回）以及上下文传递的完整性。
