go/echo
1
0
Fork 0
mirror of https://github.com/labstack/echo.git synced 2025-12-07 23:12:43 +02:00
Code Issues Releases Activity

Add custom jwt extractor to jwt config

Browse Source
This commit is contained in:
Rashad Ansari
2021-08-19 15:00:07 +02:00
committed by Martti T
parent 6b5e62b27e
commit 4fffee2ec8
2 changed files with 52 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
middleware/jwt_test.go
View File

@@ -603,3 +603,27 @@ func TestJWTConfig_custom_ParseTokenFunc_Keyfunc(t *testing.T) {
assert.Equal(t, http.StatusTeapot, res.Code)
}
func TestJWTConfig_TokenLookupFuncs(t *testing.T) {
e := echo.New()
e.GET("/", func(c echo.Context) error {
return c.String(http.StatusOK, "test")
})
e.Use(JWTWithConfig(JWTConfig{
TokenLookupFuncs: []TokenLookupFunc{
func(c echo.Context) (string, error) {
return c.Request().Header.Get("X-API-Key"), nil
},
},
SigningKey: []byte("secret"),
}))
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)
}