mirror of
https://github.com/fluencelabs/tendermint
synced 2025-04-25 14:52:17 +00:00
8481c49c82
* Add CacheDB & SimpleMap * Generic memBatch; Fix cLevelDB tests * CacheWrap() for CacheDB and MemDB * Change Iterator to match LeviGo Iterator * Fixes from review * cacheWrapWriteMutex and some race fixes * Use tmlibs/common * NewCWWMutex is exposed. DB can be CacheWrap'd * Remove GetOK, not needed * Fsdb (#72) * Add FSDB * Review fixes from Anton * Review changes * Fixes from review
49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package db
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestMemDbIterator(t *testing.T) {
|
|
db := NewMemDB()
|
|
keys := make([][]byte, 100)
|
|
for i := 0; i < 100; i++ {
|
|
keys[i] = []byte{byte(i)}
|
|
}
|
|
|
|
value := []byte{5}
|
|
for _, k := range keys {
|
|
db.Set(k, value)
|
|
}
|
|
|
|
iter := db.Iterator()
|
|
i := 0
|
|
for ; iter.Valid(); iter.Next() {
|
|
assert.Equal(t, db.Get(iter.Key()), iter.Value(), "values dont match for key")
|
|
i += 1
|
|
}
|
|
assert.Equal(t, i, len(db.db), "iterator didnt cover whole db")
|
|
}
|
|
|
|
func TestMemDBClose(t *testing.T) {
|
|
db := NewMemDB()
|
|
copyDB := func(orig map[string][]byte) map[string][]byte {
|
|
copy := make(map[string][]byte)
|
|
for k, v := range orig {
|
|
copy[k] = v
|
|
}
|
|
return copy
|
|
}
|
|
k, v := []byte("foo"), []byte("bar")
|
|
db.Set(k, v)
|
|
require.Equal(t, db.Get(k), v, "expecting a successful get")
|
|
copyBefore := copyDB(db.db)
|
|
db.Close()
|
|
require.Equal(t, db.Get(k), v, "Close is a noop, expecting a successful get")
|
|
copyAfter := copyDB(db.db)
|
|
require.Equal(t, copyBefore, copyAfter, "Close is a noop and shouldn't modify any internal data")
|
|
}
|