diff --git a/mock/scraper_service.go b/mock/scraper_service.go new file mode 100644 index 0000000000..bf0dbdb715 --- /dev/null +++ b/mock/scraper_service.go @@ -0,0 +1,43 @@ +package mock + +import ( + "context" + + "github.com/influxdata/platform" +) + +var _ platform.ScraperTargetStoreService = &ScraperTargetStoreService{} + +// ScraperTargetStoreService is a mock implementation of a platform.ScraperTargetStoreService. +type ScraperTargetStoreService struct { + ListTargetsF func(ctx context.Context) ([]platform.ScraperTarget, error) + AddTargetF func(ctx context.Context, t *platform.ScraperTarget) error + GetTargetByIDF func(ctx context.Context, id platform.ID) (*platform.ScraperTarget, error) + RemoveTargetF func(ctx context.Context, id platform.ID) error + UpdateTargetF func(ctx context.Context, t *platform.ScraperTarget) (*platform.ScraperTarget, error) +} + +// ListTargets lists all the scraper targets. +func (s *ScraperTargetStoreService) ListTargets(ctx context.Context) ([]platform.ScraperTarget, error) { + return s.ListTargetsF(ctx) +} + +// AddTarget adds a scraper target. +func (s *ScraperTargetStoreService) AddTarget(ctx context.Context, t *platform.ScraperTarget) error { + return s.AddTargetF(ctx, t) +} + +// GetTargetByID retrieves a scraper target by id. +func (s *ScraperTargetStoreService) GetTargetByID(ctx context.Context, id platform.ID) (*platform.ScraperTarget, error) { + return s.GetTargetByIDF(ctx, id) +} + +// RemoveTarget deletes a scraper target. +func (s *ScraperTargetStoreService) RemoveTarget(ctx context.Context, id platform.ID) error { + return s.RemoveTargetF(ctx, id) +} + +// UpdateTarget updates a scraper target. +func (s *ScraperTargetStoreService) UpdateTarget(ctx context.Context, t *platform.ScraperTarget) (*platform.ScraperTarget, error) { + return s.UpdateTargetF(ctx, t) +} diff --git a/mock/task_service.go b/mock/task_service.go index f9f9ffde95..111b4d481a 100644 --- a/mock/task_service.go +++ b/mock/task_service.go @@ -24,7 +24,6 @@ type TaskService struct { func (s *TaskService) FindTaskByID(ctx context.Context, id platform.ID) (*platform.Task, error) { return s.FindTaskByIDFn(ctx, id) - } func (s *TaskService) FindTasks(ctx context.Context, filter platform.TaskFilter) ([]*platform.Task, int, error) {