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 }