mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
6195a0bfa1
* Forgot clear cache needs to be within the loop * space
137 lines
3.3 KiB
Go
137 lines
3.3 KiB
Go
package forkchoice
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
testDB "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/hashutil"
|
|
"gopkg.in/yaml.v2"
|
|
)
|
|
|
|
type Config struct {
|
|
TestCases []struct {
|
|
Blocks []struct {
|
|
ID string `yaml:"id"`
|
|
Parent string `yaml:"parent"`
|
|
} `yaml:"blocks"`
|
|
Weights map[string]int `yaml:"weights"`
|
|
Head string `yaml:"head"`
|
|
} `yaml:"test_cases"`
|
|
}
|
|
|
|
func TestGetHeadFromYaml(t *testing.T) {
|
|
ctx := context.Background()
|
|
filename, _ := filepath.Abs("./lmd_ghost_test.yaml")
|
|
yamlFile, err := ioutil.ReadFile(filename)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var c *Config
|
|
err = yaml.Unmarshal(yamlFile, &c)
|
|
|
|
for _, test := range c.TestCases {
|
|
db := testDB.SetupDB(t)
|
|
defer testDB.TeardownDB(t, db)
|
|
|
|
blksRoot := make(map[int][]byte)
|
|
// Construct block tree from yaml.
|
|
for _, blk := range test.Blocks {
|
|
// genesis block condition
|
|
if blk.ID == blk.Parent {
|
|
b := ðpb.BeaconBlock{Slot: 0, ParentRoot: []byte{'g'}}
|
|
if err := db.SaveBlock(ctx, b); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
root, err := ssz.SigningRoot(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
blksRoot[0] = root[:]
|
|
} else {
|
|
slot, err := strconv.Atoi(blk.ID[1:])
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
parentSlot, err := strconv.Atoi(blk.Parent[1:])
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
b := ðpb.BeaconBlock{Slot: uint64(slot), ParentRoot: blksRoot[parentSlot]}
|
|
if err := db.SaveBlock(ctx, b); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
root, err := ssz.SigningRoot(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
blksRoot[slot] = root[:]
|
|
}
|
|
}
|
|
|
|
// Assign validator votes to the blocks as weights.
|
|
count := 0
|
|
for blk, votes := range test.Weights {
|
|
slot, err := strconv.Atoi(blk[1:])
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
max := count + votes
|
|
for i := count; i < max; i++ {
|
|
if err := db.SaveValidatorLatestVote(ctx, uint64(i), &pb.ValidatorLatestVote{Root: blksRoot[slot]}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
count++
|
|
}
|
|
}
|
|
|
|
store := NewForkChoiceService(ctx, db)
|
|
validators := make([]*ethpb.Validator, count)
|
|
for i := 0; i < len(validators); i++ {
|
|
validators[i] = ðpb.Validator{ExitEpoch: 2, EffectiveBalance: 1e9}
|
|
}
|
|
|
|
s := &pb.BeaconState{Validators: validators}
|
|
|
|
if err := store.GenesisStore(ctx, s); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
store.justifiedCheckpt.Root = blksRoot[0]
|
|
h, err := hashutil.HashProto(store.justifiedCheckpt)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := store.db.SaveState(ctx, s, bytesutil.ToBytes32(blksRoot[0])); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
store.checkptBlkRoot[h] = bytesutil.ToBytes32(blksRoot[0])
|
|
|
|
head, err := store.Head(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
headSlot, err := strconv.Atoi(test.Head[1:])
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
wantedHead := blksRoot[headSlot]
|
|
|
|
if !bytes.Equal(head, wantedHead) {
|
|
t.Errorf("wanted root %#x, got root %#x", wantedHead, head)
|
|
}
|
|
|
|
helpers.ClearAllCaches()
|
|
}
|
|
}
|