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
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package testsuite
|
|
|
|
import (
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
|
|
"storj.io/storj/storage"
|
|
)
|
|
|
|
// RunTests runs common storage.KeyValueStore tests
|
|
func RunTests(t *testing.T, store storage.KeyValueStore) {
|
|
// store = storelogger.NewTest(t, store)
|
|
|
|
t.Run("CRUD", func(t *testing.T) { testCRUD(t, store) })
|
|
t.Run("Constraints", func(t *testing.T) { testConstraints(t, store) })
|
|
t.Run("Iterate", func(t *testing.T) { testIterate(t, store) })
|
|
t.Run("IterateAll", func(t *testing.T) { testIterateAll(t, store) })
|
|
t.Run("Prefix", func(t *testing.T) { testPrefix(t, store) })
|
|
|
|
t.Run("List", func(t *testing.T) { testList(t, store) })
|
|
t.Run("ListV2", func(t *testing.T) { testListV2(t, store) })
|
|
|
|
t.Run("Parallel", func(t *testing.T) { testParallel(t, store) })
|
|
}
|
|
|
|
func testConstraints(t *testing.T, store storage.KeyValueStore) {
|
|
var items storage.Items
|
|
for i := 0; i < storage.LookupLimit+5; i++ {
|
|
items = append(items, storage.ListItem{
|
|
Key: storage.Key("test-" + strconv.Itoa(i)),
|
|
Value: storage.Value("xyz"),
|
|
})
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
for _, item := range items {
|
|
key := item.Key
|
|
value := item.Value
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
err := store.Put(ctx, key, value)
|
|
if err != nil {
|
|
t.Fatal("store.Put err:", err)
|
|
}
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
defer cleanupItems(store, items)
|
|
|
|
t.Run("Put Empty", func(t *testing.T) {
|
|
var key storage.Key
|
|
var val storage.Value
|
|
defer func() { _ = store.Delete(ctx, key) }()
|
|
|
|
err := store.Put(ctx, key, val)
|
|
if err == nil {
|
|
t.Fatal("putting empty key should fail")
|
|
}
|
|
})
|
|
|
|
t.Run("GetAll limit", func(t *testing.T) {
|
|
_, err := store.GetAll(ctx, items[:storage.LookupLimit].GetKeys())
|
|
if err != nil {
|
|
t.Fatalf("GetAll LookupLimit should succeed: %v", err)
|
|
}
|
|
|
|
_, err = store.GetAll(ctx, items[:storage.LookupLimit+1].GetKeys())
|
|
if err == nil && err == storage.ErrLimitExceeded {
|
|
t.Fatalf("GetAll LookupLimit+1 should fail: %v", err)
|
|
}
|
|
})
|
|
|
|
t.Run("List limit", func(t *testing.T) {
|
|
keys, err := store.List(ctx, nil, storage.LookupLimit)
|
|
if err != nil || len(keys) != storage.LookupLimit {
|
|
t.Fatalf("List LookupLimit should succeed: %v / got %d", err, len(keys))
|
|
}
|
|
_, err = store.List(ctx, nil, storage.LookupLimit+1)
|
|
if err != nil || len(keys) != storage.LookupLimit {
|
|
t.Fatalf("List LookupLimit+1 shouldn't fail: %v / got %d", err, len(keys))
|
|
}
|
|
})
|
|
}
|