mirror of
https://github.com/status-im/nimbus-eth2.git
synced 2025-01-24 13:33:14 +00:00
54 lines
1.6 KiB
Nim
54 lines
1.6 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import options, unittest, strutils, eth/trie/[db],
|
|
../beacon_chain/[beacon_chain_db, ssz],
|
|
../beacon_chain/spec/[datatypes, digest, crypto]
|
|
|
|
suite "Beacon chain DB":
|
|
var
|
|
db = init(BeaconChainDB, newMemoryDB())
|
|
|
|
test "empty database":
|
|
check:
|
|
db.getState(Eth2Digest()).isNone
|
|
db.getBlock(Eth2Digest()).isNone
|
|
|
|
test "find ancestors":
|
|
var x: ValidatorSig
|
|
var y = init(ValidatorSig, x.getBytes())
|
|
|
|
check: x == y
|
|
|
|
let
|
|
a0 = BeaconBlock(slot: 0)
|
|
a1 = BeaconBlock(slot: 1, parent_root: hash_tree_root_final(a0))
|
|
a2 = BeaconBlock(slot: 2, parent_root: hash_tree_root_final(a1))
|
|
|
|
# TODO check completely kills compile times here
|
|
doAssert db.getAncestors(a0) == [a0]
|
|
doAssert db.getAncestors(a2) == [a2]
|
|
|
|
db.putBlock(a2)
|
|
|
|
doAssert db.getAncestors(a0) == [a0]
|
|
doAssert db.getAncestors(a2) == [a2]
|
|
|
|
db.putBlock(a1)
|
|
|
|
doAssert db.getAncestors(a0) == [a0]
|
|
doAssert db.getAncestors(a2) == [a2, a1]
|
|
|
|
db.putBlock(a0)
|
|
|
|
doAssert db.getAncestors(a0) == [a0]
|
|
doAssert db.getAncestors(a2) == [a2, a1, a0]
|
|
|
|
let tmp = db.getAncestors(a2) do (b: BeaconBlock) -> bool:
|
|
b.slot == 1
|
|
doAssert tmp == [a2, a1]
|