github.com/celestiaorg/celestia-node@v0.15.0-beta.1/nodebuilder/p2p/mocks/api.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/celestiaorg/celestia-node/nodebuilder/p2p (interfaces: Module)
     3  
     4  // Package mocks is a generated GoMock package.
     5  package mocks
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	metrics "github.com/libp2p/go-libp2p/core/metrics"
    13  	network "github.com/libp2p/go-libp2p/core/network"
    14  	peer "github.com/libp2p/go-libp2p/core/peer"
    15  	protocol "github.com/libp2p/go-libp2p/core/protocol"
    16  	rcmgr "github.com/libp2p/go-libp2p/p2p/host/resource-manager"
    17  )
    18  
    19  // MockModule is a mock of Module interface.
    20  type MockModule struct {
    21  	ctrl     *gomock.Controller
    22  	recorder *MockModuleMockRecorder
    23  }
    24  
    25  // MockModuleMockRecorder is the mock recorder for MockModule.
    26  type MockModuleMockRecorder struct {
    27  	mock *MockModule
    28  }
    29  
    30  // NewMockModule creates a new mock instance.
    31  func NewMockModule(ctrl *gomock.Controller) *MockModule {
    32  	mock := &MockModule{ctrl: ctrl}
    33  	mock.recorder = &MockModuleMockRecorder{mock}
    34  	return mock
    35  }
    36  
    37  // EXPECT returns an object that allows the caller to indicate expected use.
    38  func (m *MockModule) EXPECT() *MockModuleMockRecorder {
    39  	return m.recorder
    40  }
    41  
    42  // BandwidthForPeer mocks base method.
    43  func (m *MockModule) BandwidthForPeer(arg0 context.Context, arg1 peer.ID) (metrics.Stats, error) {
    44  	m.ctrl.T.Helper()
    45  	ret := m.ctrl.Call(m, "BandwidthForPeer", arg0, arg1)
    46  	ret0, _ := ret[0].(metrics.Stats)
    47  	ret1, _ := ret[1].(error)
    48  	return ret0, ret1
    49  }
    50  
    51  // BandwidthForPeer indicates an expected call of BandwidthForPeer.
    52  func (mr *MockModuleMockRecorder) BandwidthForPeer(arg0, arg1 interface{}) *gomock.Call {
    53  	mr.mock.ctrl.T.Helper()
    54  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BandwidthForPeer", reflect.TypeOf((*MockModule)(nil).BandwidthForPeer), arg0, arg1)
    55  }
    56  
    57  // BandwidthForProtocol mocks base method.
    58  func (m *MockModule) BandwidthForProtocol(arg0 context.Context, arg1 protocol.ID) (metrics.Stats, error) {
    59  	m.ctrl.T.Helper()
    60  	ret := m.ctrl.Call(m, "BandwidthForProtocol", arg0, arg1)
    61  	ret0, _ := ret[0].(metrics.Stats)
    62  	ret1, _ := ret[1].(error)
    63  	return ret0, ret1
    64  }
    65  
    66  // BandwidthForProtocol indicates an expected call of BandwidthForProtocol.
    67  func (mr *MockModuleMockRecorder) BandwidthForProtocol(arg0, arg1 interface{}) *gomock.Call {
    68  	mr.mock.ctrl.T.Helper()
    69  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BandwidthForProtocol", reflect.TypeOf((*MockModule)(nil).BandwidthForProtocol), arg0, arg1)
    70  }
    71  
    72  // BandwidthStats mocks base method.
    73  func (m *MockModule) BandwidthStats(arg0 context.Context) (metrics.Stats, error) {
    74  	m.ctrl.T.Helper()
    75  	ret := m.ctrl.Call(m, "BandwidthStats", arg0)
    76  	ret0, _ := ret[0].(metrics.Stats)
    77  	ret1, _ := ret[1].(error)
    78  	return ret0, ret1
    79  }
    80  
    81  // BandwidthStats indicates an expected call of BandwidthStats.
    82  func (mr *MockModuleMockRecorder) BandwidthStats(arg0 interface{}) *gomock.Call {
    83  	mr.mock.ctrl.T.Helper()
    84  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BandwidthStats", reflect.TypeOf((*MockModule)(nil).BandwidthStats), arg0)
    85  }
    86  
    87  // BlockPeer mocks base method.
    88  func (m *MockModule) BlockPeer(arg0 context.Context, arg1 peer.ID) error {
    89  	m.ctrl.T.Helper()
    90  	ret := m.ctrl.Call(m, "BlockPeer", arg0, arg1)
    91  	ret0, _ := ret[0].(error)
    92  	return ret0
    93  }
    94  
    95  // BlockPeer indicates an expected call of BlockPeer.
    96  func (mr *MockModuleMockRecorder) BlockPeer(arg0, arg1 interface{}) *gomock.Call {
    97  	mr.mock.ctrl.T.Helper()
    98  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BlockPeer", reflect.TypeOf((*MockModule)(nil).BlockPeer), arg0, arg1)
    99  }
   100  
   101  // ClosePeer mocks base method.
   102  func (m *MockModule) ClosePeer(arg0 context.Context, arg1 peer.ID) error {
   103  	m.ctrl.T.Helper()
   104  	ret := m.ctrl.Call(m, "ClosePeer", arg0, arg1)
   105  	ret0, _ := ret[0].(error)
   106  	return ret0
   107  }
   108  
   109  // ClosePeer indicates an expected call of ClosePeer.
   110  func (mr *MockModuleMockRecorder) ClosePeer(arg0, arg1 interface{}) *gomock.Call {
   111  	mr.mock.ctrl.T.Helper()
   112  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClosePeer", reflect.TypeOf((*MockModule)(nil).ClosePeer), arg0, arg1)
   113  }
   114  
   115  // Connect mocks base method.
   116  func (m *MockModule) Connect(arg0 context.Context, arg1 peer.AddrInfo) error {
   117  	m.ctrl.T.Helper()
   118  	ret := m.ctrl.Call(m, "Connect", arg0, arg1)
   119  	ret0, _ := ret[0].(error)
   120  	return ret0
   121  }
   122  
   123  // Connect indicates an expected call of Connect.
   124  func (mr *MockModuleMockRecorder) Connect(arg0, arg1 interface{}) *gomock.Call {
   125  	mr.mock.ctrl.T.Helper()
   126  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connect", reflect.TypeOf((*MockModule)(nil).Connect), arg0, arg1)
   127  }
   128  
   129  // Connectedness mocks base method.
   130  func (m *MockModule) Connectedness(arg0 context.Context, arg1 peer.ID) (network.Connectedness, error) {
   131  	m.ctrl.T.Helper()
   132  	ret := m.ctrl.Call(m, "Connectedness", arg0, arg1)
   133  	ret0, _ := ret[0].(network.Connectedness)
   134  	ret1, _ := ret[1].(error)
   135  	return ret0, ret1
   136  }
   137  
   138  // Connectedness indicates an expected call of Connectedness.
   139  func (mr *MockModuleMockRecorder) Connectedness(arg0, arg1 interface{}) *gomock.Call {
   140  	mr.mock.ctrl.T.Helper()
   141  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Connectedness", reflect.TypeOf((*MockModule)(nil).Connectedness), arg0, arg1)
   142  }
   143  
   144  // Info mocks base method.
   145  func (m *MockModule) Info(arg0 context.Context) (peer.AddrInfo, error) {
   146  	m.ctrl.T.Helper()
   147  	ret := m.ctrl.Call(m, "Info", arg0)
   148  	ret0, _ := ret[0].(peer.AddrInfo)
   149  	ret1, _ := ret[1].(error)
   150  	return ret0, ret1
   151  }
   152  
   153  // Info indicates an expected call of Info.
   154  func (mr *MockModuleMockRecorder) Info(arg0 interface{}) *gomock.Call {
   155  	mr.mock.ctrl.T.Helper()
   156  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Info", reflect.TypeOf((*MockModule)(nil).Info), arg0)
   157  }
   158  
   159  // IsProtected mocks base method.
   160  func (m *MockModule) IsProtected(arg0 context.Context, arg1 peer.ID, arg2 string) (bool, error) {
   161  	m.ctrl.T.Helper()
   162  	ret := m.ctrl.Call(m, "IsProtected", arg0, arg1, arg2)
   163  	ret0, _ := ret[0].(bool)
   164  	ret1, _ := ret[1].(error)
   165  	return ret0, ret1
   166  }
   167  
   168  // IsProtected indicates an expected call of IsProtected.
   169  func (mr *MockModuleMockRecorder) IsProtected(arg0, arg1, arg2 interface{}) *gomock.Call {
   170  	mr.mock.ctrl.T.Helper()
   171  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsProtected", reflect.TypeOf((*MockModule)(nil).IsProtected), arg0, arg1, arg2)
   172  }
   173  
   174  // ListBlockedPeers mocks base method.
   175  func (m *MockModule) ListBlockedPeers(arg0 context.Context) ([]peer.ID, error) {
   176  	m.ctrl.T.Helper()
   177  	ret := m.ctrl.Call(m, "ListBlockedPeers", arg0)
   178  	ret0, _ := ret[0].([]peer.ID)
   179  	ret1, _ := ret[1].(error)
   180  	return ret0, ret1
   181  }
   182  
   183  // ListBlockedPeers indicates an expected call of ListBlockedPeers.
   184  func (mr *MockModuleMockRecorder) ListBlockedPeers(arg0 interface{}) *gomock.Call {
   185  	mr.mock.ctrl.T.Helper()
   186  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBlockedPeers", reflect.TypeOf((*MockModule)(nil).ListBlockedPeers), arg0)
   187  }
   188  
   189  // NATStatus mocks base method.
   190  func (m *MockModule) NATStatus(arg0 context.Context) (network.Reachability, error) {
   191  	m.ctrl.T.Helper()
   192  	ret := m.ctrl.Call(m, "NATStatus", arg0)
   193  	ret0, _ := ret[0].(network.Reachability)
   194  	ret1, _ := ret[1].(error)
   195  	return ret0, ret1
   196  }
   197  
   198  // NATStatus indicates an expected call of NATStatus.
   199  func (mr *MockModuleMockRecorder) NATStatus(arg0 interface{}) *gomock.Call {
   200  	mr.mock.ctrl.T.Helper()
   201  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NATStatus", reflect.TypeOf((*MockModule)(nil).NATStatus), arg0)
   202  }
   203  
   204  // PeerInfo mocks base method.
   205  func (m *MockModule) PeerInfo(arg0 context.Context, arg1 peer.ID) (peer.AddrInfo, error) {
   206  	m.ctrl.T.Helper()
   207  	ret := m.ctrl.Call(m, "PeerInfo", arg0, arg1)
   208  	ret0, _ := ret[0].(peer.AddrInfo)
   209  	ret1, _ := ret[1].(error)
   210  	return ret0, ret1
   211  }
   212  
   213  // PeerInfo indicates an expected call of PeerInfo.
   214  func (mr *MockModuleMockRecorder) PeerInfo(arg0, arg1 interface{}) *gomock.Call {
   215  	mr.mock.ctrl.T.Helper()
   216  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PeerInfo", reflect.TypeOf((*MockModule)(nil).PeerInfo), arg0, arg1)
   217  }
   218  
   219  // Peers mocks base method.
   220  func (m *MockModule) Peers(arg0 context.Context) ([]peer.ID, error) {
   221  	m.ctrl.T.Helper()
   222  	ret := m.ctrl.Call(m, "Peers", arg0)
   223  	ret0, _ := ret[0].([]peer.ID)
   224  	ret1, _ := ret[1].(error)
   225  	return ret0, ret1
   226  }
   227  
   228  // Peers indicates an expected call of Peers.
   229  func (mr *MockModuleMockRecorder) Peers(arg0 interface{}) *gomock.Call {
   230  	mr.mock.ctrl.T.Helper()
   231  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Peers", reflect.TypeOf((*MockModule)(nil).Peers), arg0)
   232  }
   233  
   234  // Protect mocks base method.
   235  func (m *MockModule) Protect(arg0 context.Context, arg1 peer.ID, arg2 string) error {
   236  	m.ctrl.T.Helper()
   237  	ret := m.ctrl.Call(m, "Protect", arg0, arg1, arg2)
   238  	ret0, _ := ret[0].(error)
   239  	return ret0
   240  }
   241  
   242  // Protect indicates an expected call of Protect.
   243  func (mr *MockModuleMockRecorder) Protect(arg0, arg1, arg2 interface{}) *gomock.Call {
   244  	mr.mock.ctrl.T.Helper()
   245  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Protect", reflect.TypeOf((*MockModule)(nil).Protect), arg0, arg1, arg2)
   246  }
   247  
   248  // PubSubPeers mocks base method.
   249  func (m *MockModule) PubSubPeers(arg0 context.Context, arg1 string) ([]peer.ID, error) {
   250  	m.ctrl.T.Helper()
   251  	ret := m.ctrl.Call(m, "PubSubPeers", arg0, arg1)
   252  	ret0, _ := ret[0].([]peer.ID)
   253  	ret1, _ := ret[1].(error)
   254  	return ret0, ret1
   255  }
   256  
   257  // PubSubPeers indicates an expected call of PubSubPeers.
   258  func (mr *MockModuleMockRecorder) PubSubPeers(arg0, arg1 interface{}) *gomock.Call {
   259  	mr.mock.ctrl.T.Helper()
   260  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PubSubPeers", reflect.TypeOf((*MockModule)(nil).PubSubPeers), arg0, arg1)
   261  }
   262  
   263  // ResourceState mocks base method.
   264  func (m *MockModule) ResourceState(arg0 context.Context) (rcmgr.ResourceManagerStat, error) {
   265  	m.ctrl.T.Helper()
   266  	ret := m.ctrl.Call(m, "ResourceState", arg0)
   267  	ret0, _ := ret[0].(rcmgr.ResourceManagerStat)
   268  	ret1, _ := ret[1].(error)
   269  	return ret0, ret1
   270  }
   271  
   272  // ResourceState indicates an expected call of ResourceState.
   273  func (mr *MockModuleMockRecorder) ResourceState(arg0 interface{}) *gomock.Call {
   274  	mr.mock.ctrl.T.Helper()
   275  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResourceState", reflect.TypeOf((*MockModule)(nil).ResourceState), arg0)
   276  }
   277  
   278  // UnblockPeer mocks base method.
   279  func (m *MockModule) UnblockPeer(arg0 context.Context, arg1 peer.ID) error {
   280  	m.ctrl.T.Helper()
   281  	ret := m.ctrl.Call(m, "UnblockPeer", arg0, arg1)
   282  	ret0, _ := ret[0].(error)
   283  	return ret0
   284  }
   285  
   286  // UnblockPeer indicates an expected call of UnblockPeer.
   287  func (mr *MockModuleMockRecorder) UnblockPeer(arg0, arg1 interface{}) *gomock.Call {
   288  	mr.mock.ctrl.T.Helper()
   289  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UnblockPeer", reflect.TypeOf((*MockModule)(nil).UnblockPeer), arg0, arg1)
   290  }
   291  
   292  // Unprotect mocks base method.
   293  func (m *MockModule) Unprotect(arg0 context.Context, arg1 peer.ID, arg2 string) (bool, error) {
   294  	m.ctrl.T.Helper()
   295  	ret := m.ctrl.Call(m, "Unprotect", arg0, arg1, arg2)
   296  	ret0, _ := ret[0].(bool)
   297  	ret1, _ := ret[1].(error)
   298  	return ret0, ret1
   299  }
   300  
   301  // Unprotect indicates an expected call of Unprotect.
   302  func (mr *MockModuleMockRecorder) Unprotect(arg0, arg1, arg2 interface{}) *gomock.Call {
   303  	mr.mock.ctrl.T.Helper()
   304  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Unprotect", reflect.TypeOf((*MockModule)(nil).Unprotect), arg0, arg1, arg2)
   305  }