// Copyright (c) 2014, Suryandaru Triandana <*****@*****.**> // All rights reserved. // // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. package iterator_test import ( . "github.com/onsi/ginkgo" . "github.com/maybebtc/interplanetary/Godeps/_workspace/src/github.com/syndtr/goleveldb/leveldb/iterator" "github.com/maybebtc/interplanetary/Godeps/_workspace/src/github.com/syndtr/goleveldb/leveldb/testutil" ) var _ = testutil.Defer(func() { Describe("Array iterator", func() { It("Should iterates and seeks correctly", func() { // Build key/value. kv := testutil.KeyValue_Generate(nil, 70, 1, 5, 3, 3) // Test the iterator. t := testutil.IteratorTesting{ KeyValue: kv.Clone(), Iter: NewArrayIterator(kv), } testutil.DoIteratorTesting(&t) }) }) })
"github.com/maybebtc/interplanetary/Godeps/_workspace/src/github.com/syndtr/goleveldb/leveldb/comparer" . "github.com/maybebtc/interplanetary/Godeps/_workspace/src/github.com/syndtr/goleveldb/leveldb/iterator" "github.com/maybebtc/interplanetary/Godeps/_workspace/src/github.com/syndtr/goleveldb/leveldb/testutil" ) var _ = testutil.Defer(func() { Describe("Merged iterator", func() { Test := func(filled int, empty int) func() { return func() { It("Should iterates and seeks correctly", func(done Done) { rnd := testutil.NewRand() // Build key/value. filledKV := make([]testutil.KeyValue, filled) kv := testutil.KeyValue_Generate(nil, 100, 1, 10, 4, 4) kv.Iterate(func(i int, key, value []byte) { filledKV[rnd.Intn(filled)].Put(key, value) }) // Create itearators. iters := make([]Iterator, filled+empty) for i := range iters { if empty == 0 || (rnd.Int()%2 == 0 && filled > 0) { filled-- Expect(filledKV[filled].Len()).ShouldNot(BeZero()) iters[i] = NewArrayIterator(filledKV[filled]) } else { empty-- iters[i] = NewEmptyIterator(nil) }
kv.Iterate(func(i int, key, value []byte) { tw.Append(key, value) }) tw.Close() // Opening the table. tr := NewReader(bytes.NewReader(buf.Bytes()), int64(buf.Len()), nil, nil, o) return tableWrapper{tr} } Test := func(kv *testutil.KeyValue, body func(r *Reader)) func() { return func() { db := Build(*kv) if body != nil { body(db.(tableWrapper).Reader) } testutil.KeyValueTesting(nil, db, *kv) } } testutil.AllKeyValueTesting(nil, Build) Describe("with one key per block", Test(testutil.KeyValue_Generate(nil, 9, 1, 10, 512, 512), func(r *Reader) { It("should have correct blocks number", func() { indexBlock, err := r.readBlock(r.indexBH, true) Expect(err).To(BeNil()) Expect(indexBlock.restartsLen).Should(Equal(9)) }) })) }) }) })
func (p *DB) TestGet(key []byte) (value []byte, err error) { return p.Get(key) } func (p *DB) TestNewIterator(slice *util.Range) iterator.Iterator { return p.NewIterator(slice) } var _ = testutil.Defer(func() { Describe("Memdb", func() { Describe("write test", func() { It("should do write correctly", func() { db := New(comparer.DefaultComparer, 0) t := testutil.DBTesting{ DB: db, Deleted: testutil.KeyValue_Generate(nil, 1000, 1, 30, 5, 5).Clone(), PostFn: func(t *testutil.DBTesting) { Expect(db.Len()).Should(Equal(t.Present.Len())) Expect(db.Size()).Should(Equal(t.Present.Size())) switch t.Act { case testutil.DBPut, testutil.DBOverwrite: Expect(db.Contains(t.ActKey)).Should(BeTrue()) default: Expect(db.Contains(t.ActKey)).Should(BeFalse()) } }, } testutil.DoDBTesting(&t) }) })