82 lines
2.5 KiB
Nim
82 lines
2.5 KiB
Nim
# beacon_chain
|
|
# Copyright (c) 2018-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.
|
|
|
|
{.push raises: [Defect].}
|
|
{.used.}
|
|
|
|
import
|
|
std/[algorithm, strformat, stats, tables, times],
|
|
testutils/markdown_reports,
|
|
unittest2,
|
|
../beacon_chain/spec/presets
|
|
|
|
export unittest2
|
|
|
|
type
|
|
TestDuration = tuple[duration: float, label: string]
|
|
|
|
func preset*(): string =
|
|
" [Preset: " & const_preset & ']'
|
|
|
|
# For state_sim
|
|
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]]()
|
|
|
|
type TimingCollector = ref object of OutputFormatter
|
|
|
|
func toFloatSeconds(duration: Duration): float =
|
|
duration.inNanoseconds().float / 1_000_000_000.0
|
|
|
|
method testEnded*(formatter: TimingCollector, testResult: TestResult) =
|
|
{.gcsafe.}: # Lie!
|
|
status.mgetOrPut(testResult.suiteName, initOrderedTable[string, Status]())[testResult.testName] =
|
|
case testResult.status
|
|
of TestStatus.OK: Status.OK
|
|
of TestStatus.FAILED: Status.Fail
|
|
of TestStatus.SKIPPED: Status.Skip
|
|
testTimes.add (testResult.duration.toFloatSeconds, testResult.testName)
|
|
|
|
proc summarizeLongTests*(name: string) =
|
|
# TODO clean-up and make machine-readable/storable the output
|
|
sort(testTimes, system.cmp, SortOrder.Descending)
|
|
|
|
try:
|
|
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)
|
|
except CatchableError as exc:
|
|
raiseAssert exc.msg
|
|
|
|
addOutputFormatter(new TimingCollector)
|