4b95f611c2
* Chore: Add AwaitHealthy to ModuleEngine and Server * switch from fmt.Errorf to errors.New Co-authored-by: Will Browne <wbrowne@users.noreply.github.com> --------- Co-authored-by: Will Browne <wbrowne@users.noreply.github.com>
84 lines
1.9 KiB
Go
84 lines
1.9 KiB
Go
package modules
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/grafana/dskit/services"
|
|
)
|
|
|
|
var _ Manager = (*MockModuleManager)(nil)
|
|
var _ Engine = (*MockModuleEngine)(nil)
|
|
|
|
type MockModuleManager struct {
|
|
RegisterModuleFunc func(name string, initFn func() (services.Service, error))
|
|
RegisterInvisibleModuleFunc func(name string, initFn func() (services.Service, error))
|
|
}
|
|
|
|
func (m *MockModuleManager) RegisterModule(name string, initFn func() (services.Service, error)) {
|
|
if m.RegisterModuleFunc != nil {
|
|
m.RegisterModuleFunc(name, initFn)
|
|
}
|
|
}
|
|
|
|
func (m *MockModuleManager) RegisterInvisibleModule(name string, initFn func() (services.Service, error)) {
|
|
if m.RegisterInvisibleModuleFunc != nil {
|
|
m.RegisterInvisibleModuleFunc(name, initFn)
|
|
}
|
|
}
|
|
|
|
type MockModuleEngine struct {
|
|
AwaitHealthyFunc func(context.Context) error
|
|
InitFunc func(context.Context) error
|
|
RunFunc func(context.Context) error
|
|
ShutdownFunc func(context.Context) error
|
|
}
|
|
|
|
func (m *MockModuleEngine) AwaitHealthy(ctx context.Context) error {
|
|
if m.AwaitHealthyFunc != nil {
|
|
return m.AwaitHealthyFunc(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockModuleEngine) Init(ctx context.Context) error {
|
|
if m.InitFunc != nil {
|
|
return m.InitFunc(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockModuleEngine) Run(ctx context.Context) error {
|
|
if m.RunFunc != nil {
|
|
return m.RunFunc(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *MockModuleEngine) Shutdown(ctx context.Context) error {
|
|
if m.ShutdownFunc != nil {
|
|
return m.ShutdownFunc(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func stringsContain(values []string, search string) bool {
|
|
for _, v := range values {
|
|
if search == v {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
type MockNamedService struct {
|
|
*services.BasicService
|
|
}
|
|
|
|
func NewMockNamedService(name string) *MockNamedService {
|
|
startFn := func(_ context.Context) error { return nil }
|
|
return &MockNamedService{
|
|
BasicService: services.NewIdleService(startFn, nil).WithName(name),
|
|
}
|
|
}
|