mirror of
https://github.com/netbirdio/netbird.git
synced 2026-03-31 06:24:18 -04:00
241 lines
9.5 KiB
Go
241 lines
9.5 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)
|
|
}
|
|
|
|
// CountStreams mocks base method.
|
|
func (m *MockController) CountStreams() int {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CountStreams")
|
|
ret0, _ := ret[0].(int)
|
|
return ret0
|
|
}
|
|
|
|
// CountStreams indicates an expected call of CountStreams.
|
|
func (mr *MockControllerMockRecorder) CountStreams() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CountStreams", reflect.TypeOf((*MockController)(nil).CountStreams))
|
|
}
|
|
|
|
// DisconnectPeers mocks base method.
|
|
func (m *MockController) DisconnectPeers(ctx context.Context, accountId string, peerIDs []string) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "DisconnectPeers", ctx, accountId, peerIDs)
|
|
}
|
|
|
|
// DisconnectPeers indicates an expected call of DisconnectPeers.
|
|
func (mr *MockControllerMockRecorder) DisconnectPeers(ctx, accountId, peerIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DisconnectPeers", reflect.TypeOf((*MockController)(nil).DisconnectPeers), ctx, accountId, 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)
|
|
}
|
|
|
|
// OnPeerConnected mocks base method.
|
|
func (m *MockController) OnPeerConnected(ctx context.Context, accountID, peerID string) (chan *UpdateMessage, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeerConnected", ctx, accountID, peerID)
|
|
ret0, _ := ret[0].(chan *UpdateMessage)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// OnPeerConnected indicates an expected call of OnPeerConnected.
|
|
func (mr *MockControllerMockRecorder) OnPeerConnected(ctx, accountID, peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeerConnected", reflect.TypeOf((*MockController)(nil).OnPeerConnected), ctx, accountID, peerID)
|
|
}
|
|
|
|
// OnPeerDisconnected mocks base method.
|
|
func (m *MockController) OnPeerDisconnected(ctx context.Context, accountID, peerID string) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "OnPeerDisconnected", ctx, accountID, peerID)
|
|
}
|
|
|
|
// OnPeerDisconnected indicates an expected call of OnPeerDisconnected.
|
|
func (mr *MockControllerMockRecorder) OnPeerDisconnected(ctx, accountID, peerID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeerDisconnected", reflect.TypeOf((*MockController)(nil).OnPeerDisconnected), ctx, accountID, peerID)
|
|
}
|
|
|
|
// OnPeersAdded mocks base method.
|
|
func (m *MockController) OnPeersAdded(ctx context.Context, accountID string, peerIDs []string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeersAdded", ctx, accountID, peerIDs)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OnPeersAdded indicates an expected call of OnPeersAdded.
|
|
func (mr *MockControllerMockRecorder) OnPeersAdded(ctx, accountID, peerIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeersAdded", reflect.TypeOf((*MockController)(nil).OnPeersAdded), ctx, accountID, peerIDs)
|
|
}
|
|
|
|
// OnPeersDeleted mocks base method.
|
|
func (m *MockController) OnPeersDeleted(ctx context.Context, accountID string, peerIDs []string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeersDeleted", ctx, accountID, peerIDs)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OnPeersDeleted indicates an expected call of OnPeersDeleted.
|
|
func (mr *MockControllerMockRecorder) OnPeersDeleted(ctx, accountID, peerIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeersDeleted", reflect.TypeOf((*MockController)(nil).OnPeersDeleted), ctx, accountID, peerIDs)
|
|
}
|
|
|
|
// OnPeersUpdated mocks base method.
|
|
func (m *MockController) OnPeersUpdated(ctx context.Context, accountId string, peerIDs []string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "OnPeersUpdated", ctx, accountId, peerIDs)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// OnPeersUpdated indicates an expected call of OnPeersUpdated.
|
|
func (mr *MockControllerMockRecorder) OnPeersUpdated(ctx, accountId, peerIDs any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnPeersUpdated", reflect.TypeOf((*MockController)(nil).OnPeersUpdated), ctx, accountId, peerIDs)
|
|
}
|
|
|
|
// 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)
|
|
}
|