github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/dm/pbmock/dmworker.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/pingcap/tiflow/dm/pb (interfaces: WorkerClient,WorkerServer)
     3  
     4  // Package pbmock is a generated GoMock package.
     5  package pbmock
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	pb "github.com/pingcap/tiflow/dm/pb"
    13  	grpc "google.golang.org/grpc"
    14  )
    15  
    16  // MockWorkerClient is a mock of WorkerClient interface.
    17  type MockWorkerClient struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockWorkerClientMockRecorder
    20  }
    21  
    22  // MockWorkerClientMockRecorder is the mock recorder for MockWorkerClient.
    23  type MockWorkerClientMockRecorder struct {
    24  	mock *MockWorkerClient
    25  }
    26  
    27  // NewMockWorkerClient creates a new mock instance.
    28  func NewMockWorkerClient(ctrl *gomock.Controller) *MockWorkerClient {
    29  	mock := &MockWorkerClient{ctrl: ctrl}
    30  	mock.recorder = &MockWorkerClientMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockWorkerClient) EXPECT() *MockWorkerClientMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // CheckSubtasksCanUpdate mocks base method.
    40  func (m *MockWorkerClient) CheckSubtasksCanUpdate(arg0 context.Context, arg1 *pb.CheckSubtasksCanUpdateRequest, arg2 ...grpc.CallOption) (*pb.CheckSubtasksCanUpdateResponse, error) {
    41  	m.ctrl.T.Helper()
    42  	varargs := []interface{}{arg0, arg1}
    43  	for _, a := range arg2 {
    44  		varargs = append(varargs, a)
    45  	}
    46  	ret := m.ctrl.Call(m, "CheckSubtasksCanUpdate", varargs...)
    47  	ret0, _ := ret[0].(*pb.CheckSubtasksCanUpdateResponse)
    48  	ret1, _ := ret[1].(error)
    49  	return ret0, ret1
    50  }
    51  
    52  // CheckSubtasksCanUpdate indicates an expected call of CheckSubtasksCanUpdate.
    53  func (mr *MockWorkerClientMockRecorder) CheckSubtasksCanUpdate(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    54  	mr.mock.ctrl.T.Helper()
    55  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    56  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckSubtasksCanUpdate", reflect.TypeOf((*MockWorkerClient)(nil).CheckSubtasksCanUpdate), varargs...)
    57  }
    58  
    59  // GetValidatorError mocks base method.
    60  func (m *MockWorkerClient) GetValidatorError(arg0 context.Context, arg1 *pb.GetValidationErrorRequest, arg2 ...grpc.CallOption) (*pb.GetValidationErrorResponse, error) {
    61  	m.ctrl.T.Helper()
    62  	varargs := []interface{}{arg0, arg1}
    63  	for _, a := range arg2 {
    64  		varargs = append(varargs, a)
    65  	}
    66  	ret := m.ctrl.Call(m, "GetValidatorError", varargs...)
    67  	ret0, _ := ret[0].(*pb.GetValidationErrorResponse)
    68  	ret1, _ := ret[1].(error)
    69  	return ret0, ret1
    70  }
    71  
    72  // GetValidatorError indicates an expected call of GetValidatorError.
    73  func (mr *MockWorkerClientMockRecorder) GetValidatorError(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    74  	mr.mock.ctrl.T.Helper()
    75  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    76  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetValidatorError", reflect.TypeOf((*MockWorkerClient)(nil).GetValidatorError), varargs...)
    77  }
    78  
    79  // GetWorkerCfg mocks base method.
    80  func (m *MockWorkerClient) GetWorkerCfg(arg0 context.Context, arg1 *pb.GetWorkerCfgRequest, arg2 ...grpc.CallOption) (*pb.GetWorkerCfgResponse, error) {
    81  	m.ctrl.T.Helper()
    82  	varargs := []interface{}{arg0, arg1}
    83  	for _, a := range arg2 {
    84  		varargs = append(varargs, a)
    85  	}
    86  	ret := m.ctrl.Call(m, "GetWorkerCfg", varargs...)
    87  	ret0, _ := ret[0].(*pb.GetWorkerCfgResponse)
    88  	ret1, _ := ret[1].(error)
    89  	return ret0, ret1
    90  }
    91  
    92  // GetWorkerCfg indicates an expected call of GetWorkerCfg.
    93  func (mr *MockWorkerClientMockRecorder) GetWorkerCfg(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    94  	mr.mock.ctrl.T.Helper()
    95  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    96  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerCfg", reflect.TypeOf((*MockWorkerClient)(nil).GetWorkerCfg), varargs...)
    97  }
    98  
    99  // GetWorkerValidatorStatus mocks base method.
   100  func (m *MockWorkerClient) GetWorkerValidatorStatus(arg0 context.Context, arg1 *pb.GetValidationStatusRequest, arg2 ...grpc.CallOption) (*pb.GetValidationStatusResponse, error) {
   101  	m.ctrl.T.Helper()
   102  	varargs := []interface{}{arg0, arg1}
   103  	for _, a := range arg2 {
   104  		varargs = append(varargs, a)
   105  	}
   106  	ret := m.ctrl.Call(m, "GetWorkerValidatorStatus", varargs...)
   107  	ret0, _ := ret[0].(*pb.GetValidationStatusResponse)
   108  	ret1, _ := ret[1].(error)
   109  	return ret0, ret1
   110  }
   111  
   112  // GetWorkerValidatorStatus indicates an expected call of GetWorkerValidatorStatus.
   113  func (mr *MockWorkerClientMockRecorder) GetWorkerValidatorStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   114  	mr.mock.ctrl.T.Helper()
   115  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   116  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerValidatorStatus", reflect.TypeOf((*MockWorkerClient)(nil).GetWorkerValidatorStatus), varargs...)
   117  }
   118  
   119  // HandleError mocks base method.
   120  func (m *MockWorkerClient) HandleError(arg0 context.Context, arg1 *pb.HandleWorkerErrorRequest, arg2 ...grpc.CallOption) (*pb.CommonWorkerResponse, error) {
   121  	m.ctrl.T.Helper()
   122  	varargs := []interface{}{arg0, arg1}
   123  	for _, a := range arg2 {
   124  		varargs = append(varargs, a)
   125  	}
   126  	ret := m.ctrl.Call(m, "HandleError", varargs...)
   127  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   128  	ret1, _ := ret[1].(error)
   129  	return ret0, ret1
   130  }
   131  
   132  // HandleError indicates an expected call of HandleError.
   133  func (mr *MockWorkerClientMockRecorder) HandleError(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   134  	mr.mock.ctrl.T.Helper()
   135  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   136  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleError", reflect.TypeOf((*MockWorkerClient)(nil).HandleError), varargs...)
   137  }
   138  
   139  // OperateSchema mocks base method.
   140  func (m *MockWorkerClient) OperateSchema(arg0 context.Context, arg1 *pb.OperateWorkerSchemaRequest, arg2 ...grpc.CallOption) (*pb.CommonWorkerResponse, error) {
   141  	m.ctrl.T.Helper()
   142  	varargs := []interface{}{arg0, arg1}
   143  	for _, a := range arg2 {
   144  		varargs = append(varargs, a)
   145  	}
   146  	ret := m.ctrl.Call(m, "OperateSchema", varargs...)
   147  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   148  	ret1, _ := ret[1].(error)
   149  	return ret0, ret1
   150  }
   151  
   152  // OperateSchema indicates an expected call of OperateSchema.
   153  func (mr *MockWorkerClientMockRecorder) OperateSchema(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   154  	mr.mock.ctrl.T.Helper()
   155  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   156  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateSchema", reflect.TypeOf((*MockWorkerClient)(nil).OperateSchema), varargs...)
   157  }
   158  
   159  // OperateV1Meta mocks base method.
   160  func (m *MockWorkerClient) OperateV1Meta(arg0 context.Context, arg1 *pb.OperateV1MetaRequest, arg2 ...grpc.CallOption) (*pb.OperateV1MetaResponse, error) {
   161  	m.ctrl.T.Helper()
   162  	varargs := []interface{}{arg0, arg1}
   163  	for _, a := range arg2 {
   164  		varargs = append(varargs, a)
   165  	}
   166  	ret := m.ctrl.Call(m, "OperateV1Meta", varargs...)
   167  	ret0, _ := ret[0].(*pb.OperateV1MetaResponse)
   168  	ret1, _ := ret[1].(error)
   169  	return ret0, ret1
   170  }
   171  
   172  // OperateV1Meta indicates an expected call of OperateV1Meta.
   173  func (mr *MockWorkerClientMockRecorder) OperateV1Meta(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   174  	mr.mock.ctrl.T.Helper()
   175  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   176  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateV1Meta", reflect.TypeOf((*MockWorkerClient)(nil).OperateV1Meta), varargs...)
   177  }
   178  
   179  // OperateValidatorError mocks base method.
   180  func (m *MockWorkerClient) OperateValidatorError(arg0 context.Context, arg1 *pb.OperateValidationErrorRequest, arg2 ...grpc.CallOption) (*pb.OperateValidationErrorResponse, error) {
   181  	m.ctrl.T.Helper()
   182  	varargs := []interface{}{arg0, arg1}
   183  	for _, a := range arg2 {
   184  		varargs = append(varargs, a)
   185  	}
   186  	ret := m.ctrl.Call(m, "OperateValidatorError", varargs...)
   187  	ret0, _ := ret[0].(*pb.OperateValidationErrorResponse)
   188  	ret1, _ := ret[1].(error)
   189  	return ret0, ret1
   190  }
   191  
   192  // OperateValidatorError indicates an expected call of OperateValidatorError.
   193  func (mr *MockWorkerClientMockRecorder) OperateValidatorError(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   194  	mr.mock.ctrl.T.Helper()
   195  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   196  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateValidatorError", reflect.TypeOf((*MockWorkerClient)(nil).OperateValidatorError), varargs...)
   197  }
   198  
   199  // PurgeRelay mocks base method.
   200  func (m *MockWorkerClient) PurgeRelay(arg0 context.Context, arg1 *pb.PurgeRelayRequest, arg2 ...grpc.CallOption) (*pb.CommonWorkerResponse, error) {
   201  	m.ctrl.T.Helper()
   202  	varargs := []interface{}{arg0, arg1}
   203  	for _, a := range arg2 {
   204  		varargs = append(varargs, a)
   205  	}
   206  	ret := m.ctrl.Call(m, "PurgeRelay", varargs...)
   207  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   208  	ret1, _ := ret[1].(error)
   209  	return ret0, ret1
   210  }
   211  
   212  // PurgeRelay indicates an expected call of PurgeRelay.
   213  func (mr *MockWorkerClientMockRecorder) PurgeRelay(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   214  	mr.mock.ctrl.T.Helper()
   215  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   216  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PurgeRelay", reflect.TypeOf((*MockWorkerClient)(nil).PurgeRelay), varargs...)
   217  }
   218  
   219  // QueryStatus mocks base method.
   220  func (m *MockWorkerClient) QueryStatus(arg0 context.Context, arg1 *pb.QueryStatusRequest, arg2 ...grpc.CallOption) (*pb.QueryStatusResponse, error) {
   221  	m.ctrl.T.Helper()
   222  	varargs := []interface{}{arg0, arg1}
   223  	for _, a := range arg2 {
   224  		varargs = append(varargs, a)
   225  	}
   226  	ret := m.ctrl.Call(m, "QueryStatus", varargs...)
   227  	ret0, _ := ret[0].(*pb.QueryStatusResponse)
   228  	ret1, _ := ret[1].(error)
   229  	return ret0, ret1
   230  }
   231  
   232  // QueryStatus indicates an expected call of QueryStatus.
   233  func (mr *MockWorkerClientMockRecorder) QueryStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   234  	mr.mock.ctrl.T.Helper()
   235  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   236  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryStatus", reflect.TypeOf((*MockWorkerClient)(nil).QueryStatus), varargs...)
   237  }
   238  
   239  // UpdateValidator mocks base method.
   240  func (m *MockWorkerClient) UpdateValidator(arg0 context.Context, arg1 *pb.UpdateValidationWorkerRequest, arg2 ...grpc.CallOption) (*pb.CommonWorkerResponse, error) {
   241  	m.ctrl.T.Helper()
   242  	varargs := []interface{}{arg0, arg1}
   243  	for _, a := range arg2 {
   244  		varargs = append(varargs, a)
   245  	}
   246  	ret := m.ctrl.Call(m, "UpdateValidator", varargs...)
   247  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   248  	ret1, _ := ret[1].(error)
   249  	return ret0, ret1
   250  }
   251  
   252  // UpdateValidator indicates an expected call of UpdateValidator.
   253  func (mr *MockWorkerClientMockRecorder) UpdateValidator(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
   254  	mr.mock.ctrl.T.Helper()
   255  	varargs := append([]interface{}{arg0, arg1}, arg2...)
   256  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateValidator", reflect.TypeOf((*MockWorkerClient)(nil).UpdateValidator), varargs...)
   257  }
   258  
   259  // MockWorkerServer is a mock of WorkerServer interface.
   260  type MockWorkerServer struct {
   261  	ctrl     *gomock.Controller
   262  	recorder *MockWorkerServerMockRecorder
   263  }
   264  
   265  // MockWorkerServerMockRecorder is the mock recorder for MockWorkerServer.
   266  type MockWorkerServerMockRecorder struct {
   267  	mock *MockWorkerServer
   268  }
   269  
   270  // NewMockWorkerServer creates a new mock instance.
   271  func NewMockWorkerServer(ctrl *gomock.Controller) *MockWorkerServer {
   272  	mock := &MockWorkerServer{ctrl: ctrl}
   273  	mock.recorder = &MockWorkerServerMockRecorder{mock}
   274  	return mock
   275  }
   276  
   277  // EXPECT returns an object that allows the caller to indicate expected use.
   278  func (m *MockWorkerServer) EXPECT() *MockWorkerServerMockRecorder {
   279  	return m.recorder
   280  }
   281  
   282  // CheckSubtasksCanUpdate mocks base method.
   283  func (m *MockWorkerServer) CheckSubtasksCanUpdate(arg0 context.Context, arg1 *pb.CheckSubtasksCanUpdateRequest) (*pb.CheckSubtasksCanUpdateResponse, error) {
   284  	m.ctrl.T.Helper()
   285  	ret := m.ctrl.Call(m, "CheckSubtasksCanUpdate", arg0, arg1)
   286  	ret0, _ := ret[0].(*pb.CheckSubtasksCanUpdateResponse)
   287  	ret1, _ := ret[1].(error)
   288  	return ret0, ret1
   289  }
   290  
   291  // CheckSubtasksCanUpdate indicates an expected call of CheckSubtasksCanUpdate.
   292  func (mr *MockWorkerServerMockRecorder) CheckSubtasksCanUpdate(arg0, arg1 interface{}) *gomock.Call {
   293  	mr.mock.ctrl.T.Helper()
   294  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckSubtasksCanUpdate", reflect.TypeOf((*MockWorkerServer)(nil).CheckSubtasksCanUpdate), arg0, arg1)
   295  }
   296  
   297  // GetValidatorError mocks base method.
   298  func (m *MockWorkerServer) GetValidatorError(arg0 context.Context, arg1 *pb.GetValidationErrorRequest) (*pb.GetValidationErrorResponse, error) {
   299  	m.ctrl.T.Helper()
   300  	ret := m.ctrl.Call(m, "GetValidatorError", arg0, arg1)
   301  	ret0, _ := ret[0].(*pb.GetValidationErrorResponse)
   302  	ret1, _ := ret[1].(error)
   303  	return ret0, ret1
   304  }
   305  
   306  // GetValidatorError indicates an expected call of GetValidatorError.
   307  func (mr *MockWorkerServerMockRecorder) GetValidatorError(arg0, arg1 interface{}) *gomock.Call {
   308  	mr.mock.ctrl.T.Helper()
   309  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetValidatorError", reflect.TypeOf((*MockWorkerServer)(nil).GetValidatorError), arg0, arg1)
   310  }
   311  
   312  // GetWorkerCfg mocks base method.
   313  func (m *MockWorkerServer) GetWorkerCfg(arg0 context.Context, arg1 *pb.GetWorkerCfgRequest) (*pb.GetWorkerCfgResponse, error) {
   314  	m.ctrl.T.Helper()
   315  	ret := m.ctrl.Call(m, "GetWorkerCfg", arg0, arg1)
   316  	ret0, _ := ret[0].(*pb.GetWorkerCfgResponse)
   317  	ret1, _ := ret[1].(error)
   318  	return ret0, ret1
   319  }
   320  
   321  // GetWorkerCfg indicates an expected call of GetWorkerCfg.
   322  func (mr *MockWorkerServerMockRecorder) GetWorkerCfg(arg0, arg1 interface{}) *gomock.Call {
   323  	mr.mock.ctrl.T.Helper()
   324  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerCfg", reflect.TypeOf((*MockWorkerServer)(nil).GetWorkerCfg), arg0, arg1)
   325  }
   326  
   327  // GetWorkerValidatorStatus mocks base method.
   328  func (m *MockWorkerServer) GetWorkerValidatorStatus(arg0 context.Context, arg1 *pb.GetValidationStatusRequest) (*pb.GetValidationStatusResponse, error) {
   329  	m.ctrl.T.Helper()
   330  	ret := m.ctrl.Call(m, "GetWorkerValidatorStatus", arg0, arg1)
   331  	ret0, _ := ret[0].(*pb.GetValidationStatusResponse)
   332  	ret1, _ := ret[1].(error)
   333  	return ret0, ret1
   334  }
   335  
   336  // GetWorkerValidatorStatus indicates an expected call of GetWorkerValidatorStatus.
   337  func (mr *MockWorkerServerMockRecorder) GetWorkerValidatorStatus(arg0, arg1 interface{}) *gomock.Call {
   338  	mr.mock.ctrl.T.Helper()
   339  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerValidatorStatus", reflect.TypeOf((*MockWorkerServer)(nil).GetWorkerValidatorStatus), arg0, arg1)
   340  }
   341  
   342  // HandleError mocks base method.
   343  func (m *MockWorkerServer) HandleError(arg0 context.Context, arg1 *pb.HandleWorkerErrorRequest) (*pb.CommonWorkerResponse, error) {
   344  	m.ctrl.T.Helper()
   345  	ret := m.ctrl.Call(m, "HandleError", arg0, arg1)
   346  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   347  	ret1, _ := ret[1].(error)
   348  	return ret0, ret1
   349  }
   350  
   351  // HandleError indicates an expected call of HandleError.
   352  func (mr *MockWorkerServerMockRecorder) HandleError(arg0, arg1 interface{}) *gomock.Call {
   353  	mr.mock.ctrl.T.Helper()
   354  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HandleError", reflect.TypeOf((*MockWorkerServer)(nil).HandleError), arg0, arg1)
   355  }
   356  
   357  // OperateSchema mocks base method.
   358  func (m *MockWorkerServer) OperateSchema(arg0 context.Context, arg1 *pb.OperateWorkerSchemaRequest) (*pb.CommonWorkerResponse, error) {
   359  	m.ctrl.T.Helper()
   360  	ret := m.ctrl.Call(m, "OperateSchema", arg0, arg1)
   361  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   362  	ret1, _ := ret[1].(error)
   363  	return ret0, ret1
   364  }
   365  
   366  // OperateSchema indicates an expected call of OperateSchema.
   367  func (mr *MockWorkerServerMockRecorder) OperateSchema(arg0, arg1 interface{}) *gomock.Call {
   368  	mr.mock.ctrl.T.Helper()
   369  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateSchema", reflect.TypeOf((*MockWorkerServer)(nil).OperateSchema), arg0, arg1)
   370  }
   371  
   372  // OperateV1Meta mocks base method.
   373  func (m *MockWorkerServer) OperateV1Meta(arg0 context.Context, arg1 *pb.OperateV1MetaRequest) (*pb.OperateV1MetaResponse, error) {
   374  	m.ctrl.T.Helper()
   375  	ret := m.ctrl.Call(m, "OperateV1Meta", arg0, arg1)
   376  	ret0, _ := ret[0].(*pb.OperateV1MetaResponse)
   377  	ret1, _ := ret[1].(error)
   378  	return ret0, ret1
   379  }
   380  
   381  // OperateV1Meta indicates an expected call of OperateV1Meta.
   382  func (mr *MockWorkerServerMockRecorder) OperateV1Meta(arg0, arg1 interface{}) *gomock.Call {
   383  	mr.mock.ctrl.T.Helper()
   384  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateV1Meta", reflect.TypeOf((*MockWorkerServer)(nil).OperateV1Meta), arg0, arg1)
   385  }
   386  
   387  // OperateValidatorError mocks base method.
   388  func (m *MockWorkerServer) OperateValidatorError(arg0 context.Context, arg1 *pb.OperateValidationErrorRequest) (*pb.OperateValidationErrorResponse, error) {
   389  	m.ctrl.T.Helper()
   390  	ret := m.ctrl.Call(m, "OperateValidatorError", arg0, arg1)
   391  	ret0, _ := ret[0].(*pb.OperateValidationErrorResponse)
   392  	ret1, _ := ret[1].(error)
   393  	return ret0, ret1
   394  }
   395  
   396  // OperateValidatorError indicates an expected call of OperateValidatorError.
   397  func (mr *MockWorkerServerMockRecorder) OperateValidatorError(arg0, arg1 interface{}) *gomock.Call {
   398  	mr.mock.ctrl.T.Helper()
   399  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperateValidatorError", reflect.TypeOf((*MockWorkerServer)(nil).OperateValidatorError), arg0, arg1)
   400  }
   401  
   402  // PurgeRelay mocks base method.
   403  func (m *MockWorkerServer) PurgeRelay(arg0 context.Context, arg1 *pb.PurgeRelayRequest) (*pb.CommonWorkerResponse, error) {
   404  	m.ctrl.T.Helper()
   405  	ret := m.ctrl.Call(m, "PurgeRelay", arg0, arg1)
   406  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   407  	ret1, _ := ret[1].(error)
   408  	return ret0, ret1
   409  }
   410  
   411  // PurgeRelay indicates an expected call of PurgeRelay.
   412  func (mr *MockWorkerServerMockRecorder) PurgeRelay(arg0, arg1 interface{}) *gomock.Call {
   413  	mr.mock.ctrl.T.Helper()
   414  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PurgeRelay", reflect.TypeOf((*MockWorkerServer)(nil).PurgeRelay), arg0, arg1)
   415  }
   416  
   417  // QueryStatus mocks base method.
   418  func (m *MockWorkerServer) QueryStatus(arg0 context.Context, arg1 *pb.QueryStatusRequest) (*pb.QueryStatusResponse, error) {
   419  	m.ctrl.T.Helper()
   420  	ret := m.ctrl.Call(m, "QueryStatus", arg0, arg1)
   421  	ret0, _ := ret[0].(*pb.QueryStatusResponse)
   422  	ret1, _ := ret[1].(error)
   423  	return ret0, ret1
   424  }
   425  
   426  // QueryStatus indicates an expected call of QueryStatus.
   427  func (mr *MockWorkerServerMockRecorder) QueryStatus(arg0, arg1 interface{}) *gomock.Call {
   428  	mr.mock.ctrl.T.Helper()
   429  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryStatus", reflect.TypeOf((*MockWorkerServer)(nil).QueryStatus), arg0, arg1)
   430  }
   431  
   432  // UpdateValidator mocks base method.
   433  func (m *MockWorkerServer) UpdateValidator(arg0 context.Context, arg1 *pb.UpdateValidationWorkerRequest) (*pb.CommonWorkerResponse, error) {
   434  	m.ctrl.T.Helper()
   435  	ret := m.ctrl.Call(m, "UpdateValidator", arg0, arg1)
   436  	ret0, _ := ret[0].(*pb.CommonWorkerResponse)
   437  	ret1, _ := ret[1].(error)
   438  	return ret0, ret1
   439  }
   440  
   441  // UpdateValidator indicates an expected call of UpdateValidator.
   442  func (mr *MockWorkerServerMockRecorder) UpdateValidator(arg0, arg1 interface{}) *gomock.Call {
   443  	mr.mock.ctrl.T.Helper()
   444  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateValidator", reflect.TypeOf((*MockWorkerServer)(nil).UpdateValidator), arg0, arg1)
   445  }