// Code generated by MockGen. DO NOT EDIT. // Source: management/internals/controllers/network_map/interface.go // // Generated by this command: // // mockgen -package network_map -destination=management/internals/controllers/network_map/interface_mock.go -source=management/internals/controllers/network_map/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) } // TrackEphemeralPeer mocks base method. func (m *MockController) TrackEphemeralPeer(ctx context.Context, arg1 *peer.Peer) { m.ctrl.T.Helper() m.ctrl.Call(m, "TrackEphemeralPeer", ctx, arg1) } // TrackEphemeralPeer indicates an expected call of TrackEphemeralPeer. func (mr *MockControllerMockRecorder) TrackEphemeralPeer(ctx, arg1 any) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TrackEphemeralPeer", reflect.TypeOf((*MockController)(nil).TrackEphemeralPeer), ctx, arg1) } // 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) }