package kv import ( "errors" "strings" "github.com/abronan/valkeyrie/store" ) func newProviderMock(kvPairs []*store.KVPair) *Provider { return &Provider{ Prefix: "traefik", kvClient: &Mock{ KVPairs: kvPairs, }, } } // Override Get/List to return a error type KvError struct { Get error List error } // Extremely limited mock store so we can test initialization type Mock struct { Error KvError KVPairs []*store.KVPair WatchTreeMethod func() <-chan []*store.KVPair } func newKvClientMock(kvPairs []*store.KVPair, err error) *Mock { mock := &Mock{ KVPairs: kvPairs, } if err != nil { mock.Error = KvError{ Get: err, List: err, } } return mock } func (s *Mock) Put(key string, value []byte, opts *store.WriteOptions) error { return errors.New("put not supported") } func (s *Mock) Get(key string, options *store.ReadOptions) (*store.KVPair, error) { if err := s.Error.Get; err != nil { return nil, err } for _, kvPair := range s.KVPairs { if kvPair.Key == key { return kvPair, nil } } return nil, store.ErrKeyNotFound } func (s *Mock) Delete(key string) error { return errors.New("delete not supported") } // Exists mock func (s *Mock) Exists(key string, options *store.ReadOptions) (bool, error) { if err := s.Error.Get; err != nil { return false, err } for _, kvPair := range s.KVPairs { if strings.HasPrefix(kvPair.Key, key) { return true, nil } } return false, store.ErrKeyNotFound } // Watch mock func (s *Mock) Watch(key string, stopCh <-chan struct{}, options *store.ReadOptions) (<-chan *store.KVPair, error) { return nil, errors.New("watch not supported") } // WatchTree mock func (s *Mock) WatchTree(prefix string, stopCh <-chan struct{}, options *store.ReadOptions) (<-chan []*store.KVPair, error) { return s.WatchTreeMethod(), nil } // NewLock mock func (s *Mock) NewLock(key string, options *store.LockOptions) (store.Locker, error) { return nil, errors.New("NewLock not supported") } // List mock func (s *Mock) List(prefix string, options *store.ReadOptions) ([]*store.KVPair, error) { if err := s.Error.List; err != nil { return nil, err } var kv []*store.KVPair for _, kvPair := range s.KVPairs { if strings.HasPrefix(kvPair.Key, prefix) && !strings.ContainsAny(strings.TrimPrefix(kvPair.Key, prefix), pathSeparator) { kv = append(kv, kvPair) } } return kv, nil } // DeleteTree mock func (s *Mock) DeleteTree(prefix string) error { return errors.New("DeleteTree not supported") } // AtomicPut mock func (s *Mock) AtomicPut(key string, value []byte, previous *store.KVPair, opts *store.WriteOptions) (bool, *store.KVPair, error) { return false, nil, errors.New("AtomicPut not supported") } // AtomicDelete mock func (s *Mock) AtomicDelete(key string, previous *store.KVPair) (bool, error) { return false, errors.New("AtomicDelete not supported") } // Close mock func (s *Mock) Close() {}