prysm-pulse/slasher/detection/testing/utils.go

// Package testing includes useful helpers for slasher-related
// unit tests.
package testing

import (
	ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
	"github.com/prysmaticlabs/prysm/shared/params"
	"github.com/prysmaticlabs/prysm/shared/rand"
)

// SignedBlockHeader given slot, proposer index this function generates signed block header.
// with random bytes as its signature.
func SignedBlockHeader(slot uint64, proposerIdx uint64) (*ethpb.SignedBeaconBlockHeader, error) {
	sig, err := genRandomByteArray(96)
	if err != nil {
		return nil, err
	}
	root := [32]byte{1, 2, 3}
	return &ethpb.SignedBeaconBlockHeader{
		Header: &ethpb.BeaconBlockHeader{
			ProposerIndex: proposerIdx,
			Slot:          slot,
			ParentRoot:    root[:],
			StateRoot:     root[:],
			BodyRoot:      root[:],
		},
		Signature: sig,
	}, nil
}

// BlockHeader given slot, proposer index this function generates block header.
func BlockHeader(slot uint64, proposerIdx uint64) (*ethpb.BeaconBlockHeader, error) {
	root := [32]byte{1, 2, 3}
	return &ethpb.BeaconBlockHeader{
		ProposerIndex: proposerIdx,
		Slot:          slot,
		ParentRoot:    root[:],
		StateRoot:     root[:],
		BodyRoot:      root[:],
	}, nil
}

func genRandomByteArray(length int) ([]byte, error) {
	blk := make([]byte, length)
	randGen := rand.NewDeterministicGenerator()
	_, err := randGen.Read(blk)
	return blk, err
}

// StartSlot returns the first slot of a given epoch.
func StartSlot(epoch uint64) uint64 {
	return epoch * params.BeaconConfig().SlotsPerEpoch
}