33 lines
980 B
Go
33 lines
980 B
Go
package port_forwarding
|
|
|
|
import (
|
|
"context"
|
|
|
|
nbtypes "github.com/netbirdio/netbird/management/server/types"
|
|
)
|
|
|
|
type Controller interface {
|
|
SendUpdate(ctx context.Context, accountID string, affectedProxyID string, affectedPeerIDs []string)
|
|
GetProxyNetworkMaps(ctx context.Context, accountID string) (map[string]*nbtypes.NetworkMap, error)
|
|
IsPeerInIngressPorts(ctx context.Context, accountID, peerID string) (bool, error)
|
|
}
|
|
|
|
type ControllerMock struct {
|
|
}
|
|
|
|
func NewControllerMock() *ControllerMock {
|
|
return &ControllerMock{}
|
|
}
|
|
|
|
func (c *ControllerMock) SendUpdate(ctx context.Context, accountID string, affectedProxyID string, affectedPeerIDs []string) {
|
|
// noop
|
|
}
|
|
|
|
func (c *ControllerMock) GetProxyNetworkMaps(ctx context.Context, accountID string) (map[string]*nbtypes.NetworkMap, error) {
|
|
return make(map[string]*nbtypes.NetworkMap), nil
|
|
}
|
|
|
|
func (c *ControllerMock) IsPeerInIngressPorts(ctx context.Context, accountID, peerID string) (bool, error) {
|
|
return false, nil
|
|
}
|