prysm-pulse/shared/testutil/helpers_test.go
Victor Farazdagi 748d513c62
proper error checking and type assertions (#5424)
* proper error checking and type assertions
2020-04-14 16:41:09 +00:00

77 lines
2.1 KiB
Go

package testutil
import (
"bytes"
"encoding/binary"
"testing"
"github.com/prysmaticlabs/go-ssz"
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
"github.com/prysmaticlabs/prysm/shared/params"
)
func TestBlockSignature(t *testing.T) {
beaconState, privKeys := DeterministicGenesisState(t, 100)
block, err := GenerateFullBlock(beaconState, privKeys, nil, 0)
if err != nil {
t.Fatal(err)
}
if err := beaconState.SetSlot(beaconState.Slot() + 1); err != nil {
t.Fatal(err)
}
proposerIdx, err := helpers.BeaconProposerIndex(beaconState)
if err != nil {
t.Error(err)
}
if err := beaconState.SetSlot(beaconState.Slot() - 1); err != nil {
t.Fatal(err)
}
signingRoot, err := ssz.HashTreeRoot(block.Block)
if err != nil {
t.Error(err)
}
epoch := helpers.SlotToEpoch(block.Block.Slot)
domain, err := helpers.Domain(beaconState.Fork(), epoch, params.BeaconConfig().DomainBeaconProposer)
if err != nil {
t.Fatal(err)
}
blockSig := privKeys[proposerIdx].Sign(signingRoot[:], domain).Marshal()
signature, err := BlockSignature(beaconState, block.Block, privKeys)
if err != nil {
t.Error(err)
}
if !bytes.Equal(blockSig[:], signature.Marshal()) {
t.Errorf("Expected block signatures to be equal, received %#x != %#x", blockSig[:], signature.Marshal())
}
}
func TestRandaoReveal(t *testing.T) {
beaconState, privKeys := DeterministicGenesisState(t, 100)
epoch := helpers.CurrentEpoch(beaconState)
randaoReveal, err := RandaoReveal(beaconState, epoch, privKeys)
if err != nil {
t.Error(err)
}
proposerIdx, err := helpers.BeaconProposerIndex(beaconState)
if err != nil {
t.Error(err)
}
buf := make([]byte, 32)
binary.LittleEndian.PutUint64(buf, epoch)
domain, err := helpers.Domain(beaconState.Fork(), epoch, params.BeaconConfig().DomainRandao)
if err != nil {
t.Fatal(err)
}
// We make the previous validator's index sign the message instead of the proposer.
epochSignature := privKeys[proposerIdx].Sign(buf, domain).Marshal()
if !bytes.Equal(randaoReveal[:], epochSignature[:]) {
t.Errorf("Expected randao reveals to be equal, received %#x != %#x", randaoReveal[:], epochSignature[:])
}
}