forked from status-im/nimbus-eth2
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_spec.nim
132 lines (110 loc) · 5.05 KB
/
test_spec.nim
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
# beacon_chain
# Copyright (c) 2018-2022 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.
{.used.}
# Test for spec functions and helpers outside of the EF test vectors - mainly
# helpers that extend or make the spec functions usable outside of the state
# transition functions
import
unittest2,
../beacon_chain/spec/datatypes/phase0,
../beacon_chain/spec/[beaconstate, state_transition],
./testutil, ./testblockutil
suite "Beacon state" & preset():
setup:
let cfg = defaultRuntimeConfig
test "Smoke test initialize_beacon_state_from_eth1" & preset():
let state = newClone(initialize_beacon_state_from_eth1(
cfg, ZERO_HASH, 0, makeInitialDeposits(SLOTS_PER_EPOCH, {}), {}))
check: state.validators.lenu64 == SLOTS_PER_EPOCH
test "process_slots":
var
state = (ref ForkedHashedBeaconState)(
kind: BeaconStateFork.Phase0,
phase0Data: initialize_hashed_beacon_state_from_eth1(
defaultRuntimeConfig, ZERO_HASH, 0,
makeInitialDeposits(SLOTS_PER_EPOCH, {}), {skipBlsValidation}))
cache: StateCache
info: ForkedEpochInfo
check:
process_slots(cfg, state[], Slot 1, cache, info, {}).isOk()
process_slots(cfg, state[], Slot 1, cache, info, {}).isErr()
process_slots(cfg, state[], Slot 1, cache, info, {slotProcessed}).isOk()
test "latest_block_root":
var
state = (ref ForkedHashedBeaconState)(
kind: BeaconStateFork.Phase0,
phase0Data: initialize_hashed_beacon_state_from_eth1(
defaultRuntimeConfig, ZERO_HASH, 0,
makeInitialDeposits(SLOTS_PER_EPOCH, {}), {skipBlsValidation}))
genBlock = get_initial_beacon_block(state[])
cache: StateCache
info: ForkedEpochInfo
check: # Works for genesis block
state[].phase0Data.latest_block_root == genBlock.root
state[].phase0Data.latest_block_id == genBlock.toBlockId()
process_slots(cfg, state[], Slot 1, cache, info, {}).isOk()
state[].phase0Data.latest_block_root == genBlock.root
let blck = addTestBlock(
state[], cache, nextSlot = false, flags = {skipBlsValidation}).phase0Data
check: # Works for random blocks
state[].phase0Data.latest_block_root == blck.root
process_slots(cfg, state[], Slot 2, cache, info, {}).isOk()
state[].phase0Data.latest_block_root == blck.root
test "get_beacon_proposer_index":
var
state = (ref ForkedHashedBeaconState)(
kind: BeaconStateFork.Phase0,
phase0Data: initialize_hashed_beacon_state_from_eth1(
defaultRuntimeConfig, ZERO_HASH, 0,
makeInitialDeposits(SLOTS_PER_EPOCH, {}), {skipBlsValidation}))
cache: StateCache
info: ForkedEpochInfo
check:
get_beacon_proposer_index(state[].phase0Data.data, cache, Slot 1).isSome()
get_beacon_proposer_index(
state[].phase0Data.data, cache, Epoch(1).start_slot()).isNone()
get_beacon_proposer_index(
state[].phase0Data.data, cache, Epoch(2).start_slot()).isNone()
check:
process_slots(cfg, state[], Epoch(1).start_slot(), cache, info, {}).isOk()
get_beacon_proposer_index(state[].phase0Data.data, cache, Slot 1).isNone()
get_beacon_proposer_index(
state[].phase0Data.data, cache, Epoch(1).start_slot()).isSome()
get_beacon_proposer_index(
state[].phase0Data.data, cache, Epoch(2).start_slot()).isNone()
test "dependent_root":
var
state = (ref ForkedHashedBeaconState)(
kind: BeaconStateFork.Phase0,
phase0Data: initialize_hashed_beacon_state_from_eth1(
defaultRuntimeConfig, ZERO_HASH, 0,
makeInitialDeposits(SLOTS_PER_EPOCH, {}), {skipBlsValidation}))
genBlock = get_initial_beacon_block(state[])
cache: StateCache
info: ForkedEpochInfo
check:
state[].phase0Data.dependent_root(Epoch(0)) == genBlock.root
while getStateField(state[], slot).epoch < Epoch(1):
discard addTestBlock(state[], cache)
check:
state[].phase0Data.dependent_root(Epoch(1)) ==
state[].phase0Data.data.get_block_root_at_slot(Epoch(1).start_slot - 1)
state[].phase0Data.dependent_root(Epoch(0)) == genBlock.root
while getStateField(state[], slot).epoch < Epoch(2):
discard addTestBlock(state[], cache)
check:
state[].phase0Data.dependent_root(Epoch(2)) ==
state[].phase0Data.data.get_block_root_at_slot(Epoch(2).start_slot - 1)
state[].phase0Data.dependent_root(Epoch(1)) ==
state[].phase0Data.data.get_block_root_at_slot(Epoch(1).start_slot - 1)
state[].phase0Data.dependent_root(Epoch(0)) == genBlock.root
test "merklizer state roundtrip":
let
dcs = DepositContractState()
merkleizer = DepositsMerkleizer.init(dcs)
check:
dcs == merkleizer.toDepositContractState()