mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-16 07:16:38 +00:00
226 lines
8.7 KiB
Go
226 lines
8.7 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: ./interface.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package network_map -destination=interface_mock.go -source=./interface.go -build_flags=-mod=mod
|
|
//
|
|
|
|
// Package network_map is a generated GoMock package.
|
|
package network_map
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
peer "github.com/netbirdio/netbird/management/server/peer"
|
|
posture "github.com/netbirdio/netbird/management/server/posture"
|
|
types "github.com/netbirdio/netbird/management/server/types"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockController is a mock of Controller interface.
|
|
type MockController struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockControllerMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockControllerMockRecorder is the mock recorder for MockController.
|
|
type MockControllerMockRecorder struct {
|
|
mock *MockController
|
|
}
|
|
|
|
// NewMockController creates a new mock instance.
|
|
func NewMockController(ctrl *gomock.Controller) *MockController {
|
|
mock := &MockController{ctrl: ctrl}
|
|
mock.recorder = &MockControllerMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockController) EXPECT() *MockControllerMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// BufferUpdateAccountPeers mocks base method.
|
|
func (m *MockController) BufferUpdateAccountPeers(ctx context.Context, accountID string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "BufferUpdateAccountPeers", ctx, accountID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// BufferUpdateAccountPeers indicates an expected call of BufferUpdateAccountPeers.
|
|
func (mr *MockControllerMockRecorder) BufferUpdateAccountPeers(ctx, accountID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BufferUpdateAccountPeers", reflect.TypeOf((*MockController)(nil).BufferUpdateAccountPeers), ctx, accountID)
|
|
}
|
|
|
|
// DeletePeer mocks base method.
|
|
func (m *MockController) DeletePeer(ctx context.Context, accountId, peerId string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeletePeer", ctx, accountId, peerId)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// DeletePeer indicates an expected call of DeletePeer.
|
|
func (mr *MockControllerMockRecorder) DeletePeer(ctx, accountId, peerId any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeletePeer", reflect.TypeOf((*MockController)(nil).DeletePeer), ctx, accountId, peerId)
|
|
}
|
|
|
|
// DisconnectPeers mocks base method.
|
|
func (m *MockController) DisconnectPeers(ctx context.Context, peerIDs []string) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "DisconnectPeers", ctx, peerIDs)
|
|
}
|
|
|
|
// DisconnectPeers indicates an expected call of DisconnectPeers.
|
|
func (mr *MockControllerMockRecorder) DisconnectPeers(ctx, peerIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DisconnectPeers", reflect.TypeOf((*MockController)(nil).DisconnectPeers), ctx, peerIDs)
|
|
}
|
|
|
|
// GetDNSDomain mocks base method.
|
|
func (m *MockController) GetDNSDomain(settings *types.Settings) string {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetDNSDomain", settings)
|
|
ret0, _ := ret[0].(string)
|
|
return ret0
|
|
}
|
|
|
|
// GetDNSDomain indicates an expected call of GetDNSDomain.
|
|
func (mr *MockControllerMockRecorder) GetDNSDomain(settings any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDNSDomain", reflect.TypeOf((*MockController)(nil).GetDNSDomain), settings)
|
|
}
|
|
|
|
// GetNetworkMap mocks base method.
|
|
func (m *MockController) GetNetworkMap(ctx context.Context, peerID string) (*types.NetworkMap, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetNetworkMap", ctx, peerID)
|
|
ret0, _ := ret[0].(*types.NetworkMap)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// GetNetworkMap indicates an expected call of GetNetworkMap.
|
|
func (mr *MockControllerMockRecorder) GetNetworkMap(ctx, peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNetworkMap", reflect.TypeOf((*MockController)(nil).GetNetworkMap), ctx, peerID)
|
|
}
|
|
|
|
// GetValidatedPeerWithMap mocks base method.
|
|
func (m *MockController) GetValidatedPeerWithMap(ctx context.Context, isRequiresApproval bool, accountID string, p *peer.Peer) (*peer.Peer, *types.NetworkMap, []*posture.Checks, int64, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetValidatedPeerWithMap", ctx, isRequiresApproval, accountID, p)
|
|
ret0, _ := ret[0].(*peer.Peer)
|
|
ret1, _ := ret[1].(*types.NetworkMap)
|
|
ret2, _ := ret[2].([]*posture.Checks)
|
|
ret3, _ := ret[3].(int64)
|
|
ret4, _ := ret[4].(error)
|
|
return ret0, ret1, ret2, ret3, ret4
|
|
}
|
|
|
|
// GetValidatedPeerWithMap indicates an expected call of GetValidatedPeerWithMap.
|
|
func (mr *MockControllerMockRecorder) GetValidatedPeerWithMap(ctx, isRequiresApproval, accountID, p any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetValidatedPeerWithMap", reflect.TypeOf((*MockController)(nil).GetValidatedPeerWithMap), ctx, isRequiresApproval, accountID, p)
|
|
}
|
|
|
|
// IsConnected mocks base method.
|
|
func (m *MockController) IsConnected(peerID string) bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsConnected", peerID)
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// IsConnected indicates an expected call of IsConnected.
|
|
func (mr *MockControllerMockRecorder) IsConnected(peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsConnected", reflect.TypeOf((*MockController)(nil).IsConnected), peerID)
|
|
}
|
|
|
|
// OnPeerAdded mocks base method.
|
|
func (m *MockController) OnPeerAdded(ctx context.Context, accountID, peerID string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeerAdded", ctx, accountID, peerID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OnPeerAdded indicates an expected call of OnPeerAdded.
|
|
func (mr *MockControllerMockRecorder) OnPeerAdded(ctx, accountID, peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeerAdded", reflect.TypeOf((*MockController)(nil).OnPeerAdded), ctx, accountID, peerID)
|
|
}
|
|
|
|
// OnPeerDeleted mocks base method.
|
|
func (m *MockController) OnPeerDeleted(ctx context.Context, accountID, peerID string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeerDeleted", ctx, accountID, peerID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OnPeerDeleted indicates an expected call of OnPeerDeleted.
|
|
func (mr *MockControllerMockRecorder) OnPeerDeleted(ctx, accountID, peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeerDeleted", reflect.TypeOf((*MockController)(nil).OnPeerDeleted), ctx, accountID, peerID)
|
|
}
|
|
|
|
// OnPeerUpdated mocks base method.
|
|
func (m *MockController) OnPeerUpdated(accountId string, peer *peer.Peer) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "OnPeerUpdated", accountId, peer)
|
|
}
|
|
|
|
// OnPeerUpdated indicates an expected call of OnPeerUpdated.
|
|
func (mr *MockControllerMockRecorder) OnPeerUpdated(accountId, peer any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeerUpdated", reflect.TypeOf((*MockController)(nil).OnPeerUpdated), accountId, peer)
|
|
}
|
|
|
|
// StartWarmup mocks base method.
|
|
func (m *MockController) StartWarmup(arg0 context.Context) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "StartWarmup", arg0)
|
|
}
|
|
|
|
// StartWarmup indicates an expected call of StartWarmup.
|
|
func (mr *MockControllerMockRecorder) StartWarmup(arg0 any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartWarmup", reflect.TypeOf((*MockController)(nil).StartWarmup), arg0)
|
|
}
|
|
|
|
// UpdateAccountPeer mocks base method.
|
|
func (m *MockController) UpdateAccountPeer(ctx context.Context, accountId, peerId string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "UpdateAccountPeer", ctx, accountId, peerId)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// UpdateAccountPeer indicates an expected call of UpdateAccountPeer.
|
|
func (mr *MockControllerMockRecorder) UpdateAccountPeer(ctx, accountId, peerId any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAccountPeer", reflect.TypeOf((*MockController)(nil).UpdateAccountPeer), ctx, accountId, peerId)
|
|
}
|
|
|
|
// UpdateAccountPeers mocks base method.
|
|
func (m *MockController) UpdateAccountPeers(ctx context.Context, accountID string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "UpdateAccountPeers", ctx, accountID)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// UpdateAccountPeers indicates an expected call of UpdateAccountPeers.
|
|
func (mr *MockControllerMockRecorder) UpdateAccountPeers(ctx, accountID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAccountPeers", reflect.TypeOf((*MockController)(nil).UpdateAccountPeers), ctx, accountID)
|
|
}
|