gibheer/bleve
0
0
Fork 0
Code

store_test: deduplicate store initialization

Browse Source
This commit is contained in:
Patrick Mezard 2015-10-20 19:21:01 +02:00
parent 873f483804
commit da72d0c2b9
1 changed files with 36 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
index/store/test/iterator.go
View File

@ -11,12 +11,39 @@ import (
// tests around the correct behavior of iterators // tests around the correct behavior of iterators
type testRow struct {
key []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) { func CommonTestPrefixIterator(t *testing.T, s store.KVStore) {
data := []struct { data := []testRow{
key []byte
val []byte
}{
{[]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)
} }