ConfigAPI and Query API improvements

[multicloud/k8s.git] / src / k8splugin / internal / db / etcd_testing.go

diff --git a/src/k8splugin/internal/db/etcd_testing.go b/src/k8splugin/internal/db/etcd_testing.go
index 4b4dfe3..2f62d36 100644 (file)
--- a/src/k8splugin/internal/db/etcd_testing.go
+++ b/src/k8splugin/internal/db/etcd_testing.go
@@ -41,7 +41,17 @@ func (c *MockEtcdClient) Get(key string) ([]byte, error) {
        return nil, pkgerrors.Errorf("Key doesn't exist")
 }
 
-func (c *MockEtcdClient) GetAll(key string) ([][]byte, error) {
+func (c *MockEtcdClient) GetKeys(key string) ([]string, error) {
+       result := make([]string, 0)
+       for kvKey := range c.Items {
+               if strings.HasPrefix(kvKey, key) {
+                       result = append(result, kvKey)
+               }
+       }
+       return result, nil
+}
+
+func (c *MockEtcdClient) GetValues(key string) ([][]byte, error) {
        result := make([][]byte, 0)
        for kvKey, kvValue := range c.Items {
                if strings.HasPrefix(kvKey, key) {