diff options
author | Ritu Sood <ritu.sood@intel.com> | 2020-02-07 18:36:54 -0800 |
---|---|---|
committer | Marcus Williams <marcus.williams@intel.com> | 2020-02-14 21:57:46 +0000 |
commit | c3e1c9a5fefde3fcb6aaf05c19b18f211c1a43ba (patch) | |
tree | 52de09121293ad1fec9c0be1d1f24439f9b73fc0 /src/orchestrator/pkg/infra/contextdb/mock.go | |
parent | a75d489bbf87712371d67dce0753577bdacce0c3 (diff) |
Add etcd support in infrastructure layer
Create a etcd client and provide minimal
functionality needed by the orchestrator
for using etcd
Issue-ID: MULTICLOUD-871
Signed-off-by: Ritu Sood <ritu.sood@intel.com>
Change-Id: I56fb4643addf43cdc59366e7163b66bb1618876d
Diffstat (limited to 'src/orchestrator/pkg/infra/contextdb/mock.go')
-rw-r--r-- | src/orchestrator/pkg/infra/contextdb/mock.go | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/src/orchestrator/pkg/infra/contextdb/mock.go b/src/orchestrator/pkg/infra/contextdb/mock.go new file mode 100644 index 00000000..fc0f8ff7 --- /dev/null +++ b/src/orchestrator/pkg/infra/contextdb/mock.go @@ -0,0 +1,58 @@ +/* +Copyright 2020 Intel Corporation. +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + http://www.apache.org/licenses/LICENSE-2.0 +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +*/ + +package contextdb + +import ( + pkgerrors "github.com/pkg/errors" +) + +type MockEtcd struct { + Items map[string]interface{} + Err error +} + +func (c *MockEtcd) Put(key string, value interface{}) error { + if c.Items == nil { + c.Items = make(map[string]interface{}) + } + c.Items[key] = value + return c.Err +} + +func (c *MockEtcd) Get(key string, value interface{}) error { + for kvKey, kvValue := range c.Items { + if kvKey == key { + value = kvValue + return nil + } + } + return pkgerrors.Errorf("Key doesn't exist") +} + +func (c *MockEtcd) Delete(key string) error { + delete(c.Items, key) + return c.Err +} + +func (c *MockEtcd) GetAllKeys(path string) ([]string, error) { + var keys []string + for k, _ := range c.Items { + keys = append(keys, string(k)) + } + return keys, nil +} + +func (e *MockEtcd) HealthCheck() error { + return nil +} |