// Code generated by MockGen. DO NOT EDIT. // Source: git.blender.org/flamenco/internal/manager/task_state_machine (interfaces: PersistenceService,ChangeBroadcaster,LogStorage) // Package mocks is a generated GoMock package. package mocks import ( context "context" reflect "reflect" persistence "git.blender.org/flamenco/internal/manager/persistence" api "git.blender.org/flamenco/pkg/api" gomock "github.com/golang/mock/gomock" zerolog "github.com/rs/zerolog" ) // 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 } // CountTasksOfJobInStatus mocks base method. func (m *MockPersistenceService) CountTasksOfJobInStatus(arg0 context.Context, arg1 *persistence.Job, arg2 ...api.TaskStatus) (int, int, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0, arg1} for _, a := range arg2 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "CountTasksOfJobInStatus", varargs...) ret0, _ := ret[0].(int) ret1, _ := ret[1].(int) ret2, _ := ret[2].(error) return ret0, ret1, ret2 } // CountTasksOfJobInStatus indicates an expected call of CountTasksOfJobInStatus. func (mr *MockPersistenceServiceMockRecorder) CountTasksOfJobInStatus(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0, arg1}, arg2...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CountTasksOfJobInStatus", reflect.TypeOf((*MockPersistenceService)(nil).CountTasksOfJobInStatus), varargs...) } // FetchJobsInStatus mocks base method. func (m *MockPersistenceService) FetchJobsInStatus(arg0 context.Context, arg1 ...api.JobStatus) ([]*persistence.Job, error) { m.ctrl.T.Helper() varargs := []interface{}{arg0} for _, a := range arg1 { varargs = append(varargs, a) } ret := m.ctrl.Call(m, "FetchJobsInStatus", varargs...) ret0, _ := ret[0].([]*persistence.Job) ret1, _ := ret[1].(error) return ret0, ret1 } // FetchJobsInStatus indicates an expected call of FetchJobsInStatus. func (mr *MockPersistenceServiceMockRecorder) FetchJobsInStatus(arg0 interface{}, arg1 ...interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() varargs := append([]interface{}{arg0}, arg1...) return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchJobsInStatus", reflect.TypeOf((*MockPersistenceService)(nil).FetchJobsInStatus), varargs...) } // FetchTasksOfWorkerInStatus mocks base method. func (m *MockPersistenceService) FetchTasksOfWorkerInStatus(arg0 context.Context, arg1 *persistence.Worker, arg2 api.TaskStatus) ([]*persistence.Task, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "FetchTasksOfWorkerInStatus", arg0, arg1, arg2) ret0, _ := ret[0].([]*persistence.Task) ret1, _ := ret[1].(error) return ret0, ret1 } // FetchTasksOfWorkerInStatus indicates an expected call of FetchTasksOfWorkerInStatus. func (mr *MockPersistenceServiceMockRecorder) FetchTasksOfWorkerInStatus(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchTasksOfWorkerInStatus", reflect.TypeOf((*MockPersistenceService)(nil).FetchTasksOfWorkerInStatus), arg0, arg1, arg2) } // FetchTasksOfWorkerInStatusOfJob mocks base method. func (m *MockPersistenceService) FetchTasksOfWorkerInStatusOfJob(arg0 context.Context, arg1 *persistence.Worker, arg2 api.TaskStatus, arg3 *persistence.Job) ([]*persistence.Task, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "FetchTasksOfWorkerInStatusOfJob", arg0, arg1, arg2, arg3) ret0, _ := ret[0].([]*persistence.Task) ret1, _ := ret[1].(error) return ret0, ret1 } // FetchTasksOfWorkerInStatusOfJob indicates an expected call of FetchTasksOfWorkerInStatusOfJob. func (mr *MockPersistenceServiceMockRecorder) FetchTasksOfWorkerInStatusOfJob(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "FetchTasksOfWorkerInStatusOfJob", reflect.TypeOf((*MockPersistenceService)(nil).FetchTasksOfWorkerInStatusOfJob), arg0, arg1, arg2, arg3) } // JobHasTasksInStatus mocks base method. func (m *MockPersistenceService) JobHasTasksInStatus(arg0 context.Context, arg1 *persistence.Job, arg2 api.TaskStatus) (bool, error) { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "JobHasTasksInStatus", arg0, arg1, arg2) ret0, _ := ret[0].(bool) ret1, _ := ret[1].(error) return ret0, ret1 } // JobHasTasksInStatus indicates an expected call of JobHasTasksInStatus. func (mr *MockPersistenceServiceMockRecorder) JobHasTasksInStatus(arg0, arg1, arg2 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "JobHasTasksInStatus", reflect.TypeOf((*MockPersistenceService)(nil).JobHasTasksInStatus), arg0, arg1, arg2) } // SaveJobStatus mocks base method. func (m *MockPersistenceService) SaveJobStatus(arg0 context.Context, arg1 *persistence.Job) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SaveJobStatus", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // SaveJobStatus indicates an expected call of SaveJobStatus. func (mr *MockPersistenceServiceMockRecorder) SaveJobStatus(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveJobStatus", reflect.TypeOf((*MockPersistenceService)(nil).SaveJobStatus), arg0, arg1) } // SaveTask mocks base method. func (m *MockPersistenceService) SaveTask(arg0 context.Context, arg1 *persistence.Task) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SaveTask", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // SaveTask indicates an expected call of SaveTask. func (mr *MockPersistenceServiceMockRecorder) SaveTask(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTask", reflect.TypeOf((*MockPersistenceService)(nil).SaveTask), arg0, arg1) } // SaveTaskActivity mocks base method. func (m *MockPersistenceService) SaveTaskActivity(arg0 context.Context, arg1 *persistence.Task) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SaveTaskActivity", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // SaveTaskActivity indicates an expected call of SaveTaskActivity. func (mr *MockPersistenceServiceMockRecorder) SaveTaskActivity(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTaskActivity", reflect.TypeOf((*MockPersistenceService)(nil).SaveTaskActivity), arg0, arg1) } // SaveTaskStatus mocks base method. func (m *MockPersistenceService) SaveTaskStatus(arg0 context.Context, arg1 *persistence.Task) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "SaveTaskStatus", arg0, arg1) ret0, _ := ret[0].(error) return ret0 } // SaveTaskStatus indicates an expected call of SaveTaskStatus. func (mr *MockPersistenceServiceMockRecorder) SaveTaskStatus(arg0, arg1 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SaveTaskStatus", reflect.TypeOf((*MockPersistenceService)(nil).SaveTaskStatus), arg0, arg1) } // UpdateJobsTaskStatuses mocks base method. func (m *MockPersistenceService) UpdateJobsTaskStatuses(arg0 context.Context, arg1 *persistence.Job, arg2 api.TaskStatus, arg3 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateJobsTaskStatuses", arg0, arg1, arg2, arg3) ret0, _ := ret[0].(error) return ret0 } // UpdateJobsTaskStatuses indicates an expected call of UpdateJobsTaskStatuses. func (mr *MockPersistenceServiceMockRecorder) UpdateJobsTaskStatuses(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateJobsTaskStatuses", reflect.TypeOf((*MockPersistenceService)(nil).UpdateJobsTaskStatuses), arg0, arg1, arg2, arg3) } // UpdateJobsTaskStatusesConditional mocks base method. func (m *MockPersistenceService) UpdateJobsTaskStatusesConditional(arg0 context.Context, arg1 *persistence.Job, arg2 []api.TaskStatus, arg3 api.TaskStatus, arg4 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "UpdateJobsTaskStatusesConditional", arg0, arg1, arg2, arg3, arg4) ret0, _ := ret[0].(error) return ret0 } // UpdateJobsTaskStatusesConditional indicates an expected call of UpdateJobsTaskStatusesConditional. func (mr *MockPersistenceServiceMockRecorder) UpdateJobsTaskStatusesConditional(arg0, arg1, arg2, arg3, arg4 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateJobsTaskStatusesConditional", reflect.TypeOf((*MockPersistenceService)(nil).UpdateJobsTaskStatusesConditional), arg0, arg1, arg2, arg3, arg4) } // 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.SocketIOJobUpdate) { 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) } // BroadcastTaskUpdate mocks base method. func (m *MockChangeBroadcaster) BroadcastTaskUpdate(arg0 api.SocketIOTaskUpdate) { m.ctrl.T.Helper() m.ctrl.Call(m, "BroadcastTaskUpdate", arg0) } // BroadcastTaskUpdate indicates an expected call of BroadcastTaskUpdate. func (mr *MockChangeBroadcasterMockRecorder) BroadcastTaskUpdate(arg0 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BroadcastTaskUpdate", reflect.TypeOf((*MockChangeBroadcaster)(nil).BroadcastTaskUpdate), arg0) } // MockLogStorage is a mock of LogStorage interface. type MockLogStorage struct { ctrl *gomock.Controller recorder *MockLogStorageMockRecorder } // MockLogStorageMockRecorder is the mock recorder for MockLogStorage. type MockLogStorageMockRecorder struct { mock *MockLogStorage } // NewMockLogStorage creates a new mock instance. func NewMockLogStorage(ctrl *gomock.Controller) *MockLogStorage { mock := &MockLogStorage{ctrl: ctrl} mock.recorder = &MockLogStorageMockRecorder{mock} return mock } // EXPECT returns an object that allows the caller to indicate expected use. func (m *MockLogStorage) EXPECT() *MockLogStorageMockRecorder { return m.recorder } // WriteTimestamped mocks base method. func (m *MockLogStorage) WriteTimestamped(arg0 zerolog.Logger, arg1, arg2, arg3 string) error { m.ctrl.T.Helper() ret := m.ctrl.Call(m, "WriteTimestamped", arg0, arg1, arg2, arg3) ret0, _ := ret[0].(error) return ret0 } // WriteTimestamped indicates an expected call of WriteTimestamped. func (mr *MockLogStorageMockRecorder) WriteTimestamped(arg0, arg1, arg2, arg3 interface{}) *gomock.Call { mr.mock.ctrl.T.Helper() return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteTimestamped", reflect.TypeOf((*MockLogStorage)(nil).WriteTimestamped), arg0, arg1, arg2, arg3) }