From 34c73f0b34d61dbc738fb7c70529120de3f189a5 Mon Sep 17 00:00:00 2001 From: braginini Date: Wed, 1 Mar 2023 09:44:52 +0100 Subject: [PATCH] Fix account manager mock --- management/server/mock_server/account_mock.go | 9 +++++++++ 1 file changed, 9 insertions(+) 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") +}