def test_get_state_by_slot(valid_chain, genesis_block, genesis_state, keymap): # First, skip block and check if `get_state_by_slot` returns the expected state state_machine = valid_chain.get_state_machine(genesis_block.slot) config = state_machine.config state = valid_chain.get_head_state() block_skipped_slot = genesis_block.slot + 1 block_skipped_state, _ = state_machine.apply_state_transition( state, future_slot=block_skipped_slot) valid_chain.chaindb.persist_state(block_skipped_state) with pytest.raises(StateNotFound): valid_chain.get_state_by_slot(block_skipped_slot) # Next, import proposed block and check if `get_state_by_slot` returns the expected state proposed_slot = block_skipped_slot + 1 future_state, _ = state_machine.apply_state_transition( block_skipped_state, future_slot=proposed_slot) proposer_index = get_beacon_proposer_index(future_state, config) proposer = block_skipped_state.validators[proposer_index].pubkey private_key = keymap[proposer] randao_reveal = generate_randao_reveal(private_key, proposed_slot, block_skipped_state, config) block = create_block( proposed_slot, genesis_block.message.hash_tree_root, randao_reveal, state.eth1_data, (), block_skipped_state, state_machine, private_key, ) valid_chain.import_block(block) state = valid_chain.get_head_state() assert (valid_chain.get_state_by_slot(proposed_slot).hash_tree_root == state.hash_tree_root)
def test_import_blocks(valid_chain, genesis_block, genesis_state, config, keymap): state = genesis_state blocks = (genesis_block, ) valid_chain_2 = copy.deepcopy(valid_chain) for _ in range(3): state_machine = valid_chain.get_state_machine(blocks[-1].slot) parent_block = blocks[-1] slot = state.slot + 2 future_state, _ = state_machine.apply_state_transition( state, future_slot=state.slot + 2) proposer_index = get_beacon_proposer_index(future_state, config) proposer = state.validators[proposer_index].pubkey private_key = keymap[proposer] randao_reveal = generate_randao_reveal(private_key, slot, state, config) block = create_block( slot, parent_block.message.hash_tree_root, randao_reveal, state.eth1_data, (), state, state_machine, private_key, ) valid_chain.import_block(block) assert valid_chain.get_canonical_head() == block state = valid_chain.get_head_state() assert block == valid_chain.get_canonical_block_by_slot(block.slot) assert block.message.hash_tree_root == valid_chain.get_canonical_block_root( block.slot) blocks += (block, ) assert valid_chain.get_canonical_head( ) != valid_chain_2.get_canonical_head() for block in blocks[1:]: valid_chain_2.import_block(block) assert valid_chain.get_canonical_head( ) == valid_chain_2.get_canonical_head() assert valid_chain.get_head_state().slot != 0 assert valid_chain.get_head_state() == valid_chain_2.get_head_state()
def _build_chain_of_blocks_with_states( chain, state, parent_block, slots, config, keymap, attestation_participation=1.0, eth1_block_hash=ZERO_HASH32, ): blocks = () states = () for slot in range(parent_block.slot + 1, parent_block.slot + 1 + slots): sm = chain.get_state_machine(state.slot) pre_state, _ = sm.apply_state_transition(state, future_slot=slot) proposer_index = get_beacon_proposer_index(pre_state, config) public_key = state.validators[proposer_index].pubkey private_key = keymap[public_key] randao_reveal = generate_randao_reveal(private_key, slot, pre_state, config) attestations = create_mock_signed_attestations_at_slot( state, config, sm, slot - 1, parent_block.hash_tree_root, keymap, voted_attesters_ratio=attestation_participation, ) block = create_block( slot, parent_block.hash_tree_root, randao_reveal, Eth1Data.create(block_hash=eth1_block_hash), attestations, state, sm, private_key, ) parent_block = block.message state, block = sm.apply_state_transition(state, block) blocks += (block, ) states += (state, ) return blocks, states
def test_get_attestation_root(valid_chain, genesis_block, genesis_state, keymap, min_attestation_inclusion_delay): state_machine = valid_chain.get_state_machine() config = state_machine.config attestations = create_mock_signed_attestations_at_slot( state=genesis_state, config=config, state_machine=state_machine, attestation_slot=genesis_block.slot, beacon_block_root=genesis_block.message.hash_tree_root, keymap=keymap, ) state = genesis_state slot = genesis_state.slot + 1 future_state, _ = state_machine.apply_state_transition(state, future_slot=slot) proposer_index = get_beacon_proposer_index(future_state, config) proposer = state.validators[proposer_index].pubkey private_key = keymap[proposer] randao_reveal = generate_randao_reveal(private_key, slot, state, config) block = create_block( slot, genesis_block.message.hash_tree_root, randao_reveal, state.eth1_data, attestations, state, state_machine, private_key, ) valid_chain.import_block(block) # Only one attestation in attestations, so just check that one a0 = attestations[0] assert valid_chain.get_attestation_by_root(a0.hash_tree_root) == a0 assert valid_chain.attestation_exists(a0.hash_tree_root) fake_attestation = a0.set("signature", b"\x78" * 96) with pytest.raises(AttestationRootNotFound): valid_chain.get_attestation_by_root(fake_attestation.hash_tree_root) assert not valid_chain.attestation_exists(fake_attestation.hash_tree_root)