c1a9b1e877
This introduces the concept of 'event listener', which is now used by the farm status service to respond to events on the event bus. This makes it possible to reduce the regular poll period from 5 to 30 seconds. That's now only necessary as backup, just in case events are missed or otherwise things change without the event bus logic noticing.
116 lines
4.3 KiB
Go
Generated
116 lines
4.3 KiB
Go
Generated
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: projects.blender.org/studio/flamenco/internal/manager/farmstatus (interfaces: PersistenceService,EventBus)
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
context "context"
|
|
reflect "reflect"
|
|
|
|
gomock "github.com/golang/mock/gomock"
|
|
eventbus "projects.blender.org/studio/flamenco/internal/manager/eventbus"
|
|
persistence "projects.blender.org/studio/flamenco/internal/manager/persistence"
|
|
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
|
|
}
|
|
|
|
// SummarizeJobStatuses mocks base method.
|
|
func (m *MockPersistenceService) SummarizeJobStatuses(arg0 context.Context) (persistence.JobStatusCount, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SummarizeJobStatuses", arg0)
|
|
ret0, _ := ret[0].(persistence.JobStatusCount)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SummarizeJobStatuses indicates an expected call of SummarizeJobStatuses.
|
|
func (mr *MockPersistenceServiceMockRecorder) SummarizeJobStatuses(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SummarizeJobStatuses", reflect.TypeOf((*MockPersistenceService)(nil).SummarizeJobStatuses), arg0)
|
|
}
|
|
|
|
// SummarizeWorkerStatuses mocks base method.
|
|
func (m *MockPersistenceService) SummarizeWorkerStatuses(arg0 context.Context) (persistence.WorkerStatusCount, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "SummarizeWorkerStatuses", arg0)
|
|
ret0, _ := ret[0].(persistence.WorkerStatusCount)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// SummarizeWorkerStatuses indicates an expected call of SummarizeWorkerStatuses.
|
|
func (mr *MockPersistenceServiceMockRecorder) SummarizeWorkerStatuses(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SummarizeWorkerStatuses", reflect.TypeOf((*MockPersistenceService)(nil).SummarizeWorkerStatuses), arg0)
|
|
}
|
|
|
|
// MockEventBus is a mock of EventBus interface.
|
|
type MockEventBus struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockEventBusMockRecorder
|
|
}
|
|
|
|
// MockEventBusMockRecorder is the mock recorder for MockEventBus.
|
|
type MockEventBusMockRecorder struct {
|
|
mock *MockEventBus
|
|
}
|
|
|
|
// NewMockEventBus creates a new mock instance.
|
|
func NewMockEventBus(ctrl *gomock.Controller) *MockEventBus {
|
|
mock := &MockEventBus{ctrl: ctrl}
|
|
mock.recorder = &MockEventBusMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockEventBus) EXPECT() *MockEventBusMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// AddListener mocks base method.
|
|
func (m *MockEventBus) AddListener(arg0 eventbus.Listener) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "AddListener", arg0)
|
|
}
|
|
|
|
// AddListener indicates an expected call of AddListener.
|
|
func (mr *MockEventBusMockRecorder) AddListener(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddListener", reflect.TypeOf((*MockEventBus)(nil).AddListener), arg0)
|
|
}
|
|
|
|
// BroadcastFarmStatusEvent mocks base method.
|
|
func (m *MockEventBus) BroadcastFarmStatusEvent(arg0 api.EventFarmStatus) {
|
|
m.ctrl.T.Helper()
|
|
m.ctrl.Call(m, "BroadcastFarmStatusEvent", arg0)
|
|
}
|
|
|
|
// BroadcastFarmStatusEvent indicates an expected call of BroadcastFarmStatusEvent.
|
|
func (mr *MockEventBusMockRecorder) BroadcastFarmStatusEvent(arg0 interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BroadcastFarmStatusEvent", reflect.TypeOf((*MockEventBus)(nil).BroadcastFarmStatusEvent), arg0)
|
|
}
|