Fix refactor of storage KV

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)
 }

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)
 	}