139 lines
4.2 KiB
Nim
139 lines
4.2 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2018 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or https://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or https://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.used.}
|
|
|
|
import options, unittest, sequtils,
|
|
../beacon_chain/[beacon_chain_db, extras, interop, ssz],
|
|
../beacon_chain/spec/[
|
|
beaconstate, datatypes, digest, crypto, state_transition, presets],
|
|
eth/db/kvstore,
|
|
# test utilies
|
|
./testutil, ./testblockutil
|
|
|
|
proc getStateRef(db: BeaconChainDB, root: Eth2Digest): NilableBeaconStateRef =
|
|
# load beaconstate the way the block pool does it - into an existing instance
|
|
let res = BeaconStateRef()
|
|
if db.getState(root, res[], noRollback):
|
|
return res
|
|
|
|
template wrappedTimedTest(name: string, body: untyped) =
|
|
# `check` macro takes a copy of whatever it's checking, on the stack!
|
|
block: # Symbol namespacing
|
|
proc wrappedTest() =
|
|
timedTest name:
|
|
body
|
|
wrappedTest()
|
|
|
|
func withDigest(blck: TrustedBeaconBlock): TrustedSignedBeaconBlock =
|
|
TrustedSignedBeaconBlock(
|
|
message: blck,
|
|
root: hash_tree_root(blck)
|
|
)
|
|
|
|
suiteReport "Beacon chain DB" & preset():
|
|
wrappedTimedTest "empty database" & preset():
|
|
var
|
|
db = init(BeaconChainDB, kvStore MemStoreRef.init())
|
|
check:
|
|
db.getStateRef(Eth2Digest()).isNil
|
|
db.getBlock(Eth2Digest()).isNone
|
|
|
|
wrappedTimedTest "sanity check blocks" & preset():
|
|
var
|
|
db = init(BeaconChainDB, kvStore MemStoreRef.init())
|
|
|
|
let
|
|
signedBlock = withDigest(TrustedBeaconBlock())
|
|
root = hash_tree_root(signedBlock.message)
|
|
|
|
db.putBlock(signedBlock)
|
|
|
|
check:
|
|
db.containsBlock(root)
|
|
db.getBlock(root).get() == signedBlock
|
|
|
|
db.putStateRoot(root, signedBlock.message.slot, root)
|
|
var root2 = root
|
|
root2.data[0] = root.data[0] + 1
|
|
db.putStateRoot(root, signedBlock.message.slot + 1, root2)
|
|
|
|
check:
|
|
db.getStateRoot(root, signedBlock.message.slot).get() == root
|
|
db.getStateRoot(root, signedBlock.message.slot + 1).get() == root2
|
|
|
|
db.close()
|
|
|
|
wrappedTimedTest "sanity check states" & preset():
|
|
var
|
|
db = init(BeaconChainDB, kvStore MemStoreRef.init())
|
|
|
|
let
|
|
state = BeaconStateRef()
|
|
root = hash_tree_root(state[])
|
|
|
|
db.putState(state[])
|
|
|
|
check:
|
|
db.containsState(root)
|
|
hash_tree_root(db.getStateRef(root)[]) == root
|
|
|
|
db.close()
|
|
|
|
wrappedTimedTest "find ancestors" & preset():
|
|
var
|
|
db = init(BeaconChainDB, kvStore MemStoreRef.init())
|
|
|
|
let
|
|
a0 = withDigest(
|
|
TrustedBeaconBlock(slot: GENESIS_SLOT + 0))
|
|
a1 = withDigest(
|
|
TrustedBeaconBlock(slot: GENESIS_SLOT + 1, parent_root: a0.root))
|
|
a2 = withDigest(
|
|
TrustedBeaconBlock(slot: GENESIS_SLOT + 2, parent_root: a1.root))
|
|
|
|
doAssert toSeq(db.getAncestors(a0.root)) == []
|
|
doAssert toSeq(db.getAncestors(a2.root)) == []
|
|
|
|
db.putBlock(a2)
|
|
|
|
doAssert toSeq(db.getAncestors(a0.root)) == []
|
|
doAssert toSeq(db.getAncestors(a2.root)) == [a2]
|
|
|
|
db.putBlock(a1)
|
|
|
|
doAssert toSeq(db.getAncestors(a0.root)) == []
|
|
doAssert toSeq(db.getAncestors(a2.root)) == [a2, a1]
|
|
|
|
db.putBlock(a0)
|
|
|
|
doAssert toSeq(db.getAncestors(a0.root)) == [a0]
|
|
doAssert toSeq(db.getAncestors(a2.root)) == [a2, a1, a0]
|
|
|
|
wrappedTimedTest "sanity check genesis roundtrip" & preset():
|
|
# This is a really dumb way of checking that we can roundtrip a genesis
|
|
# state. We've been bit by this because we've had a bug in the BLS
|
|
# serialization where an all-zero default-initialized bls signature could
|
|
# not be deserialized because the deserialization was too strict.
|
|
var
|
|
db = init(BeaconChainDB, kvStore MemStoreRef.init())
|
|
|
|
let
|
|
state = initialize_beacon_state_from_eth1(
|
|
defaultRuntimePreset, eth1BlockHash, 0,
|
|
makeInitialDeposits(SLOTS_PER_EPOCH), {skipBlsValidation})
|
|
root = hash_tree_root(state[])
|
|
|
|
db.putState(state[])
|
|
|
|
check db.containsState(root)
|
|
let state2 = db.getStateRef(root)
|
|
db.close()
|
|
|
|
check:
|
|
hash_tree_root(state2[]) == root
|