diff --git a/jwtauth.go b/jwtauth.go index 38104fe..e6534e8 100644 --- a/jwtauth.go +++ b/jwtauth.go @@ -61,9 +61,7 @@ func New(alg string, signKey interface{}, verifyKey interface{}) *JWTAuth { // which checks the request context jwt token and error to prepare a custom // http response. func Verifier(ja *JWTAuth) func(http.Handler) http.Handler { - return func(next http.Handler) http.Handler { - return Verify(ja, TokenFromHeader, TokenFromCookie)(next) - } + return Verify(ja, TokenFromHeader, TokenFromCookie) } func Verify(ja *JWTAuth, findTokenFns ...func(r *http.Request) string) func(http.Handler) http.Handler { @@ -165,12 +163,12 @@ func Authenticator(next http.Handler) http.Handler { token, _, err := FromContext(r.Context()) if err != nil { - http.Error(w, err.Error(), 401) + http.Error(w, err.Error(), http.StatusUnauthorized) return } if token == nil || jwt.Validate(token) != nil { - http.Error(w, http.StatusText(401), 401) + http.Error(w, http.StatusText(http.StatusUnauthorized), http.StatusUnauthorized) return } diff --git a/jwtauth_test.go b/jwtauth_test.go index c8cd84c..dd2bc57 100644 --- a/jwtauth_test.go +++ b/jwtauth_test.go @@ -153,12 +153,12 @@ func TestMore(t *testing.T) { token, _, err := jwtauth.FromContext(r.Context()) if err != nil { - http.Error(w, jwtauth.ErrorReason(err).Error(), 401) + http.Error(w, jwtauth.ErrorReason(err).Error(), http.StatusUnauthorized) return } if err := jwt.Validate(token); err != nil { - http.Error(w, jwtauth.ErrorReason(err).Error(), 401) + http.Error(w, jwtauth.ErrorReason(err).Error(), http.StatusUnauthorized) return } @@ -242,10 +242,8 @@ func testRequest(t *testing.T, ts *httptest.Server, method, path string, header return 0, "" } - if header != nil { - for k, v := range header { - req.Header.Set(k, v[0]) - } + for k, v := range header { + req.Header.Set(k, v[0]) } resp, err := http.DefaultClient.Do(req)