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

Add management peer logout

Browse Source
This commit is contained in:
bcmmbaga
2023-12-07 20:03:53 +03:00
parent b03343bc4d
commit 97350ff7b7
3 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
management/server/mock_server/account_mock.go
+9
View File
@@ -78,6 +78,7 @@ type MockAccountManager struct {
GetPeerFunc func(accountID, peerID, userID string) (*nbpeer.Peer, error)
UpdateAccountSettingsFunc func(accountID, userID string, newSettings *server.Settings) (*server.Account, error)
LoginPeerFunc func(login server.PeerLogin) (*nbpeer.Peer, *server.NetworkMap, error)
LogoutPeerFunc func(peerPubKey string) error
SyncPeerFunc func(sync server.PeerSync) (*nbpeer.Peer, *server.NetworkMap, error)
InviteUserFunc func(accountID string, initiatorUserID string, targetUserEmail string) error
GetAllConnectedPeersFunc func() (map[string]struct{}, error)
@@ -607,6 +608,14 @@ func (am *MockAccountManager) LoginPeer(login server.PeerLogin) (*nbpeer.Peer, *
return nil, nil, status.Errorf(codes.Unimplemented, "method LoginPeer is not implemented")
}
// LogoutPeer mocks LogoutPeer of the AccountManager interface
func (am *MockAccountManager) LogoutPeer(peerPubKey string) error {
if am.LogoutPeerFunc != nil {
return am.LogoutPeerFunc(peerPubKey)
}
return status.Errorf(codes.Unimplemented, "method LogoutPeer is not implemented")
}
// SyncPeer mocks SyncPeer of the AccountManager interface
func (am *MockAccountManager) SyncPeer(sync server.PeerSync) (*nbpeer.Peer, *server.NetworkMap, error) {
if am.SyncPeerFunc != nil {