65 lines
2.6 KiB
Go
65 lines
2.6 KiB
Go
package mock
|
|
|
|
import (
|
|
"context"
|
|
|
|
platform "github.com/influxdata/influxdb/v2"
|
|
platform2 "github.com/influxdata/influxdb/v2/kit/platform"
|
|
)
|
|
|
|
var _ platform.SourceService = (*SourceService)(nil)
|
|
|
|
// SourceService is a mock implementation of platform.SourceService.
|
|
type SourceService struct {
|
|
DefaultSourceFn func(context.Context) (*platform.Source, error)
|
|
FindSourceByIDFn func(context.Context, platform2.ID) (*platform.Source, error)
|
|
FindSourcesFn func(context.Context, platform.FindOptions) ([]*platform.Source, int, error)
|
|
CreateSourceFn func(context.Context, *platform.Source) error
|
|
UpdateSourceFn func(context.Context, platform2.ID, platform.SourceUpdate) (*platform.Source, error)
|
|
DeleteSourceFn func(context.Context, platform2.ID) error
|
|
}
|
|
|
|
// NewSourceService returns a mock of SourceService where its methods will return zero values.
|
|
func NewSourceService() *SourceService {
|
|
return &SourceService{
|
|
DefaultSourceFn: func(context.Context) (*platform.Source, error) { return nil, nil },
|
|
FindSourceByIDFn: func(context.Context, platform2.ID) (*platform.Source, error) { return nil, nil },
|
|
CreateSourceFn: func(context.Context, *platform.Source) error { return nil },
|
|
UpdateSourceFn: func(context.Context, platform2.ID, platform.SourceUpdate) (*platform.Source, error) { return nil, nil },
|
|
DeleteSourceFn: func(context.Context, platform2.ID) error { return nil },
|
|
FindSourcesFn: func(context.Context, platform.FindOptions) ([]*platform.Source, int, error) {
|
|
return nil, 0, nil
|
|
},
|
|
}
|
|
}
|
|
|
|
// DefaultSource retrieves the default source.
|
|
func (s *SourceService) DefaultSource(ctx context.Context) (*platform.Source, error) {
|
|
return s.DefaultSourceFn(ctx)
|
|
}
|
|
|
|
// FindSourceByID retrieves a source by its ID.
|
|
func (s *SourceService) FindSourceByID(ctx context.Context, id platform2.ID) (*platform.Source, error) {
|
|
return s.FindSourceByIDFn(ctx, id)
|
|
}
|
|
|
|
// FindSources returns a list of all sources.
|
|
func (s *SourceService) FindSources(ctx context.Context, opts platform.FindOptions) ([]*platform.Source, int, error) {
|
|
return s.FindSourcesFn(ctx, opts)
|
|
}
|
|
|
|
// CreateSource sets the sources ID and stores it.
|
|
func (s *SourceService) CreateSource(ctx context.Context, source *platform.Source) error {
|
|
return s.CreateSourceFn(ctx, source)
|
|
}
|
|
|
|
// DeleteSource removes the source.
|
|
func (s *SourceService) DeleteSource(ctx context.Context, id platform2.ID) error {
|
|
return s.DeleteSourceFn(ctx, id)
|
|
}
|
|
|
|
// UpdateSource updates the source.
|
|
func (s *SourceService) UpdateSource(ctx context.Context, id platform2.ID, upd platform.SourceUpdate) (*platform.Source, error) {
|
|
return s.UpdateSourceFn(ctx, id, upd)
|
|
}
|