92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
// Copyright (c) 2014, Suryandaru Triandana <syndtr@gmail.com>
|
|
// All rights reserved.
|
|
//
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file.
|
|
|
|
package leveldb
|
|
|
|
import (
|
|
. "github.com/onsi/gomega"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb/iterator"
|
|
"github.com/syndtr/goleveldb/leveldb/opt"
|
|
"github.com/syndtr/goleveldb/leveldb/testutil"
|
|
"github.com/syndtr/goleveldb/leveldb/util"
|
|
)
|
|
|
|
type testingDB struct {
|
|
*DB
|
|
ro *opt.ReadOptions
|
|
wo *opt.WriteOptions
|
|
stor *testutil.Storage
|
|
}
|
|
|
|
func (t *testingDB) TestPut(key []byte, value []byte) error {
|
|
return t.Put(key, value, t.wo)
|
|
}
|
|
|
|
func (t *testingDB) TestDelete(key []byte) error {
|
|
return t.Delete(key, t.wo)
|
|
}
|
|
|
|
func (t *testingDB) TestGet(key []byte) (value []byte, err error) {
|
|
return t.Get(key, t.ro)
|
|
}
|
|
|
|
func (t *testingDB) TestHas(key []byte) (ret bool, err error) {
|
|
return t.Has(key, t.ro)
|
|
}
|
|
|
|
func (t *testingDB) TestNewIterator(slice *util.Range) iterator.Iterator {
|
|
return t.NewIterator(slice, t.ro)
|
|
}
|
|
|
|
func (t *testingDB) TestClose() {
|
|
err := t.Close()
|
|
ExpectWithOffset(1, err).NotTo(HaveOccurred())
|
|
err = t.stor.Close()
|
|
ExpectWithOffset(1, err).NotTo(HaveOccurred())
|
|
}
|
|
|
|
func newTestingDB(o *opt.Options, ro *opt.ReadOptions, wo *opt.WriteOptions) *testingDB {
|
|
stor := testutil.NewStorage()
|
|
db, err := Open(stor, o)
|
|
// FIXME: This may be called from outside It, which may cause panic.
|
|
Expect(err).NotTo(HaveOccurred())
|
|
return &testingDB{
|
|
DB: db,
|
|
ro: ro,
|
|
wo: wo,
|
|
stor: stor,
|
|
}
|
|
}
|
|
|
|
type testingTransaction struct {
|
|
*Transaction
|
|
ro *opt.ReadOptions
|
|
wo *opt.WriteOptions
|
|
}
|
|
|
|
func (t *testingTransaction) TestPut(key []byte, value []byte) error {
|
|
return t.Put(key, value, t.wo)
|
|
}
|
|
|
|
func (t *testingTransaction) TestDelete(key []byte) error {
|
|
return t.Delete(key, t.wo)
|
|
}
|
|
|
|
func (t *testingTransaction) TestGet(key []byte) (value []byte, err error) {
|
|
return t.Get(key, t.ro)
|
|
}
|
|
|
|
func (t *testingTransaction) TestHas(key []byte) (ret bool, err error) {
|
|
return t.Has(key, t.ro)
|
|
}
|
|
|
|
func (t *testingTransaction) TestNewIterator(slice *util.Range) iterator.Iterator {
|
|
return t.NewIterator(slice, t.ro)
|
|
}
|
|
|
|
func (t *testingTransaction) TestClose() {}
|