diff --git a/management/server/mock_server/account_mock.go b/management/server/mock_server/account_mock.go index 1776c62aa..48c5cb2c7 100644 --- a/management/server/mock_server/account_mock.go +++ b/management/server/mock_server/account_mock.go @@ -73,6 +73,7 @@ type MockAccountManager struct { GetAccountByPeerIDFunc func(peerID string) (*server.Account, error) UpdatePeerLastLoginFunc func(peerID string) error UpdateAccountSettingsFunc func(accountID, userID string, newSettings *server.Settings) (*server.Account, error) + LoginPeerFunc func(login server.PeerLogin) (*server.Peer, error) } // GetUsersFromAccount mock implementation of GetUsersFromAccount from server.AccountManager interface @@ -564,3 +565,11 @@ func (am *MockAccountManager) UpdateAccountSettings(accountID, userID string, ne } return nil, status.Errorf(codes.Unimplemented, "method UpdateAccountSettings is not implemented") } + +// LoginPeer mocks LoginPeer of the AccountManager interface +func (am *MockAccountManager) LoginPeer(login server.PeerLogin) (*server.Peer, error) { + if am.LoginPeerFunc != nil { + return am.LoginPeerFunc(login) + } + return nil, status.Errorf(codes.Unimplemented, "method LoginPeer is not implemented") +}