2019-09-03 18:02:21 +00:00
|
|
|
# beacon_chain
|
|
|
|
# Copyright (c) 2018-Present Status Research & Development GmbH
|
|
|
|
# Licensed and distributed under either of
|
2019-11-25 15:30:02 +00:00
|
|
|
# * 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).
|
2019-09-03 18:02:21 +00:00
|
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
|
2019-11-14 10:47:55 +00:00
|
|
|
{.used.}
|
|
|
|
|
2019-09-03 18:02:21 +00:00
|
|
|
import
|
|
|
|
# Standard library
|
2020-07-30 19:18:17 +00:00
|
|
|
os, sequtils, unittest, chronicles,
|
2019-09-03 18:02:21 +00:00
|
|
|
# Beacon chain internals
|
2020-07-07 23:02:14 +00:00
|
|
|
../../beacon_chain/spec/[crypto, datatypes, state_transition, presets],
|
2020-06-23 13:54:24 +00:00
|
|
|
../../beacon_chain/ssz,
|
2019-09-03 18:02:21 +00:00
|
|
|
# Test utilities
|
2020-12-22 15:54:49 +00:00
|
|
|
../testreportutils,
|
2020-04-20 17:27:52 +00:00
|
|
|
./fixtures_utils
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
const
|
|
|
|
FinalityDir = SszTestsDir/const_preset/"phase0"/"finality"/"finality"/"pyspec_tests"
|
|
|
|
SanityBlocksDir = SszTestsDir/const_preset/"phase0"/"sanity"/"blocks"/"pyspec_tests"
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
proc runTest(testName, testDir, unitTestName: string) =
|
2019-09-03 18:02:21 +00:00
|
|
|
# We wrap the tests in a proc to avoid running out of globals
|
|
|
|
# in the future: Nim supports up to 3500 globals
|
|
|
|
# but unittest with the macro/templates put everything as globals
|
|
|
|
# https://github.com/nim-lang/Nim/issues/12084#issue-486866402
|
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
let testPath = testDir / unitTestName
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
proc `testImpl _ blck _ testName`() =
|
|
|
|
let
|
|
|
|
hasPostState = existsFile(testPath/"post.ssz")
|
|
|
|
prefix = if hasPostState: "[Valid] " else: "[Invalid] "
|
2020-02-07 07:11:26 +00:00
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
timedTest prefix & testName & " - " & unitTestName & preset():
|
2020-04-30 16:27:17 +00:00
|
|
|
var
|
2020-08-17 07:19:48 +00:00
|
|
|
preState = newClone(parseTest(testPath/"pre.ssz", SSZ, BeaconState))
|
2020-05-28 08:28:14 +00:00
|
|
|
hashedPreState = (ref HashedBeaconState)(
|
2020-04-30 16:27:17 +00:00
|
|
|
data: preState[], root: hash_tree_root(preState[]))
|
2020-11-12 19:24:07 +00:00
|
|
|
cache = StateCache()
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-02-10 12:56:53 +00:00
|
|
|
# In test cases with more than 10 blocks the first 10 aren't 0-prefixed,
|
|
|
|
# so purely lexicographic sorting wouldn't sort properly.
|
2020-08-17 07:19:48 +00:00
|
|
|
let numBlocks = toSeq(walkPattern(testPath/"blocks_*.ssz")).len
|
2020-05-19 14:37:29 +00:00
|
|
|
for i in 0 ..< numBlocks:
|
2020-12-22 15:54:49 +00:00
|
|
|
let blck = parseTest(testPath/"blocks_" & $i & ".ssz", SSZ, SignedBeaconBlock[Unchecked])
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-04-22 23:35:55 +00:00
|
|
|
if hasPostState:
|
2020-04-26 19:13:33 +00:00
|
|
|
let success = state_transition(
|
2020-11-12 19:24:07 +00:00
|
|
|
defaultRuntimePreset, hashedPreState[], blck, cache, flags = {},
|
|
|
|
noRollback)
|
2020-02-07 07:11:26 +00:00
|
|
|
doAssert success, "Failure when applying block " & $i
|
2020-04-22 23:35:55 +00:00
|
|
|
else:
|
2020-04-26 19:13:33 +00:00
|
|
|
let success = state_transition(
|
2020-11-12 19:24:07 +00:00
|
|
|
defaultRuntimePreset, hashedPreState[], blck, cache, flags = {},
|
|
|
|
noRollback)
|
2020-05-19 14:37:29 +00:00
|
|
|
doAssert (i + 1 < numBlocks) or not success,
|
|
|
|
"We didn't expect these invalid blocks to be processed"
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-04-22 23:35:55 +00:00
|
|
|
if hasPostState:
|
2020-08-17 07:19:48 +00:00
|
|
|
let postState = newClone(parseTest(testPath/"post.ssz", SSZ, BeaconState))
|
2020-04-17 22:09:34 +00:00
|
|
|
when false:
|
2020-04-30 16:27:17 +00:00
|
|
|
reportDiff(hashedPreState.data, postState)
|
|
|
|
doAssert hashedPreState.root == postState[].hash_tree_root()
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-08-17 07:19:48 +00:00
|
|
|
`testImpl _ blck _ testName`()
|
2019-09-03 18:02:21 +00:00
|
|
|
|
2020-03-10 04:00:19 +00:00
|
|
|
suiteReport "Official - Sanity - Blocks " & preset():
|
2020-02-10 12:56:53 +00:00
|
|
|
for kind, path in walkDir(SanityBlocksDir, true):
|
2020-08-17 07:19:48 +00:00
|
|
|
runTest("Official - Sanity - Blocks", SanityBlocksDir, path)
|
|
|
|
|
|
|
|
suiteReport "Official - Finality " & preset():
|
|
|
|
for kind, path in walkDir(FinalityDir, true):
|
|
|
|
runTest("Official - Finality", FinalityDir, path)
|