github.com/onflow/flow-go@v0.35.7-crescendo-preview.23-atree-inlining/network/mocknetwork/mock_network.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/onflow/flow-go/network (interfaces: EngineRegistry)
     3  
     4  // Package mocknetwork is a generated GoMock package.
     5  package mocknetwork
     6  
     7  import (
     8  	reflect "reflect"
     9  
    10  	gomock "github.com/golang/mock/gomock"
    11  	datastore "github.com/ipfs/go-datastore"
    12  	protocol "github.com/libp2p/go-libp2p/core/protocol"
    13  	irrecoverable "github.com/onflow/flow-go/module/irrecoverable"
    14  	network "github.com/onflow/flow-go/network"
    15  	channels "github.com/onflow/flow-go/network/channels"
    16  )
    17  
    18  // MockEngineRegistry is a mock of EngineRegistry interface.
    19  type MockEngineRegistry struct {
    20  	ctrl     *gomock.Controller
    21  	recorder *MockEngineRegistryMockRecorder
    22  }
    23  
    24  // MockEngineRegistryMockRecorder is the mock recorder for MockEngineRegistry.
    25  type MockEngineRegistryMockRecorder struct {
    26  	mock *MockEngineRegistry
    27  }
    28  
    29  // NewMockEngineRegistry creates a new mock instance.
    30  func NewMockEngineRegistry(ctrl *gomock.Controller) *MockEngineRegistry {
    31  	mock := &MockEngineRegistry{ctrl: ctrl}
    32  	mock.recorder = &MockEngineRegistryMockRecorder{mock}
    33  	return mock
    34  }
    35  
    36  // EXPECT returns an object that allows the caller to indicate expected use.
    37  func (m *MockEngineRegistry) EXPECT() *MockEngineRegistryMockRecorder {
    38  	return m.recorder
    39  }
    40  
    41  // Done mocks base method.
    42  func (m *MockEngineRegistry) Done() <-chan struct{} {
    43  	m.ctrl.T.Helper()
    44  	ret := m.ctrl.Call(m, "Done")
    45  	ret0, _ := ret[0].(<-chan struct{})
    46  	return ret0
    47  }
    48  
    49  // Done indicates an expected call of Done.
    50  func (mr *MockEngineRegistryMockRecorder) Done() *gomock.Call {
    51  	mr.mock.ctrl.T.Helper()
    52  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Done", reflect.TypeOf((*MockEngineRegistry)(nil).Done))
    53  }
    54  
    55  // Ready mocks base method.
    56  func (m *MockEngineRegistry) Ready() <-chan struct{} {
    57  	m.ctrl.T.Helper()
    58  	ret := m.ctrl.Call(m, "Ready")
    59  	ret0, _ := ret[0].(<-chan struct{})
    60  	return ret0
    61  }
    62  
    63  // Ready indicates an expected call of Ready.
    64  func (mr *MockEngineRegistryMockRecorder) Ready() *gomock.Call {
    65  	mr.mock.ctrl.T.Helper()
    66  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ready", reflect.TypeOf((*MockEngineRegistry)(nil).Ready))
    67  }
    68  
    69  // Register mocks base method.
    70  func (m *MockEngineRegistry) Register(arg0 channels.Channel, arg1 network.MessageProcessor) (network.Conduit, error) {
    71  	m.ctrl.T.Helper()
    72  	ret := m.ctrl.Call(m, "Register", arg0, arg1)
    73  	ret0, _ := ret[0].(network.Conduit)
    74  	ret1, _ := ret[1].(error)
    75  	return ret0, ret1
    76  }
    77  
    78  // Register indicates an expected call of Register.
    79  func (mr *MockEngineRegistryMockRecorder) Register(arg0, arg1 interface{}) *gomock.Call {
    80  	mr.mock.ctrl.T.Helper()
    81  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Register", reflect.TypeOf((*MockEngineRegistry)(nil).Register), arg0, arg1)
    82  }
    83  
    84  // RegisterBlobService mocks base method.
    85  func (m *MockEngineRegistry) RegisterBlobService(arg0 channels.Channel, arg1 datastore.Batching, arg2 ...network.BlobServiceOption) (network.BlobService, error) {
    86  	m.ctrl.T.Helper()
    87  	varargs := []interface{}{arg0, arg1}
    88  	for _, a := range arg2 {
    89  		varargs = append(varargs, a)
    90  	}
    91  	ret := m.ctrl.Call(m, "RegisterBlobService", varargs...)
    92  	ret0, _ := ret[0].(network.BlobService)
    93  	ret1, _ := ret[1].(error)
    94  	return ret0, ret1
    95  }
    96  
    97  // RegisterBlobService indicates an expected call of RegisterBlobService.
    98  func (mr *MockEngineRegistryMockRecorder) RegisterBlobService(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    99  	mr.mock.ctrl.T.Helper()
   100  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   101  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterBlobService", reflect.TypeOf((*MockEngineRegistry)(nil).RegisterBlobService), varargs...)
   102  }
   103  
   104  // RegisterPingService mocks base method.
   105  func (m *MockEngineRegistry) RegisterPingService(arg0 protocol.ID, arg1 network.PingInfoProvider) (network.PingService, error) {
   106  	m.ctrl.T.Helper()
   107  	ret := m.ctrl.Call(m, "RegisterPingService", arg0, arg1)
   108  	ret0, _ := ret[0].(network.PingService)
   109  	ret1, _ := ret[1].(error)
   110  	return ret0, ret1
   111  }
   112  
   113  // RegisterPingService indicates an expected call of RegisterPingService.
   114  func (mr *MockEngineRegistryMockRecorder) RegisterPingService(arg0, arg1 interface{}) *gomock.Call {
   115  	mr.mock.ctrl.T.Helper()
   116  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RegisterPingService", reflect.TypeOf((*MockEngineRegistry)(nil).RegisterPingService), arg0, arg1)
   117  }
   118  
   119  // Start mocks base method.
   120  func (m *MockEngineRegistry) Start(arg0 irrecoverable.SignalerContext) {
   121  	m.ctrl.T.Helper()
   122  	m.ctrl.Call(m, "Start", arg0)
   123  }
   124  
   125  // Start indicates an expected call of Start.
   126  func (mr *MockEngineRegistryMockRecorder) Start(arg0 interface{}) *gomock.Call {
   127  	mr.mock.ctrl.T.Helper()
   128  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Start", reflect.TypeOf((*MockEngineRegistry)(nil).Start), arg0)
   129  }