mirror of
https://github.com/netbirdio/netbird.git
synced 2024-11-30 12:05:29 +01:00
47 lines
1.6 KiB
Go
47 lines
1.6 KiB
Go
package mock_server
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/wiretrustee/wiretrustee/management/proto"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
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)
|
|
}
|
|
|
|
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")
|
|
}
|