status-im
/
eth2.0-specs
mirror of https://github.com/status-im/eth2.0-specs.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Randomize validator index in partial withdrawal test

This commit is contained in:
NC 2024-08-22 00:17:56 +08:00
parent 01aab85412
commit 4c9be8978f
1 changed files with 25 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
tests/core/pyspec/eth2spec/test/electra/block_processing/test_process_withdrawal_request.py
View File

@ -1,3 +1,4 @@
import random
from eth2spec.test.context import ( from eth2spec.test.context import (
spec_state_test, spec_state_test,
expect_assertion_error, expect_assertion_error,
@ -23,7 +24,7 @@ def test_basic_withdrawal_request(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_eth1_withdrawal_credential_with_balance( set_eth1_withdrawal_credential_with_balance(
@ -47,7 +48,7 @@ def test_basic_withdrawal_request_with_compounding_credentials(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_compounding_withdrawal_credential(spec, state, validator_index, address=address) set_compounding_withdrawal_credential(spec, state, validator_index, address=address)
@ -68,7 +69,7 @@ def test_basic_withdrawal_request_with_compounding_credentials(spec, state):
def test_basic_withdrawal_request_with_full_partial_withdrawal_queue(spec, state): def test_basic_withdrawal_request_with_full_partial_withdrawal_queue(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_eth1_withdrawal_credential_with_balance( set_eth1_withdrawal_credential_with_balance(
@ -106,7 +107,7 @@ def test_incorrect_source_address(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
incorrect_address = b"\x33" * 20 incorrect_address = b"\x33" * 20
@ -131,7 +132,7 @@ def test_incorrect_withdrawal_credential_prefix(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_eth1_withdrawal_credential_with_balance( set_eth1_withdrawal_credential_with_balance(
@ -160,7 +161,7 @@ def test_on_withdrawal_request_initiated_validator(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_eth1_withdrawal_credential_with_balance( set_eth1_withdrawal_credential_with_balance(
@ -183,7 +184,7 @@ def test_on_withdrawal_request_initiated_validator(spec, state):
@spec_state_test @spec_state_test
def test_activation_epoch_less_than_shard_committee_period(spec, state): def test_activation_epoch_less_than_shard_committee_period(spec, state):
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
set_eth1_withdrawal_credential_with_balance( set_eth1_withdrawal_credential_with_balance(
@ -213,7 +214,7 @@ def test_activation_epoch_less_than_shard_committee_period(spec, state):
def test_basic_partial_withdrawal_request(spec, state): def test_basic_partial_withdrawal_request(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -245,7 +246,7 @@ def test_basic_partial_withdrawal_request(spec, state):
def test_basic_partial_withdrawal_request_higher_excess_balance(spec, state): def test_basic_partial_withdrawal_request_higher_excess_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -277,7 +278,7 @@ def test_basic_partial_withdrawal_request_higher_excess_balance(spec, state):
def test_basic_partial_withdrawal_request_lower_than_excess_balance(spec, state): def test_basic_partial_withdrawal_request_lower_than_excess_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
excess_balance = spec.EFFECTIVE_BALANCE_INCREMENT excess_balance = spec.EFFECTIVE_BALANCE_INCREMENT
@ -310,7 +311,7 @@ def test_basic_partial_withdrawal_request_lower_than_excess_balance(spec, state)
def test_partial_withdrawal_request_with_pending_withdrawals(spec, state): def test_partial_withdrawal_request_with_pending_withdrawals(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -351,7 +352,7 @@ def test_partial_withdrawal_request_with_pending_withdrawals_and_high_amount(
): ):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.UINT64_MAX amount = spec.UINT64_MAX
@ -389,7 +390,7 @@ def test_partial_withdrawal_request_with_pending_withdrawals_and_high_amount(
def test_partial_withdrawal_request_with_high_balance(spec, state): def test_partial_withdrawal_request_with_high_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.MAX_EFFECTIVE_BALANCE_ELECTRA amount = spec.MAX_EFFECTIVE_BALANCE_ELECTRA
@ -426,7 +427,7 @@ def test_partial_withdrawal_request_with_high_balance(spec, state):
def test_partial_withdrawal_request_with_high_amount(spec, state): def test_partial_withdrawal_request_with_high_amount(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
# Set high amount requested to withdraw # Set high amount requested to withdraw
@ -459,7 +460,7 @@ def test_partial_withdrawal_request_with_high_amount(spec, state):
def test_partial_withdrawal_request_with_low_amount(spec, state): def test_partial_withdrawal_request_with_low_amount(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = 1 amount = 1
@ -494,7 +495,7 @@ def test_partial_withdrawal_request_with_low_amount(spec, state):
def test_partial_withdrawal_queue_full(spec, state): def test_partial_withdrawal_queue_full(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -524,7 +525,7 @@ def test_partial_withdrawal_queue_full(spec, state):
def test_no_compounding_credentials(spec, state): def test_no_compounding_credentials(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -553,7 +554,7 @@ def test_no_compounding_credentials(spec, state):
def test_no_excess_balance(spec, state): def test_no_excess_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -575,7 +576,7 @@ def test_no_excess_balance(spec, state):
def test_pending_withdrawals_consume_all_excess_balance(spec, state): def test_pending_withdrawals_consume_all_excess_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -605,7 +606,7 @@ def test_pending_withdrawals_consume_all_excess_balance(spec, state):
def test_insufficient_effective_balance(spec, state): def test_insufficient_effective_balance(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -636,7 +637,7 @@ def test_partial_withdrawal_incorrect_source_address(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
incorrect_address = b"\x33" * 20 incorrect_address = b"\x33" * 20
@ -662,7 +663,7 @@ def test_partial_withdrawal_incorrect_withdrawal_credential_prefix(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -691,7 +692,7 @@ def test_partial_withdrawal_on_exit_initiated_validator(spec, state):
state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH state.slot += spec.config.SHARD_COMMITTEE_PERIOD * spec.SLOTS_PER_EPOCH
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT
@ -716,7 +717,7 @@ def test_partial_withdrawal_activation_epoch_less_than_shard_committee_period(
spec, state spec, state
): ):
current_epoch = spec.get_current_epoch(state) current_epoch = spec.get_current_epoch(state)
validator_index = spec.get_active_validator_indices(state, current_epoch)[0] validator_index = random.choice(spec.get_active_validator_indices(state, current_epoch))
validator_pubkey = state.validators[validator_index].pubkey validator_pubkey = state.validators[validator_index].pubkey
address = b"\x22" * 20 address = b"\x22" * 20
amount = spec.EFFECTIVE_BALANCE_INCREMENT amount = spec.EFFECTIVE_BALANCE_INCREMENT