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  }