nimbus-eth1/fluffy/tests/test_content_db.nim

88 lines
2.0 KiB
Nim

# Nimbus
# Copyright (c) 2021 Status Research & Development GmbH
# Licensed and distributed under either of
# * MIT license (license terms in the root directory or at https://opensource.org/licenses/MIT).
# * Apache v2 license (license terms in the root directory or at https://www.apache.org/licenses/LICENSE-2.0).
# at your option. This file may not be copied, modified, or distributed except according to those terms.
{.used.}
import
unittest2, stint,
../network/state/state_content,
../content_db
proc genByteSeq(length: int): seq[byte] =
var i = 0
var resultSeq = newSeq[byte](length)
while i < length:
resultSeq[i] = byte(i)
inc i
return resultSeq
suite "Content Database":
# Note: We are currently not really testing something new here just basic
# underlying kvstore.
test "ContentDB basic API":
let
db = ContentDB.new("", inMemory = true)
key = ContentId(UInt256.high()) # Some key
block:
let val = db.get(key)
check:
val.isNone()
db.contains(key) == false
block:
db.put(key, [byte 0, 1, 2, 3])
let val = db.get(key)
check:
val.isSome()
val.get() == [byte 0, 1, 2, 3]
db.contains(key) == true
block:
db.del(key)
let val = db.get(key)
check:
val.isNone()
db.contains(key) == false
test "ContentDB size":
let
db = ContentDB.new("", inMemory = true)
let numBytes = 10000
let size1 = db.size()
db.put(@[1'u8], genByteSeq(numBytes))
let size2 = db.size()
db.put(@[2'u8], genByteSeq(numBytes))
let size3 = db.size()
db.put(@[2'u8], genByteSeq(numBytes))
let size4 = db.size()
check:
size2 > size1
size3 > size2
size3 == size4
db.del(@[2'u8])
db.del(@[1'u8])
let size5 = db.size()
check:
size4 == size5
db.reclaimSpace()
let size6 = db.size()
check:
# After space reclamation size of db should be equal to initial size
size6 == size1