Use `EMPTY_SIGNATURE` and `ZERO_HASH` in the tests

This commit is contained in:
Hsiao-Wei Wang 2019-03-19 11:46:32 +08:00
parent a07219c570
commit 8fc1fe5f20
No known key found for this signature in database
GPG Key ID: 95B070122902DEA4
2 changed files with 15 additions and 8 deletions

View File

@ -5,11 +5,15 @@ from py_ecc import bls
import build.phase0.spec as spec import build.phase0.spec as spec
from build.phase0.utils.minimal_ssz import signed_root from build.phase0.utils.minimal_ssz import signed_root
from build.phase0.spec import ( from build.phase0.spec import (
# constants
EMPTY_SIGNATURE,
# SSZ
AttestationData, AttestationData,
Deposit, Deposit,
DepositInput, DepositInput,
DepositData, DepositData,
Eth1Data, Eth1Data,
# functions
get_block_root, get_block_root,
get_current_epoch, get_current_epoch,
get_domain, get_domain,
@ -92,7 +96,7 @@ def build_deposit_data(state, pubkey, privkey, amount):
deposit_input = DepositInput( deposit_input = DepositInput(
pubkey=pubkey, pubkey=pubkey,
withdrawal_credentials=privkey.to_bytes(32, byteorder='big'), withdrawal_credentials=privkey.to_bytes(32, byteorder='big'),
proof_of_possession=b'\x00' * 96, proof_of_possession=EMPTY_SIGNATURE,
) )
proof_of_possession = bls.sign( proof_of_possession = bls.sign(
message_hash=signed_root(deposit_input), message_hash=signed_root(deposit_input),

View File

@ -7,6 +7,9 @@ import build.phase0.spec as spec
from build.phase0.utils.minimal_ssz import signed_root from build.phase0.utils.minimal_ssz import signed_root
from build.phase0.spec import ( from build.phase0.spec import (
# constants
EMPTY_SIGNATURE,
ZERO_HASH,
# SSZ # SSZ
Attestation, Attestation,
AttestationDataAndCustodyBit, AttestationDataAndCustodyBit,
@ -117,10 +120,10 @@ def test_proposer_slashing(state, pubkeys, privkeys):
slot = spec.GENESIS_SLOT slot = spec.GENESIS_SLOT
header_1 = BeaconBlockHeader( header_1 = BeaconBlockHeader(
slot=slot, slot=slot,
previous_block_root=b'\x00' * 32, previous_block_root=ZERO_HASH,
state_root=b'\x00' * 32, state_root=ZERO_HASH,
block_body_root=b'\x00' * 32, block_body_root=ZERO_HASH,
signature=b'\x00' * 96 signature=EMPTY_SIGNATURE,
) )
header_2 = deepcopy(header_1) header_2 = deepcopy(header_1)
header_2.previous_block_root = b'\x02' * 32 header_2.previous_block_root = b'\x02' * 32
@ -259,7 +262,7 @@ def test_attestation(state, pubkeys, privkeys):
aggregation_bitfield=aggregation_bitfield, aggregation_bitfield=aggregation_bitfield,
data=attestation_data, data=attestation_data,
custody_bitfield=custody_bitfield, custody_bitfield=custody_bitfield,
aggregate_signature=b'\x00' * 96, aggregate_signature=EMPTY_SIGNATURE,
) )
participants = get_attestation_participants( participants = get_attestation_participants(
test_state, test_state,
@ -325,7 +328,7 @@ def test_voluntary_exit(state, pubkeys, privkeys):
voluntary_exit = VoluntaryExit( voluntary_exit = VoluntaryExit(
epoch=get_current_epoch(pre_state), epoch=get_current_epoch(pre_state),
validator_index=validator_index, validator_index=validator_index,
signature=b'\x00' * 96, signature=EMPTY_SIGNATURE,
) )
voluntary_exit.signature = bls.sign( voluntary_exit.signature = bls.sign(
message_hash=signed_root(voluntary_exit), message_hash=signed_root(voluntary_exit),
@ -376,7 +379,7 @@ def test_transfer(state, pubkeys, privkeys):
fee=0, fee=0,
slot=pre_state.slot + 1, slot=pre_state.slot + 1,
pubkey=transfer_pubkey, pubkey=transfer_pubkey,
signature=b'\x00' * 96, signature=EMPTY_SIGNATURE,
) )
transfer.signature = bls.sign( transfer.signature = bls.sign(
message_hash=signed_root(transfer), message_hash=signed_root(transfer),