diff --git a/storage/clientVersion/store.go b/storage/clientVersion/store.go index 344b138bd8cf328bfa17722bce8bec8c6fbfd860..3f1eca48aca688113ba8c240ad8e747029947a90 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 8ff38a27599f2d43b8576e97a58e0843192bb814..22da1257c04f0b7780d63b46c0cf28a86bd6143d 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) }