github.com/MetalBlockchain/metalgo@v1.11.9/snow/engine/common/mock_sender.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: snow/engine/common/sender.go
     3  //
     4  // Generated by this command:
     5  //
     6  //	mockgen -source=snow/engine/common/sender.go -destination=snow/engine/common/mock_sender.go -package=common -exclude_interfaces=StateSummarySender,AcceptedStateSummarySender,FrontierSender,AcceptedSender,FetchSender,AppSender,QuerySender,CrossChainAppSender,NetworkAppSender,Gossiper
     7  //
     8  
     9  // Package common is a generated GoMock package.
    10  package common
    11  
    12  import (
    13  	context "context"
    14  	reflect "reflect"
    15  
    16  	ids "github.com/MetalBlockchain/metalgo/ids"
    17  	set "github.com/MetalBlockchain/metalgo/utils/set"
    18  	gomock "go.uber.org/mock/gomock"
    19  )
    20  
    21  // MockSender is a mock of Sender interface.
    22  type MockSender struct {
    23  	ctrl     *gomock.Controller
    24  	recorder *MockSenderMockRecorder
    25  }
    26  
    27  // MockSenderMockRecorder is the mock recorder for MockSender.
    28  type MockSenderMockRecorder struct {
    29  	mock *MockSender
    30  }
    31  
    32  // NewMockSender creates a new mock instance.
    33  func NewMockSender(ctrl *gomock.Controller) *MockSender {
    34  	mock := &MockSender{ctrl: ctrl}
    35  	mock.recorder = &MockSenderMockRecorder{mock}
    36  	return mock
    37  }
    38  
    39  // EXPECT returns an object that allows the caller to indicate expected use.
    40  func (m *MockSender) EXPECT() *MockSenderMockRecorder {
    41  	return m.recorder
    42  }
    43  
    44  // SendAccepted mocks base method.
    45  func (m *MockSender) SendAccepted(ctx context.Context, nodeID ids.NodeID, requestID uint32, containerIDs []ids.ID) {
    46  	m.ctrl.T.Helper()
    47  	m.ctrl.Call(m, "SendAccepted", ctx, nodeID, requestID, containerIDs)
    48  }
    49  
    50  // SendAccepted indicates an expected call of SendAccepted.
    51  func (mr *MockSenderMockRecorder) SendAccepted(ctx, nodeID, requestID, containerIDs any) *gomock.Call {
    52  	mr.mock.ctrl.T.Helper()
    53  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAccepted", reflect.TypeOf((*MockSender)(nil).SendAccepted), ctx, nodeID, requestID, containerIDs)
    54  }
    55  
    56  // SendAcceptedFrontier mocks base method.
    57  func (m *MockSender) SendAcceptedFrontier(ctx context.Context, nodeID ids.NodeID, requestID uint32, containerID ids.ID) {
    58  	m.ctrl.T.Helper()
    59  	m.ctrl.Call(m, "SendAcceptedFrontier", ctx, nodeID, requestID, containerID)
    60  }
    61  
    62  // SendAcceptedFrontier indicates an expected call of SendAcceptedFrontier.
    63  func (mr *MockSenderMockRecorder) SendAcceptedFrontier(ctx, nodeID, requestID, containerID any) *gomock.Call {
    64  	mr.mock.ctrl.T.Helper()
    65  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAcceptedFrontier", reflect.TypeOf((*MockSender)(nil).SendAcceptedFrontier), ctx, nodeID, requestID, containerID)
    66  }
    67  
    68  // SendAcceptedStateSummary mocks base method.
    69  func (m *MockSender) SendAcceptedStateSummary(ctx context.Context, nodeID ids.NodeID, requestID uint32, summaryIDs []ids.ID) {
    70  	m.ctrl.T.Helper()
    71  	m.ctrl.Call(m, "SendAcceptedStateSummary", ctx, nodeID, requestID, summaryIDs)
    72  }
    73  
    74  // SendAcceptedStateSummary indicates an expected call of SendAcceptedStateSummary.
    75  func (mr *MockSenderMockRecorder) SendAcceptedStateSummary(ctx, nodeID, requestID, summaryIDs any) *gomock.Call {
    76  	mr.mock.ctrl.T.Helper()
    77  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAcceptedStateSummary", reflect.TypeOf((*MockSender)(nil).SendAcceptedStateSummary), ctx, nodeID, requestID, summaryIDs)
    78  }
    79  
    80  // SendAncestors mocks base method.
    81  func (m *MockSender) SendAncestors(ctx context.Context, nodeID ids.NodeID, requestID uint32, containers [][]byte) {
    82  	m.ctrl.T.Helper()
    83  	m.ctrl.Call(m, "SendAncestors", ctx, nodeID, requestID, containers)
    84  }
    85  
    86  // SendAncestors indicates an expected call of SendAncestors.
    87  func (mr *MockSenderMockRecorder) SendAncestors(ctx, nodeID, requestID, containers any) *gomock.Call {
    88  	mr.mock.ctrl.T.Helper()
    89  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAncestors", reflect.TypeOf((*MockSender)(nil).SendAncestors), ctx, nodeID, requestID, containers)
    90  }
    91  
    92  // SendAppError mocks base method.
    93  func (m *MockSender) SendAppError(ctx context.Context, nodeID ids.NodeID, requestID uint32, errorCode int32, errorMessage string) error {
    94  	m.ctrl.T.Helper()
    95  	ret := m.ctrl.Call(m, "SendAppError", ctx, nodeID, requestID, errorCode, errorMessage)
    96  	ret0, _ := ret[0].(error)
    97  	return ret0
    98  }
    99  
   100  // SendAppError indicates an expected call of SendAppError.
   101  func (mr *MockSenderMockRecorder) SendAppError(ctx, nodeID, requestID, errorCode, errorMessage any) *gomock.Call {
   102  	mr.mock.ctrl.T.Helper()
   103  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAppError", reflect.TypeOf((*MockSender)(nil).SendAppError), ctx, nodeID, requestID, errorCode, errorMessage)
   104  }
   105  
   106  // SendAppGossip mocks base method.
   107  func (m *MockSender) SendAppGossip(ctx context.Context, config SendConfig, appGossipBytes []byte) error {
   108  	m.ctrl.T.Helper()
   109  	ret := m.ctrl.Call(m, "SendAppGossip", ctx, config, appGossipBytes)
   110  	ret0, _ := ret[0].(error)
   111  	return ret0
   112  }
   113  
   114  // SendAppGossip indicates an expected call of SendAppGossip.
   115  func (mr *MockSenderMockRecorder) SendAppGossip(ctx, config, appGossipBytes any) *gomock.Call {
   116  	mr.mock.ctrl.T.Helper()
   117  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAppGossip", reflect.TypeOf((*MockSender)(nil).SendAppGossip), ctx, config, appGossipBytes)
   118  }
   119  
   120  // SendAppRequest mocks base method.
   121  func (m *MockSender) SendAppRequest(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, appRequestBytes []byte) error {
   122  	m.ctrl.T.Helper()
   123  	ret := m.ctrl.Call(m, "SendAppRequest", ctx, nodeIDs, requestID, appRequestBytes)
   124  	ret0, _ := ret[0].(error)
   125  	return ret0
   126  }
   127  
   128  // SendAppRequest indicates an expected call of SendAppRequest.
   129  func (mr *MockSenderMockRecorder) SendAppRequest(ctx, nodeIDs, requestID, appRequestBytes any) *gomock.Call {
   130  	mr.mock.ctrl.T.Helper()
   131  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAppRequest", reflect.TypeOf((*MockSender)(nil).SendAppRequest), ctx, nodeIDs, requestID, appRequestBytes)
   132  }
   133  
   134  // SendAppResponse mocks base method.
   135  func (m *MockSender) SendAppResponse(ctx context.Context, nodeID ids.NodeID, requestID uint32, appResponseBytes []byte) error {
   136  	m.ctrl.T.Helper()
   137  	ret := m.ctrl.Call(m, "SendAppResponse", ctx, nodeID, requestID, appResponseBytes)
   138  	ret0, _ := ret[0].(error)
   139  	return ret0
   140  }
   141  
   142  // SendAppResponse indicates an expected call of SendAppResponse.
   143  func (mr *MockSenderMockRecorder) SendAppResponse(ctx, nodeID, requestID, appResponseBytes any) *gomock.Call {
   144  	mr.mock.ctrl.T.Helper()
   145  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendAppResponse", reflect.TypeOf((*MockSender)(nil).SendAppResponse), ctx, nodeID, requestID, appResponseBytes)
   146  }
   147  
   148  // SendChits mocks base method.
   149  func (m *MockSender) SendChits(ctx context.Context, nodeID ids.NodeID, requestID uint32, preferredID, preferredIDAtHeight, acceptedID ids.ID) {
   150  	m.ctrl.T.Helper()
   151  	m.ctrl.Call(m, "SendChits", ctx, nodeID, requestID, preferredID, preferredIDAtHeight, acceptedID)
   152  }
   153  
   154  // SendChits indicates an expected call of SendChits.
   155  func (mr *MockSenderMockRecorder) SendChits(ctx, nodeID, requestID, preferredID, preferredIDAtHeight, acceptedID any) *gomock.Call {
   156  	mr.mock.ctrl.T.Helper()
   157  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendChits", reflect.TypeOf((*MockSender)(nil).SendChits), ctx, nodeID, requestID, preferredID, preferredIDAtHeight, acceptedID)
   158  }
   159  
   160  // SendCrossChainAppError mocks base method.
   161  func (m *MockSender) SendCrossChainAppError(ctx context.Context, chainID ids.ID, requestID uint32, errorCode int32, errorMessage string) error {
   162  	m.ctrl.T.Helper()
   163  	ret := m.ctrl.Call(m, "SendCrossChainAppError", ctx, chainID, requestID, errorCode, errorMessage)
   164  	ret0, _ := ret[0].(error)
   165  	return ret0
   166  }
   167  
   168  // SendCrossChainAppError indicates an expected call of SendCrossChainAppError.
   169  func (mr *MockSenderMockRecorder) SendCrossChainAppError(ctx, chainID, requestID, errorCode, errorMessage any) *gomock.Call {
   170  	mr.mock.ctrl.T.Helper()
   171  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendCrossChainAppError", reflect.TypeOf((*MockSender)(nil).SendCrossChainAppError), ctx, chainID, requestID, errorCode, errorMessage)
   172  }
   173  
   174  // SendCrossChainAppRequest mocks base method.
   175  func (m *MockSender) SendCrossChainAppRequest(ctx context.Context, chainID ids.ID, requestID uint32, appRequestBytes []byte) error {
   176  	m.ctrl.T.Helper()
   177  	ret := m.ctrl.Call(m, "SendCrossChainAppRequest", ctx, chainID, requestID, appRequestBytes)
   178  	ret0, _ := ret[0].(error)
   179  	return ret0
   180  }
   181  
   182  // SendCrossChainAppRequest indicates an expected call of SendCrossChainAppRequest.
   183  func (mr *MockSenderMockRecorder) SendCrossChainAppRequest(ctx, chainID, requestID, appRequestBytes any) *gomock.Call {
   184  	mr.mock.ctrl.T.Helper()
   185  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendCrossChainAppRequest", reflect.TypeOf((*MockSender)(nil).SendCrossChainAppRequest), ctx, chainID, requestID, appRequestBytes)
   186  }
   187  
   188  // SendCrossChainAppResponse mocks base method.
   189  func (m *MockSender) SendCrossChainAppResponse(ctx context.Context, chainID ids.ID, requestID uint32, appResponseBytes []byte) error {
   190  	m.ctrl.T.Helper()
   191  	ret := m.ctrl.Call(m, "SendCrossChainAppResponse", ctx, chainID, requestID, appResponseBytes)
   192  	ret0, _ := ret[0].(error)
   193  	return ret0
   194  }
   195  
   196  // SendCrossChainAppResponse indicates an expected call of SendCrossChainAppResponse.
   197  func (mr *MockSenderMockRecorder) SendCrossChainAppResponse(ctx, chainID, requestID, appResponseBytes any) *gomock.Call {
   198  	mr.mock.ctrl.T.Helper()
   199  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendCrossChainAppResponse", reflect.TypeOf((*MockSender)(nil).SendCrossChainAppResponse), ctx, chainID, requestID, appResponseBytes)
   200  }
   201  
   202  // SendGet mocks base method.
   203  func (m *MockSender) SendGet(ctx context.Context, nodeID ids.NodeID, requestID uint32, containerID ids.ID) {
   204  	m.ctrl.T.Helper()
   205  	m.ctrl.Call(m, "SendGet", ctx, nodeID, requestID, containerID)
   206  }
   207  
   208  // SendGet indicates an expected call of SendGet.
   209  func (mr *MockSenderMockRecorder) SendGet(ctx, nodeID, requestID, containerID any) *gomock.Call {
   210  	mr.mock.ctrl.T.Helper()
   211  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGet", reflect.TypeOf((*MockSender)(nil).SendGet), ctx, nodeID, requestID, containerID)
   212  }
   213  
   214  // SendGetAccepted mocks base method.
   215  func (m *MockSender) SendGetAccepted(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, containerIDs []ids.ID) {
   216  	m.ctrl.T.Helper()
   217  	m.ctrl.Call(m, "SendGetAccepted", ctx, nodeIDs, requestID, containerIDs)
   218  }
   219  
   220  // SendGetAccepted indicates an expected call of SendGetAccepted.
   221  func (mr *MockSenderMockRecorder) SendGetAccepted(ctx, nodeIDs, requestID, containerIDs any) *gomock.Call {
   222  	mr.mock.ctrl.T.Helper()
   223  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGetAccepted", reflect.TypeOf((*MockSender)(nil).SendGetAccepted), ctx, nodeIDs, requestID, containerIDs)
   224  }
   225  
   226  // SendGetAcceptedFrontier mocks base method.
   227  func (m *MockSender) SendGetAcceptedFrontier(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32) {
   228  	m.ctrl.T.Helper()
   229  	m.ctrl.Call(m, "SendGetAcceptedFrontier", ctx, nodeIDs, requestID)
   230  }
   231  
   232  // SendGetAcceptedFrontier indicates an expected call of SendGetAcceptedFrontier.
   233  func (mr *MockSenderMockRecorder) SendGetAcceptedFrontier(ctx, nodeIDs, requestID any) *gomock.Call {
   234  	mr.mock.ctrl.T.Helper()
   235  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGetAcceptedFrontier", reflect.TypeOf((*MockSender)(nil).SendGetAcceptedFrontier), ctx, nodeIDs, requestID)
   236  }
   237  
   238  // SendGetAcceptedStateSummary mocks base method.
   239  func (m *MockSender) SendGetAcceptedStateSummary(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, heights []uint64) {
   240  	m.ctrl.T.Helper()
   241  	m.ctrl.Call(m, "SendGetAcceptedStateSummary", ctx, nodeIDs, requestID, heights)
   242  }
   243  
   244  // SendGetAcceptedStateSummary indicates an expected call of SendGetAcceptedStateSummary.
   245  func (mr *MockSenderMockRecorder) SendGetAcceptedStateSummary(ctx, nodeIDs, requestID, heights any) *gomock.Call {
   246  	mr.mock.ctrl.T.Helper()
   247  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGetAcceptedStateSummary", reflect.TypeOf((*MockSender)(nil).SendGetAcceptedStateSummary), ctx, nodeIDs, requestID, heights)
   248  }
   249  
   250  // SendGetAncestors mocks base method.
   251  func (m *MockSender) SendGetAncestors(ctx context.Context, nodeID ids.NodeID, requestID uint32, containerID ids.ID) {
   252  	m.ctrl.T.Helper()
   253  	m.ctrl.Call(m, "SendGetAncestors", ctx, nodeID, requestID, containerID)
   254  }
   255  
   256  // SendGetAncestors indicates an expected call of SendGetAncestors.
   257  func (mr *MockSenderMockRecorder) SendGetAncestors(ctx, nodeID, requestID, containerID any) *gomock.Call {
   258  	mr.mock.ctrl.T.Helper()
   259  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGetAncestors", reflect.TypeOf((*MockSender)(nil).SendGetAncestors), ctx, nodeID, requestID, containerID)
   260  }
   261  
   262  // SendGetStateSummaryFrontier mocks base method.
   263  func (m *MockSender) SendGetStateSummaryFrontier(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32) {
   264  	m.ctrl.T.Helper()
   265  	m.ctrl.Call(m, "SendGetStateSummaryFrontier", ctx, nodeIDs, requestID)
   266  }
   267  
   268  // SendGetStateSummaryFrontier indicates an expected call of SendGetStateSummaryFrontier.
   269  func (mr *MockSenderMockRecorder) SendGetStateSummaryFrontier(ctx, nodeIDs, requestID any) *gomock.Call {
   270  	mr.mock.ctrl.T.Helper()
   271  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendGetStateSummaryFrontier", reflect.TypeOf((*MockSender)(nil).SendGetStateSummaryFrontier), ctx, nodeIDs, requestID)
   272  }
   273  
   274  // SendPullQuery mocks base method.
   275  func (m *MockSender) SendPullQuery(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, containerID ids.ID, requestedHeight uint64) {
   276  	m.ctrl.T.Helper()
   277  	m.ctrl.Call(m, "SendPullQuery", ctx, nodeIDs, requestID, containerID, requestedHeight)
   278  }
   279  
   280  // SendPullQuery indicates an expected call of SendPullQuery.
   281  func (mr *MockSenderMockRecorder) SendPullQuery(ctx, nodeIDs, requestID, containerID, requestedHeight any) *gomock.Call {
   282  	mr.mock.ctrl.T.Helper()
   283  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendPullQuery", reflect.TypeOf((*MockSender)(nil).SendPullQuery), ctx, nodeIDs, requestID, containerID, requestedHeight)
   284  }
   285  
   286  // SendPushQuery mocks base method.
   287  func (m *MockSender) SendPushQuery(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, container []byte, requestedHeight uint64) {
   288  	m.ctrl.T.Helper()
   289  	m.ctrl.Call(m, "SendPushQuery", ctx, nodeIDs, requestID, container, requestedHeight)
   290  }
   291  
   292  // SendPushQuery indicates an expected call of SendPushQuery.
   293  func (mr *MockSenderMockRecorder) SendPushQuery(ctx, nodeIDs, requestID, container, requestedHeight any) *gomock.Call {
   294  	mr.mock.ctrl.T.Helper()
   295  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendPushQuery", reflect.TypeOf((*MockSender)(nil).SendPushQuery), ctx, nodeIDs, requestID, container, requestedHeight)
   296  }
   297  
   298  // SendPut mocks base method.
   299  func (m *MockSender) SendPut(ctx context.Context, nodeID ids.NodeID, requestID uint32, container []byte) {
   300  	m.ctrl.T.Helper()
   301  	m.ctrl.Call(m, "SendPut", ctx, nodeID, requestID, container)
   302  }
   303  
   304  // SendPut indicates an expected call of SendPut.
   305  func (mr *MockSenderMockRecorder) SendPut(ctx, nodeID, requestID, container any) *gomock.Call {
   306  	mr.mock.ctrl.T.Helper()
   307  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendPut", reflect.TypeOf((*MockSender)(nil).SendPut), ctx, nodeID, requestID, container)
   308  }
   309  
   310  // SendStateSummaryFrontier mocks base method.
   311  func (m *MockSender) SendStateSummaryFrontier(ctx context.Context, nodeID ids.NodeID, requestID uint32, summary []byte) {
   312  	m.ctrl.T.Helper()
   313  	m.ctrl.Call(m, "SendStateSummaryFrontier", ctx, nodeID, requestID, summary)
   314  }
   315  
   316  // SendStateSummaryFrontier indicates an expected call of SendStateSummaryFrontier.
   317  func (mr *MockSenderMockRecorder) SendStateSummaryFrontier(ctx, nodeID, requestID, summary any) *gomock.Call {
   318  	mr.mock.ctrl.T.Helper()
   319  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SendStateSummaryFrontier", reflect.TypeOf((*MockSender)(nil).SendStateSummaryFrontier), ctx, nodeID, requestID, summary)
   320  }