mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-04 02:04:03 +00:00
## What changed? Add businessID at namespace.ClusterNames() api ## Why? To allow different replication configuration for different workflows. ## How did you test it? - [x] built - [x] run locally and tested manually - [x] covered by existing tests - [ ] added new unit test(s) - [ ] added new functional test(s) ## Potential risks No risk.
349 lines
13 KiB
Go
349 lines
13 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: executable_task.go
|
|
//
|
|
// Generated by this command:
|
|
//
|
|
// mockgen -package replication -source executable_task.go -destination executable_task_mock.go
|
|
//
|
|
|
|
// Package replication is a generated GoMock package.
|
|
package replication
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
enums "go.temporal.io/server/api/enums/v1"
|
|
repication "go.temporal.io/server/api/replication/v1"
|
|
backoff "go.temporal.io/server/common/backoff"
|
|
definition "go.temporal.io/server/common/definition"
|
|
serviceerror "go.temporal.io/server/common/serviceerror"
|
|
tasks "go.temporal.io/server/common/tasks"
|
|
gomock "go.uber.org/mock/gomock"
|
|
)
|
|
|
|
// MockExecutableTask is a mock of ExecutableTask interface.
|
|
type MockExecutableTask struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockExecutableTaskMockRecorder
|
|
isgomock struct{}
|
|
}
|
|
|
|
// MockExecutableTaskMockRecorder is the mock recorder for MockExecutableTask.
|
|
type MockExecutableTaskMockRecorder struct {
|
|
mock *MockExecutableTask
|
|
}
|
|
|
|
// NewMockExecutableTask creates a new mock instance.
|
|
func NewMockExecutableTask(ctrl *gomock.Controller) *MockExecutableTask {
|
|
mock := &MockExecutableTask{ctrl: ctrl}
|
|
mock.recorder = &MockExecutableTaskMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockExecutableTask) EXPECT() *MockExecutableTaskMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Abort mocks base method.
|
|
func (m *MockExecutableTask) Abort() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Abort")
|
|
}
|
|
|
|
// Abort indicates an expected call of Abort.
|
|
func (mr *MockExecutableTaskMockRecorder) Abort() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Abort", reflect.TypeOf((*MockExecutableTask)(nil).Abort))
|
|
}
|
|
|
|
// Ack mocks base method.
|
|
func (m *MockExecutableTask) Ack() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Ack")
|
|
}
|
|
|
|
// Ack indicates an expected call of Ack.
|
|
func (mr *MockExecutableTaskMockRecorder) Ack() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ack", reflect.TypeOf((*MockExecutableTask)(nil).Ack))
|
|
}
|
|
|
|
// Attempt mocks base method.
|
|
func (m *MockExecutableTask) Attempt() int {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Attempt")
|
|
ret0, _ := ret[0].(int)
|
|
return ret0
|
|
}
|
|
|
|
// Attempt indicates an expected call of Attempt.
|
|
func (mr *MockExecutableTaskMockRecorder) Attempt() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Attempt", reflect.TypeOf((*MockExecutableTask)(nil).Attempt))
|
|
}
|
|
|
|
// BackFillEvents mocks base method.
|
|
func (m *MockExecutableTask) BackFillEvents(ctx context.Context, remoteCluster string, workflowKey definition.WorkflowKey, startEventId, startEventVersion, endEventId, endEventVersion int64, newRunId string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "BackFillEvents", ctx, remoteCluster, workflowKey, startEventId, startEventVersion, endEventId, endEventVersion, newRunId)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// BackFillEvents indicates an expected call of BackFillEvents.
|
|
func (mr *MockExecutableTaskMockRecorder) BackFillEvents(ctx, remoteCluster, workflowKey, startEventId, startEventVersion, endEventId, endEventVersion, newRunId any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BackFillEvents", reflect.TypeOf((*MockExecutableTask)(nil).BackFillEvents), ctx, remoteCluster, workflowKey, startEventId, startEventVersion, endEventId, endEventVersion, newRunId)
|
|
}
|
|
|
|
// Cancel mocks base method.
|
|
func (m *MockExecutableTask) Cancel() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Cancel")
|
|
}
|
|
|
|
// Cancel indicates an expected call of Cancel.
|
|
func (mr *MockExecutableTaskMockRecorder) Cancel() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Cancel", reflect.TypeOf((*MockExecutableTask)(nil).Cancel))
|
|
}
|
|
|
|
// DeleteWorkflow mocks base method.
|
|
func (m *MockExecutableTask) DeleteWorkflow(ctx context.Context, workflowKey definition.WorkflowKey) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteWorkflow", ctx, workflowKey)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// DeleteWorkflow indicates an expected call of DeleteWorkflow.
|
|
func (mr *MockExecutableTaskMockRecorder) DeleteWorkflow(ctx, workflowKey any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteWorkflow", reflect.TypeOf((*MockExecutableTask)(nil).DeleteWorkflow), ctx, workflowKey)
|
|
}
|
|
|
|
// GetNamespaceInfo mocks base method.
|
|
func (m *MockExecutableTask) GetNamespaceInfo(ctx context.Context, namespaceID, businessID string) (string, bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetNamespaceInfo", ctx, namespaceID, businessID)
|
|
ret0, _ := ret[0].(string)
|
|
ret1, _ := ret[1].(bool)
|
|
ret2, _ := ret[2].(error)
|
|
return ret0, ret1, ret2
|
|
}
|
|
|
|
// GetNamespaceInfo indicates an expected call of GetNamespaceInfo.
|
|
func (mr *MockExecutableTaskMockRecorder) GetNamespaceInfo(ctx, namespaceID, businessID any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNamespaceInfo", reflect.TypeOf((*MockExecutableTask)(nil).GetNamespaceInfo), ctx, namespaceID, businessID)
|
|
}
|
|
|
|
// GetPriority mocks base method.
|
|
func (m *MockExecutableTask) GetPriority() enums.TaskPriority {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "GetPriority")
|
|
ret0, _ := ret[0].(enums.TaskPriority)
|
|
return ret0
|
|
}
|
|
|
|
// GetPriority indicates an expected call of GetPriority.
|
|
func (mr *MockExecutableTaskMockRecorder) GetPriority() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPriority", reflect.TypeOf((*MockExecutableTask)(nil).GetPriority))
|
|
}
|
|
|
|
// IsRetryableError mocks base method.
|
|
func (m *MockExecutableTask) IsRetryableError(err error) bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsRetryableError", err)
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// IsRetryableError indicates an expected call of IsRetryableError.
|
|
func (mr *MockExecutableTaskMockRecorder) IsRetryableError(err any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsRetryableError", reflect.TypeOf((*MockExecutableTask)(nil).IsRetryableError), err)
|
|
}
|
|
|
|
// MarkPoisonPill mocks base method.
|
|
func (m *MockExecutableTask) MarkPoisonPill() error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "MarkPoisonPill")
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// MarkPoisonPill indicates an expected call of MarkPoisonPill.
|
|
func (mr *MockExecutableTaskMockRecorder) MarkPoisonPill() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MarkPoisonPill", reflect.TypeOf((*MockExecutableTask)(nil).MarkPoisonPill))
|
|
}
|
|
|
|
// MarkTaskDuplicated mocks base method.
|
|
func (m *MockExecutableTask) MarkTaskDuplicated() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "MarkTaskDuplicated")
|
|
}
|
|
|
|
// MarkTaskDuplicated indicates an expected call of MarkTaskDuplicated.
|
|
func (mr *MockExecutableTaskMockRecorder) MarkTaskDuplicated() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MarkTaskDuplicated", reflect.TypeOf((*MockExecutableTask)(nil).MarkTaskDuplicated))
|
|
}
|
|
|
|
// Nack mocks base method.
|
|
func (m *MockExecutableTask) Nack(err error) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Nack", err)
|
|
}
|
|
|
|
// Nack indicates an expected call of Nack.
|
|
func (mr *MockExecutableTaskMockRecorder) Nack(err any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Nack", reflect.TypeOf((*MockExecutableTask)(nil).Nack), err)
|
|
}
|
|
|
|
// ReplicationTask mocks base method.
|
|
func (m *MockExecutableTask) ReplicationTask() *repication.ReplicationTask {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "ReplicationTask")
|
|
ret0, _ := ret[0].(*repication.ReplicationTask)
|
|
return ret0
|
|
}
|
|
|
|
// ReplicationTask indicates an expected call of ReplicationTask.
|
|
func (mr *MockExecutableTaskMockRecorder) ReplicationTask() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReplicationTask", reflect.TypeOf((*MockExecutableTask)(nil).ReplicationTask))
|
|
}
|
|
|
|
// Reschedule mocks base method.
|
|
func (m *MockExecutableTask) Reschedule() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "Reschedule")
|
|
}
|
|
|
|
// Reschedule indicates an expected call of Reschedule.
|
|
func (mr *MockExecutableTaskMockRecorder) Reschedule() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reschedule", reflect.TypeOf((*MockExecutableTask)(nil).Reschedule))
|
|
}
|
|
|
|
// Resend mocks base method.
|
|
func (m *MockExecutableTask) Resend(ctx context.Context, remoteCluster string, retryErr *serviceerror.RetryReplication, remainingAttempt int) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Resend", ctx, remoteCluster, retryErr, remainingAttempt)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Resend indicates an expected call of Resend.
|
|
func (mr *MockExecutableTaskMockRecorder) Resend(ctx, remoteCluster, retryErr, remainingAttempt any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resend", reflect.TypeOf((*MockExecutableTask)(nil).Resend), ctx, remoteCluster, retryErr, remainingAttempt)
|
|
}
|
|
|
|
// RetryPolicy mocks base method.
|
|
func (m *MockExecutableTask) RetryPolicy() backoff.RetryPolicy {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RetryPolicy")
|
|
ret0, _ := ret[0].(backoff.RetryPolicy)
|
|
return ret0
|
|
}
|
|
|
|
// RetryPolicy indicates an expected call of RetryPolicy.
|
|
func (mr *MockExecutableTaskMockRecorder) RetryPolicy() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RetryPolicy", reflect.TypeOf((*MockExecutableTask)(nil).RetryPolicy))
|
|
}
|
|
|
|
// SourceClusterName mocks base method.
|
|
func (m *MockExecutableTask) SourceClusterName() string {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SourceClusterName")
|
|
ret0, _ := ret[0].(string)
|
|
return ret0
|
|
}
|
|
|
|
// SourceClusterName indicates an expected call of SourceClusterName.
|
|
func (mr *MockExecutableTaskMockRecorder) SourceClusterName() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SourceClusterName", reflect.TypeOf((*MockExecutableTask)(nil).SourceClusterName))
|
|
}
|
|
|
|
// State mocks base method.
|
|
func (m *MockExecutableTask) State() tasks.State {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "State")
|
|
ret0, _ := ret[0].(tasks.State)
|
|
return ret0
|
|
}
|
|
|
|
// State indicates an expected call of State.
|
|
func (mr *MockExecutableTaskMockRecorder) State() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "State", reflect.TypeOf((*MockExecutableTask)(nil).State))
|
|
}
|
|
|
|
// SyncState mocks base method.
|
|
func (m *MockExecutableTask) SyncState(ctx context.Context, syncStateErr *serviceerror.SyncState, remainingAttempt int) (bool, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SyncState", ctx, syncStateErr, remainingAttempt)
|
|
ret0, _ := ret[0].(bool)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SyncState indicates an expected call of SyncState.
|
|
func (mr *MockExecutableTaskMockRecorder) SyncState(ctx, syncStateErr, remainingAttempt any) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncState", reflect.TypeOf((*MockExecutableTask)(nil).SyncState), ctx, syncStateErr, remainingAttempt)
|
|
}
|
|
|
|
// TaskCreationTime mocks base method.
|
|
func (m *MockExecutableTask) TaskCreationTime() time.Time {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "TaskCreationTime")
|
|
ret0, _ := ret[0].(time.Time)
|
|
return ret0
|
|
}
|
|
|
|
// TaskCreationTime indicates an expected call of TaskCreationTime.
|
|
func (mr *MockExecutableTaskMockRecorder) TaskCreationTime() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TaskCreationTime", reflect.TypeOf((*MockExecutableTask)(nil).TaskCreationTime))
|
|
}
|
|
|
|
// TaskID mocks base method.
|
|
func (m *MockExecutableTask) TaskID() int64 {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "TaskID")
|
|
ret0, _ := ret[0].(int64)
|
|
return ret0
|
|
}
|
|
|
|
// TaskID indicates an expected call of TaskID.
|
|
func (mr *MockExecutableTaskMockRecorder) TaskID() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TaskID", reflect.TypeOf((*MockExecutableTask)(nil).TaskID))
|
|
}
|
|
|
|
// TerminalState mocks base method.
|
|
func (m *MockExecutableTask) TerminalState() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "TerminalState")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// TerminalState indicates an expected call of TerminalState.
|
|
func (mr *MockExecutableTaskMockRecorder) TerminalState() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TerminalState", reflect.TypeOf((*MockExecutableTask)(nil).TerminalState))
|
|
}
|