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  }