
Replace old used-to-be-GORM datastructures (#104305) with sqlc-generated structs. This also makes it possible to use more specific structs that are more taylored to the specific queries, increasing efficiency. This commit deals with the remaining areas, like the job deleter, task timeout checker, and task state machine. And anything else to get things running again. Functional changes are kept to a minimum, as the API still serves the same data. Because this work covers so much of Flamenco's code, it's been split up into different commits. Each commit brings Flamenco to a state where it compiles and unit tests pass. Only the result of the final commit has actually been tested properly. Ref: #104343
262 lines
9.4 KiB
Go
Generated
262 lines
9.4 KiB
Go
Generated
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: projects.blender.org/studio/flamenco/internal/manager/job_deleter (interfaces: PersistenceService,Storage,ChangeBroadcaster,Shaman)
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
time "time"
|
|
|
|
gomock "github.com/golang/mock/gomock"
|
|
sqlc "projects.blender.org/studio/flamenco/internal/manager/persistence/sqlc"
|
|
api "projects.blender.org/studio/flamenco/pkg/api"
|
|
)
|
|
|
|
// MockPersistenceService is a mock of PersistenceService interface.
|
|
type MockPersistenceService struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockPersistenceServiceMockRecorder
|
|
}
|
|
|
|
// MockPersistenceServiceMockRecorder is the mock recorder for MockPersistenceService.
|
|
type MockPersistenceServiceMockRecorder struct {
|
|
mock *MockPersistenceService
|
|
}
|
|
|
|
// NewMockPersistenceService creates a new mock instance.
|
|
func NewMockPersistenceService(ctrl *gomock.Controller) *MockPersistenceService {
|
|
mock := &MockPersistenceService{ctrl: ctrl}
|
|
mock.recorder = &MockPersistenceServiceMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockPersistenceService) EXPECT() *MockPersistenceServiceMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// DeleteJob mocks base method.
|
|
func (m *MockPersistenceService) DeleteJob(arg0 context.Context, arg1 string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteJob", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// DeleteJob indicates an expected call of DeleteJob.
|
|
func (mr *MockPersistenceServiceMockRecorder) DeleteJob(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteJob", reflect.TypeOf((*MockPersistenceService)(nil).DeleteJob), arg0, arg1)
|
|
}
|
|
|
|
// FetchJob mocks base method.
|
|
func (m *MockPersistenceService) FetchJob(arg0 context.Context, arg1 string) (*sqlc.Job, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FetchJob", arg0, arg1)
|
|
ret0, _ := ret[0].(*sqlc.Job)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// FetchJob indicates an expected call of FetchJob.
|
|
func (mr *MockPersistenceServiceMockRecorder) FetchJob(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchJob", reflect.TypeOf((*MockPersistenceService)(nil).FetchJob), arg0, arg1)
|
|
}
|
|
|
|
// FetchJobShamanCheckoutID mocks base method.
|
|
func (m *MockPersistenceService) FetchJobShamanCheckoutID(arg0 context.Context, arg1 string) (string, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FetchJobShamanCheckoutID", arg0, arg1)
|
|
ret0, _ := ret[0].(string)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// FetchJobShamanCheckoutID indicates an expected call of FetchJobShamanCheckoutID.
|
|
func (mr *MockPersistenceServiceMockRecorder) FetchJobShamanCheckoutID(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchJobShamanCheckoutID", reflect.TypeOf((*MockPersistenceService)(nil).FetchJobShamanCheckoutID), arg0, arg1)
|
|
}
|
|
|
|
// FetchJobsDeletionRequested mocks base method.
|
|
func (m *MockPersistenceService) FetchJobsDeletionRequested(arg0 context.Context) ([]string, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "FetchJobsDeletionRequested", arg0)
|
|
ret0, _ := ret[0].([]string)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// FetchJobsDeletionRequested indicates an expected call of FetchJobsDeletionRequested.
|
|
func (mr *MockPersistenceServiceMockRecorder) FetchJobsDeletionRequested(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchJobsDeletionRequested", reflect.TypeOf((*MockPersistenceService)(nil).FetchJobsDeletionRequested), arg0)
|
|
}
|
|
|
|
// RequestIntegrityCheck mocks base method.
|
|
func (m *MockPersistenceService) RequestIntegrityCheck() {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "RequestIntegrityCheck")
|
|
}
|
|
|
|
// RequestIntegrityCheck indicates an expected call of RequestIntegrityCheck.
|
|
func (mr *MockPersistenceServiceMockRecorder) RequestIntegrityCheck() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestIntegrityCheck", reflect.TypeOf((*MockPersistenceService)(nil).RequestIntegrityCheck))
|
|
}
|
|
|
|
// RequestJobDeletion mocks base method.
|
|
func (m *MockPersistenceService) RequestJobDeletion(arg0 context.Context, arg1 *sqlc.Job) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RequestJobDeletion", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// RequestJobDeletion indicates an expected call of RequestJobDeletion.
|
|
func (mr *MockPersistenceServiceMockRecorder) RequestJobDeletion(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestJobDeletion", reflect.TypeOf((*MockPersistenceService)(nil).RequestJobDeletion), arg0, arg1)
|
|
}
|
|
|
|
// RequestJobMassDeletion mocks base method.
|
|
func (m *MockPersistenceService) RequestJobMassDeletion(arg0 context.Context, arg1 time.Time) ([]string, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RequestJobMassDeletion", arg0, arg1)
|
|
ret0, _ := ret[0].([]string)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// RequestJobMassDeletion indicates an expected call of RequestJobMassDeletion.
|
|
func (mr *MockPersistenceServiceMockRecorder) RequestJobMassDeletion(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RequestJobMassDeletion", reflect.TypeOf((*MockPersistenceService)(nil).RequestJobMassDeletion), arg0, arg1)
|
|
}
|
|
|
|
// MockStorage is a mock of Storage interface.
|
|
type MockStorage struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockStorageMockRecorder
|
|
}
|
|
|
|
// MockStorageMockRecorder is the mock recorder for MockStorage.
|
|
type MockStorageMockRecorder struct {
|
|
mock *MockStorage
|
|
}
|
|
|
|
// NewMockStorage creates a new mock instance.
|
|
func NewMockStorage(ctrl *gomock.Controller) *MockStorage {
|
|
mock := &MockStorage{ctrl: ctrl}
|
|
mock.recorder = &MockStorageMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockStorage) EXPECT() *MockStorageMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// RemoveJobStorage mocks base method.
|
|
func (m *MockStorage) RemoveJobStorage(arg0 context.Context, arg1 string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "RemoveJobStorage", arg0, arg1)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// RemoveJobStorage indicates an expected call of RemoveJobStorage.
|
|
func (mr *MockStorageMockRecorder) RemoveJobStorage(arg0, arg1 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveJobStorage", reflect.TypeOf((*MockStorage)(nil).RemoveJobStorage), arg0, arg1)
|
|
}
|
|
|
|
// MockChangeBroadcaster is a mock of ChangeBroadcaster interface.
|
|
type MockChangeBroadcaster struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockChangeBroadcasterMockRecorder
|
|
}
|
|
|
|
// MockChangeBroadcasterMockRecorder is the mock recorder for MockChangeBroadcaster.
|
|
type MockChangeBroadcasterMockRecorder struct {
|
|
mock *MockChangeBroadcaster
|
|
}
|
|
|
|
// NewMockChangeBroadcaster creates a new mock instance.
|
|
func NewMockChangeBroadcaster(ctrl *gomock.Controller) *MockChangeBroadcaster {
|
|
mock := &MockChangeBroadcaster{ctrl: ctrl}
|
|
mock.recorder = &MockChangeBroadcasterMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockChangeBroadcaster) EXPECT() *MockChangeBroadcasterMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// BroadcastJobUpdate mocks base method.
|
|
func (m *MockChangeBroadcaster) BroadcastJobUpdate(arg0 api.EventJobUpdate) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "BroadcastJobUpdate", arg0)
|
|
}
|
|
|
|
// BroadcastJobUpdate indicates an expected call of BroadcastJobUpdate.
|
|
func (mr *MockChangeBroadcasterMockRecorder) BroadcastJobUpdate(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BroadcastJobUpdate", reflect.TypeOf((*MockChangeBroadcaster)(nil).BroadcastJobUpdate), arg0)
|
|
}
|
|
|
|
// MockShaman is a mock of Shaman interface.
|
|
type MockShaman struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockShamanMockRecorder
|
|
}
|
|
|
|
// MockShamanMockRecorder is the mock recorder for MockShaman.
|
|
type MockShamanMockRecorder struct {
|
|
mock *MockShaman
|
|
}
|
|
|
|
// NewMockShaman creates a new mock instance.
|
|
func NewMockShaman(ctrl *gomock.Controller) *MockShaman {
|
|
mock := &MockShaman{ctrl: ctrl}
|
|
mock.recorder = &MockShamanMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockShaman) EXPECT() *MockShamanMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// EraseCheckout mocks base method.
|
|
func (m *MockShaman) EraseCheckout(arg0 string) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "EraseCheckout", arg0)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// EraseCheckout indicates an expected call of EraseCheckout.
|
|
func (mr *MockShamanMockRecorder) EraseCheckout(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EraseCheckout", reflect.TypeOf((*MockShaman)(nil).EraseCheckout), arg0)
|
|
}
|
|
|
|
// IsEnabled mocks base method.
|
|
func (m *MockShaman) IsEnabled() bool {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "IsEnabled")
|
|
ret0, _ := ret[0].(bool)
|
|
return ret0
|
|
}
|
|
|
|
// IsEnabled indicates an expected call of IsEnabled.
|
|
func (mr *MockShamanMockRecorder) IsEnabled() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsEnabled", reflect.TypeOf((*MockShaman)(nil).IsEnabled))
|
|
}
|