mirror of https://github.com/status-im/nim-eth.git
64 lines
1.4 KiB
Nim
64 lines
1.4 KiB
Nim
|
import
|
||
|
|
unittest, macros, os,
|
||
|
|
eth/trie/backends/[rocksdb_backend, sqlite_backend, lmdb_backend]
|
||
|
|
|
||
|
|
template dummyInstance(T: type SqliteChainDB): auto =
|
||
|
|
sqlite_backend.newChainDB(getTempDir(), inMemory = true)
|
||
|
|
|
||
|
|
template dummyInstance(T: type RocksChainDB): auto =
|
||
|
|
let tmp = getTempDir() / "nimbus-test-db"
|
||
|
|
removeDir(tmp)
|
||
|
|
rocksdb_backend.newChainDB(tmp)
|
||
|
|
|
||
|
|
template dummyInstance(T: type LmdbChainDB): auto =
|
||
|
|
# remove sqlite created database
|
||
|
|
let tmp = getTempDir() / "nimbus.db"
|
||
|
|
removeFile(tmp)
|
||
|
|
lmdb_backend.newChainDB(getTempDir())
|
||
|
|
|
||
|
|
template backendTests(DB) =
|
||
|
|
suite("storage tests: " & astToStr(DB)):
|
||
|
|
setup:
|
||
|
|
var db = dummyInstance(DB)
|
||
|
|
|
||
|
|
teardown:
|
||
|
|
close(db)
|
||
|
|
|
||
|
|
test "basic insertions and deletions":
|
||
|
|
var keyA = [1.byte, 2, 3]
|
||
|
|
var keyB = [1.byte, 2, 4]
|
||
|
|
var value1 = @[1.byte, 2, 3, 4, 5]
|
||
|
|
var value2 = @[7.byte, 8, 9, 10]
|
||
|
|
|
||
|
|
db.put(keyA, value1)
|
||
|
|
|
||
|
|
check:
|
||
|
|
keyA in db
|
||
|
|
keyB notin db
|
||
|
|
|
||
|
|
db.put(keyB, value2)
|
||
|
|
|
||
|
|
check:
|
||
|
|
keyA in db
|
||
|
|
keyB in db
|
||
|
|
|
||
|
|
check:
|
||
|
|
db.get(keyA) == value1
|
||
|
|
db.get(keyB) == value2
|
||
|
|
|
||
|
|
db.del(keyA)
|
||
|
|
db.put(keyB, value1)
|
||
|
|
|
||
|
|
check:
|
||
|
|
keyA notin db
|
||
|
|
keyB in db
|
||
|
|
|
||
|
|
check db.get(keyA) == @[]
|
||
|
|
|
||
|
|
check db.get(keyB) == value1
|
||
|
|
db.del(keyA)
|
||
|
|
|
||
|
|
backendTests(RocksChainDB)
|
||
|
|
backendTests(SqliteChainDB)
|
||
|
|
backendTests(LmdbChainDB)
|