f1641af802
* storage: add monkit task to missing places Change-Id: I9e17a6b14f7c25bbf698eeecf32785e9add3f26e * fix tests Change-Id: Id078276fa3de61a28eb3d01d4e751732ecbb173f * import order Change-Id: I814e33755b9f10b5219af37cd828cd75eb3da1a4 * remove part of other commit Change-Id: Idaa4c95cd65e97567fb466de49718db8203cfbe1
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package testsuite
|
|
|
|
import (
|
|
"bytes"
|
|
"math/rand"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"storj.io/storj/storage"
|
|
)
|
|
|
|
func testParallel(t *testing.T, store storage.KeyValueStore) {
|
|
items := storage.Items{
|
|
newItem("a", "1", false),
|
|
newItem("b", "2", false),
|
|
newItem("c", "3", false),
|
|
}
|
|
rand.Shuffle(len(items), items.Swap)
|
|
defer cleanupItems(store, items)
|
|
|
|
for idx := range items {
|
|
i := idx
|
|
item := items[i]
|
|
t.Run(strconv.Itoa(i), func(t *testing.T) {
|
|
t.Parallel()
|
|
// Put
|
|
err := store.Put(ctx, item.Key, item.Value)
|
|
if err != nil {
|
|
t.Fatalf("failed to put %q = %v: %v", item.Key, item.Value, err)
|
|
}
|
|
|
|
// Get
|
|
value, err := store.Get(ctx, item.Key)
|
|
if err != nil {
|
|
t.Fatalf("failed to get %q = %v: %v", item.Key, item.Value, err)
|
|
}
|
|
if !bytes.Equal([]byte(value), []byte(item.Value)) {
|
|
t.Fatalf("invalid value for %q = %v: got %v", item.Key, item.Value, value)
|
|
}
|
|
|
|
// GetAll
|
|
values, err := store.GetAll(ctx, []storage.Key{item.Key})
|
|
if len(values) != 1 {
|
|
t.Fatalf("failed to GetAll: %v", err)
|
|
}
|
|
|
|
if !bytes.Equal([]byte(values[0]), []byte(item.Value)) {
|
|
t.Fatalf("invalid GetAll %q = %v: got %v", item.Key, item.Value, values[i])
|
|
}
|
|
|
|
// Update value
|
|
nextValue := storage.Value(string(item.Value) + "X")
|
|
err = store.Put(ctx, item.Key, nextValue)
|
|
if err != nil {
|
|
t.Fatalf("failed to update %q = %v: %v", item.Key, nextValue, err)
|
|
}
|
|
|
|
value, err = store.Get(ctx, item.Key)
|
|
if err != nil {
|
|
t.Fatalf("failed to get %q = %v: %v", item.Key, nextValue, err)
|
|
}
|
|
if !bytes.Equal([]byte(value), []byte(nextValue)) {
|
|
t.Fatalf("invalid updated value for %q = %v: got %v", item.Key, nextValue, value)
|
|
}
|
|
|
|
err = store.Delete(ctx, item.Key)
|
|
if err != nil {
|
|
t.Fatalf("failed to delete %v: %v", item.Key, err)
|
|
}
|
|
})
|
|
}
|
|
}
|