mirror of
https://github.com/netbirdio/netbird.git
synced 2025-07-15 05:45:42 +02:00
c6ab215d9d
* Add method to retrieve peer's applied posture checks * Add posture checks in server response and update proto messages * Refactor * Extends peer metadata synchronization through SyncRequest and propagate posture changes on syncResponse * Remove account lock * Pass system info on sync * Fix tests * Refactor * resolve merge * Evaluate process check on client (#1749) * implement server and client sync peer meta alongside mocks * wip: add check file and process * Add files to peer metadata for process check * wip: update peer meta on first sync * Add files to peer's metadata * Evaluate process check using files from peer metadata * Fix panic and append windows path to files * Fix check network address and files equality * Evaluate active process on darwin * Evaluate active process on linux * Skip processing processes if no paths are set * Return network map on peer meta-sync and update account peer's * Update client network map on meta sync * Get system info with applied checks * Add windows package * Remove a network map from sync meta-response * Update checks proto message * Keep client checks state and sync meta on checks change * Evaluate a running process * skip build for android and ios * skip check file and process for android and ios * bump gopsutil version * fix tests * move process check to separate os file * refactor * evaluate info with checks on receiving management events * skip meta-update for an old client with no meta-sync support * Check if peer meta is empty without reflection
72 lines
3.0 KiB
Go
72 lines
3.0 KiB
Go
package mock_server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
|
|
"github.com/netbirdio/netbird/management/proto"
|
|
)
|
|
|
|
type ManagementServiceServerMock struct {
|
|
proto.UnimplementedManagementServiceServer
|
|
|
|
LoginFunc func(context.Context, *proto.EncryptedMessage) (*proto.EncryptedMessage, error)
|
|
SyncFunc func(*proto.EncryptedMessage, proto.ManagementService_SyncServer)
|
|
GetServerKeyFunc func(context.Context, *proto.Empty) (*proto.ServerKeyResponse, error)
|
|
IsHealthyFunc func(context.Context, *proto.Empty) (*proto.Empty, error)
|
|
GetDeviceAuthorizationFlowFunc func(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error)
|
|
GetPKCEAuthorizationFlowFunc func(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error)
|
|
SyncMetaFunc func(ctx context.Context, req *proto.EncryptedMessage) (*proto.Empty, error)
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) Login(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error) {
|
|
if m.LoginFunc != nil {
|
|
return m.LoginFunc(ctx, req)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method Login not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) Sync(msg *proto.EncryptedMessage, sync proto.ManagementService_SyncServer) error {
|
|
if m.SyncFunc != nil {
|
|
return m.Sync(msg, sync)
|
|
}
|
|
return status.Errorf(codes.Unimplemented, "method Sync not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) GetServerKey(ctx context.Context, empty *proto.Empty) (*proto.ServerKeyResponse, error) {
|
|
if m.GetServerKeyFunc != nil {
|
|
return m.GetServerKeyFunc(ctx, empty)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method GetServerKey not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) IsHealthy(ctx context.Context, empty *proto.Empty) (*proto.Empty, error) {
|
|
if m.IsHealthyFunc != nil {
|
|
return m.IsHealthyFunc(ctx, empty)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method IsHealthy not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) GetDeviceAuthorizationFlow(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error) {
|
|
if m.GetDeviceAuthorizationFlowFunc != nil {
|
|
return m.GetDeviceAuthorizationFlowFunc(ctx, req)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method GetDeviceAuthorizationFlow not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) GetPKCEAuthorizationFlow(ctx context.Context, req *proto.EncryptedMessage) (*proto.EncryptedMessage, error) {
|
|
if m.GetPKCEAuthorizationFlowFunc != nil {
|
|
return m.GetPKCEAuthorizationFlowFunc(ctx, req)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method GetPKCEAuthorizationFlow not implemented")
|
|
}
|
|
|
|
func (m ManagementServiceServerMock) SyncMeta(ctx context.Context, req *proto.EncryptedMessage) (*proto.Empty, error) {
|
|
if m.SyncMetaFunc != nil {
|
|
return m.SyncMetaFunc(ctx, req)
|
|
}
|
|
return nil, status.Errorf(codes.Unimplemented, "method SyncMeta not implemented")
|
|
}
|