def test_transfer(state): pre_state = deepcopy(state) current_epoch = get_current_epoch(pre_state) sender_index = get_active_validator_indices(pre_state.validator_registry, current_epoch)[-1] recipient_index = get_active_validator_indices( pre_state.validator_registry, current_epoch)[0] transfer_pubkey = pubkeys[-1] transfer_privkey = pubkey_to_privkey[transfer_pubkey] amount = pre_state.validator_balances[sender_index] pre_transfer_recipient_balance = pre_state.validator_balances[ recipient_index] transfer = Transfer( sender=sender_index, recipient=recipient_index, amount=amount, fee=0, slot=pre_state.slot + 1, pubkey=transfer_pubkey, signature=b'\x00' * 96, ) transfer.signature = bls.sign(message_hash=signed_root(transfer), privkey=transfer_privkey, domain=get_domain( fork=pre_state.fork, epoch=get_current_epoch(pre_state), domain_type=spec.DOMAIN_TRANSFER, )) # ensure withdrawal_credentials reproducable pre_state.validator_registry[sender_index].withdrawal_credentials = ( spec.BLS_WITHDRAWAL_PREFIX_BYTE + hash(transfer_pubkey)[1:]) # un-activate so validator can transfer pre_state.validator_registry[ sender_index].activation_epoch = spec.FAR_FUTURE_EPOCH post_state = deepcopy(pre_state) # # Add to state via block transition # block = construct_empty_block_for_next_slot(post_state) block.body.transfers.append(transfer) state_transition(post_state, block) sender_balance = post_state.validator_balances[sender_index] recipient_balance = post_state.validator_balances[recipient_index] assert sender_balance == 0 assert recipient_balance == pre_transfer_recipient_balance + amount return pre_state, [block], post_state
def test_ejection(state): pre_state = deepcopy(state) current_epoch = get_current_epoch(pre_state) validator_index = get_active_validator_indices( pre_state.validator_registry, current_epoch)[-1] assert pre_state.validator_registry[ validator_index].exit_epoch == spec.FAR_FUTURE_EPOCH # set validator balance to below ejection threshold pre_state.validator_balances[validator_index] = spec.EJECTION_BALANCE - 1 post_state = deepcopy(pre_state) # # trigger epoch transition # block = construct_empty_block_for_next_slot(post_state) block.slot += spec.SLOTS_PER_EPOCH state_transition(post_state, block) assert post_state.validator_registry[ validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH return pre_state, [block], post_state
def test_voluntary_exit(state): test_state = deepcopy(state) current_epoch = get_current_epoch(test_state) validator_index = get_active_validator_indices(test_state.validator_registry, current_epoch)[-1] voluntary_exit = VoluntaryExit( epoch=current_epoch, validator_index=validator_index, signature=b'\x00'*96, ) # # Add to state via block transition # block = construct_empty_block_for_next_slot(test_state) block.body.voluntary_exits.append(voluntary_exit) state_transition(test_state, block) assert not state.validator_registry[validator_index].initiated_exit assert test_state.validator_registry[validator_index].initiated_exit assert test_state.validator_registry[validator_index].exit_epoch == FAR_FUTURE_EPOCH # # Process within epoch transition # # artificially trigger registry update test_state.validator_registry_update_epoch -= 1 block = construct_empty_block_for_next_slot(test_state) block.slot += SLOTS_PER_EPOCH state_transition(test_state, block) assert test_state.validator_registry[validator_index].exit_epoch < FAR_FUTURE_EPOCH
def test_proposer_slashing(state): test_state = deepcopy(state) current_epoch = get_current_epoch(test_state) validator_index = get_active_validator_indices( test_state.validator_registry, current_epoch)[-1] pubkey = pubkeys[validator_index] privkey = pubkey_to_privkey[pubkey] slot = spec.GENESIS_SLOT header_1 = BeaconBlockHeader(slot=slot, previous_block_root=b'\x00' * 32, state_root=b'\x00' * 32, block_body_root=b'\x00' * 32, signature=b'\x00' * 96) header_2 = deepcopy(header_1) header_2.previous_block_root = b'\x02' * 32 header_2.slot = slot + 1 domain = get_domain( fork=test_state.fork, epoch=get_current_epoch(test_state), domain_type=spec.DOMAIN_BEACON_BLOCK, ) header_1.signature = bls.sign( message_hash=signed_root(header_1), privkey=privkey, domain=domain, ) header_2.signature = bls.sign( message_hash=signed_root(header_2), privkey=privkey, domain=domain, ) proposer_slashing = ProposerSlashing( proposer_index=validator_index, header_1=header_1, header_2=header_2, ) # # Add to state via block transition # block = construct_empty_block_for_next_slot(test_state) block.body.proposer_slashings.append(proposer_slashing) state_transition(test_state, block) assert not state.validator_registry[validator_index].initiated_exit assert not state.validator_registry[validator_index].slashed slashed_validator = test_state.validator_registry[validator_index] assert not slashed_validator.initiated_exit assert slashed_validator.slashed assert slashed_validator.exit_epoch < spec.FAR_FUTURE_EPOCH assert slashed_validator.withdrawable_epoch < spec.FAR_FUTURE_EPOCH # lost whistleblower reward assert test_state.validator_balances[ validator_index] < state.validator_balances[validator_index] return state, [block], test_state
def test_transfer(state): test_state = deepcopy(state) current_epoch = get_current_epoch(test_state) sender_index = get_active_validator_indices(test_state.validator_registry, current_epoch)[-1] recipient_index = get_active_validator_indices( test_state.validator_registry, current_epoch)[0] pubkey = b'\x00' * 48 amount = test_state.validator_balances[sender_index] pre_transfer_recipient_balance = test_state.validator_balances[ recipient_index] transfer = Transfer( sender=sender_index, recipient=recipient_index, amount=amount, fee=0, slot=test_state.slot + 1, pubkey=pubkey, signature=b'\x42' * 96, ) # ensure withdrawal_credentials reproducable test_state.validator_registry[sender_index].withdrawal_credentials = ( BLS_WITHDRAWAL_PREFIX_BYTE + hash(pubkey)[1:]) # un-activate so validator can transfer test_state.validator_registry[ sender_index].activation_epoch = FAR_FUTURE_EPOCH # # Add to state via block transition # block = construct_empty_block_for_next_slot(test_state) block.body.transfers.append(transfer) state_transition(test_state, block) sender_balance = test_state.validator_balances[sender_index] recipient_balance = test_state.validator_balances[recipient_index] assert sender_balance == 0 assert recipient_balance == pre_transfer_recipient_balance + amount
def test_voluntary_exit(state): pre_state = deepcopy(state) validator_index = get_active_validator_indices( pre_state.validator_registry, get_current_epoch(pre_state))[-1] pubkey = pubkeys[validator_index] # move state forward PERSISTENT_COMMITTEE_PERIOD epochs to allow for exit pre_state.slot += spec.PERSISTENT_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH # artificially trigger registry update at next epoch transition pre_state.validator_registry_update_epoch -= 1 post_state = deepcopy(pre_state) voluntary_exit = VoluntaryExit( epoch=get_current_epoch(pre_state), validator_index=validator_index, signature=b'\x00' * 96, ) voluntary_exit.signature = bls.sign( message_hash=signed_root(voluntary_exit), privkey=pubkey_to_privkey[pubkey], domain=get_domain( fork=pre_state.fork, epoch=get_current_epoch(pre_state), domain_type=spec.DOMAIN_VOLUNTARY_EXIT, )) # # Add to state via block transition # initiate_exit_block = construct_empty_block_for_next_slot(post_state) initiate_exit_block.body.voluntary_exits.append(voluntary_exit) state_transition(post_state, initiate_exit_block) assert not pre_state.validator_registry[validator_index].initiated_exit assert post_state.validator_registry[validator_index].initiated_exit assert post_state.validator_registry[ validator_index].exit_epoch == spec.FAR_FUTURE_EPOCH # # Process within epoch transition # exit_block = construct_empty_block_for_next_slot(post_state) exit_block.slot += spec.SLOTS_PER_EPOCH state_transition(post_state, exit_block) assert post_state.validator_registry[ validator_index].exit_epoch < spec.FAR_FUTURE_EPOCH return pre_state, [initiate_exit_block, exit_block], post_state
def test_ejection(state): test_state = deepcopy(state) current_epoch = get_current_epoch(test_state) validator_index = get_active_validator_indices(test_state.validator_registry, current_epoch)[-1] assert test_state.validator_registry[validator_index].exit_epoch == FAR_FUTURE_EPOCH # set validator balance to below ejection threshold test_state.validator_balances[validator_index] = EJECTION_BALANCE - 1 # # trigger epoch transition # block = construct_empty_block_for_next_slot(test_state) block.slot += SLOTS_PER_EPOCH state_transition(test_state, block) assert test_state.validator_registry[validator_index].exit_epoch < FAR_FUTURE_EPOCH
def test_proposer_slashing(state): test_state = deepcopy(state) current_epoch = get_current_epoch(test_state) validator_index = get_active_validator_indices(test_state.validator_registry, current_epoch)[-1] slot = GENESIS_SLOT header_1 = BeaconBlockHeader( slot=slot, previous_block_root=b'\x00'*32, state_root=b'\x00'*32, block_body_root=b'\x00'*32, signature=b'\x00'*96 ) header_2 = deepcopy(header_1) header_2.previous_block_root = b'\x02'*32 proposer_slashing = ProposerSlashing( proposer_index=validator_index, header_1=header_1, header_2=header_2, ) # # Add to state via block transition # block = construct_empty_block_for_next_slot(test_state) block.body.proposer_slashings.append(proposer_slashing) state_transition(test_state, block) assert not state.validator_registry[validator_index].initiated_exit assert not state.validator_registry[validator_index].slashed slashed_validator = test_state.validator_registry[validator_index] assert not slashed_validator.initiated_exit assert slashed_validator.slashed assert slashed_validator.exit_epoch < FAR_FUTURE_EPOCH assert slashed_validator.withdrawable_epoch < FAR_FUTURE_EPOCH # lost whistleblower reward assert test_state.validator_balances[validator_index] < state.validator_balances[validator_index]