Function TestAuthFromAuthorizationHeader

pkg/middlewares/http/jwt_test.go:87–127

(t *testing.T)

}

func TestAuthFromAuthorizationHeader(t *testing.T) {
	validToken := generateValidToken()

	tests := []struct {
		name       string
		header     string
		wantStatus int
		wantClaims map[string]interface{}
	}{
		{"Valid Token", "Bearer " + validToken, http.StatusOK, map[string]interface{}{"foo": "bar"}},
		{"Missing Header", "", http.StatusUnauthorized, nil},
		{"Invalid Header Format", "Bearer", http.StatusUnauthorized, nil},
		{"Invalid Token", "Bearer invalidtoken", http.StatusUnauthorized, nil},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			req, _ := http.NewRequest("GET", "/", nil)
			if tt.header != "" {
				req.Header.Set(authorizationKey, tt.header)
			}
			rr := httptest.NewRecorder()
			handler := AuthFromAuthorizationHeader(mockKeyFunc, genericClaimsFunc(), mockSigningMethod, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				claims, ok := jwtmiddleware.FromContext(r.Context())
				mapClaims := claims.(*jwt.MapClaims)
				if tt.wantClaims != nil {
					assert.True(t, ok, "claims not found in context")
					for key, value := range tt.wantClaims {
						claimsVal, exists := (*mapClaims)[key]
						assert.True(t, exists, "claims missing key: %v", key)
						assert.Equal(t, value, claimsVal, "claims value mismatch for key: %v", key)
					}
				}
				w.WriteHeader(http.StatusOK)
			}))
			handler.ServeHTTP(rr, req)

			assert.Equal(t, tt.wantStatus, rr.Code, "handler returned wrong status code")
		})
	}
}

// CustomClaimsA represents the first custom JWT claims type
type CustomClaimsA struct {