store_test: deduplicate store initialization
This commit is contained in:
parent
873f483804
commit
da72d0c2b9
|
@ -11,12 +11,39 @@ import (
|
||||||
|
|
||||||
// tests around the correct behavior of iterators
|
// tests around the correct behavior of iterators
|
||||||
|
|
||||||
func CommonTestPrefixIterator(t *testing.T, s store.KVStore) {
|
type testRow struct {
|
||||||
|
|
||||||
data := []struct {
|
|
||||||
key []byte
|
key []byte
|
||||||
val []byte
|
val []byte
|
||||||
}{
|
}
|
||||||
|
|
||||||
|
func batchWriteRows(s store.KVStore, rows []testRow) error {
|
||||||
|
// open a writer
|
||||||
|
writer, err := s.Writer()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// write the data
|
||||||
|
batch := writer.NewBatch()
|
||||||
|
for _, row := range rows {
|
||||||
|
batch.Set(row.key, row.val)
|
||||||
|
}
|
||||||
|
err = writer.ExecuteBatch(batch)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// close the writer
|
||||||
|
err = writer.Close()
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func CommonTestPrefixIterator(t *testing.T, s store.KVStore) {
|
||||||
|
|
||||||
|
data := []testRow{
|
||||||
{[]byte("apple"), []byte("val")},
|
{[]byte("apple"), []byte("val")},
|
||||||
{[]byte("cat1"), []byte("val")},
|
{[]byte("cat1"), []byte("val")},
|
||||||
{[]byte("cat2"), []byte("val")},
|
{[]byte("cat2"), []byte("val")},
|
||||||
|
@ -40,24 +67,7 @@ func CommonTestPrefixIterator(t *testing.T, s store.KVStore) {
|
||||||
[]byte("dog4"),
|
[]byte("dog4"),
|
||||||
}
|
}
|
||||||
|
|
||||||
// open a writer
|
err := batchWriteRows(s, data)
|
||||||
writer, err := s.Writer()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// write the data
|
|
||||||
batch := writer.NewBatch()
|
|
||||||
for _, row := range data {
|
|
||||||
batch.Set(row.key, row.val)
|
|
||||||
}
|
|
||||||
err = writer.ExecuteBatch(batch)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// close the writer
|
|
||||||
err = writer.Close()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
@ -123,10 +133,7 @@ func CommonTestPrefixIterator(t *testing.T, s store.KVStore) {
|
||||||
|
|
||||||
func CommonTestRangeIterator(t *testing.T, s store.KVStore) {
|
func CommonTestRangeIterator(t *testing.T, s store.KVStore) {
|
||||||
|
|
||||||
data := []struct {
|
data := []testRow{
|
||||||
key []byte
|
|
||||||
val []byte
|
|
||||||
}{
|
|
||||||
{[]byte("a1"), []byte("val")},
|
{[]byte("a1"), []byte("val")},
|
||||||
{[]byte("b1"), []byte("val")},
|
{[]byte("b1"), []byte("val")},
|
||||||
{[]byte("b2"), []byte("val")},
|
{[]byte("b2"), []byte("val")},
|
||||||
|
@ -154,24 +161,7 @@ func CommonTestRangeIterator(t *testing.T, s store.KVStore) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// open a writer
|
err := batchWriteRows(s, data)
|
||||||
writer, err := s.Writer()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// write the data
|
|
||||||
batch := writer.NewBatch()
|
|
||||||
for _, row := range data {
|
|
||||||
batch.Set(row.key, row.val)
|
|
||||||
}
|
|
||||||
err = writer.ExecuteBatch(batch)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// close the writer
|
|
||||||
err = writer.Close()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
@ -279,10 +269,7 @@ func CommonTestRangeIterator(t *testing.T, s store.KVStore) {
|
||||||
|
|
||||||
func CommonTestRangeIteratorSeek(t *testing.T, s store.KVStore) {
|
func CommonTestRangeIteratorSeek(t *testing.T, s store.KVStore) {
|
||||||
|
|
||||||
data := []struct {
|
data := []testRow{
|
||||||
key []byte
|
|
||||||
val []byte
|
|
||||||
}{
|
|
||||||
{[]byte("a1"), []byte("val")},
|
{[]byte("a1"), []byte("val")},
|
||||||
{[]byte("b1"), []byte("val")},
|
{[]byte("b1"), []byte("val")},
|
||||||
{[]byte("c1"), []byte("val")},
|
{[]byte("c1"), []byte("val")},
|
||||||
|
@ -290,24 +277,7 @@ func CommonTestRangeIteratorSeek(t *testing.T, s store.KVStore) {
|
||||||
{[]byte("e1"), []byte("val")},
|
{[]byte("e1"), []byte("val")},
|
||||||
}
|
}
|
||||||
|
|
||||||
// open a writer
|
err := batchWriteRows(s, data)
|
||||||
writer, err := s.Writer()
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// write the data
|
|
||||||
batch := writer.NewBatch()
|
|
||||||
for _, row := range data {
|
|
||||||
batch.Set(row.key, row.val)
|
|
||||||
}
|
|
||||||
err = writer.ExecuteBatch(batch)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
// close the writer
|
|
||||||
err = writer.Close()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user