From 5961c8330e350a553ed59007fd59a480d5df9f17 Mon Sep 17 00:00:00 2001 From: Yury Gargay Date: Mon, 11 Dec 2023 17:32:10 +0100 Subject: [PATCH] Fix SaveOrAddUser and GetPeers methods in MockAccountManager (#1374) --- management/server/mock_server/account_mock.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/management/server/mock_server/account_mock.go b/management/server/mock_server/account_mock.go index ae3b0ebbdeb..3f8d7c40b20 100644 --- a/management/server/mock_server/account_mock.go +++ b/management/server/mock_server/account_mock.go @@ -464,7 +464,7 @@ func (am *MockAccountManager) SaveUser(accountID, userID string, user *server.Us // SaveOrAddUser mocks SaveOrAddUser of the AccountManager interface func (am *MockAccountManager) SaveOrAddUser(accountID, userID string, user *server.User, addIfNotExists bool) (*server.UserInfo, error) { - if am.SaveUserFunc != nil { + if am.SaveOrAddUserFunc != nil { return am.SaveOrAddUserFunc(accountID, userID, user, addIfNotExists) } return nil, status.Errorf(codes.Unimplemented, "method SaveOrAddUser is not implemented") @@ -545,7 +545,7 @@ func (am *MockAccountManager) GetAccountFromToken(claims jwtclaims.Authorization // GetPeers mocks GetPeers of the AccountManager interface func (am *MockAccountManager) GetPeers(accountID, userID string) ([]*nbpeer.Peer, error) { - if am.GetAccountFromTokenFunc != nil { + if am.GetPeersFunc != nil { return am.GetPeersFunc(accountID, userID) } return nil, status.Errorf(codes.Unimplemented, "method GetPeers is not implemented")