package mocks

import (
	"context"

	"github.com/influxdata/influxdb/chronograf"
)

var _ chronograf.OrganizationsStore = &OrganizationsStore{}

type OrganizationsStore struct {
	AllF                 func(context.Context) ([]chronograf.Organization, error)
	AddF                 func(context.Context, *chronograf.Organization) (*chronograf.Organization, error)
	DeleteF              func(context.Context, *chronograf.Organization) error
	GetF                 func(ctx context.Context, q chronograf.OrganizationQuery) (*chronograf.Organization, error)
	UpdateF              func(context.Context, *chronograf.Organization) error
	CreateDefaultF       func(context.Context) error
	DefaultOrganizationF func(context.Context) (*chronograf.Organization, error)
}

func (s *OrganizationsStore) CreateDefault(ctx context.Context) error {
	return s.CreateDefaultF(ctx)
}

func (s *OrganizationsStore) DefaultOrganization(ctx context.Context) (*chronograf.Organization, error) {
	return s.DefaultOrganizationF(ctx)
}

func (s *OrganizationsStore) Add(ctx context.Context, o *chronograf.Organization) (*chronograf.Organization, error) {
	return s.AddF(ctx, o)
}

func (s *OrganizationsStore) All(ctx context.Context) ([]chronograf.Organization, error) {
	return s.AllF(ctx)
}

func (s *OrganizationsStore) Delete(ctx context.Context, o *chronograf.Organization) error {
	return s.DeleteF(ctx, o)
}

func (s *OrganizationsStore) Get(ctx context.Context, q chronograf.OrganizationQuery) (*chronograf.Organization, error) {
	return s.GetF(ctx, q)
}

func (s *OrganizationsStore) Update(ctx context.Context, o *chronograf.Organization) error {
	return s.UpdateF(ctx, o)
}