fix: session tests

This commit is contained in:
Lakhan Samani 2022-06-11 19:24:53 +05:30
parent 926ab07c07
commit ac49b5bb70
2 changed files with 18 additions and 7 deletions

View File

@ -29,10 +29,16 @@ func logoutTests(t *testing.T, s TestSetup) {
})
token := *verifyRes.AccessToken
session, err := memorystore.Provider.GetUserSession(verifyRes.User.ID, token)
sessions, err := memorystore.Provider.GetAllUserSessions(verifyRes.User.ID)
assert.NoError(t, err)
assert.NotEmpty(t, session)
cookie := fmt.Sprintf("%s=%s;", constants.AppCookieName+"_session", session)
assert.NotEmpty(t, sessions)
cookie := ""
// set all they keys in cookie one of them should be session cookie
for key := range sessions {
if key != token {
cookie += fmt.Sprintf("%s=%s;", constants.AppCookieName+"_session", key)
}
}
cookie = strings.TrimSuffix(cookie, ";")
req.Header.Set("Cookie", cookie)

View File

@ -34,14 +34,19 @@ func sessionTests(t *testing.T, s TestSetup) {
})
token := *verifyRes.AccessToken
session, err := memorystore.Provider.GetUserSession(verifyRes.User.ID, token)
sessions, err := memorystore.Provider.GetAllUserSessions(verifyRes.User.ID)
assert.NoError(t, err)
assert.NotEmpty(t, session)
cookie := fmt.Sprintf("%s=%s;", constants.AppCookieName+"_session", session)
assert.NotEmpty(t, sessions)
cookie := ""
// set all they keys in cookie one of them should be session cookie
for key := range sessions {
if key != token {
cookie += fmt.Sprintf("%s=%s;", constants.AppCookieName+"_session", key)
}
}
cookie = strings.TrimSuffix(cookie, ";")
req.Header.Set("Cookie", cookie)
_, err = resolvers.SessionResolver(ctx, &model.SessionQueryInput{})
assert.Nil(t, err)