github.com/influxdata/influxdb/v2@v2.7.6/mock/org_service.go (about) 1 package mock 2 3 import ( 4 "context" 5 6 platform "github.com/influxdata/influxdb/v2" 7 platform2 "github.com/influxdata/influxdb/v2/kit/platform" 8 ) 9 10 var _ platform.OrganizationService = &OrganizationService{} 11 12 // OrganizationService is a mock organization server. 13 type OrganizationService struct { 14 FindOrganizationByIDF func(ctx context.Context, id platform2.ID) (*platform.Organization, error) 15 FindOrganizationF func(ctx context.Context, filter platform.OrganizationFilter) (*platform.Organization, error) 16 FindOrganizationsF func(ctx context.Context, filter platform.OrganizationFilter, opt ...platform.FindOptions) ([]*platform.Organization, int, error) 17 CreateOrganizationF func(ctx context.Context, b *platform.Organization) error 18 UpdateOrganizationF func(ctx context.Context, id platform2.ID, upd platform.OrganizationUpdate) (*platform.Organization, error) 19 DeleteOrganizationF func(ctx context.Context, id platform2.ID) error 20 FindResourceOrganizationIDF func(ctx context.Context, rt platform.ResourceType, id platform2.ID) (platform2.ID, error) 21 } 22 23 // NewOrganizationService returns a mock OrganizationService where its methods will return 24 // zero values. 25 func NewOrganizationService() *OrganizationService { 26 return &OrganizationService{ 27 FindOrganizationByIDF: func(ctx context.Context, id platform2.ID) (*platform.Organization, error) { return nil, nil }, 28 FindOrganizationF: func(ctx context.Context, filter platform.OrganizationFilter) (*platform.Organization, error) { 29 return nil, nil 30 }, 31 FindOrganizationsF: func(ctx context.Context, filter platform.OrganizationFilter, opt ...platform.FindOptions) ([]*platform.Organization, int, error) { 32 return nil, 0, nil 33 }, 34 CreateOrganizationF: func(ctx context.Context, b *platform.Organization) error { return nil }, 35 UpdateOrganizationF: func(ctx context.Context, id platform2.ID, upd platform.OrganizationUpdate) (*platform.Organization, error) { 36 return nil, nil 37 }, 38 DeleteOrganizationF: func(ctx context.Context, id platform2.ID) error { return nil }, 39 } 40 } 41 42 // FindOrganizationByID calls FindOrganizationByIDF. 43 func (s *OrganizationService) FindOrganizationByID(ctx context.Context, id platform2.ID) (*platform.Organization, error) { 44 return s.FindOrganizationByIDF(ctx, id) 45 } 46 47 // FindOrganization calls FindOrganizationF. 48 func (s *OrganizationService) FindOrganization(ctx context.Context, filter platform.OrganizationFilter) (*platform.Organization, error) { 49 return s.FindOrganizationF(ctx, filter) 50 } 51 52 // FindOrganizations calls FindOrganizationsF. 53 func (s *OrganizationService) FindOrganizations(ctx context.Context, filter platform.OrganizationFilter, opt ...platform.FindOptions) ([]*platform.Organization, int, error) { 54 return s.FindOrganizationsF(ctx, filter, opt...) 55 } 56 57 // CreateOrganization calls CreateOrganizationF. 58 func (s *OrganizationService) CreateOrganization(ctx context.Context, b *platform.Organization) error { 59 return s.CreateOrganizationF(ctx, b) 60 } 61 62 // UpdateOrganization calls UpdateOrganizationF. 63 func (s *OrganizationService) UpdateOrganization(ctx context.Context, id platform2.ID, upd platform.OrganizationUpdate) (*platform.Organization, error) { 64 return s.UpdateOrganizationF(ctx, id, upd) 65 } 66 67 // DeleteOrganization calls DeleteOrganizationF. 68 func (s *OrganizationService) DeleteOrganization(ctx context.Context, id platform2.ID) error { 69 return s.DeleteOrganizationF(ctx, id) 70 } 71 72 // FindResourceOrganizationID calls FindResourceOrganizationIDF. 73 func (s *OrganizationService) FindResourceOrganizationID(ctx context.Context, rt platform.ResourceType, id platform2.ID) (platform2.ID, error) { 74 return s.FindResourceOrganizationIDF(ctx, rt, id) 75 }