56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package bolt_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/v2/kv"
|
|
"github.com/influxdata/influxdb/v2/kv/migration"
|
|
platformtesting "github.com/influxdata/influxdb/v2/testing"
|
|
)
|
|
|
|
func initKVStore(f platformtesting.KVStoreFields, t *testing.T) (kv.Store, func()) {
|
|
s, closeFn, err := NewTestKVStore(t)
|
|
if err != nil {
|
|
t.Fatalf("failed to create new kv store: %v", err)
|
|
}
|
|
|
|
mustCreateBucket(t, s, f.Bucket)
|
|
|
|
err = s.Update(context.Background(), func(tx kv.Tx) error {
|
|
b, err := tx.Bucket(f.Bucket)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, p := range f.Pairs {
|
|
if err := b.Put(p.Key, p.Value); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("failed to put keys: %v", err)
|
|
}
|
|
return s, func() {
|
|
closeFn()
|
|
}
|
|
}
|
|
|
|
func TestKVStore(t *testing.T) {
|
|
platformtesting.KVStore(initKVStore, t)
|
|
}
|
|
|
|
func mustCreateBucket(t testing.TB, store kv.SchemaStore, bucket []byte) {
|
|
t.Helper()
|
|
|
|
migrationName := fmt.Sprintf("create bucket %q", string(bucket))
|
|
|
|
if err := migration.CreateBuckets(migrationName, bucket).Up(context.Background(), store); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|