110 lines
3.3 KiB
Nim
110 lines
3.3 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, tables, std/monotimes, stew/endians2,
|
|
testutils/markdown_reports, chronicles,
|
|
../beacon_chain/[beacon_chain_db, block_pool, extras, ssz, beacon_node_types],
|
|
../beacon_chain/spec/[digest, beaconstate, datatypes],
|
|
eth/db/kvstore,
|
|
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]
|
|
var status = initOrderedTable[string, OrderedTable[string, Status]]()
|
|
var last: string
|
|
|
|
proc summarizeLongTests*(name: string) =
|
|
# 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
|
|
|
|
status.sort do (a: (string, OrderedTable[string, Status]),
|
|
b: (string, OrderedTable[string, Status])) -> int: cmp(a[0], b[0])
|
|
|
|
generateReport(name & "-" & const_preset, status, width=90)
|
|
|
|
template suiteReport*(name, body) =
|
|
last = name
|
|
status[last] = initOrderedTable[string, Status]()
|
|
suite name:
|
|
body
|
|
|
|
template timedTest*(name, body) =
|
|
var f: float
|
|
test name:
|
|
status[last][name] = Status.Fail
|
|
|
|
withTimer f:
|
|
body
|
|
|
|
status[last][name] = case testStatusIMPL
|
|
of OK: Status.OK
|
|
of FAILED: Status.Fail
|
|
of SKIPPED: Status.Skip
|
|
|
|
# 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, kvStore MemStoreRef.init())
|
|
BlockPool.preInit(result, tailState, tailBlock)
|
|
|
|
proc makeTestDB*(validators: int): BeaconChainDB =
|
|
let
|
|
genState = initialize_beacon_state_from_eth1(
|
|
Eth2Digest(), 0,
|
|
makeInitialDeposits(validators, flags = {skipBlsValidation}),
|
|
{skipBlsValidation})
|
|
genBlock = get_initial_beacon_block(genState[])
|
|
makeTestDB(genState[], genBlock)
|
|
|
|
export inMicroseconds
|