add mock implementation of source service

mock/source_service.go

@@ -0,0 +1,63 @@
+package mock
+
+import (
+	"context"
+
+	"github.com/influxdata/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, platform.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, platform.ID, platform.SourceUpdate) (*platform.Source, error)
+	DeleteSourceFn   func(context.Context, platform.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, platform.ID) (*platform.Source, error) { return nil, nil },
+		CreateSourceFn:   func(context.Context, *platform.Source) error { return nil },
+		UpdateSourceFn:   func(context.Context, platform.ID, platform.SourceUpdate) (*platform.Source, error) { return nil, nil },
+		DeleteSourceFn:   func(context.Context, platform.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 platform.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 platform.ID) error {
+	return s.DeleteSourceFn(ctx, id)
+}
+
+// UpdateSource updates the source.
+func (s *SourceService) UpdateSource(ctx context.Context, id platform.ID, upd platform.SourceUpdate) (*platform.Source, error) {
+	return s.UpdateSourceFn(ctx, id, upd)
+}