mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-03 08:37:37 +00:00
d5ec248691
* Rename getter functions * Rename new * Radek's feedback Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
40 lines
1.0 KiB
Go
40 lines
1.0 KiB
Go
package testing
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
)
|
|
|
|
const fileBase = "0-11-0/mainnet/beaconstate"
|
|
const fileBaseENV = "BEACONSTATES_PATH"
|
|
|
|
// BeaconFuzzState returns a beacon state by ID using the beacon-fuzz corpora.
|
|
func BeaconFuzzState(ID uint64) (*pb.BeaconState, error) {
|
|
base := fileBase
|
|
// Using an environment variable allows a host image to specify the path when only the binary
|
|
// executable was uploaded (without the runfiles). i.e. fuzzit's platform.
|
|
if p, ok := os.LookupEnv(fileBaseENV); ok {
|
|
base = p
|
|
}
|
|
ok, err := testutil.BazelDirectoryNonEmpty(base)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
if !ok {
|
|
panic(fmt.Sprintf("Beacon states directory (%s) does not exist or has no files.", base))
|
|
}
|
|
b, err := testutil.BazelFileBytes(base, strconv.Itoa(int(ID)))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
st := &pb.BeaconState{}
|
|
if err := st.UnmarshalSSZ(b); err != nil {
|
|
return nil, err
|
|
}
|
|
return st, nil
|
|
}
|