88 lines
2.6 KiB
Nim
88 lines
2.6 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-2019 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.
|
|
|
|
import
|
|
algorithm, strformat, stats, times, std/monotimes, stew/endians2,
|
|
chronicles, eth/trie/[db],
|
|
../beacon_chain/[beacon_chain_db, block_pool, extras, ssz, beacon_node_types],
|
|
../beacon_chain/spec/[digest, beaconstate, datatypes],
|
|
testblockutil
|
|
|
|
type
|
|
TestDuration = tuple[duration: float, label: string]
|
|
|
|
func preset*(): string =
|
|
" [Preset: " & const_preset & ']'
|
|
|
|
# For state_sim
|
|
template withTimer*(stats: var RunningStat, body: untyped) =
|
|
let start = getMonoTime()
|
|
|
|
block:
|
|
body
|
|
|
|
let stop = getMonoTime()
|
|
stats.push (stop - start).inMicroseconds.float / 1000000.0
|
|
|
|
template withTimer*(duration: var float, body: untyped) =
|
|
let start = getMonoTime()
|
|
|
|
block:
|
|
body
|
|
|
|
duration = (getMonoTime() - start).inMicroseconds.float / 1000000.0
|
|
|
|
# For state_sim
|
|
template withTimerRet*(stats: var RunningStat, body: untyped): untyped =
|
|
let start = getMonoTime()
|
|
let tmp = block:
|
|
body
|
|
let stop = getMonoTime()
|
|
stats.push (stop - start).inMicroseconds.float / 1000000.0
|
|
|
|
tmp
|
|
|
|
var testTimes: seq[TestDuration]
|
|
|
|
proc summarizeLongTests*() =
|
|
# TODO clean-up and make machine-readable/storable the output
|
|
# TODO this is too hard-coded and mostly a demo for using the
|
|
# timedTest wrapper template for unittest
|
|
sort(testTimes, system.cmp, SortOrder.Descending)
|
|
|
|
echo ""
|
|
echo "10 longest individual test durations"
|
|
echo "------------------------------------"
|
|
for i, item in testTimes:
|
|
echo &"{item.duration:6.2f}s for {item.label}"
|
|
if i >= 10:
|
|
break
|
|
|
|
template timedTest*(name, body) =
|
|
var f: float
|
|
test name:
|
|
withTimer f:
|
|
body
|
|
|
|
# TODO reached for a failed test; maybe defer or similar
|
|
# TODO noto thread-safe as-is
|
|
testTimes.add (f, name)
|
|
|
|
proc makeTestDB*(tailState: BeaconState, tailBlock: SignedBeaconBlock): BeaconChainDB =
|
|
result = init(BeaconChainDB, newMemoryDB())
|
|
BlockPool.preInit(result, tailState, tailBlock)
|
|
|
|
proc makeTestDB*(validators: int): BeaconChainDB =
|
|
let
|
|
genState = initialize_beacon_state_from_eth1(
|
|
Eth2Digest(), 0,
|
|
makeInitialDeposits(validators, flags = {skipValidation}), {skipValidation})
|
|
genBlock = get_initial_beacon_block(genState)
|
|
makeTestDB(genState, genBlock)
|
|
|
|
export inMicroseconds
|