github.com/prysmaticlabs/prysm@v1.4.4/shared/mock/node_service_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/prysmaticlabs/prysm/proto/eth/v1alpha1 (interfaces: NodeClient)
     3  
     4  // Package mock is a generated GoMock package.
     5  package mock
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	v1alpha1 "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
    13  	grpc "google.golang.org/grpc"
    14  	emptypb "google.golang.org/protobuf/types/known/emptypb"
    15  )
    16  
    17  // MockNodeClient is a mock of NodeClient interface
    18  type MockNodeClient struct {
    19  	ctrl     *gomock.Controller
    20  	recorder *MockNodeClientMockRecorder
    21  }
    22  
    23  // MockNodeClientMockRecorder is the mock recorder for MockNodeClient
    24  type MockNodeClientMockRecorder struct {
    25  	mock *MockNodeClient
    26  }
    27  
    28  // NewMockNodeClient creates a new mock instance
    29  func NewMockNodeClient(ctrl *gomock.Controller) *MockNodeClient {
    30  	mock := &MockNodeClient{ctrl: ctrl}
    31  	mock.recorder = &MockNodeClientMockRecorder{mock}
    32  	return mock
    33  }
    34  
    35  // EXPECT returns an object that allows the caller to indicate expected use
    36  func (m *MockNodeClient) EXPECT() *MockNodeClientMockRecorder {
    37  	return m.recorder
    38  }
    39  
    40  // GetGenesis mocks base method
    41  func (m *MockNodeClient) GetGenesis(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.Genesis, error) {
    42  	m.ctrl.T.Helper()
    43  	varargs := []interface{}{arg0, arg1}
    44  	for _, a := range arg2 {
    45  		varargs = append(varargs, a)
    46  	}
    47  	ret := m.ctrl.Call(m, "GetGenesis", varargs...)
    48  	ret0, _ := ret[0].(*v1alpha1.Genesis)
    49  	ret1, _ := ret[1].(error)
    50  	return ret0, ret1
    51  }
    52  
    53  // GetGenesis indicates an expected call of GetGenesis
    54  func (mr *MockNodeClientMockRecorder) GetGenesis(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    55  	mr.mock.ctrl.T.Helper()
    56  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    57  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetGenesis", reflect.TypeOf((*MockNodeClient)(nil).GetGenesis), varargs...)
    58  }
    59  
    60  // GetHost mocks base method
    61  func (m *MockNodeClient) GetHost(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.HostData, error) {
    62  	m.ctrl.T.Helper()
    63  	varargs := []interface{}{arg0, arg1}
    64  	for _, a := range arg2 {
    65  		varargs = append(varargs, a)
    66  	}
    67  	ret := m.ctrl.Call(m, "GetHost", varargs...)
    68  	ret0, _ := ret[0].(*v1alpha1.HostData)
    69  	ret1, _ := ret[1].(error)
    70  	return ret0, ret1
    71  }
    72  
    73  // GetHost indicates an expected call of GetHost
    74  func (mr *MockNodeClientMockRecorder) GetHost(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    75  	mr.mock.ctrl.T.Helper()
    76  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    77  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHost", reflect.TypeOf((*MockNodeClient)(nil).GetHost), varargs...)
    78  }
    79  
    80  // GetPeer mocks base method
    81  func (m *MockNodeClient) GetPeer(arg0 context.Context, arg1 *v1alpha1.PeerRequest, arg2 ...grpc.CallOption) (*v1alpha1.Peer, error) {
    82  	m.ctrl.T.Helper()
    83  	varargs := []interface{}{arg0, arg1}
    84  	for _, a := range arg2 {
    85  		varargs = append(varargs, a)
    86  	}
    87  	ret := m.ctrl.Call(m, "GetPeer", varargs...)
    88  	ret0, _ := ret[0].(*v1alpha1.Peer)
    89  	ret1, _ := ret[1].(error)
    90  	return ret0, ret1
    91  }
    92  
    93  // GetPeer indicates an expected call of GetPeer
    94  func (mr *MockNodeClientMockRecorder) GetPeer(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    95  	mr.mock.ctrl.T.Helper()
    96  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    97  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPeer", reflect.TypeOf((*MockNodeClient)(nil).GetPeer), varargs...)
    98  }
    99  
   100  // GetSyncStatus mocks base method
   101  func (m *MockNodeClient) GetSyncStatus(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.SyncStatus, error) {
   102  	m.ctrl.T.Helper()
   103  	varargs := []interface{}{arg0, arg1}
   104  	for _, a := range arg2 {
   105  		varargs = append(varargs, a)
   106  	}
   107  	ret := m.ctrl.Call(m, "GetSyncStatus", varargs...)
   108  	ret0, _ := ret[0].(*v1alpha1.SyncStatus)
   109  	ret1, _ := ret[1].(error)
   110  	return ret0, ret1
   111  }
   112  
   113  // GetSyncStatus indicates an expected call of GetSyncStatus
   114  func (mr *MockNodeClientMockRecorder) GetSyncStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   115  	mr.mock.ctrl.T.Helper()
   116  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   117  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncStatus", reflect.TypeOf((*MockNodeClient)(nil).GetSyncStatus), varargs...)
   118  }
   119  
   120  // GetVersion mocks base method
   121  func (m *MockNodeClient) GetVersion(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.Version, error) {
   122  	m.ctrl.T.Helper()
   123  	varargs := []interface{}{arg0, arg1}
   124  	for _, a := range arg2 {
   125  		varargs = append(varargs, a)
   126  	}
   127  	ret := m.ctrl.Call(m, "GetVersion", varargs...)
   128  	ret0, _ := ret[0].(*v1alpha1.Version)
   129  	ret1, _ := ret[1].(error)
   130  	return ret0, ret1
   131  }
   132  
   133  // GetVersion indicates an expected call of GetVersion
   134  func (mr *MockNodeClientMockRecorder) GetVersion(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   135  	mr.mock.ctrl.T.Helper()
   136  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   137  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVersion", reflect.TypeOf((*MockNodeClient)(nil).GetVersion), varargs...)
   138  }
   139  
   140  // ListImplementedServices mocks base method
   141  func (m *MockNodeClient) ListImplementedServices(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.ImplementedServices, error) {
   142  	m.ctrl.T.Helper()
   143  	varargs := []interface{}{arg0, arg1}
   144  	for _, a := range arg2 {
   145  		varargs = append(varargs, a)
   146  	}
   147  	ret := m.ctrl.Call(m, "ListImplementedServices", varargs...)
   148  	ret0, _ := ret[0].(*v1alpha1.ImplementedServices)
   149  	ret1, _ := ret[1].(error)
   150  	return ret0, ret1
   151  }
   152  
   153  // ListImplementedServices indicates an expected call of ListImplementedServices
   154  func (mr *MockNodeClientMockRecorder) ListImplementedServices(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   155  	mr.mock.ctrl.T.Helper()
   156  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   157  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListImplementedServices", reflect.TypeOf((*MockNodeClient)(nil).ListImplementedServices), varargs...)
   158  }
   159  
   160  // ListPeers mocks base method
   161  func (m *MockNodeClient) ListPeers(arg0 context.Context, arg1 *emptypb.Empty, arg2 ...grpc.CallOption) (*v1alpha1.Peers, error) {
   162  	m.ctrl.T.Helper()
   163  	varargs := []interface{}{arg0, arg1}
   164  	for _, a := range arg2 {
   165  		varargs = append(varargs, a)
   166  	}
   167  	ret := m.ctrl.Call(m, "ListPeers", varargs...)
   168  	ret0, _ := ret[0].(*v1alpha1.Peers)
   169  	ret1, _ := ret[1].(error)
   170  	return ret0, ret1
   171  }
   172  
   173  // ListPeers indicates an expected call of ListPeers
   174  func (mr *MockNodeClientMockRecorder) ListPeers(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   175  	mr.mock.ctrl.T.Helper()
   176  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   177  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPeers", reflect.TypeOf((*MockNodeClient)(nil).ListPeers), varargs...)
   178  }