github.com/influxdata/influxdb/v2@v2.7.6/mock/variable_service.go (about) 1 package mock 2 3 import ( 4 "context" 5 6 "github.com/influxdata/influxdb/v2" 7 "github.com/influxdata/influxdb/v2/kit/platform" 8 ) 9 10 var _ influxdb.VariableService = &VariableService{} 11 12 type VariableService struct { 13 CreateVariableF func(context.Context, *influxdb.Variable) error 14 CreateVariableCalls SafeCount 15 DeleteVariableF func(context.Context, platform.ID) error 16 DeleteVariableCalls SafeCount 17 FindVariableByIDF func(context.Context, platform.ID) (*influxdb.Variable, error) 18 FindVariableByIDCalls SafeCount 19 FindVariablesF func(context.Context, influxdb.VariableFilter, ...influxdb.FindOptions) ([]*influxdb.Variable, error) 20 FindVariablesCalls SafeCount 21 ReplaceVariableF func(context.Context, *influxdb.Variable) error 22 ReplaceVariableCalls SafeCount 23 UpdateVariableF func(ctx context.Context, id platform.ID, update *influxdb.VariableUpdate) (*influxdb.Variable, error) 24 UpdateVariableCalls SafeCount 25 } 26 27 // NewVariableService returns a mock of VariableService where its methods will return zero values. 28 func NewVariableService() *VariableService { 29 return &VariableService{ 30 CreateVariableF: func(context.Context, *influxdb.Variable) error { return nil }, 31 DeleteVariableF: func(context.Context, platform.ID) error { return nil }, 32 FindVariableByIDF: func(context.Context, platform.ID) (*influxdb.Variable, error) { return nil, nil }, 33 FindVariablesF: func(context.Context, influxdb.VariableFilter, ...influxdb.FindOptions) ([]*influxdb.Variable, error) { 34 return nil, nil 35 }, 36 ReplaceVariableF: func(context.Context, *influxdb.Variable) error { return nil }, 37 UpdateVariableF: func(ctx context.Context, id platform.ID, update *influxdb.VariableUpdate) (*influxdb.Variable, error) { 38 return nil, nil 39 }, 40 } 41 } 42 43 func (s *VariableService) CreateVariable(ctx context.Context, variable *influxdb.Variable) error { 44 defer s.CreateVariableCalls.IncrFn()() 45 return s.CreateVariableF(ctx, variable) 46 } 47 48 func (s *VariableService) ReplaceVariable(ctx context.Context, variable *influxdb.Variable) error { 49 defer s.ReplaceVariableCalls.IncrFn()() 50 return s.ReplaceVariableF(ctx, variable) 51 } 52 53 func (s *VariableService) FindVariables(ctx context.Context, filter influxdb.VariableFilter, opts ...influxdb.FindOptions) ([]*influxdb.Variable, error) { 54 defer s.FindVariablesCalls.IncrFn()() 55 return s.FindVariablesF(ctx, filter, opts...) 56 } 57 58 func (s *VariableService) FindVariableByID(ctx context.Context, id platform.ID) (*influxdb.Variable, error) { 59 defer s.FindVariableByIDCalls.IncrFn()() 60 return s.FindVariableByIDF(ctx, id) 61 } 62 63 func (s *VariableService) DeleteVariable(ctx context.Context, id platform.ID) error { 64 defer s.DeleteVariableCalls.IncrFn()() 65 return s.DeleteVariableF(ctx, id) 66 } 67 68 func (s *VariableService) UpdateVariable(ctx context.Context, id platform.ID, update *influxdb.VariableUpdate) (*influxdb.Variable, error) { 69 defer s.UpdateVariableCalls.IncrFn()() 70 return s.UpdateVariableF(ctx, id, update) 71 }