Files
tasks/service/history/replication/sync_state_retriever_mock.go
Yichao YangandGitHub 63839f1a9d CHASM: Propagate ArchetypeID (#8693)
## What changed?
- Use archetypeID everywhere in history service and pass them to
persistence

## Why?
- Required for separate ID space work.

## How did you test it?
- [x] built
- [ ] run locally and tested manually
- [x] covered by existing tests
- [x] added new unit test(s)
- [ ] added new functional test(s)
2025-11-26 16:04:44 -08:00

130 lines
7.0 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: sync_state_retriever.go
//
// Generated by this command:
//
// mockgen -package replication -source sync_state_retriever.go -destination sync_state_retriever_mock.go
//
// Package replication is a generated GoMock package.
package replication
import (
context "context"
reflect "reflect"
common "go.temporal.io/api/common/v1"
history "go.temporal.io/server/api/history/v1"
persistence "go.temporal.io/server/api/persistence/v1"
chasm "go.temporal.io/server/chasm"
interfaces "go.temporal.io/server/service/history/interfaces"
gomock "go.uber.org/mock/gomock"
)
// MockSyncStateRetriever is a mock of SyncStateRetriever interface.
type MockSyncStateRetriever struct {
ctrl *gomock.Controller
recorder *MockSyncStateRetrieverMockRecorder
isgomock struct{}
}
// MockSyncStateRetrieverMockRecorder is the mock recorder for MockSyncStateRetriever.
type MockSyncStateRetrieverMockRecorder struct {
mock *MockSyncStateRetriever
}
// NewMockSyncStateRetriever creates a new mock instance.
func NewMockSyncStateRetriever(ctrl *gomock.Controller) *MockSyncStateRetriever {
mock := &MockSyncStateRetriever{ctrl: ctrl}
mock.recorder = &MockSyncStateRetrieverMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockSyncStateRetriever) EXPECT() *MockSyncStateRetrieverMockRecorder {
return m.recorder
}
// GetSyncWorkflowStateArtifact mocks base method.
func (m *MockSyncStateRetriever) GetSyncWorkflowStateArtifact(ctx context.Context, namespaceID string, execution *common.WorkflowExecution, archetypeID chasm.ArchetypeID, targetVersionedTransition *persistence.VersionedTransition, targetVersionHistories *history.VersionHistories) (*SyncStateResult, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetSyncWorkflowStateArtifact", ctx, namespaceID, execution, archetypeID, targetVersionedTransition, targetVersionHistories)
ret0, _ := ret[0].(*SyncStateResult)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetSyncWorkflowStateArtifact indicates an expected call of GetSyncWorkflowStateArtifact.
func (mr *MockSyncStateRetrieverMockRecorder) GetSyncWorkflowStateArtifact(ctx, namespaceID, execution, archetypeID, targetVersionedTransition, targetVersionHistories any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncWorkflowStateArtifact", reflect.TypeOf((*MockSyncStateRetriever)(nil).GetSyncWorkflowStateArtifact), ctx, namespaceID, execution, archetypeID, targetVersionedTransition, targetVersionHistories)
}
// GetSyncWorkflowStateArtifactFromMutableState mocks base method.
func (m *MockSyncStateRetriever) GetSyncWorkflowStateArtifactFromMutableState(ctx context.Context, namespaceID string, execution *common.WorkflowExecution, mutableState interfaces.MutableState, targetVersionedTransition *persistence.VersionedTransition, targetVersionHistories [][]*history.VersionHistoryItem, releaseFunc interfaces.ReleaseWorkflowContextFunc) (*SyncStateResult, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetSyncWorkflowStateArtifactFromMutableState", ctx, namespaceID, execution, mutableState, targetVersionedTransition, targetVersionHistories, releaseFunc)
ret0, _ := ret[0].(*SyncStateResult)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetSyncWorkflowStateArtifactFromMutableState indicates an expected call of GetSyncWorkflowStateArtifactFromMutableState.
func (mr *MockSyncStateRetrieverMockRecorder) GetSyncWorkflowStateArtifactFromMutableState(ctx, namespaceID, execution, mutableState, targetVersionedTransition, targetVersionHistories, releaseFunc any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncWorkflowStateArtifactFromMutableState", reflect.TypeOf((*MockSyncStateRetriever)(nil).GetSyncWorkflowStateArtifactFromMutableState), ctx, namespaceID, execution, mutableState, targetVersionedTransition, targetVersionHistories, releaseFunc)
}
// GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow mocks base method.
func (m *MockSyncStateRetriever) GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow(ctx context.Context, namespaceID string, execution *common.WorkflowExecution, mutableState interfaces.MutableState, releaseFunc interfaces.ReleaseWorkflowContextFunc, taskVersionedTransition *persistence.VersionedTransition) (*SyncStateResult, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow", ctx, namespaceID, execution, mutableState, releaseFunc, taskVersionedTransition)
ret0, _ := ret[0].(*SyncStateResult)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow indicates an expected call of GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow.
func (mr *MockSyncStateRetrieverMockRecorder) GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow(ctx, namespaceID, execution, mutableState, releaseFunc, taskVersionedTransition any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow", reflect.TypeOf((*MockSyncStateRetriever)(nil).GetSyncWorkflowStateArtifactFromMutableStateForNewWorkflow), ctx, namespaceID, execution, mutableState, releaseFunc, taskVersionedTransition)
}
// MocklastUpdatedStateTransitionGetter is a mock of lastUpdatedStateTransitionGetter interface.
type MocklastUpdatedStateTransitionGetter struct {
ctrl *gomock.Controller
recorder *MocklastUpdatedStateTransitionGetterMockRecorder
isgomock struct{}
}
// MocklastUpdatedStateTransitionGetterMockRecorder is the mock recorder for MocklastUpdatedStateTransitionGetter.
type MocklastUpdatedStateTransitionGetterMockRecorder struct {
mock *MocklastUpdatedStateTransitionGetter
}
// NewMocklastUpdatedStateTransitionGetter creates a new mock instance.
func NewMocklastUpdatedStateTransitionGetter(ctrl *gomock.Controller) *MocklastUpdatedStateTransitionGetter {
mock := &MocklastUpdatedStateTransitionGetter{ctrl: ctrl}
mock.recorder = &MocklastUpdatedStateTransitionGetterMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MocklastUpdatedStateTransitionGetter) EXPECT() *MocklastUpdatedStateTransitionGetterMockRecorder {
return m.recorder
}
// GetLastUpdateVersionedTransition mocks base method.
func (m *MocklastUpdatedStateTransitionGetter) GetLastUpdateVersionedTransition() *persistence.VersionedTransition {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetLastUpdateVersionedTransition")
ret0, _ := ret[0].(*persistence.VersionedTransition)
return ret0
}
// GetLastUpdateVersionedTransition indicates an expected call of GetLastUpdateVersionedTransition.
func (mr *MocklastUpdatedStateTransitionGetterMockRecorder) GetLastUpdateVersionedTransition() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLastUpdateVersionedTransition", reflect.TypeOf((*MocklastUpdatedStateTransitionGetter)(nil).GetLastUpdateVersionedTransition))
}