go.temporal.io/server@v1.23.0/common/testing/mocksdk/client_mock.go (about) 1 // The MIT License 2 // 3 // Copyright (c) 2020 Temporal Technologies Inc. All rights reserved. 4 // 5 // Copyright (c) 2020 Uber Technologies, Inc. 6 // 7 // Permission is hereby granted, free of charge, to any person obtaining a copy 8 // of this software and associated documentation files (the "Software"), to deal 9 // in the Software without restriction, including without limitation the rights 10 // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 11 // copies of the Software, and to permit persons to whom the Software is 12 // furnished to do so, subject to the following conditions: 13 // 14 // The above copyright notice and this permission notice shall be included in 15 // all copies or substantial portions of the Software. 16 // 17 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 18 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 19 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 20 // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 21 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 22 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN 23 // THE SOFTWARE. 24 25 // Code generated by MockGen. DO NOT EDIT. 26 // Source: go.temporal.io/sdk/client (interfaces: Client) 27 28 // Package mocksdk is a generated GoMock package. 29 package mocksdk 30 31 import ( 32 context "context" 33 reflect "reflect" 34 35 gomock "github.com/golang/mock/gomock" 36 enums "go.temporal.io/api/enums/v1" 37 operatorservice "go.temporal.io/api/operatorservice/v1" 38 workflowservice "go.temporal.io/api/workflowservice/v1" 39 client "go.temporal.io/sdk/client" 40 converter "go.temporal.io/sdk/converter" 41 ) 42 43 // MockClient is a mock of Client interface. 44 type MockClient struct { 45 ctrl *gomock.Controller 46 recorder *MockClientMockRecorder 47 } 48 49 // MockClientMockRecorder is the mock recorder for MockClient. 50 type MockClientMockRecorder struct { 51 mock *MockClient 52 } 53 54 // NewMockClient creates a new mock instance. 55 func NewMockClient(ctrl *gomock.Controller) *MockClient { 56 mock := &MockClient{ctrl: ctrl} 57 mock.recorder = &MockClientMockRecorder{mock} 58 return mock 59 } 60 61 // EXPECT returns an object that allows the caller to indicate expected use. 62 func (m *MockClient) EXPECT() *MockClientMockRecorder { 63 return m.recorder 64 } 65 66 // CancelWorkflow mocks base method. 67 func (m *MockClient) CancelWorkflow(arg0 context.Context, arg1, arg2 string) error { 68 m.ctrl.T.Helper() 69 ret := m.ctrl.Call(m, "CancelWorkflow", arg0, arg1, arg2) 70 ret0, _ := ret[0].(error) 71 return ret0 72 } 73 74 // CancelWorkflow indicates an expected call of CancelWorkflow. 75 func (mr *MockClientMockRecorder) CancelWorkflow(arg0, arg1, arg2 interface{}) *gomock.Call { 76 mr.mock.ctrl.T.Helper() 77 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelWorkflow", reflect.TypeOf((*MockClient)(nil).CancelWorkflow), arg0, arg1, arg2) 78 } 79 80 // CheckHealth mocks base method. 81 func (m *MockClient) CheckHealth(arg0 context.Context, arg1 *client.CheckHealthRequest) (*client.CheckHealthResponse, error) { 82 m.ctrl.T.Helper() 83 ret := m.ctrl.Call(m, "CheckHealth", arg0, arg1) 84 ret0, _ := ret[0].(*client.CheckHealthResponse) 85 ret1, _ := ret[1].(error) 86 return ret0, ret1 87 } 88 89 // CheckHealth indicates an expected call of CheckHealth. 90 func (mr *MockClientMockRecorder) CheckHealth(arg0, arg1 interface{}) *gomock.Call { 91 mr.mock.ctrl.T.Helper() 92 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CheckHealth", reflect.TypeOf((*MockClient)(nil).CheckHealth), arg0, arg1) 93 } 94 95 // Close mocks base method. 96 func (m *MockClient) Close() { 97 m.ctrl.T.Helper() 98 m.ctrl.Call(m, "Close") 99 } 100 101 // Close indicates an expected call of Close. 102 func (mr *MockClientMockRecorder) Close() *gomock.Call { 103 mr.mock.ctrl.T.Helper() 104 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockClient)(nil).Close)) 105 } 106 107 // CompleteActivity mocks base method. 108 func (m *MockClient) CompleteActivity(arg0 context.Context, arg1 []byte, arg2 interface{}, arg3 error) error { 109 m.ctrl.T.Helper() 110 ret := m.ctrl.Call(m, "CompleteActivity", arg0, arg1, arg2, arg3) 111 ret0, _ := ret[0].(error) 112 return ret0 113 } 114 115 // CompleteActivity indicates an expected call of CompleteActivity. 116 func (mr *MockClientMockRecorder) CompleteActivity(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { 117 mr.mock.ctrl.T.Helper() 118 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteActivity", reflect.TypeOf((*MockClient)(nil).CompleteActivity), arg0, arg1, arg2, arg3) 119 } 120 121 // CompleteActivityByID mocks base method. 122 func (m *MockClient) CompleteActivityByID(arg0 context.Context, arg1, arg2, arg3, arg4 string, arg5 interface{}, arg6 error) error { 123 m.ctrl.T.Helper() 124 ret := m.ctrl.Call(m, "CompleteActivityByID", arg0, arg1, arg2, arg3, arg4, arg5, arg6) 125 ret0, _ := ret[0].(error) 126 return ret0 127 } 128 129 // CompleteActivityByID indicates an expected call of CompleteActivityByID. 130 func (mr *MockClientMockRecorder) CompleteActivityByID(arg0, arg1, arg2, arg3, arg4, arg5, arg6 interface{}) *gomock.Call { 131 mr.mock.ctrl.T.Helper() 132 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CompleteActivityByID", reflect.TypeOf((*MockClient)(nil).CompleteActivityByID), arg0, arg1, arg2, arg3, arg4, arg5, arg6) 133 } 134 135 // CountWorkflow mocks base method. 136 func (m *MockClient) CountWorkflow(arg0 context.Context, arg1 *workflowservice.CountWorkflowExecutionsRequest) (*workflowservice.CountWorkflowExecutionsResponse, error) { 137 m.ctrl.T.Helper() 138 ret := m.ctrl.Call(m, "CountWorkflow", arg0, arg1) 139 ret0, _ := ret[0].(*workflowservice.CountWorkflowExecutionsResponse) 140 ret1, _ := ret[1].(error) 141 return ret0, ret1 142 } 143 144 // CountWorkflow indicates an expected call of CountWorkflow. 145 func (mr *MockClientMockRecorder) CountWorkflow(arg0, arg1 interface{}) *gomock.Call { 146 mr.mock.ctrl.T.Helper() 147 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CountWorkflow", reflect.TypeOf((*MockClient)(nil).CountWorkflow), arg0, arg1) 148 } 149 150 // DescribeTaskQueue mocks base method. 151 func (m *MockClient) DescribeTaskQueue(arg0 context.Context, arg1 string, arg2 enums.TaskQueueType) (*workflowservice.DescribeTaskQueueResponse, error) { 152 m.ctrl.T.Helper() 153 ret := m.ctrl.Call(m, "DescribeTaskQueue", arg0, arg1, arg2) 154 ret0, _ := ret[0].(*workflowservice.DescribeTaskQueueResponse) 155 ret1, _ := ret[1].(error) 156 return ret0, ret1 157 } 158 159 // DescribeTaskQueue indicates an expected call of DescribeTaskQueue. 160 func (mr *MockClientMockRecorder) DescribeTaskQueue(arg0, arg1, arg2 interface{}) *gomock.Call { 161 mr.mock.ctrl.T.Helper() 162 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DescribeTaskQueue", reflect.TypeOf((*MockClient)(nil).DescribeTaskQueue), arg0, arg1, arg2) 163 } 164 165 // DescribeWorkflowExecution mocks base method. 166 func (m *MockClient) DescribeWorkflowExecution(arg0 context.Context, arg1, arg2 string) (*workflowservice.DescribeWorkflowExecutionResponse, error) { 167 m.ctrl.T.Helper() 168 ret := m.ctrl.Call(m, "DescribeWorkflowExecution", arg0, arg1, arg2) 169 ret0, _ := ret[0].(*workflowservice.DescribeWorkflowExecutionResponse) 170 ret1, _ := ret[1].(error) 171 return ret0, ret1 172 } 173 174 // DescribeWorkflowExecution indicates an expected call of DescribeWorkflowExecution. 175 func (mr *MockClientMockRecorder) DescribeWorkflowExecution(arg0, arg1, arg2 interface{}) *gomock.Call { 176 mr.mock.ctrl.T.Helper() 177 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DescribeWorkflowExecution", reflect.TypeOf((*MockClient)(nil).DescribeWorkflowExecution), arg0, arg1, arg2) 178 } 179 180 // ExecuteWorkflow mocks base method. 181 func (m *MockClient) ExecuteWorkflow(arg0 context.Context, arg1 client.StartWorkflowOptions, arg2 interface{}, arg3 ...interface{}) (client.WorkflowRun, error) { 182 m.ctrl.T.Helper() 183 varargs := []interface{}{arg0, arg1, arg2} 184 for _, a := range arg3 { 185 varargs = append(varargs, a) 186 } 187 ret := m.ctrl.Call(m, "ExecuteWorkflow", varargs...) 188 ret0, _ := ret[0].(client.WorkflowRun) 189 ret1, _ := ret[1].(error) 190 return ret0, ret1 191 } 192 193 // ExecuteWorkflow indicates an expected call of ExecuteWorkflow. 194 func (mr *MockClientMockRecorder) ExecuteWorkflow(arg0, arg1, arg2 interface{}, arg3 ...interface{}) *gomock.Call { 195 mr.mock.ctrl.T.Helper() 196 varargs := append([]interface{}{arg0, arg1, arg2}, arg3...) 197 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExecuteWorkflow", reflect.TypeOf((*MockClient)(nil).ExecuteWorkflow), varargs...) 198 } 199 200 // GetSearchAttributes mocks base method. 201 func (m *MockClient) GetSearchAttributes(arg0 context.Context) (*workflowservice.GetSearchAttributesResponse, error) { 202 m.ctrl.T.Helper() 203 ret := m.ctrl.Call(m, "GetSearchAttributes", arg0) 204 ret0, _ := ret[0].(*workflowservice.GetSearchAttributesResponse) 205 ret1, _ := ret[1].(error) 206 return ret0, ret1 207 } 208 209 // GetSearchAttributes indicates an expected call of GetSearchAttributes. 210 func (mr *MockClientMockRecorder) GetSearchAttributes(arg0 interface{}) *gomock.Call { 211 mr.mock.ctrl.T.Helper() 212 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSearchAttributes", reflect.TypeOf((*MockClient)(nil).GetSearchAttributes), arg0) 213 } 214 215 // GetWorkerBuildIdCompatibility mocks base method. 216 func (m *MockClient) GetWorkerBuildIdCompatibility(arg0 context.Context, arg1 *client.GetWorkerBuildIdCompatibilityOptions) (*client.WorkerBuildIDVersionSets, error) { 217 m.ctrl.T.Helper() 218 ret := m.ctrl.Call(m, "GetWorkerBuildIdCompatibility", arg0, arg1) 219 ret0, _ := ret[0].(*client.WorkerBuildIDVersionSets) 220 ret1, _ := ret[1].(error) 221 return ret0, ret1 222 } 223 224 // GetWorkerBuildIdCompatibility indicates an expected call of GetWorkerBuildIdCompatibility. 225 func (mr *MockClientMockRecorder) GetWorkerBuildIdCompatibility(arg0, arg1 interface{}) *gomock.Call { 226 mr.mock.ctrl.T.Helper() 227 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerBuildIdCompatibility", reflect.TypeOf((*MockClient)(nil).GetWorkerBuildIdCompatibility), arg0, arg1) 228 } 229 230 // GetWorkerTaskReachability mocks base method. 231 func (m *MockClient) GetWorkerTaskReachability(arg0 context.Context, arg1 *client.GetWorkerTaskReachabilityOptions) (*client.WorkerTaskReachability, error) { 232 m.ctrl.T.Helper() 233 ret := m.ctrl.Call(m, "GetWorkerTaskReachability", arg0, arg1) 234 ret0, _ := ret[0].(*client.WorkerTaskReachability) 235 ret1, _ := ret[1].(error) 236 return ret0, ret1 237 } 238 239 // GetWorkerTaskReachability indicates an expected call of GetWorkerTaskReachability. 240 func (mr *MockClientMockRecorder) GetWorkerTaskReachability(arg0, arg1 interface{}) *gomock.Call { 241 mr.mock.ctrl.T.Helper() 242 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkerTaskReachability", reflect.TypeOf((*MockClient)(nil).GetWorkerTaskReachability), arg0, arg1) 243 } 244 245 // GetWorkflow mocks base method. 246 func (m *MockClient) GetWorkflow(arg0 context.Context, arg1, arg2 string) client.WorkflowRun { 247 m.ctrl.T.Helper() 248 ret := m.ctrl.Call(m, "GetWorkflow", arg0, arg1, arg2) 249 ret0, _ := ret[0].(client.WorkflowRun) 250 return ret0 251 } 252 253 // GetWorkflow indicates an expected call of GetWorkflow. 254 func (mr *MockClientMockRecorder) GetWorkflow(arg0, arg1, arg2 interface{}) *gomock.Call { 255 mr.mock.ctrl.T.Helper() 256 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflow", reflect.TypeOf((*MockClient)(nil).GetWorkflow), arg0, arg1, arg2) 257 } 258 259 // GetWorkflowHistory mocks base method. 260 func (m *MockClient) GetWorkflowHistory(arg0 context.Context, arg1, arg2 string, arg3 bool, arg4 enums.HistoryEventFilterType) client.HistoryEventIterator { 261 m.ctrl.T.Helper() 262 ret := m.ctrl.Call(m, "GetWorkflowHistory", arg0, arg1, arg2, arg3, arg4) 263 ret0, _ := ret[0].(client.HistoryEventIterator) 264 return ret0 265 } 266 267 // GetWorkflowHistory indicates an expected call of GetWorkflowHistory. 268 func (mr *MockClientMockRecorder) GetWorkflowHistory(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { 269 mr.mock.ctrl.T.Helper() 270 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflowHistory", reflect.TypeOf((*MockClient)(nil).GetWorkflowHistory), arg0, arg1, arg2, arg3, arg4) 271 } 272 273 // GetWorkflowUpdateHandle mocks base method. 274 func (m *MockClient) GetWorkflowUpdateHandle(arg0 client.GetWorkflowUpdateHandleOptions) client.WorkflowUpdateHandle { 275 m.ctrl.T.Helper() 276 ret := m.ctrl.Call(m, "GetWorkflowUpdateHandle", arg0) 277 ret0, _ := ret[0].(client.WorkflowUpdateHandle) 278 return ret0 279 } 280 281 // GetWorkflowUpdateHandle indicates an expected call of GetWorkflowUpdateHandle. 282 func (mr *MockClientMockRecorder) GetWorkflowUpdateHandle(arg0 interface{}) *gomock.Call { 283 mr.mock.ctrl.T.Helper() 284 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkflowUpdateHandle", reflect.TypeOf((*MockClient)(nil).GetWorkflowUpdateHandle), arg0) 285 } 286 287 // ListArchivedWorkflow mocks base method. 288 func (m *MockClient) ListArchivedWorkflow(arg0 context.Context, arg1 *workflowservice.ListArchivedWorkflowExecutionsRequest) (*workflowservice.ListArchivedWorkflowExecutionsResponse, error) { 289 m.ctrl.T.Helper() 290 ret := m.ctrl.Call(m, "ListArchivedWorkflow", arg0, arg1) 291 ret0, _ := ret[0].(*workflowservice.ListArchivedWorkflowExecutionsResponse) 292 ret1, _ := ret[1].(error) 293 return ret0, ret1 294 } 295 296 // ListArchivedWorkflow indicates an expected call of ListArchivedWorkflow. 297 func (mr *MockClientMockRecorder) ListArchivedWorkflow(arg0, arg1 interface{}) *gomock.Call { 298 mr.mock.ctrl.T.Helper() 299 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListArchivedWorkflow", reflect.TypeOf((*MockClient)(nil).ListArchivedWorkflow), arg0, arg1) 300 } 301 302 // ListClosedWorkflow mocks base method. 303 func (m *MockClient) ListClosedWorkflow(arg0 context.Context, arg1 *workflowservice.ListClosedWorkflowExecutionsRequest) (*workflowservice.ListClosedWorkflowExecutionsResponse, error) { 304 m.ctrl.T.Helper() 305 ret := m.ctrl.Call(m, "ListClosedWorkflow", arg0, arg1) 306 ret0, _ := ret[0].(*workflowservice.ListClosedWorkflowExecutionsResponse) 307 ret1, _ := ret[1].(error) 308 return ret0, ret1 309 } 310 311 // ListClosedWorkflow indicates an expected call of ListClosedWorkflow. 312 func (mr *MockClientMockRecorder) ListClosedWorkflow(arg0, arg1 interface{}) *gomock.Call { 313 mr.mock.ctrl.T.Helper() 314 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListClosedWorkflow", reflect.TypeOf((*MockClient)(nil).ListClosedWorkflow), arg0, arg1) 315 } 316 317 // ListOpenWorkflow mocks base method. 318 func (m *MockClient) ListOpenWorkflow(arg0 context.Context, arg1 *workflowservice.ListOpenWorkflowExecutionsRequest) (*workflowservice.ListOpenWorkflowExecutionsResponse, error) { 319 m.ctrl.T.Helper() 320 ret := m.ctrl.Call(m, "ListOpenWorkflow", arg0, arg1) 321 ret0, _ := ret[0].(*workflowservice.ListOpenWorkflowExecutionsResponse) 322 ret1, _ := ret[1].(error) 323 return ret0, ret1 324 } 325 326 // ListOpenWorkflow indicates an expected call of ListOpenWorkflow. 327 func (mr *MockClientMockRecorder) ListOpenWorkflow(arg0, arg1 interface{}) *gomock.Call { 328 mr.mock.ctrl.T.Helper() 329 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListOpenWorkflow", reflect.TypeOf((*MockClient)(nil).ListOpenWorkflow), arg0, arg1) 330 } 331 332 // ListWorkflow mocks base method. 333 func (m *MockClient) ListWorkflow(arg0 context.Context, arg1 *workflowservice.ListWorkflowExecutionsRequest) (*workflowservice.ListWorkflowExecutionsResponse, error) { 334 m.ctrl.T.Helper() 335 ret := m.ctrl.Call(m, "ListWorkflow", arg0, arg1) 336 ret0, _ := ret[0].(*workflowservice.ListWorkflowExecutionsResponse) 337 ret1, _ := ret[1].(error) 338 return ret0, ret1 339 } 340 341 // ListWorkflow indicates an expected call of ListWorkflow. 342 func (mr *MockClientMockRecorder) ListWorkflow(arg0, arg1 interface{}) *gomock.Call { 343 mr.mock.ctrl.T.Helper() 344 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListWorkflow", reflect.TypeOf((*MockClient)(nil).ListWorkflow), arg0, arg1) 345 } 346 347 // OperatorService mocks base method. 348 func (m *MockClient) OperatorService() operatorservice.OperatorServiceClient { 349 m.ctrl.T.Helper() 350 ret := m.ctrl.Call(m, "OperatorService") 351 ret0, _ := ret[0].(operatorservice.OperatorServiceClient) 352 return ret0 353 } 354 355 // OperatorService indicates an expected call of OperatorService. 356 func (mr *MockClientMockRecorder) OperatorService() *gomock.Call { 357 mr.mock.ctrl.T.Helper() 358 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OperatorService", reflect.TypeOf((*MockClient)(nil).OperatorService)) 359 } 360 361 // QueryWorkflow mocks base method. 362 func (m *MockClient) QueryWorkflow(arg0 context.Context, arg1, arg2, arg3 string, arg4 ...interface{}) (converter.EncodedValue, error) { 363 m.ctrl.T.Helper() 364 varargs := []interface{}{arg0, arg1, arg2, arg3} 365 for _, a := range arg4 { 366 varargs = append(varargs, a) 367 } 368 ret := m.ctrl.Call(m, "QueryWorkflow", varargs...) 369 ret0, _ := ret[0].(converter.EncodedValue) 370 ret1, _ := ret[1].(error) 371 return ret0, ret1 372 } 373 374 // QueryWorkflow indicates an expected call of QueryWorkflow. 375 func (mr *MockClientMockRecorder) QueryWorkflow(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { 376 mr.mock.ctrl.T.Helper() 377 varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) 378 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryWorkflow", reflect.TypeOf((*MockClient)(nil).QueryWorkflow), varargs...) 379 } 380 381 // QueryWorkflowWithOptions mocks base method. 382 func (m *MockClient) QueryWorkflowWithOptions(arg0 context.Context, arg1 *client.QueryWorkflowWithOptionsRequest) (*client.QueryWorkflowWithOptionsResponse, error) { 383 m.ctrl.T.Helper() 384 ret := m.ctrl.Call(m, "QueryWorkflowWithOptions", arg0, arg1) 385 ret0, _ := ret[0].(*client.QueryWorkflowWithOptionsResponse) 386 ret1, _ := ret[1].(error) 387 return ret0, ret1 388 } 389 390 // QueryWorkflowWithOptions indicates an expected call of QueryWorkflowWithOptions. 391 func (mr *MockClientMockRecorder) QueryWorkflowWithOptions(arg0, arg1 interface{}) *gomock.Call { 392 mr.mock.ctrl.T.Helper() 393 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryWorkflowWithOptions", reflect.TypeOf((*MockClient)(nil).QueryWorkflowWithOptions), arg0, arg1) 394 } 395 396 // RecordActivityHeartbeat mocks base method. 397 func (m *MockClient) RecordActivityHeartbeat(arg0 context.Context, arg1 []byte, arg2 ...interface{}) error { 398 m.ctrl.T.Helper() 399 varargs := []interface{}{arg0, arg1} 400 for _, a := range arg2 { 401 varargs = append(varargs, a) 402 } 403 ret := m.ctrl.Call(m, "RecordActivityHeartbeat", varargs...) 404 ret0, _ := ret[0].(error) 405 return ret0 406 } 407 408 // RecordActivityHeartbeat indicates an expected call of RecordActivityHeartbeat. 409 func (mr *MockClientMockRecorder) RecordActivityHeartbeat(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { 410 mr.mock.ctrl.T.Helper() 411 varargs := append([]interface{}{arg0, arg1}, arg2...) 412 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecordActivityHeartbeat", reflect.TypeOf((*MockClient)(nil).RecordActivityHeartbeat), varargs...) 413 } 414 415 // RecordActivityHeartbeatByID mocks base method. 416 func (m *MockClient) RecordActivityHeartbeatByID(arg0 context.Context, arg1, arg2, arg3, arg4 string, arg5 ...interface{}) error { 417 m.ctrl.T.Helper() 418 varargs := []interface{}{arg0, arg1, arg2, arg3, arg4} 419 for _, a := range arg5 { 420 varargs = append(varargs, a) 421 } 422 ret := m.ctrl.Call(m, "RecordActivityHeartbeatByID", varargs...) 423 ret0, _ := ret[0].(error) 424 return ret0 425 } 426 427 // RecordActivityHeartbeatByID indicates an expected call of RecordActivityHeartbeatByID. 428 func (mr *MockClientMockRecorder) RecordActivityHeartbeatByID(arg0, arg1, arg2, arg3, arg4 interface{}, arg5 ...interface{}) *gomock.Call { 429 mr.mock.ctrl.T.Helper() 430 varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4}, arg5...) 431 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RecordActivityHeartbeatByID", reflect.TypeOf((*MockClient)(nil).RecordActivityHeartbeatByID), varargs...) 432 } 433 434 // ResetWorkflowExecution mocks base method. 435 func (m *MockClient) ResetWorkflowExecution(arg0 context.Context, arg1 *workflowservice.ResetWorkflowExecutionRequest) (*workflowservice.ResetWorkflowExecutionResponse, error) { 436 m.ctrl.T.Helper() 437 ret := m.ctrl.Call(m, "ResetWorkflowExecution", arg0, arg1) 438 ret0, _ := ret[0].(*workflowservice.ResetWorkflowExecutionResponse) 439 ret1, _ := ret[1].(error) 440 return ret0, ret1 441 } 442 443 // ResetWorkflowExecution indicates an expected call of ResetWorkflowExecution. 444 func (mr *MockClientMockRecorder) ResetWorkflowExecution(arg0, arg1 interface{}) *gomock.Call { 445 mr.mock.ctrl.T.Helper() 446 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetWorkflowExecution", reflect.TypeOf((*MockClient)(nil).ResetWorkflowExecution), arg0, arg1) 447 } 448 449 // ScanWorkflow mocks base method. 450 func (m *MockClient) ScanWorkflow(arg0 context.Context, arg1 *workflowservice.ScanWorkflowExecutionsRequest) (*workflowservice.ScanWorkflowExecutionsResponse, error) { 451 m.ctrl.T.Helper() 452 ret := m.ctrl.Call(m, "ScanWorkflow", arg0, arg1) 453 ret0, _ := ret[0].(*workflowservice.ScanWorkflowExecutionsResponse) 454 ret1, _ := ret[1].(error) 455 return ret0, ret1 456 } 457 458 // ScanWorkflow indicates an expected call of ScanWorkflow. 459 func (mr *MockClientMockRecorder) ScanWorkflow(arg0, arg1 interface{}) *gomock.Call { 460 mr.mock.ctrl.T.Helper() 461 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScanWorkflow", reflect.TypeOf((*MockClient)(nil).ScanWorkflow), arg0, arg1) 462 } 463 464 // ScheduleClient mocks base method. 465 func (m *MockClient) ScheduleClient() client.ScheduleClient { 466 m.ctrl.T.Helper() 467 ret := m.ctrl.Call(m, "ScheduleClient") 468 ret0, _ := ret[0].(client.ScheduleClient) 469 return ret0 470 } 471 472 // ScheduleClient indicates an expected call of ScheduleClient. 473 func (mr *MockClientMockRecorder) ScheduleClient() *gomock.Call { 474 mr.mock.ctrl.T.Helper() 475 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ScheduleClient", reflect.TypeOf((*MockClient)(nil).ScheduleClient)) 476 } 477 478 // SignalWithStartWorkflow mocks base method. 479 func (m *MockClient) SignalWithStartWorkflow(arg0 context.Context, arg1, arg2 string, arg3 interface{}, arg4 client.StartWorkflowOptions, arg5 interface{}, arg6 ...interface{}) (client.WorkflowRun, error) { 480 m.ctrl.T.Helper() 481 varargs := []interface{}{arg0, arg1, arg2, arg3, arg4, arg5} 482 for _, a := range arg6 { 483 varargs = append(varargs, a) 484 } 485 ret := m.ctrl.Call(m, "SignalWithStartWorkflow", varargs...) 486 ret0, _ := ret[0].(client.WorkflowRun) 487 ret1, _ := ret[1].(error) 488 return ret0, ret1 489 } 490 491 // SignalWithStartWorkflow indicates an expected call of SignalWithStartWorkflow. 492 func (mr *MockClientMockRecorder) SignalWithStartWorkflow(arg0, arg1, arg2, arg3, arg4, arg5 interface{}, arg6 ...interface{}) *gomock.Call { 493 mr.mock.ctrl.T.Helper() 494 varargs := append([]interface{}{arg0, arg1, arg2, arg3, arg4, arg5}, arg6...) 495 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SignalWithStartWorkflow", reflect.TypeOf((*MockClient)(nil).SignalWithStartWorkflow), varargs...) 496 } 497 498 // SignalWorkflow mocks base method. 499 func (m *MockClient) SignalWorkflow(arg0 context.Context, arg1, arg2, arg3 string, arg4 interface{}) error { 500 m.ctrl.T.Helper() 501 ret := m.ctrl.Call(m, "SignalWorkflow", arg0, arg1, arg2, arg3, arg4) 502 ret0, _ := ret[0].(error) 503 return ret0 504 } 505 506 // SignalWorkflow indicates an expected call of SignalWorkflow. 507 func (mr *MockClientMockRecorder) SignalWorkflow(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { 508 mr.mock.ctrl.T.Helper() 509 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SignalWorkflow", reflect.TypeOf((*MockClient)(nil).SignalWorkflow), arg0, arg1, arg2, arg3, arg4) 510 } 511 512 // TerminateWorkflow mocks base method. 513 func (m *MockClient) TerminateWorkflow(arg0 context.Context, arg1, arg2, arg3 string, arg4 ...interface{}) error { 514 m.ctrl.T.Helper() 515 varargs := []interface{}{arg0, arg1, arg2, arg3} 516 for _, a := range arg4 { 517 varargs = append(varargs, a) 518 } 519 ret := m.ctrl.Call(m, "TerminateWorkflow", varargs...) 520 ret0, _ := ret[0].(error) 521 return ret0 522 } 523 524 // TerminateWorkflow indicates an expected call of TerminateWorkflow. 525 func (mr *MockClientMockRecorder) TerminateWorkflow(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { 526 mr.mock.ctrl.T.Helper() 527 varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) 528 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TerminateWorkflow", reflect.TypeOf((*MockClient)(nil).TerminateWorkflow), varargs...) 529 } 530 531 // UpdateWorkerBuildIdCompatibility mocks base method. 532 func (m *MockClient) UpdateWorkerBuildIdCompatibility(arg0 context.Context, arg1 *client.UpdateWorkerBuildIdCompatibilityOptions) error { 533 m.ctrl.T.Helper() 534 ret := m.ctrl.Call(m, "UpdateWorkerBuildIdCompatibility", arg0, arg1) 535 ret0, _ := ret[0].(error) 536 return ret0 537 } 538 539 // UpdateWorkerBuildIdCompatibility indicates an expected call of UpdateWorkerBuildIdCompatibility. 540 func (mr *MockClientMockRecorder) UpdateWorkerBuildIdCompatibility(arg0, arg1 interface{}) *gomock.Call { 541 mr.mock.ctrl.T.Helper() 542 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkerBuildIdCompatibility", reflect.TypeOf((*MockClient)(nil).UpdateWorkerBuildIdCompatibility), arg0, arg1) 543 } 544 545 // UpdateWorkflow mocks base method. 546 func (m *MockClient) UpdateWorkflow(arg0 context.Context, arg1, arg2, arg3 string, arg4 ...interface{}) (client.WorkflowUpdateHandle, error) { 547 m.ctrl.T.Helper() 548 varargs := []interface{}{arg0, arg1, arg2, arg3} 549 for _, a := range arg4 { 550 varargs = append(varargs, a) 551 } 552 ret := m.ctrl.Call(m, "UpdateWorkflow", varargs...) 553 ret0, _ := ret[0].(client.WorkflowUpdateHandle) 554 ret1, _ := ret[1].(error) 555 return ret0, ret1 556 } 557 558 // UpdateWorkflow indicates an expected call of UpdateWorkflow. 559 func (mr *MockClientMockRecorder) UpdateWorkflow(arg0, arg1, arg2, arg3 interface{}, arg4 ...interface{}) *gomock.Call { 560 mr.mock.ctrl.T.Helper() 561 varargs := append([]interface{}{arg0, arg1, arg2, arg3}, arg4...) 562 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkflow", reflect.TypeOf((*MockClient)(nil).UpdateWorkflow), varargs...) 563 } 564 565 // UpdateWorkflowWithOptions mocks base method. 566 func (m *MockClient) UpdateWorkflowWithOptions(arg0 context.Context, arg1 *client.UpdateWorkflowWithOptionsRequest) (client.WorkflowUpdateHandle, error) { 567 m.ctrl.T.Helper() 568 ret := m.ctrl.Call(m, "UpdateWorkflowWithOptions", arg0, arg1) 569 ret0, _ := ret[0].(client.WorkflowUpdateHandle) 570 ret1, _ := ret[1].(error) 571 return ret0, ret1 572 } 573 574 // UpdateWorkflowWithOptions indicates an expected call of UpdateWorkflowWithOptions. 575 func (mr *MockClientMockRecorder) UpdateWorkflowWithOptions(arg0, arg1 interface{}) *gomock.Call { 576 mr.mock.ctrl.T.Helper() 577 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateWorkflowWithOptions", reflect.TypeOf((*MockClient)(nil).UpdateWorkflowWithOptions), arg0, arg1) 578 } 579 580 // WorkflowService mocks base method. 581 func (m *MockClient) WorkflowService() workflowservice.WorkflowServiceClient { 582 m.ctrl.T.Helper() 583 ret := m.ctrl.Call(m, "WorkflowService") 584 ret0, _ := ret[0].(workflowservice.WorkflowServiceClient) 585 return ret0 586 } 587 588 // WorkflowService indicates an expected call of WorkflowService. 589 func (mr *MockClientMockRecorder) WorkflowService() *gomock.Call { 590 mr.mock.ctrl.T.Helper() 591 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WorkflowService", reflect.TypeOf((*MockClient)(nil).WorkflowService)) 592 }