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  }