Files
tasks/common/archiver/provider/provider_mock.go
Yichao YangandGitHub 429f90fbb6 Add server option for customizing archiver implementation (#9090)
## What changed?
- Add server option for injecting customHistory/VisibilityArchiverFactory

## Why?
- Add community user to use different archiver implementations

## How did you test it?
- [x] built
- [ ] run locally and tested manually
- [ ] covered by existing tests
- [x] added new unit test(s)
- [x] added new functional test(s)
2026-03-24 00:48:30 +00:00

150 lines
5.8 KiB
Go

// Code generated by MockGen. DO NOT EDIT.
// Source: provider.go
//
// Generated by this command:
//
// mockgen -package provider -source provider.go -destination provider_mock.go
//
// Package provider is a generated GoMock package.
package provider
import (
reflect "reflect"
archiver "go.temporal.io/server/common/archiver"
gomock "go.uber.org/mock/gomock"
)
// MockArchiverProvider is a mock of ArchiverProvider interface.
type MockArchiverProvider struct {
ctrl *gomock.Controller
recorder *MockArchiverProviderMockRecorder
isgomock struct{}
}
// MockArchiverProviderMockRecorder is the mock recorder for MockArchiverProvider.
type MockArchiverProviderMockRecorder struct {
mock *MockArchiverProvider
}
// NewMockArchiverProvider creates a new mock instance.
func NewMockArchiverProvider(ctrl *gomock.Controller) *MockArchiverProvider {
mock := &MockArchiverProvider{ctrl: ctrl}
mock.recorder = &MockArchiverProviderMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockArchiverProvider) EXPECT() *MockArchiverProviderMockRecorder {
return m.recorder
}
// GetHistoryArchiver mocks base method.
func (m *MockArchiverProvider) GetHistoryArchiver(scheme string) (archiver.HistoryArchiver, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetHistoryArchiver", scheme)
ret0, _ := ret[0].(archiver.HistoryArchiver)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetHistoryArchiver indicates an expected call of GetHistoryArchiver.
func (mr *MockArchiverProviderMockRecorder) GetHistoryArchiver(scheme any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHistoryArchiver", reflect.TypeOf((*MockArchiverProvider)(nil).GetHistoryArchiver), scheme)
}
// GetVisibilityArchiver mocks base method.
func (m *MockArchiverProvider) GetVisibilityArchiver(scheme string) (archiver.VisibilityArchiver, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "GetVisibilityArchiver", scheme)
ret0, _ := ret[0].(archiver.VisibilityArchiver)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// GetVisibilityArchiver indicates an expected call of GetVisibilityArchiver.
func (mr *MockArchiverProviderMockRecorder) GetVisibilityArchiver(scheme any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetVisibilityArchiver", reflect.TypeOf((*MockArchiverProvider)(nil).GetVisibilityArchiver), scheme)
}
// MockCustomHistoryArchiverFactory is a mock of CustomHistoryArchiverFactory interface.
type MockCustomHistoryArchiverFactory struct {
ctrl *gomock.Controller
recorder *MockCustomHistoryArchiverFactoryMockRecorder
isgomock struct{}
}
// MockCustomHistoryArchiverFactoryMockRecorder is the mock recorder for MockCustomHistoryArchiverFactory.
type MockCustomHistoryArchiverFactoryMockRecorder struct {
mock *MockCustomHistoryArchiverFactory
}
// NewMockCustomHistoryArchiverFactory creates a new mock instance.
func NewMockCustomHistoryArchiverFactory(ctrl *gomock.Controller) *MockCustomHistoryArchiverFactory {
mock := &MockCustomHistoryArchiverFactory{ctrl: ctrl}
mock.recorder = &MockCustomHistoryArchiverFactoryMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockCustomHistoryArchiverFactory) EXPECT() *MockCustomHistoryArchiverFactoryMockRecorder {
return m.recorder
}
// NewCustomHistoryArchiver mocks base method.
func (m *MockCustomHistoryArchiverFactory) NewCustomHistoryArchiver(arg0 NewCustomHistoryArchiverParams) (archiver.HistoryArchiver, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "NewCustomHistoryArchiver", arg0)
ret0, _ := ret[0].(archiver.HistoryArchiver)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// NewCustomHistoryArchiver indicates an expected call of NewCustomHistoryArchiver.
func (mr *MockCustomHistoryArchiverFactoryMockRecorder) NewCustomHistoryArchiver(arg0 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewCustomHistoryArchiver", reflect.TypeOf((*MockCustomHistoryArchiverFactory)(nil).NewCustomHistoryArchiver), arg0)
}
// MockCustomVisibilityArchiverFactory is a mock of CustomVisibilityArchiverFactory interface.
type MockCustomVisibilityArchiverFactory struct {
ctrl *gomock.Controller
recorder *MockCustomVisibilityArchiverFactoryMockRecorder
isgomock struct{}
}
// MockCustomVisibilityArchiverFactoryMockRecorder is the mock recorder for MockCustomVisibilityArchiverFactory.
type MockCustomVisibilityArchiverFactoryMockRecorder struct {
mock *MockCustomVisibilityArchiverFactory
}
// NewMockCustomVisibilityArchiverFactory creates a new mock instance.
func NewMockCustomVisibilityArchiverFactory(ctrl *gomock.Controller) *MockCustomVisibilityArchiverFactory {
mock := &MockCustomVisibilityArchiverFactory{ctrl: ctrl}
mock.recorder = &MockCustomVisibilityArchiverFactoryMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockCustomVisibilityArchiverFactory) EXPECT() *MockCustomVisibilityArchiverFactoryMockRecorder {
return m.recorder
}
// NewCustomVisibilityArchiver mocks base method.
func (m *MockCustomVisibilityArchiverFactory) NewCustomVisibilityArchiver(arg0 NewCustomVisibilityArchiverParams) (archiver.VisibilityArchiver, error) {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "NewCustomVisibilityArchiver", arg0)
ret0, _ := ret[0].(archiver.VisibilityArchiver)
ret1, _ := ret[1].(error)
return ret0, ret1
}
// NewCustomVisibilityArchiver indicates an expected call of NewCustomVisibilityArchiver.
func (mr *MockCustomVisibilityArchiverFactoryMockRecorder) NewCustomVisibilityArchiver(arg0 any) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewCustomVisibilityArchiver", reflect.TypeOf((*MockCustomVisibilityArchiverFactory)(nil).NewCustomVisibilityArchiver), arg0)
}