mirror of
https://github.com/labstack/echo.git
synced 2024-11-24 08:22:21 +02:00
Minor refactor and format
Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
parent
2557d33c4b
commit
87dbea59f7
@ -28,6 +28,10 @@ type (
|
||||
// Optional. Default value "user".
|
||||
ContextKey string `json:"context_key"`
|
||||
|
||||
// Claims are extendable claims data defining token content.
|
||||
// Optional. Default value jwt.MapClaims
|
||||
Claims jwt.Claims
|
||||
|
||||
// TokenLookup is a string in the form of "<source>:<name>" that is used
|
||||
// to extract token from the request.
|
||||
// Optional. Default value "header:Authorization".
|
||||
@ -35,10 +39,6 @@ type (
|
||||
// - "header:<name>"
|
||||
// - "query:<name>"
|
||||
TokenLookup string `json:"token_lookup"`
|
||||
|
||||
// Claims are extendable claims data defining token content.
|
||||
// Optional. Default value jwt.MapClaims
|
||||
Claims jwt.Claims
|
||||
}
|
||||
|
||||
jwtExtractor func(echo.Context) (string, error)
|
||||
|
@ -10,16 +10,16 @@ import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
// CustomInfo defines some custom types we're going to use within our tokens
|
||||
type CustomInfo struct {
|
||||
// jwtCustomInfo defines some custom types we're going to use within our tokens.
|
||||
type jwtCustomInfo struct {
|
||||
Name string `json:"name"`
|
||||
Admin bool `json:"admin"`
|
||||
}
|
||||
|
||||
// MyCustomClaims are custom claims expanding default ones
|
||||
type MyCustomClaims struct {
|
||||
// jwtCustomClaims are custom claims expanding default ones.
|
||||
type jwtCustomClaims struct {
|
||||
*jwt.StandardClaims
|
||||
CustomInfo
|
||||
jwtCustomInfo
|
||||
}
|
||||
|
||||
func TestJWT(t *testing.T) {
|
||||
@ -63,13 +63,13 @@ func TestJWT(t *testing.T) {
|
||||
|
||||
// Valid JWT with custom claims
|
||||
config = JWTConfig{
|
||||
Claims: &MyCustomClaims{},
|
||||
Claims: &jwtCustomClaims{},
|
||||
SigningKey: []byte("secret"),
|
||||
}
|
||||
h = JWTWithConfig(config)(handler)
|
||||
if assert.NoError(t, h(c)) {
|
||||
user := c.Get("user").(*jwt.Token)
|
||||
claims := user.Claims.(*MyCustomClaims)
|
||||
claims := user.Claims.(*jwtCustomClaims)
|
||||
assert.Equal(t, claims.Name, "John Doe")
|
||||
assert.Equal(t, claims.Admin, true)
|
||||
}
|
||||
@ -79,5 +79,4 @@ func TestJWT(t *testing.T) {
|
||||
h = JWT([]byte("secret"))(handler)
|
||||
he = h(c).(*echo.HTTPError)
|
||||
assert.Equal(t, http.StatusBadRequest, he.Code)
|
||||
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user