From 453643683ddeb140bd5222e55659090678e342bf Mon Sep 17 00:00:00 2001 From: Pascal Fischer Date: Thu, 16 Mar 2023 16:44:05 +0100 Subject: [PATCH] add method to account 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 9da748757..4f2580679 100644 --- a/management/server/mock_server/account_mock.go +++ b/management/server/mock_server/account_mock.go @@ -47,6 +47,7 @@ type MockAccountManager struct { DeletePolicyFunc func(accountID, policyID, userID string) error ListPoliciesFunc func(accountID, userID string) ([]*server.Policy, error) GetUsersFromAccountFunc func(accountID, userID string) ([]*server.UserInfo, error) + GetAccountFromPATFunc func(pat string) (*server.Account, *server.User, error) UpdatePeerMetaFunc func(peerID string, meta server.PeerSystemMeta) error UpdatePeerSSHKeyFunc func(peerID string, sshKey string) error UpdatePeerFunc func(accountID, userID string, peer *server.Peer) (*server.Peer, error) @@ -175,6 +176,14 @@ func (am *MockAccountManager) GetPeerByIP(accountId string, peerIP string) (*ser return nil, status.Errorf(codes.Unimplemented, "method GetPeerByIP is not implemented") } +// GetAccountFromPAT mock implementation of GetAccountFromPAT from server.AccountManager interface +func (am *MockAccountManager) GetAccountFromPAT(pat string) (*server.Account, *server.User, error) { + if am.GetAccountFromPATFunc != nil { + return am.GetAccountFromPATFunc(pat) + } + return nil, nil, status.Errorf(codes.Unimplemented, "method GetAccountFromPAT is not implemented") +} + // GetNetworkMap mock implementation of GetNetworkMap from server.AccountManager interface func (am *MockAccountManager) GetNetworkMap(peerKey string) (*server.NetworkMap, error) { if am.GetNetworkMapFunc != nil {