Dynamic Configuration Refactoring

provider/kv/kv_mock_test.go

@@ -1,124 +0,0 @@
-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() {}