mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
70f3fcdbd9
* add ssz rules * regenerate protos to include deposit message * built to match the deposit cli * derived test * fix up nil item * fix up more nil items * preston suggestion * comment * add missing tests to build file * fix busted tests Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> Co-authored-by: Victor Farazdagi <simple.square@gmail.com>
154 lines
4.6 KiB
Go
154 lines
4.6 KiB
Go
package testing
|
|
|
|
import (
|
|
"context"
|
|
"encoding/hex"
|
|
"errors"
|
|
"path"
|
|
"testing"
|
|
|
|
fssz "github.com/ferranbt/fastssz"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
webpb "github.com/prysmaticlabs/prysm/proto/validator/accounts/v2"
|
|
"github.com/prysmaticlabs/prysm/shared/params/spectest"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
// SSZRoots --
|
|
type SSZRoots struct {
|
|
Root string `json:"root"`
|
|
SigningRoot string `json:"signing_root"`
|
|
}
|
|
|
|
func runSSZStaticTests(t *testing.T, config string) {
|
|
require.NoError(t, spectest.SetConfig(t, config))
|
|
|
|
testFolders, _ := testutil.TestFolders(t, config, "ssz_static")
|
|
for _, folder := range testFolders {
|
|
innerPath := path.Join("ssz_static", folder.Name(), "ssz_random")
|
|
innerTestFolders, innerTestsFolderPath := testutil.TestFolders(t, config, innerPath)
|
|
|
|
for _, innerFolder := range innerTestFolders {
|
|
t.Run(path.Join(folder.Name(), innerFolder.Name()), func(t *testing.T) {
|
|
serializedBytes, err := testutil.BazelFileBytes(innerTestsFolderPath, innerFolder.Name(), "serialized.ssz")
|
|
require.NoError(t, err)
|
|
object, err := UnmarshalledSSZ(t, serializedBytes, folder.Name())
|
|
require.NoError(t, err, "Could not unmarshall serialized SSZ")
|
|
|
|
rootsYamlFile, err := testutil.BazelFileBytes(innerTestsFolderPath, innerFolder.Name(), "roots.yaml")
|
|
require.NoError(t, err)
|
|
rootsYaml := &SSZRoots{}
|
|
require.NoError(t, testutil.UnmarshalYaml(rootsYamlFile, rootsYaml), "Failed to Unmarshal")
|
|
|
|
// Custom hash tree root for beacon state.
|
|
var htr func(interface{}) ([32]byte, error)
|
|
if _, ok := object.(*pb.BeaconState); ok {
|
|
htr = func(s interface{}) ([32]byte, error) {
|
|
beaconState, err := state.InitializeFromProto(s.(*pb.BeaconState))
|
|
require.NoError(t, err)
|
|
return beaconState.HashTreeRoot(context.Background())
|
|
}
|
|
} else {
|
|
htr = ssz.HashTreeRoot
|
|
}
|
|
|
|
root, err := htr(object)
|
|
require.NoError(t, err)
|
|
rootBytes, err := hex.DecodeString(rootsYaml.Root[2:])
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, rootBytes, root[:], "Did not receive expected hash tree root")
|
|
|
|
if rootsYaml.SigningRoot == "" {
|
|
return
|
|
}
|
|
|
|
var signingRoot [32]byte
|
|
if v, ok := object.(fssz.HashRoot); ok {
|
|
signingRoot, err = v.HashTreeRoot()
|
|
} else {
|
|
t.Fatal("object does not meet fssz.HashRoot")
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
signingRootBytes, err := hex.DecodeString(rootsYaml.SigningRoot[2:])
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, signingRootBytes, signingRoot[:], "Did not receive expected signing root")
|
|
})
|
|
}
|
|
}
|
|
}
|
|
|
|
func UnmarshalledSSZ(t *testing.T, serializedBytes []byte, folderName string) (interface{}, error) {
|
|
var obj interface{}
|
|
switch folderName {
|
|
case "Attestation":
|
|
obj = ðpb.Attestation{}
|
|
case "AttestationData":
|
|
obj = ðpb.AttestationData{}
|
|
case "AttesterSlashing":
|
|
obj = ðpb.AttesterSlashing{}
|
|
case "AggregateAndProof":
|
|
obj = ðpb.AggregateAttestationAndProof{}
|
|
case "BeaconBlock":
|
|
obj = ðpb.BeaconBlock{}
|
|
case "BeaconBlockBody":
|
|
obj = ðpb.BeaconBlockBody{}
|
|
case "BeaconBlockHeader":
|
|
obj = ðpb.BeaconBlockHeader{}
|
|
case "BeaconState":
|
|
obj = &pb.BeaconState{}
|
|
case "Checkpoint":
|
|
obj = ðpb.Checkpoint{}
|
|
case "Deposit":
|
|
obj = ðpb.Deposit{}
|
|
case "DepositData":
|
|
obj = ðpb.Deposit_Data{}
|
|
case "DepositMessage":
|
|
obj = &webpb.DepositMessage{}
|
|
case "Eth1Data":
|
|
obj = ðpb.Eth1Data{}
|
|
case "Eth1Block":
|
|
t.Skip("Unused type")
|
|
return nil, nil
|
|
case "Fork":
|
|
obj = &pb.Fork{}
|
|
case "ForkData":
|
|
obj = &pb.ForkData{}
|
|
case "HistoricalBatch":
|
|
obj = &pb.HistoricalBatch{}
|
|
case "IndexedAttestation":
|
|
obj = ðpb.IndexedAttestation{}
|
|
case "PendingAttestation":
|
|
obj = &pb.PendingAttestation{}
|
|
case "ProposerSlashing":
|
|
obj = ðpb.ProposerSlashing{}
|
|
case "SignedAggregateAndProof":
|
|
obj = ðpb.SignedAggregateAttestationAndProof{}
|
|
case "SignedBeaconBlock":
|
|
obj = ðpb.SignedBeaconBlock{}
|
|
case "SignedBeaconBlockHeader":
|
|
obj = ðpb.SignedBeaconBlockHeader{}
|
|
case "SignedVoluntaryExit":
|
|
obj = ðpb.SignedVoluntaryExit{}
|
|
case "SigningData":
|
|
obj = &pb.SigningData{}
|
|
case "Validator":
|
|
obj = ðpb.Validator{}
|
|
case "VoluntaryExit":
|
|
obj = ðpb.VoluntaryExit{}
|
|
default:
|
|
return nil, errors.New("type not found")
|
|
}
|
|
var err error
|
|
if o, ok := obj.(fssz.Unmarshaler); ok {
|
|
err = o.UnmarshalSSZ(serializedBytes)
|
|
} else {
|
|
err = ssz.Unmarshal(serializedBytes, obj)
|
|
}
|
|
return obj, err
|
|
}
|