From ac49b5bb707b0a85301345f9929e8839f6920d54 Mon Sep 17 00:00:00 2001 From: Lakhan Samani Date: Sat, 11 Jun 2022 19:24:53 +0530 Subject: [PATCH] fix: session tests --- server/test/logout_test.go | 12 +++++++++--- server/test/session_test.go | 13 +++++++++---- 2 files changed, 18 insertions(+), 7 deletions(-) diff --git a/server/test/logout_test.go b/server/test/logout_test.go index 18ca45a..736bd98 100644 --- a/server/test/logout_test.go +++ b/server/test/logout_test.go @@ -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) diff --git a/server/test/session_test.go b/server/test/session_test.go index affcde9..66e2c89 100644 --- a/server/test/session_test.go +++ b/server/test/session_test.go @@ -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)