History client mock

Reviewers: maxim

Reviewed By: maxim

Subscribers: jenkins

Differential Revision: https://code.uberinternal.com/D712781

client/history/mocks/Client.go

@@ -0,0 +1,170 @@
+package mocks
+
+import gohistory "code.uber.internal/devexp/minions/.gen/go/history"
+import history "code.uber.internal/devexp/minions/client/history"
+import mock "github.com/stretchr/testify/mock"
+import shared "code.uber.internal/devexp/minions/.gen/go/shared"
+
+// Client is an autogenerated mock type for the Client type
+type Client struct {
+	mock.Mock
+}
+
+// GetWorkflowExecutionHistory provides a mock function with given fields: request
+func (_m *Client) GetWorkflowExecutionHistory(request *shared.GetWorkflowExecutionHistoryRequest) (*shared.GetWorkflowExecutionHistoryResponse, error) {
+	ret := _m.Called(request)
+
+	var r0 *shared.GetWorkflowExecutionHistoryResponse
+	if rf, ok := ret.Get(0).(func(*shared.GetWorkflowExecutionHistoryRequest) *shared.GetWorkflowExecutionHistoryResponse); ok {
+		r0 = rf(request)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*shared.GetWorkflowExecutionHistoryResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(*shared.GetWorkflowExecutionHistoryRequest) error); ok {
+		r1 = rf(request)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// RecordActivityTaskHeartbeat provides a mock function with given fields: request
+func (_m *Client) RecordActivityTaskHeartbeat(request *shared.RecordActivityTaskHeartbeatRequest) (*shared.RecordActivityTaskHeartbeatResponse, error) {
+	ret := _m.Called(request)
+
+	var r0 *shared.RecordActivityTaskHeartbeatResponse
+	if rf, ok := ret.Get(0).(func(*shared.RecordActivityTaskHeartbeatRequest) *shared.RecordActivityTaskHeartbeatResponse); ok {
+		r0 = rf(request)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*shared.RecordActivityTaskHeartbeatResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(*shared.RecordActivityTaskHeartbeatRequest) error); ok {
+		r1 = rf(request)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// RecordActivityTaskStarted provides a mock function with given fields: request
+func (_m *Client) RecordActivityTaskStarted(request *gohistory.RecordActivityTaskStartedRequest) (*gohistory.RecordActivityTaskStartedResponse, error) {
+	ret := _m.Called(request)
+
+	var r0 *gohistory.RecordActivityTaskStartedResponse
+	if rf, ok := ret.Get(0).(func(*gohistory.RecordActivityTaskStartedRequest) *gohistory.RecordActivityTaskStartedResponse); ok {
+		r0 = rf(request)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*gohistory.RecordActivityTaskStartedResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(*gohistory.RecordActivityTaskStartedRequest) error); ok {
+		r1 = rf(request)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// RecordDecisionTaskStarted provides a mock function with given fields: request
+func (_m *Client) RecordDecisionTaskStarted(request *gohistory.RecordDecisionTaskStartedRequest) (*gohistory.RecordDecisionTaskStartedResponse, error) {
+	ret := _m.Called(request)
+
+	var r0 *gohistory.RecordDecisionTaskStartedResponse
+	if rf, ok := ret.Get(0).(func(*gohistory.RecordDecisionTaskStartedRequest) *gohistory.RecordDecisionTaskStartedResponse); ok {
+		r0 = rf(request)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*gohistory.RecordDecisionTaskStartedResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(*gohistory.RecordDecisionTaskStartedRequest) error); ok {
+		r1 = rf(request)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+// RespondActivityTaskCompleted provides a mock function with given fields: request
+func (_m *Client) RespondActivityTaskCompleted(request *shared.RespondActivityTaskCompletedRequest) error {
+	ret := _m.Called(request)
+
+	var r0 error
+	if rf, ok := ret.Get(0).(func(*shared.RespondActivityTaskCompletedRequest) error); ok {
+		r0 = rf(request)
+	} else {
+		r0 = ret.Error(0)
+	}
+
+	return r0
+}
+
+// RespondActivityTaskFailed provides a mock function with given fields: request
+func (_m *Client) RespondActivityTaskFailed(request *shared.RespondActivityTaskFailedRequest) error {
+	ret := _m.Called(request)
+
+	var r0 error
+	if rf, ok := ret.Get(0).(func(*shared.RespondActivityTaskFailedRequest) error); ok {
+		r0 = rf(request)
+	} else {
+		r0 = ret.Error(0)
+	}
+
+	return r0
+}
+
+// RespondDecisionTaskCompleted provides a mock function with given fields: request
+func (_m *Client) RespondDecisionTaskCompleted(request *shared.RespondDecisionTaskCompletedRequest) error {
+	ret := _m.Called(request)
+
+	var r0 error
+	if rf, ok := ret.Get(0).(func(*shared.RespondDecisionTaskCompletedRequest) error); ok {
+		r0 = rf(request)
+	} else {
+		r0 = ret.Error(0)
+	}
+
+	return r0
+}
+
+// StartWorkflowExecution provides a mock function with given fields: request
+func (_m *Client) StartWorkflowExecution(request *shared.StartWorkflowExecutionRequest) (*shared.StartWorkflowExecutionResponse, error) {
+	ret := _m.Called(request)
+
+	var r0 *shared.StartWorkflowExecutionResponse
+	if rf, ok := ret.Get(0).(func(*shared.StartWorkflowExecutionRequest) *shared.StartWorkflowExecutionResponse); ok {
+		r0 = rf(request)
+	} else {
+		if ret.Get(0) != nil {
+			r0 = ret.Get(0).(*shared.StartWorkflowExecutionResponse)
+		}
+	}
+
+	var r1 error
+	if rf, ok := ret.Get(1).(func(*shared.StartWorkflowExecutionRequest) error); ok {
+		r1 = rf(request)
+	} else {
+		r1 = ret.Error(1)
+	}
+
+	return r0, r1
+}
+
+var _ history.Client = (*Client)(nil)