如何在Golang中测试HTTP中间件逻辑_Golang中间件测试示例

3次阅读

go中测试http中间件需解耦中间件与路由,用httptest模拟请求响应;验证其修改请求头、拦截访问及上下文注入等行为,重点覆盖短路逻辑与上下文完整性。

如何在Golang中测试HTTP中间件逻辑_Golang中间件测试示例

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语言免费学习笔记(深入)”;

如何在Golang中测试HTTP中间件逻辑_Golang中间件测试示例

美图AI开放平台

美图推出的AI人脸图像处理平台

如何在Golang中测试HTTP中间件逻辑_Golang中间件测试示例 102

查看详情 如何在Golang中测试HTTP中间件逻辑_Golang中间件测试示例

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 构造请求和记录响应,就能全面覆盖各种场景。不复杂但容易忽略的是:确保测试了短路情况(如认证失败直接返回)以及上下文传递的完整性。

text=ZqhQzanResources