Early derived secret testing
This commit is contained in:
Carl Beekhuizen
parent
f3742566f7
commit
77b29f4960
|
|
@ -1,109 +1,108 @@
|
|||
# from copy import deepcopy
|
||||
# import pytest
|
||||
from copy import deepcopy
|
||||
import pytest
|
||||
|
||||
# import eth2spec.phase1.spec as spec
|
||||
# from eth2spec.phase1.spec import (
|
||||
# get_current_epoch,
|
||||
# process_early_derived_secret_reveal,
|
||||
# RANDAO_PENALTY_EPOCHS,
|
||||
# CUSTODY_PERIOD_TO_RANDAO_PADDING,
|
||||
# EARLY_DERIVED_SECRET_PENALTY_MAX_FUTURE_EPOCHS,
|
||||
# )
|
||||
# from tests.helpers_phase1 import (
|
||||
# get_valid_early_derived_secret_reveal,
|
||||
# )
|
||||
import eth2spec.phase1.spec as spec
|
||||
from eth2spec.phase1.spec import (
|
||||
get_current_epoch,
|
||||
process_early_derived_secret_reveal,
|
||||
RANDAO_PENALTY_EPOCHS,
|
||||
CUSTODY_PERIOD_TO_RANDAO_PADDING,
|
||||
EARLY_DERIVED_SECRET_PENALTY_MAX_FUTURE_EPOCHS,
|
||||
)
|
||||
from tests.phase1.helpers import (
|
||||
get_valid_early_derived_secret_reveal,
|
||||
)
|
||||
|
||||
# #mark entire file as 'randao_key_reveals'
|
||||
# pytestmark = pytest.mark.randao_key_reveals
|
||||
#mark entire file as 'randao_key_reveals'
|
||||
pytestmark = pytest.mark.randao_key_reveals
|
||||
|
||||
def run_early_derived_secret_reveal_processing(state, randao_key_reveal, valid=True):
|
||||
"""
|
||||
Run ``process_randao_key_reveal`` returning the pre and post state.
|
||||
If ``valid == False``, run expecting ``AssertionError``
|
||||
"""
|
||||
post_state = deepcopy(state)
|
||||
|
||||
if not valid:
|
||||
with pytest.raises(AssertionError):
|
||||
process_early_derived_secret_reveal(post_state, randao_key_reveal)
|
||||
return state, None
|
||||
|
||||
process_early_derived_secret_reveal(post_state, randao_key_reveal)
|
||||
|
||||
slashed_validator = post_state.validator_registry[randao_key_reveal.revealed_index]
|
||||
|
||||
if randao_key_reveal.epoch >= get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING:
|
||||
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
|
||||
# FIXME: Currently broken because get_base_reward in genesis epoch is 0
|
||||
assert (
|
||||
post_state.balances[randao_key_reveal.revealed_index] <
|
||||
state.balances[randao_key_reveal.revealed_index]
|
||||
)
|
||||
|
||||
return state, post_state
|
||||
|
||||
|
||||
# def run_early_derived_secret_reveal_processing(state, randao_key_reveal, valid=True):
|
||||
# """
|
||||
# Run ``process_randao_key_reveal`` returning the pre and post state.
|
||||
# If ``valid == False``, run expecting ``AssertionError``
|
||||
# """
|
||||
# post_state = deepcopy(state)
|
||||
def test_success(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state)
|
||||
|
||||
# if not valid:
|
||||
# with pytest.raises(AssertionError):
|
||||
# process_early_derived_secret_reveal(post_state, randao_key_reveal)
|
||||
# return state, None
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal)
|
||||
|
||||
# process_early_derived_secret_reveal(post_state, randao_key_reveal)
|
||||
|
||||
# slashed_validator = post_state.validator_registry[randao_key_reveal.revealed_index]
|
||||
|
||||
# if randao_key_reveal.epoch >= get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING:
|
||||
# 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
|
||||
# # FIXME: Currently broken because get_base_reward in genesis epoch is 0
|
||||
# assert (
|
||||
# post_state.balances[randao_key_reveal.revealed_index] <
|
||||
# state.balances[randao_key_reveal.revealed_index]
|
||||
# )
|
||||
|
||||
# return state, post_state
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
||||
|
||||
# def test_success(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state)
|
||||
def test_reveal_from_current_epoch(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state))
|
||||
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal)
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
|
||||
|
||||
# def test_reveal_from_current_epoch(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state))
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
||||
# Not currently possible as we are testing at epoch 0
|
||||
#
|
||||
#def test_reveal_from_past_epoch(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) - 1)
|
||||
#
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
|
||||
#
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
|
||||
# # Not currently possible as we are testing at epoch 0
|
||||
# #
|
||||
# #def test_reveal_from_past_epoch(state):
|
||||
# # randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) - 1)
|
||||
# #
|
||||
# # pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
# #
|
||||
# # return pre_state, randao_key_reveal, post_state
|
||||
def test_reveal_with_custody_padding(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING)
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, True)
|
||||
|
||||
# def test_reveal_with_custody_padding(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING)
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, True)
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
def test_reveal_with_custody_padding_minus_one(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING - 1)
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, True)
|
||||
|
||||
# def test_reveal_with_custody_padding_minus_one(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + CUSTODY_PERIOD_TO_RANDAO_PADDING - 1)
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, True)
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
def test_double_reveal(state):
|
||||
|
||||
# def test_double_reveal(state):
|
||||
randao_key_reveal1 = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + RANDAO_PENALTY_EPOCHS + 1)
|
||||
pre_state, intermediate_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal1)
|
||||
|
||||
# randao_key_reveal1 = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + RANDAO_PENALTY_EPOCHS + 1)
|
||||
# pre_state, intermediate_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal1)
|
||||
randao_key_reveal2 = get_valid_early_derived_secret_reveal(intermediate_state, get_current_epoch(pre_state) + RANDAO_PENALTY_EPOCHS + 1)
|
||||
intermediate_state_, post_state = run_early_derived_secret_reveal_processing(intermediate_state, randao_key_reveal2, False)
|
||||
|
||||
# randao_key_reveal2 = get_valid_early_derived_secret_reveal(intermediate_state, get_current_epoch(pre_state) + RANDAO_PENALTY_EPOCHS + 1)
|
||||
# intermediate_state_, post_state = run_early_derived_secret_reveal_processing(intermediate_state, randao_key_reveal2, False)
|
||||
return pre_state, [randao_key_reveal1, randao_key_reveal2], post_state
|
||||
|
||||
# return pre_state, [randao_key_reveal1, randao_key_reveal2], post_state
|
||||
def test_revealer_is_slashed(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state))
|
||||
state.validator_registry[randao_key_reveal.revealed_index].slashed = True
|
||||
|
||||
# def test_revealer_is_slashed(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state))
|
||||
# state.validator_registry[randao_key_reveal.revealed_index].slashed = True
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
def test_far_future_epoch(state):
|
||||
randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + EARLY_DERIVED_SECRET_PENALTY_MAX_FUTURE_EPOCHS)
|
||||
|
||||
# def test_far_future_epoch(state):
|
||||
# randao_key_reveal = get_valid_early_derived_secret_reveal(state, get_current_epoch(state) + EARLY_DERIVED_SECRET_PENALTY_MAX_FUTURE_EPOCHS)
|
||||
pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
|
||||
# pre_state, post_state = run_early_derived_secret_reveal_processing(state, randao_key_reveal, False)
|
||||
|
||||
# return pre_state, randao_key_reveal, post_state
|
||||
return pre_state, randao_key_reveal, post_state
|
||||
|
|
|
|||
|
|
@ -0,0 +1,54 @@
|
|||
from py_ecc import bls
|
||||
|
||||
import eth2spec.phase1.spec as spec
|
||||
from eth2spec.phase1.spec import (
|
||||
# constants
|
||||
ZERO_HASH,
|
||||
CUSTODY_PERIOD_TO_RANDAO_PADDING,
|
||||
# SSZ
|
||||
EarlyDerivedSecretReveal,
|
||||
# functions
|
||||
get_active_validator_indices,
|
||||
get_current_epoch,
|
||||
get_domain,
|
||||
hash_tree_root,
|
||||
)
|
||||
|
||||
from tests.phase0.helpers import (
|
||||
privkeys,
|
||||
)
|
||||
|
||||
def get_valid_early_derived_secret_reveal(state, epoch=None):
|
||||
current_epoch = get_current_epoch(state)
|
||||
revealed_index = get_active_validator_indices(state, current_epoch)[-1]
|
||||
masker_index = get_active_validator_indices(state, current_epoch)[0]
|
||||
|
||||
if epoch is None:
|
||||
epoch = current_epoch + CUSTODY_PERIOD_TO_RANDAO_PADDING
|
||||
|
||||
reveal = bls.sign(
|
||||
message_hash=hash_tree_root(epoch),
|
||||
privkey=privkeys[revealed_index],
|
||||
domain=get_domain(
|
||||
state=state,
|
||||
domain_type=spec.DOMAIN_RANDAO,
|
||||
message_epoch=epoch,
|
||||
),
|
||||
)
|
||||
mask = bls.sign(
|
||||
message_hash=hash_tree_root(epoch),
|
||||
privkey=privkeys[masker_index],
|
||||
domain=get_domain(
|
||||
state=state,
|
||||
domain_type=spec.DOMAIN_RANDAO,
|
||||
message_epoch=epoch,
|
||||
),
|
||||
)
|
||||
|
||||
return EarlyDerivedSecretReveal(
|
||||
revealed_index=revealed_index,
|
||||
epoch=epoch,
|
||||
reveal=reveal,
|
||||
masker_index=masker_index,
|
||||
mask=mask,
|
||||
)
|
||||
Loading…
Reference in New Issue