| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865 |
- package handler
- import (
- "io"
- "testing"
- "github.com/google/uuid"
- "github.com/mk6i/retro-aim-server/config"
- "github.com/mk6i/retro-aim-server/oscar"
- "github.com/mk6i/retro-aim-server/state"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/mock"
- )
- func TestAuthService_BUCPLoginRequestHandler(t *testing.T) {
- sessUUID := uuid.UUID{1, 2, 3}
- user := state.User{
- ScreenName: "screen_name",
- AuthKey: "auth_key",
- }
- assert.NoError(t, user.HashPassword("the_password"))
- userSession := newTestSession(user.ScreenName, sessOptID(sessUUID.String()))
- cases := []struct {
- // name is the unit test name
- name string
- // cfg is the app configuration
- cfg config.Config
- // inputSNAC is the SNAC sent from the client to the server
- inputSNAC oscar.SNAC_0x17_0x02_BUCPLoginRequest
- // mockParams is the list of params sent to mocks that satisfy this
- // method's dependencies
- mockParams mockParams
- // newUserFn is the function that registers a new user account
- newUserFn func(screenName string) (state.User, error)
- // expectOutput is the SNAC sent from the server to client
- expectOutput oscar.SNACMessage
- // wantErr is the error we expect from the method
- wantErr error
- }{
- {
- name: "user provides valid credentials and logs in successfully",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, user.PassHash),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: &user,
- },
- },
- },
- sessionManagerParams: sessionManagerParams{
- addSessionParams: addSessionParams{
- {
- sessID: userSession.ID(),
- screenName: user.ScreenName,
- result: userSession,
- },
- },
- },
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPLoginResponse,
- },
- Body: oscar.SNAC_0x17_0x03_BUCPLoginResponse{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- oscar.NewTLV(oscar.TLVReconnectHere, "127.0.0.1:1234"),
- oscar.NewTLV(oscar.TLVAuthorizationCookie, sessUUID.String()),
- },
- },
- },
- },
- },
- {
- name: "user logs in with non-existent screen name--account is created and logged in successfully",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- DisableAuth: true,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, user.PassHash),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: nil,
- },
- },
- upsertUserParams: upsertUserParams{
- {
- user: user,
- },
- },
- },
- sessionManagerParams: sessionManagerParams{
- addSessionParams: addSessionParams{
- {
- sessID: userSession.ID(),
- screenName: user.ScreenName,
- result: userSession,
- },
- },
- },
- },
- newUserFn: func(screenName string) (state.User, error) {
- return user, nil
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPLoginResponse,
- },
- Body: oscar.SNAC_0x17_0x03_BUCPLoginResponse{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- oscar.NewTLV(oscar.TLVReconnectHere, "127.0.0.1:1234"),
- oscar.NewTLV(oscar.TLVAuthorizationCookie, sessUUID.String()),
- },
- },
- },
- },
- },
- {
- name: "user logs in with invalid password--account is created and logged in successfully",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- DisableAuth: true,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, []byte("bad-password-hash")),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: &user,
- },
- },
- upsertUserParams: upsertUserParams{
- {
- user: user,
- },
- },
- },
- sessionManagerParams: sessionManagerParams{
- addSessionParams: addSessionParams{
- {
- sessID: userSession.ID(),
- screenName: user.ScreenName,
- result: userSession,
- },
- },
- },
- },
- newUserFn: func(screenName string) (state.User, error) {
- return user, nil
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPLoginResponse,
- },
- Body: oscar.SNAC_0x17_0x03_BUCPLoginResponse{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- oscar.NewTLV(oscar.TLVReconnectHere, "127.0.0.1:1234"),
- oscar.NewTLV(oscar.TLVAuthorizationCookie, sessUUID.String()),
- },
- },
- },
- },
- },
- {
- name: "user provides invalid password--account creation fails due to user creation runtime error",
- cfg: config.Config{
- DisableAuth: true,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, []byte("bad-password-hash")),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: &user,
- },
- },
- },
- },
- newUserFn: func(screenName string) (state.User, error) {
- return user, io.EOF
- },
- wantErr: io.EOF,
- },
- {
- name: "user provides invalid password--account creation fails due to user upsert runtime error",
- cfg: config.Config{
- DisableAuth: true,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, []byte("bad-password-hash")),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: &user,
- },
- },
- upsertUserParams: upsertUserParams{
- {
- user: user,
- err: io.EOF,
- },
- },
- },
- },
- newUserFn: func(screenName string) (state.User, error) {
- return user, nil
- },
- wantErr: io.EOF,
- },
- {
- name: "user provides invalid password and receives invalid login response",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- },
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, []byte("bad_password")),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- result: nil,
- },
- },
- },
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPLoginResponse,
- },
- Body: oscar.SNAC_0x17_0x03_BUCPLoginResponse{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- oscar.NewTLV(oscar.TLVErrorSubcode, uint16(0x01)),
- },
- },
- },
- },
- },
- {
- name: "login fails on user manager lookup",
- inputSNAC: oscar.SNAC_0x17_0x02_BUCPLoginRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVPasswordHash, user.PassHash),
- oscar.NewTLV(oscar.TLVScreenName, user.ScreenName),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: user.ScreenName,
- err: io.EOF,
- },
- },
- },
- },
- wantErr: io.EOF,
- },
- }
- for _, tc := range cases {
- t.Run(tc.name, func(t *testing.T) {
- userManager := newMockUserManager(t)
- for _, params := range tc.mockParams.getUserParams {
- userManager.EXPECT().
- User(params.screenName).
- Return(params.result, params.err)
- }
- for _, params := range tc.mockParams.upsertUserParams {
- userManager.EXPECT().
- InsertUser(params.user).
- Return(params.err)
- }
- sessionManager := newMockSessionManager(t)
- for _, params := range tc.mockParams.addSessionParams {
- sessionManager.EXPECT().
- AddSession(params.sessID, params.screenName).
- Return(params.result)
- }
- svc := AuthService{
- config: tc.cfg,
- sessionManager: sessionManager,
- userManager: userManager,
- }
- fnNewUUID := func() uuid.UUID {
- return sessUUID
- }
- outputSNAC, err := svc.BUCPLoginRequestHandler(tc.inputSNAC, fnNewUUID, tc.newUserFn)
- assert.ErrorIs(t, err, tc.wantErr)
- assert.Equal(t, tc.expectOutput, outputSNAC)
- })
- }
- }
- func TestAuthService_BUCPChallengeRequestHandler(t *testing.T) {
- sessUUID := uuid.UUID{1, 2, 3}
- cases := []struct {
- // name is the unit test name
- name string
- // cfg is the app configuration
- cfg config.Config
- // inputSNAC is the SNAC sent from the client to the server
- inputSNAC oscar.SNAC_0x17_0x06_BUCPChallengeRequest
- // mockParams is the list of params sent to mocks that satisfy this
- // method's dependencies
- mockParams mockParams
- // expectOutput is the SNAC sent from the server to client
- expectOutput oscar.SNACMessage
- // wantErr is the error we expect from the method
- wantErr error
- }{
- {
- name: "login with valid username, expect OK login response",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- },
- inputSNAC: oscar.SNAC_0x17_0x06_BUCPChallengeRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, "sn_user_a"),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: "sn_user_a",
- result: &state.User{
- ScreenName: "sn_user_a",
- AuthKey: "auth_key_user_a",
- },
- },
- },
- },
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPChallengeResponse,
- },
- Body: oscar.SNAC_0x17_0x07_BUCPChallengeResponse{
- AuthKey: "auth_key_user_a",
- },
- },
- },
- {
- name: "login with invalid username, expect OK login response (Cfg.DisableAuth=true)",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- DisableAuth: true,
- },
- inputSNAC: oscar.SNAC_0x17_0x06_BUCPChallengeRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, "sn_user_b"),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: "sn_user_b",
- result: nil,
- },
- },
- },
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPChallengeResponse,
- },
- Body: oscar.SNAC_0x17_0x07_BUCPChallengeResponse{
- AuthKey: sessUUID.String(),
- },
- },
- },
- {
- name: "login with invalid username, expect failed login response (Cfg.DisableAuth=false)",
- cfg: config.Config{
- OSCARHost: "127.0.0.1",
- BOSPort: 1234,
- },
- inputSNAC: oscar.SNAC_0x17_0x06_BUCPChallengeRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, "sn_user_b"),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: "sn_user_b",
- result: nil,
- },
- },
- },
- },
- expectOutput: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.BUCP,
- SubGroup: oscar.BUCPLoginResponse,
- },
- Body: oscar.SNAC_0x17_0x03_BUCPLoginResponse{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVErrorSubcode, uint16(0x01)),
- },
- },
- },
- },
- },
- {
- name: "login fails on user manager lookup",
- inputSNAC: oscar.SNAC_0x17_0x06_BUCPChallengeRequest{
- TLVRestBlock: oscar.TLVRestBlock{
- TLVList: oscar.TLVList{
- oscar.NewTLV(oscar.TLVScreenName, "sn_user_b"),
- },
- },
- },
- mockParams: mockParams{
- userManagerParams: userManagerParams{
- getUserParams: getUserParams{
- {
- screenName: "sn_user_b",
- err: io.EOF,
- },
- },
- },
- },
- wantErr: io.EOF,
- },
- }
- for _, tc := range cases {
- t.Run(tc.name, func(t *testing.T) {
- userManager := newMockUserManager(t)
- for _, params := range tc.mockParams.getUserParams {
- userManager.EXPECT().
- User(params.screenName).
- Return(params.result, params.err)
- }
- svc := AuthService{
- config: tc.cfg,
- userManager: userManager,
- }
- fnNewUUID := func() uuid.UUID {
- return sessUUID
- }
- outputSNAC, err := svc.BUCPChallengeRequestHandler(tc.inputSNAC, fnNewUUID)
- assert.ErrorIs(t, err, tc.wantErr)
- assert.Equal(t, tc.expectOutput, outputSNAC)
- })
- }
- }
- func TestAuthService_RetrieveChatSession_HappyPath(t *testing.T) {
- chatID := "chat-1234"
- sess := newTestSession("screen-name", sessOptCannedID)
- sessionManager := newMockSessionManager(t)
- sessionManager.EXPECT().
- RetrieveSession(sess.ID()).
- Return(sess)
- chatRegistry := newMockChatRegistry(t)
- chatRegistry.EXPECT().
- Retrieve(chatID).
- Return(state.ChatRoom{}, sessionManager, nil)
- svc := NewAuthService(config.Config{}, nil, nil, nil, nil, chatRegistry)
- have, err := svc.RetrieveChatSession(chatID, sess.ID())
- assert.NoError(t, err)
- assert.Equal(t, sess, have)
- }
- func TestAuthService_RetrieveChatSession_ChatNotFound(t *testing.T) {
- chatID := "chat-1234"
- sess := newTestSession("screen-name", sessOptCannedID)
- chatRegistry := newMockChatRegistry(t)
- chatRegistry.EXPECT().
- Retrieve(chatID).
- Return(state.ChatRoom{}, nil, state.ErrChatRoomNotFound)
- svc := NewAuthService(config.Config{}, nil, nil, nil, nil, chatRegistry)
- _, err := svc.RetrieveChatSession(chatID, sess.ID())
- assert.ErrorIs(t, err, state.ErrChatRoomNotFound)
- }
- func TestAuthService_RetrieveChatSession_SessionNotFound(t *testing.T) {
- chatID := "chat-1234"
- sess := newTestSession("screen-name", sessOptCannedID)
- sessionManager := newMockSessionManager(t)
- sessionManager.EXPECT().
- RetrieveSession(sess.ID()).
- Return(nil)
- chatRegistry := newMockChatRegistry(t)
- chatRegistry.EXPECT().
- Retrieve(chatID).
- Return(state.ChatRoom{}, sessionManager, nil)
- svc := NewAuthService(config.Config{}, nil, nil, nil, nil, chatRegistry)
- have, err := svc.RetrieveChatSession(chatID, sess.ID())
- assert.NoError(t, err)
- assert.Nil(t, have)
- }
- func TestAuthService_RetrieveBOSSession_HappyPath(t *testing.T) {
- sess := newTestSession("screen-name", sessOptCannedID)
- sessionManager := newMockSessionManager(t)
- sessionManager.EXPECT().
- RetrieveSession(sess.ID()).
- Return(sess)
- svc := NewAuthService(config.Config{}, sessionManager, nil, nil, nil, nil)
- have, err := svc.RetrieveBOSSession(sess.ID())
- assert.NoError(t, err)
- assert.Equal(t, sess, have)
- }
- func TestAuthService_RetrieveBOSSession_SessionNotFound(t *testing.T) {
- sess := newTestSession("screen-name", sessOptCannedID)
- sessionManager := newMockSessionManager(t)
- sessionManager.EXPECT().
- RetrieveSession(sess.ID()).
- Return(nil)
- svc := NewAuthService(config.Config{}, sessionManager, nil, nil, nil, nil)
- have, err := svc.RetrieveBOSSession(sess.ID())
- assert.NoError(t, err)
- assert.Nil(t, have)
- }
- func TestAuthService_SignoutChat(t *testing.T) {
- sess := newTestSession("", sessOptCannedSignonTime)
- tests := []struct {
- // name is the unit test name
- name string
- // userSession is the session of the user signing out
- userSession *state.Session
- // chatRoom is the chat room user is exiting
- chatRoom state.ChatRoom
- // wantErr is the error we expect from the method
- wantErr error
- // mockParams is the list of params sent to mocks that satisfy this
- // method's dependencies
- mockParams mockParams
- }{
- {
- name: "user signs out of chat room, room is empty after user leaves",
- userSession: sess,
- chatRoom: state.ChatRoom{
- Cookie: "the-chat-cookie",
- },
- mockParams: mockParams{
- chatMessageRelayerParams: chatMessageRelayerParams{
- broadcastExceptParams: broadcastExceptParams{
- {
- except: sess,
- message: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.Chat,
- SubGroup: oscar.ChatUsersLeft,
- },
- Body: oscar.SNAC_0x0E_0x04_ChatUsersLeft{
- Users: []oscar.TLVUserInfo{
- sess.TLVUserInfo(),
- },
- },
- },
- },
- },
- },
- sessionManagerParams: sessionManagerParams{
- emptyParams: emptyParams{
- {
- result: true,
- },
- },
- removeSessionParams: removeSessionParams{
- {
- sess: sess,
- },
- },
- },
- },
- },
- {
- name: "user signs out of chat room, room is not empty after user leaves",
- userSession: sess,
- chatRoom: state.ChatRoom{
- Cookie: "the-chat-cookie",
- },
- mockParams: mockParams{
- chatMessageRelayerParams: chatMessageRelayerParams{
- broadcastExceptParams: broadcastExceptParams{
- {
- except: sess,
- message: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.Chat,
- SubGroup: oscar.ChatUsersLeft,
- },
- Body: oscar.SNAC_0x0E_0x04_ChatUsersLeft{
- Users: []oscar.TLVUserInfo{
- sess.TLVUserInfo(),
- },
- },
- },
- },
- },
- },
- sessionManagerParams: sessionManagerParams{
- emptyParams: emptyParams{
- {
- result: false,
- },
- },
- removeSessionParams: removeSessionParams{
- {
- sess: sess,
- },
- },
- },
- },
- },
- {
- name: "user can't sign out because chat room doesn't exist",
- userSession: sess,
- chatRoom: state.ChatRoom{
- Cookie: "the-chat-cookie",
- },
- wantErr: state.ErrChatRoomNotFound,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- chatMessageRelayer := newMockChatMessageRelayer(t)
- for _, params := range tt.mockParams.broadcastExceptParams {
- chatMessageRelayer.EXPECT().
- RelayToAllExcept(nil, params.except, params.message)
- }
- sessionManager := newMockSessionManager(t)
- chatRegistry := newMockChatRegistry(t)
- for _, params := range tt.mockParams.removeSessionParams {
- sessionManager.EXPECT().RemoveSession(params.sess)
- }
- for _, params := range tt.mockParams.emptyParams {
- sessionManager.EXPECT().Empty().Return(params.result)
- if params.result {
- chatRegistry.EXPECT().Remove(tt.chatRoom.Cookie)
- }
- }
- chatSessionManager := struct {
- ChatMessageRelayer
- SessionManager
- }{
- chatMessageRelayer,
- sessionManager,
- }
- chatRegistry.EXPECT().
- Retrieve(tt.chatRoom.Cookie).
- Return(tt.chatRoom, chatSessionManager, tt.wantErr)
- svc := NewAuthService(config.Config{}, nil, nil, nil, nil, chatRegistry)
- err := svc.SignoutChat(nil, tt.userSession, tt.chatRoom.Cookie)
- assert.ErrorIs(t, err, tt.wantErr)
- })
- }
- }
- func TestAuthService_Signout(t *testing.T) {
- sess := newTestSession("user_screen_name", sessOptCannedSignonTime)
- tests := []struct {
- // name is the unit test name
- name string
- // userSession is the session of the user signing out
- userSession *state.Session
- // chatRoom is the chat room user is exiting
- chatRoom state.ChatRoom
- // wantErr is the error we expect from the method
- wantErr error
- // mockParams is the list of params sent to mocks that satisfy this
- // method's dependencies
- mockParams mockParams
- }{
- {
- name: "user signs out of chat room, room is empty after user leaves",
- userSession: sess,
- chatRoom: state.ChatRoom{
- Cookie: "the-chat-cookie",
- },
- mockParams: mockParams{
- sessionManagerParams: sessionManagerParams{
- removeSessionParams: removeSessionParams{
- {
- sess: sess,
- },
- },
- },
- feedbagManagerParams: feedbagManagerParams{
- interestedUsersParams: interestedUsersParams{
- {
- screenName: "user_screen_name",
- users: []string{"friend1", "friend2"},
- },
- },
- },
- messageRelayerParams: messageRelayerParams{
- broadcastToScreenNamesParams: broadcastToScreenNamesParams{
- {
- screenNames: []string{"friend1", "friend2"},
- message: oscar.SNACMessage{
- Frame: oscar.SNACFrame{
- FoodGroup: oscar.Buddy,
- SubGroup: oscar.BuddyDeparted,
- },
- Body: oscar.SNAC_0x03_0x0C_BuddyDeparted{
- TLVUserInfo: oscar.TLVUserInfo{
- ScreenName: sess.ScreenName(),
- WarningLevel: sess.Warning(),
- },
- },
- },
- },
- },
- },
- },
- },
- {
- name: "user signs out of chat room, room is empty after user leaves",
- userSession: sess,
- chatRoom: state.ChatRoom{
- Cookie: "the-chat-cookie",
- },
- mockParams: mockParams{
- feedbagManagerParams: feedbagManagerParams{
- interestedUsersParams: interestedUsersParams{
- {
- screenName: "user_screen_name",
- users: []string{"friend1", "friend2"},
- err: io.EOF,
- },
- },
- },
- },
- wantErr: io.EOF,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- messageRelayer := newMockMessageRelayer(t)
- for _, params := range tt.mockParams.broadcastToScreenNamesParams {
- messageRelayer.EXPECT().
- RelayToScreenNames(mock.Anything, params.screenNames, params.message)
- }
- feedbagManager := newMockFeedbagManager(t)
- for _, params := range tt.mockParams.interestedUsersParams {
- feedbagManager.EXPECT().
- AdjacentUsers(params.screenName).
- Return(params.users, params.err)
- }
- sessionManager := newMockSessionManager(t)
- for _, params := range tt.mockParams.removeSessionParams {
- sessionManager.EXPECT().RemoveSession(params.sess)
- }
- svc := NewAuthService(config.Config{}, sessionManager, messageRelayer, feedbagManager, nil, nil)
- err := svc.Signout(nil, tt.userSession)
- assert.ErrorIs(t, err, tt.wantErr)
- })
- }
- }
|