scriptoriumadmin/featurebase
1
0
Fork 0
mirror of https://github.com/featurebasedb/featurebase.git synced 2026-09-11 07:11:02 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

tests

Browse source
This commit is contained in:
Samir Patel 2021-12-16 23:39:58 -06:00
parent e5866f5f9c
commit db2263465b
2 changed files with 37 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
authn/authenticate.go
View file

@ -134,11 +134,14 @@ func (a *Auth) Redirect(w http.ResponseWriter, r *http.Request) {
token, err := a.getToken(code)
if err != nil {
errors.Wrap(err, "getting token")
http.Redirect(w, r, "/login", http.StatusTemporaryRedirect)
http.Redirect(w, r, "/login", http.StatusUnauthorized)
}
cv, err := a.newCookieValue(token)
if err != nil {
http.Error(w, "authenticating", http.StatusBadRequest)
}
fmt.Printf("TOKEN %v\n\n", token)
cv := a.newCookieValue(token)
a.setCookie(w, cv)
http.Redirect(w, r, "/", http.StatusTemporaryRedirect)
}
@ -164,7 +167,10 @@ func (a *Auth) getToken(code string) (*oauth2.Token, error) {
return token, nil
}
func (a *Auth) newCookieValue(token *oauth2.Token) *CookieValue {
func (a *Auth) newCookieValue(token *oauth2.Token) (*CookieValue, error) {
if token == nil {
return nil, errors.New("baking cookie due to nil token")
}
accessParsed, err := jwt.Parse(token.AccessToken, nil)
if token == nil {
fmt.Println(errors.Wrap(err, "parsing jwt claims from access tokens"))
@ -175,7 +181,7 @@ func (a *Auth) newCookieValue(token *oauth2.Token) *CookieValue {
if err != nil {
fmt.Println(errors.Wrap(err, "getting group memebership"))
}
// not needed anymore, and makes the encoded cookie too large
// not needed at this point in the logic and makes the encoded cookie too large
token.AccessToken = ""
// mannually setting expiry for testing ... REMOVE
token.Expiry = time.Now().Add(time.Second * time.Duration(30))
@ -184,7 +190,7 @@ func (a *Auth) newCookieValue(token *oauth2.Token) *CookieValue {
UserName: claims["name"].(string),
GroupMembership: groups.Groups,
Token: token,
}
}, nil
}
func (a *Auth) getGroupMembership(token *oauth2.Token) (Groups, error) {
@ -258,7 +264,11 @@ func (a *Auth) refreshToken(w http.ResponseWriter, cookie *CookieValue) error {
fmt.Printf("Refreshed AT: %v\n\n", newToken.AccessToken)
if newToken.Expiry != cookie.Token.Expiry {
cv := a.newCookieValue(newToken)
cv, err := a.newCookieValue(newToken)
if err != nil {
errors.New("setting cookie")
}
a.setCookie(w, cv)
fmt.Println("refreshed access token")
}

21
authn/authenticate_test.go
View file

@ -67,7 +67,6 @@ func TestAuth(t *testing.T) {
}
})
// t.Run("Logout", func(t *testing.T) {
// r := httptest.NewRequest(gohttp.MethodGet, "/login", nil)
// w := httptest.NewRecorder()
@ -99,4 +98,24 @@ func TestAuth(t *testing.T) {
// })
t.Run("Logout", func(t *testing.T) {
r := httptest.NewRequest(gohttp.MethodGet, "/login", nil)
w := httptest.NewRecorder()
a.Logout(w, r)
})
t.Run("Authenticate", func(t *testing.T) {
r := httptest.NewRequest(gohttp.MethodGet, "/login", nil)
w := httptest.NewRecorder()
a.Authenticate(w, r)
})
t.Run("Redirect", func(t *testing.T) {
r := httptest.NewRequest(gohttp.MethodGet, "/login", nil)
w := httptest.NewRecorder()
a.Redirect(w, r)
})
t.Run("GetUserInfo", func(t *testing.T) {
r := httptest.NewRequest(gohttp.MethodGet, "/login", nil)
a.GetUserInfo(r)
})
}