mirror of
https://github.com/labstack/echo.git
synced 2024-12-22 20:06:21 +02:00
667 lines
20 KiB
Go
667 lines
20 KiB
Go
package middleware
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/golang-jwt/jwt/v4"
|
|
"github.com/labstack/echo/v5"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func createTestParseTokenFuncForJWTGo(signingMethod string, signingKey interface{}) func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
// This is minimal implementation for github.com/golang-jwt/jwt as JWT parser library. good enough to get old tests running
|
|
keyFunc := func(t *jwt.Token) (interface{}, error) {
|
|
if t.Method.Alg() != signingMethod {
|
|
return nil, fmt.Errorf("unexpected jwt signing method=%v", t.Header["alg"])
|
|
}
|
|
return signingKey, nil
|
|
}
|
|
|
|
return func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
token, err := jwt.ParseWithClaims(auth, jwt.MapClaims{}, keyFunc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !token.Valid {
|
|
return nil, errors.New("invalid token")
|
|
}
|
|
return token, nil
|
|
}
|
|
}
|
|
|
|
// jwtCustomInfo defines some custom types we're going to use within our tokens.
|
|
type jwtCustomInfo struct {
|
|
Name string `json:"name"`
|
|
Admin bool `json:"admin"`
|
|
}
|
|
|
|
// jwtCustomClaims are custom claims expanding default ones.
|
|
type jwtCustomClaims struct {
|
|
*jwt.RegisteredClaims
|
|
jwtCustomInfo
|
|
}
|
|
|
|
func TestJWT(t *testing.T) {
|
|
e := echo.New()
|
|
|
|
e.GET("/", func(c echo.Context) error {
|
|
token := c.Get("user").(*jwt.Token)
|
|
return c.JSON(http.StatusOK, token.Claims)
|
|
})
|
|
|
|
e.Use(JWT(createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret"))))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Set(echo.HeaderAuthorization, "bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ")
|
|
res := httptest.NewRecorder()
|
|
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, http.StatusOK, res.Code)
|
|
assert.Equal(t, `{"admin":true,"name":"John Doe","sub":"1234567890"}`+"\n", res.Body.String())
|
|
}
|
|
|
|
func TestJWT_combinations(t *testing.T) {
|
|
e := echo.New()
|
|
handler := func(c echo.Context) error {
|
|
return c.String(http.StatusOK, "test")
|
|
}
|
|
token := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ"
|
|
validKey := []byte("secret")
|
|
invalidKey := []byte("invalid-key")
|
|
validAuth := "Bearer " + token
|
|
|
|
var testCases = []struct {
|
|
name string
|
|
config JWTConfig
|
|
reqURL string // "/" if empty
|
|
hdrAuth string
|
|
hdrCookie string // test.Request doesn't provide SetCookie(); use name=val
|
|
formValues map[string]string
|
|
expectPanic bool
|
|
expectToMiddlewareError string
|
|
expectError string
|
|
}{
|
|
{
|
|
name: "No signing key provided",
|
|
expectToMiddlewareError: "echo jwt middleware requires parse token function",
|
|
},
|
|
{
|
|
name: "invalid TokenLookup",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo("RS256", validKey),
|
|
TokenLookup: "q",
|
|
},
|
|
expectToMiddlewareError: "extractor source for lookup could not be split into needed parts: q",
|
|
},
|
|
{
|
|
name: "Unexpected signing method",
|
|
hdrAuth: validAuth,
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo("RS256", validKey),
|
|
},
|
|
expectError: "code=401, message=invalid or expired jwt, internal=unexpected jwt signing method=HS256",
|
|
},
|
|
{
|
|
name: "Invalid key",
|
|
hdrAuth: validAuth,
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, invalidKey),
|
|
},
|
|
expectError: "code=401, message=invalid or expired jwt, internal=signature is invalid",
|
|
},
|
|
{
|
|
name: "Valid JWT",
|
|
hdrAuth: validAuth,
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
},
|
|
},
|
|
{
|
|
name: "Valid JWT with custom AuthScheme",
|
|
hdrAuth: "Token" + " " + token,
|
|
config: JWTConfig{
|
|
TokenLookup: "header:" + echo.HeaderAuthorization + ":Token ",
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
},
|
|
},
|
|
{
|
|
name: "Valid JWT with custom claims",
|
|
hdrAuth: validAuth,
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret")),
|
|
},
|
|
},
|
|
{
|
|
name: "Invalid Authorization header",
|
|
hdrAuth: "invalid-auth",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
},
|
|
expectError: "code=401, message=missing or malformed jwt, internal=invalid value in request header",
|
|
},
|
|
{
|
|
name: "Empty header auth field",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
},
|
|
expectError: "code=401, message=missing or malformed jwt, internal=invalid value in request header",
|
|
},
|
|
{
|
|
name: "Valid query method",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "query:jwt",
|
|
},
|
|
reqURL: "/?a=b&jwt=" + token,
|
|
},
|
|
{
|
|
name: "Invalid query param name",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "query:jwt",
|
|
},
|
|
reqURL: "/?a=b&jwtxyz=" + token,
|
|
expectError: "code=401, message=missing or malformed jwt, internal=missing value in the query string",
|
|
},
|
|
{
|
|
name: "Invalid query param value",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "query:jwt",
|
|
},
|
|
reqURL: "/?a=b&jwt=invalid-token",
|
|
expectError: "code=401, message=invalid or expired jwt, internal=token contains an invalid number of segments",
|
|
},
|
|
{
|
|
name: "Empty query",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "query:jwt",
|
|
},
|
|
reqURL: "/?a=b",
|
|
expectError: "code=401, message=missing or malformed jwt, internal=missing value in the query string",
|
|
},
|
|
{
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "param:jwt",
|
|
},
|
|
reqURL: "/" + token,
|
|
name: "Valid param method",
|
|
},
|
|
{
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "cookie:jwt",
|
|
},
|
|
hdrCookie: "jwt=" + token,
|
|
name: "Valid cookie method",
|
|
},
|
|
{
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "query:jwt,cookie:jwt",
|
|
},
|
|
hdrCookie: "jwt=" + token,
|
|
name: "Multiple jwt lookuop",
|
|
},
|
|
{
|
|
name: "Invalid token with cookie method",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "cookie:jwt",
|
|
},
|
|
hdrCookie: "jwt=invalid",
|
|
expectError: "code=401, message=invalid or expired jwt, internal=token contains an invalid number of segments",
|
|
},
|
|
{
|
|
name: "Empty cookie",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "cookie:jwt",
|
|
},
|
|
expectError: "code=401, message=missing or malformed jwt, internal=missing value in cookies",
|
|
},
|
|
{
|
|
name: "Valid form method",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "form:jwt",
|
|
},
|
|
formValues: map[string]string{"jwt": token},
|
|
},
|
|
{
|
|
name: "Invalid token with form method",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "form:jwt",
|
|
},
|
|
formValues: map[string]string{"jwt": "invalid"},
|
|
expectError: "code=401, message=invalid or expired jwt, internal=token contains an invalid number of segments",
|
|
},
|
|
{
|
|
name: "Empty form field",
|
|
config: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, validKey),
|
|
TokenLookup: "form:jwt",
|
|
},
|
|
expectError: "code=401, message=missing or malformed jwt, internal=missing value in the form",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
if tc.reqURL == "" {
|
|
tc.reqURL = "/"
|
|
}
|
|
|
|
var req *http.Request
|
|
if len(tc.formValues) > 0 {
|
|
form := url.Values{}
|
|
for k, v := range tc.formValues {
|
|
form.Set(k, v)
|
|
}
|
|
req = httptest.NewRequest(http.MethodPost, tc.reqURL, strings.NewReader(form.Encode()))
|
|
req.Header.Set(echo.HeaderContentType, "application/x-www-form-urlencoded")
|
|
req.ParseForm()
|
|
} else {
|
|
req = httptest.NewRequest(http.MethodGet, tc.reqURL, nil)
|
|
}
|
|
res := httptest.NewRecorder()
|
|
req.Header.Set(echo.HeaderAuthorization, tc.hdrAuth)
|
|
req.Header.Set(echo.HeaderCookie, tc.hdrCookie)
|
|
c := e.NewContext(req, res)
|
|
|
|
if tc.reqURL == "/"+token {
|
|
cc := c.(echo.ServableContext)
|
|
cc.SetPathParams(echo.PathParams{
|
|
{Name: "jwt", Value: token},
|
|
})
|
|
}
|
|
|
|
mw, err := tc.config.ToMiddleware()
|
|
if tc.expectToMiddlewareError != "" {
|
|
assert.EqualError(t, err, tc.expectToMiddlewareError)
|
|
return
|
|
}
|
|
|
|
hErr := mw(handler)(c)
|
|
if tc.expectError != "" {
|
|
assert.EqualError(t, hErr, tc.expectError)
|
|
return
|
|
}
|
|
assert.NoError(t, hErr)
|
|
|
|
user := c.Get("user").(*jwt.Token)
|
|
switch claims := user.Claims.(type) {
|
|
case jwt.MapClaims:
|
|
assert.Equal(t, claims["name"], "John Doe")
|
|
case *jwtCustomClaims:
|
|
assert.Equal(t, claims.Name, "John Doe")
|
|
assert.Equal(t, claims.Admin, true)
|
|
default:
|
|
panic("unexpected type of claims")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJWTConfig_skipper(t *testing.T) {
|
|
e := echo.New()
|
|
|
|
e.Use(JWTWithConfig(JWTConfig{
|
|
Skipper: func(context echo.Context) bool {
|
|
return true // skip everything
|
|
},
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret")),
|
|
}))
|
|
|
|
isCalled := false
|
|
e.GET("/", func(c echo.Context) error {
|
|
isCalled = true
|
|
return c.String(http.StatusTeapot, "test")
|
|
})
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, http.StatusTeapot, res.Code)
|
|
assert.True(t, isCalled)
|
|
}
|
|
|
|
func TestJWTConfig_BeforeFunc(t *testing.T) {
|
|
e := echo.New()
|
|
e.GET("/", func(c echo.Context) error {
|
|
return c.String(http.StatusTeapot, "test")
|
|
})
|
|
|
|
isCalled := false
|
|
e.Use(JWTWithConfig(JWTConfig{
|
|
BeforeFunc: func(context echo.Context) {
|
|
isCalled = true
|
|
},
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret")),
|
|
}))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Set(echo.HeaderAuthorization, "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ")
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, http.StatusTeapot, res.Code)
|
|
assert.True(t, isCalled)
|
|
}
|
|
|
|
func TestJWTConfig_extractorErrorHandling(t *testing.T) {
|
|
var testCases = []struct {
|
|
name string
|
|
given JWTConfig
|
|
expectStatusCode int
|
|
}{
|
|
{
|
|
name: "ok, ErrorHandler is executed",
|
|
given: JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret")),
|
|
ErrorHandler: func(c echo.Context, err error) error {
|
|
return echo.NewHTTPError(http.StatusTeapot, "custom_error")
|
|
},
|
|
},
|
|
expectStatusCode: http.StatusTeapot,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
e := echo.New()
|
|
e.GET("/", func(c echo.Context) error {
|
|
return c.String(http.StatusNotImplemented, "should not end up here")
|
|
})
|
|
|
|
e.Use(JWTWithConfig(tc.given))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, tc.expectStatusCode, res.Code)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJWTConfig_parseTokenErrorHandling(t *testing.T) {
|
|
var testCases = []struct {
|
|
name string
|
|
given JWTConfig
|
|
expectErr string
|
|
}{
|
|
{
|
|
name: "ok, ErrorHandler is executed",
|
|
given: JWTConfig{
|
|
ErrorHandler: func(c echo.Context, err error) error {
|
|
return echo.NewHTTPError(http.StatusTeapot, "ErrorHandler: "+err.Error())
|
|
},
|
|
},
|
|
expectErr: "{\"message\":\"ErrorHandler: parsing failed\"}\n",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
e := echo.New()
|
|
//e.Debug = true
|
|
e.GET("/", func(c echo.Context) error {
|
|
return c.String(http.StatusNotImplemented, "should not end up here")
|
|
})
|
|
|
|
config := tc.given
|
|
parseTokenCalled := false
|
|
config.ParseTokenFunc = func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
parseTokenCalled = true
|
|
return nil, errors.New("parsing failed")
|
|
}
|
|
e.Use(JWTWithConfig(config))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Set(echo.HeaderAuthorization, "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ")
|
|
res := httptest.NewRecorder()
|
|
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, http.StatusTeapot, res.Code)
|
|
assert.Equal(t, tc.expectErr, res.Body.String())
|
|
assert.True(t, parseTokenCalled)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJWTConfig_custom_ParseTokenFunc_Keyfunc(t *testing.T) {
|
|
e := echo.New()
|
|
e.GET("/", func(c echo.Context) error {
|
|
return c.String(http.StatusTeapot, "test")
|
|
})
|
|
|
|
// example of minimal custom ParseTokenFunc implementation. Allows you to use different versions of `github.com/golang-jwt/jwt`
|
|
// with current JWT middleware
|
|
signingKey := []byte("secret")
|
|
|
|
var fromSource ExtractorSource
|
|
config := JWTConfig{
|
|
ParseTokenFunc: func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
fromSource = source
|
|
keyFunc := func(t *jwt.Token) (interface{}, error) {
|
|
if t.Method.Alg() != "HS256" {
|
|
return nil, fmt.Errorf("unexpected jwt signing method=%v", t.Header["alg"])
|
|
}
|
|
return signingKey, nil
|
|
}
|
|
|
|
// claims are of type `jwt.MapClaims` when token is created with `jwt.Parse`
|
|
token, err := jwt.Parse(auth, keyFunc)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !token.Valid {
|
|
return nil, errors.New("invalid token")
|
|
}
|
|
return token, nil
|
|
},
|
|
}
|
|
|
|
e.Use(JWTWithConfig(config))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Set(echo.HeaderAuthorization, "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ")
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, fromSource, ExtractorSourceHeader)
|
|
assert.Equal(t, http.StatusTeapot, res.Code)
|
|
}
|
|
|
|
func TestMustJWTWithConfig_SuccessHandler(t *testing.T) {
|
|
e := echo.New()
|
|
|
|
e.GET("/", func(c echo.Context) error {
|
|
success := c.Get("success").(string)
|
|
user := c.Get("user").(string)
|
|
return c.String(http.StatusTeapot, fmt.Sprintf("%v:%v", success, user))
|
|
})
|
|
|
|
mw, err := JWTConfig{
|
|
ParseTokenFunc: func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
return auth, nil
|
|
},
|
|
SuccessHandler: func(c echo.Context) {
|
|
c.Set("success", "yes")
|
|
},
|
|
}.ToMiddleware()
|
|
assert.NoError(t, err)
|
|
e.Use(mw)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Add(echo.HeaderAuthorization, "Bearer valid_token_base64")
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, "yes:valid_token_base64", res.Body.String())
|
|
assert.Equal(t, http.StatusTeapot, res.Code)
|
|
}
|
|
|
|
func TestJWTWithConfig_ContinueOnIgnoredError(t *testing.T) {
|
|
var testCases = []struct {
|
|
name string
|
|
givenContinueOnIgnoredError bool
|
|
givenErrorHandler JWTErrorHandlerWithContext
|
|
givenTokenLookup string
|
|
whenAuthHeaders []string
|
|
whenCookies []string
|
|
whenParseReturn string
|
|
whenParseError error
|
|
expectHandlerCalled bool
|
|
expect string
|
|
expectCode int
|
|
}{
|
|
{
|
|
name: "ok, with valid JWT from auth header",
|
|
givenContinueOnIgnoredError: true,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
return nil
|
|
},
|
|
whenAuthHeaders: []string{"Bearer valid_token_base64"},
|
|
whenParseReturn: "valid_token",
|
|
expectCode: http.StatusTeapot,
|
|
expect: "valid_token",
|
|
},
|
|
{
|
|
name: "ok, missing header, callNext and set public_token from error handler",
|
|
givenContinueOnIgnoredError: true,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
if errors.Is(err, &ValueExtractorError{}) {
|
|
panic("must get ErrJWTMissing")
|
|
}
|
|
c.Set("user", "public_token")
|
|
return nil
|
|
},
|
|
whenAuthHeaders: []string{}, // no JWT header
|
|
expectCode: http.StatusTeapot,
|
|
expect: "public_token",
|
|
},
|
|
{
|
|
name: "ok, invalid token, callNext and set public_token from error handler",
|
|
givenContinueOnIgnoredError: true,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
// this is probably not realistic usecase. on parse error you probably want to return error
|
|
if err.Error() != "parser_error" {
|
|
panic("must get parser_error")
|
|
}
|
|
c.Set("user", "public_token")
|
|
return nil
|
|
},
|
|
whenAuthHeaders: []string{"Bearer invalid_header"},
|
|
whenParseError: errors.New("parser_error"),
|
|
expectCode: http.StatusTeapot,
|
|
expect: "public_token",
|
|
},
|
|
{
|
|
name: "nok, invalid token, return error from error handler",
|
|
givenContinueOnIgnoredError: true,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
if err.Error() != "parser_error" {
|
|
panic("must get parser_error")
|
|
}
|
|
return err
|
|
},
|
|
whenAuthHeaders: []string{"Bearer invalid_header"},
|
|
whenParseError: errors.New("parser_error"),
|
|
expectCode: http.StatusInternalServerError,
|
|
expect: "{\"message\":\"Internal Server Error\"}\n",
|
|
},
|
|
{
|
|
name: "nok, ContinueOnIgnoredError but return error from error handler",
|
|
givenContinueOnIgnoredError: true,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
return echo.ErrUnauthorized.WithInternal(err)
|
|
},
|
|
whenAuthHeaders: []string{}, // no JWT header
|
|
expectCode: http.StatusUnauthorized,
|
|
expect: "{\"message\":\"Unauthorized\"}\n",
|
|
},
|
|
{
|
|
name: "nok, ContinueOnIgnoredError=false",
|
|
givenContinueOnIgnoredError: false,
|
|
givenErrorHandler: func(c echo.Context, err error) error {
|
|
return echo.ErrUnauthorized.WithInternal(err)
|
|
},
|
|
whenAuthHeaders: []string{}, // no JWT header
|
|
expectCode: http.StatusUnauthorized,
|
|
expect: "{\"message\":\"Unauthorized\"}\n",
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
e := echo.New()
|
|
|
|
e.GET("/", func(c echo.Context) error {
|
|
token := c.Get("user").(string)
|
|
return c.String(http.StatusTeapot, token)
|
|
})
|
|
|
|
mw, err := JWTConfig{
|
|
ContinueOnIgnoredError: tc.givenContinueOnIgnoredError,
|
|
TokenLookup: tc.givenTokenLookup,
|
|
ParseTokenFunc: func(c echo.Context, auth string, source ExtractorSource) (interface{}, error) {
|
|
return tc.whenParseReturn, tc.whenParseError
|
|
},
|
|
ErrorHandler: tc.givenErrorHandler,
|
|
}.ToMiddleware()
|
|
assert.NoError(t, err)
|
|
e.Use(mw)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
for _, a := range tc.whenAuthHeaders {
|
|
req.Header.Add(echo.HeaderAuthorization, a)
|
|
}
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, tc.expect, res.Body.String())
|
|
assert.Equal(t, tc.expectCode, res.Code)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestJWTConfig_TokenLookupFuncs(t *testing.T) {
|
|
e := echo.New()
|
|
|
|
e.GET("/", func(c echo.Context) error {
|
|
token := c.Get("user").(*jwt.Token)
|
|
return c.JSON(http.StatusOK, token.Claims)
|
|
})
|
|
|
|
e.Use(JWTWithConfig(JWTConfig{
|
|
ParseTokenFunc: createTestParseTokenFuncForJWTGo(AlgorithmHS256, []byte("secret")),
|
|
TokenLookupFuncs: []ValuesExtractor{
|
|
func(c echo.Context) ([]string, ExtractorSource, error) {
|
|
return []string{c.Request().Header.Get("X-API-Key")}, ExtractorSourceCustom, nil
|
|
},
|
|
},
|
|
}))
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
|
req.Header.Set("X-API-Key", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiIxMjM0NTY3ODkwIiwibmFtZSI6IkpvaG4gRG9lIiwiYWRtaW4iOnRydWV9.TJVA95OrM7E2cBab30RMHrHDcEfxjoYZgeFONFh7HgQ")
|
|
res := httptest.NewRecorder()
|
|
e.ServeHTTP(res, req)
|
|
|
|
assert.Equal(t, http.StatusOK, res.Code)
|
|
assert.Equal(t, `{"admin":true,"name":"John Doe","sub":"1234567890"}`+"\n", res.Body.String())
|
|
}
|