sendnrw/netbird
2
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2026-04-23 02:36:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix account manager mock

Browse Source
This commit is contained in:
braginini
2023-03-01 09:44:52 +01:00
parent d554da2951
commit 34c73f0b34
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
management/server/mock_server/account_mock.go
View File

@@ -73,6 +73,7 @@ type MockAccountManager struct {
GetAccountByPeerIDFunc func(peerID string) (*server.Account, error) GetAccountByPeerIDFunc func(peerID string) (*server.Account, error)
UpdatePeerLastLoginFunc func(peerID string) error UpdatePeerLastLoginFunc func(peerID string) error
UpdateAccountSettingsFunc func(accountID, userID string, newSettings *server.Settings) (*server.Account, 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 // 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") 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")
}