From 006a86d9191276221c4d8df19fd0745560b66227 Mon Sep 17 00:00:00 2001 From: josh <josh@elixxir.io> Date: Tue, 9 Mar 2021 12:01:27 -0800 Subject: [PATCH] Fix refactor of storage KV --- storage/clientVersion/store.go | 4 ++-- storage/clientVersion/store_test.go | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/storage/clientVersion/store.go b/storage/clientVersion/store.go index 344b138bd..3f1eca48a 100644 --- a/storage/clientVersion/store.go +++ b/storage/clientVersion/store.go @@ -45,7 +45,7 @@ func LoadStore(kv *versioned.KV) (*Store, error) { kv: kv.Prefix(prefix), } - obj, err := s.kv.Get(storeKey) + obj, err := s.kv.Get(storeKey, storeVersion) if err != nil { return nil, err } @@ -113,5 +113,5 @@ func (s *Store) save() error { Data: []byte(s.version.String()), } - return s.kv.Set(storeKey, &obj) + return s.kv.Set(storeKey, storeVersion, &obj) } diff --git a/storage/clientVersion/store_test.go b/storage/clientVersion/store_test.go index 8ff38a275..22da1257c 100644 --- a/storage/clientVersion/store_test.go +++ b/storage/clientVersion/store_test.go @@ -71,7 +71,7 @@ func TestLoadStore_ParseVersionError(t *testing.T) { Data: []byte("invalid version"), } - err := kv.Prefix(prefix).Set(storeKey, &obj) + err := kv.Prefix(prefix).Set(storeKey, storeVersion, &obj) if err != nil { t.Fatalf("Failed to save Store: %+v", err) } @@ -219,7 +219,7 @@ func TestStore_save(t *testing.T) { t.Errorf("save() returned an error: %+v", err) } - obj, err := s.kv.Get(storeKey) + obj, err := s.kv.Get(storeKey, storeVersion) if err != nil { t.Errorf("Failed to load clientVersion store: %+v", err) } -- GitLab