prysm-pulse/beacon-chain/core/randao/randao_test.go

package randao

import (
	"bytes"
	"testing"

	v "github.com/prysmaticlabs/prysm/beacon-chain/core/validators"
	pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
	"github.com/prysmaticlabs/prysm/shared/params"
)

func TestUpdateRandaoLayers(t *testing.T) {
	beaconState := &pb.BeaconState{}
	genesisValidatorRegistry := v.InitialValidatorRegistry()
	beaconState.ValidatorRegistry = genesisValidatorRegistry

	newState, err := UpdateRandaoLayers(beaconState, 1)
	if err != nil {
		t.Fatalf("failed to update randao layers: %v", err)
	}

	vreg := newState.ValidatorRegistry

	// Since slot 1 has proposer index 511
	if vreg[511].RandaoLayers != 2 {
		t.Fatalf("randao layers not updated %d", vreg[511].RandaoLayers)
	}

	if vreg[9].RandaoLayers != 1 {
		t.Errorf("randao layers updated when they were not supposed to %d", vreg[9].RandaoLayers)
	}
}

func TestUpdateLatestRandaoMixes(t *testing.T) {
	beaconState := &pb.BeaconState{
		LatestRandaoMixesHash32S: make([][]byte, params.BeaconConfig().LatestRandaoMixesLength),
		Slot:                     5,
	}
	beaconState.LatestRandaoMixesHash32S[4%params.BeaconConfig().LatestRandaoMixesLength] = []byte{1, 2, 3}
	beaconState.LatestRandaoMixesHash32S[5%params.BeaconConfig().LatestRandaoMixesLength] = []byte{4, 5, 6}
	newState := UpdateRandaoMixes(beaconState)
	prevSlotMix := newState.LatestRandaoMixesHash32S[4%params.BeaconConfig().LatestRandaoMixesLength]
	currSlotMix := newState.LatestRandaoMixesHash32S[5%params.BeaconConfig().LatestRandaoMixesLength]
	if !bytes.Equal(currSlotMix, prevSlotMix) {
		t.Errorf("Latest randao mix not updated, wanted %#x, received %#x", prevSlotMix, currSlotMix)
	}
}