mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
376d248c22
* Add in progress handler for committee cache * Remove debug print * Update validators.go * Fix all the tests * More tests * Update committee_disabled.go * Update committee_disabled.go * Update testing util * Update main.go Co-authored-by: Nishant Das <nishdas93@gmail.com>
398 lines
13 KiB
Go
398 lines
13 KiB
Go
package beacon
|
|
|
|
import (
|
|
"context"
|
|
"encoding/binary"
|
|
"fmt"
|
|
"strconv"
|
|
"testing"
|
|
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
mock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
dbTest "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
"github.com/prysmaticlabs/prysm/cmd"
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/wrapper"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
"github.com/prysmaticlabs/prysm/testing/util"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func TestServer_ListAssignments_CannotRequestFutureEpoch(t *testing.T) {
|
|
|
|
db := dbTest.SetupDB(t)
|
|
ctx := context.Background()
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
}
|
|
|
|
wanted := errNoEpochInfoError
|
|
_, err := bs.ListValidatorAssignments(
|
|
ctx,
|
|
ðpb.ListValidatorAssignmentsRequest{
|
|
QueryFilter: ðpb.ListValidatorAssignmentsRequest_Epoch{
|
|
Epoch: core.SlotToEpoch(bs.GenesisTimeFetcher.CurrentSlot()) + 1,
|
|
},
|
|
},
|
|
)
|
|
assert.ErrorContains(t, wanted, err)
|
|
}
|
|
|
|
func TestServer_ListAssignments_NoResults(t *testing.T) {
|
|
|
|
db := dbTest.SetupDB(t)
|
|
ctx := context.Background()
|
|
st, err := util.NewBeaconState()
|
|
require.NoError(t, err)
|
|
|
|
b := util.NewBeaconBlock()
|
|
require.NoError(t, db.SaveBlock(ctx, wrapper.WrappedPhase0SignedBeaconBlock(b)))
|
|
gRoot, err := b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, gRoot))
|
|
require.NoError(t, db.SaveState(ctx, st, gRoot))
|
|
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
StateGen: stategen.New(db),
|
|
}
|
|
wanted := ðpb.ValidatorAssignments{
|
|
Assignments: make([]*ethpb.ValidatorAssignments_CommitteeAssignment, 0),
|
|
TotalSize: int32(0),
|
|
NextPageToken: strconv.Itoa(0),
|
|
}
|
|
res, err := bs.ListValidatorAssignments(
|
|
ctx,
|
|
ðpb.ListValidatorAssignmentsRequest{
|
|
QueryFilter: ðpb.ListValidatorAssignmentsRequest_Genesis{
|
|
Genesis: true,
|
|
},
|
|
},
|
|
)
|
|
require.NoError(t, err)
|
|
if !proto.Equal(wanted, res) {
|
|
t.Errorf("Wanted %v, received %v", wanted, res)
|
|
}
|
|
}
|
|
|
|
func TestServer_ListAssignments_Pagination_InputOutOfRange(t *testing.T) {
|
|
helpers.ClearCache()
|
|
db := dbTest.SetupDB(t)
|
|
ctx := context.Background()
|
|
count := 100
|
|
validators := make([]*ethpb.Validator, 0, count)
|
|
for i := 0; i < count; i++ {
|
|
pubKey := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
withdrawalCred := make([]byte, 32)
|
|
binary.LittleEndian.PutUint64(pubKey, uint64(i))
|
|
validators = append(validators, ðpb.Validator{
|
|
PublicKey: pubKey,
|
|
WithdrawalCredentials: withdrawalCred,
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
ActivationEpoch: 0,
|
|
})
|
|
}
|
|
|
|
blk := util.NewBeaconBlock().Block
|
|
blockRoot, err := blk.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
s, err := util.NewBeaconState()
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.SetValidators(validators))
|
|
require.NoError(t, db.SaveState(ctx, s, blockRoot))
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, blockRoot))
|
|
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
HeadFetcher: &mock.ChainService{
|
|
State: s,
|
|
},
|
|
FinalizationFetcher: &mock.ChainService{
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
},
|
|
},
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
StateGen: stategen.New(db),
|
|
}
|
|
|
|
wanted := fmt.Sprintf("page start %d >= list %d", 500, count)
|
|
_, err = bs.ListValidatorAssignments(context.Background(), ðpb.ListValidatorAssignmentsRequest{
|
|
PageToken: strconv.Itoa(2),
|
|
QueryFilter: ðpb.ListValidatorAssignmentsRequest_Genesis{Genesis: true},
|
|
})
|
|
assert.ErrorContains(t, wanted, err)
|
|
}
|
|
|
|
func TestServer_ListAssignments_Pagination_ExceedsMaxPageSize(t *testing.T) {
|
|
bs := &Server{}
|
|
exceedsMax := int32(cmd.Get().MaxRPCPageSize + 1)
|
|
|
|
wanted := fmt.Sprintf("Requested page size %d can not be greater than max size %d", exceedsMax, cmd.Get().MaxRPCPageSize)
|
|
req := ðpb.ListValidatorAssignmentsRequest{
|
|
PageToken: strconv.Itoa(0),
|
|
PageSize: exceedsMax,
|
|
}
|
|
_, err := bs.ListValidatorAssignments(context.Background(), req)
|
|
assert.ErrorContains(t, wanted, err)
|
|
}
|
|
|
|
func TestServer_ListAssignments_Pagination_DefaultPageSize_NoArchive(t *testing.T) {
|
|
helpers.ClearCache()
|
|
db := dbTest.SetupDB(t)
|
|
ctx := context.Background()
|
|
count := 500
|
|
validators := make([]*ethpb.Validator, 0, count)
|
|
for i := 0; i < count; i++ {
|
|
pubKey := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
withdrawalCred := make([]byte, 32)
|
|
binary.LittleEndian.PutUint64(pubKey, uint64(i))
|
|
// Mark the validators with index divisible by 3 inactive.
|
|
if i%3 == 0 {
|
|
validators = append(validators, ðpb.Validator{
|
|
PublicKey: pubKey,
|
|
WithdrawalCredentials: withdrawalCred,
|
|
ExitEpoch: 0,
|
|
ActivationEpoch: 0,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
})
|
|
} else {
|
|
validators = append(validators, ðpb.Validator{
|
|
PublicKey: pubKey,
|
|
WithdrawalCredentials: withdrawalCred,
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
ActivationEpoch: 0,
|
|
})
|
|
}
|
|
}
|
|
|
|
blk := util.NewBeaconBlock().Block
|
|
blockRoot, err := blk.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
s, err := util.NewBeaconState()
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.SetValidators(validators))
|
|
require.NoError(t, db.SaveState(ctx, s, blockRoot))
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, blockRoot))
|
|
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
HeadFetcher: &mock.ChainService{
|
|
State: s,
|
|
},
|
|
FinalizationFetcher: &mock.ChainService{
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
},
|
|
},
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
StateGen: stategen.New(db),
|
|
}
|
|
|
|
res, err := bs.ListValidatorAssignments(context.Background(), ðpb.ListValidatorAssignmentsRequest{
|
|
QueryFilter: ðpb.ListValidatorAssignmentsRequest_Genesis{Genesis: true},
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Construct the wanted assignments.
|
|
var wanted []*ethpb.ValidatorAssignments_CommitteeAssignment
|
|
|
|
activeIndices, err := helpers.ActiveValidatorIndices(ctx, s, 0)
|
|
require.NoError(t, err)
|
|
committeeAssignments, proposerIndexToSlots, err := helpers.CommitteeAssignments(context.Background(), s, 0)
|
|
require.NoError(t, err)
|
|
for _, index := range activeIndices[0:params.BeaconConfig().DefaultPageSize] {
|
|
val, err := s.ValidatorAtIndex(index)
|
|
require.NoError(t, err)
|
|
wanted = append(wanted, ðpb.ValidatorAssignments_CommitteeAssignment{
|
|
BeaconCommittees: committeeAssignments[index].Committee,
|
|
CommitteeIndex: committeeAssignments[index].CommitteeIndex,
|
|
AttesterSlot: committeeAssignments[index].AttesterSlot,
|
|
ProposerSlots: proposerIndexToSlots[index],
|
|
PublicKey: val.PublicKey,
|
|
ValidatorIndex: index,
|
|
})
|
|
}
|
|
assert.DeepSSZEqual(t, wanted, res.Assignments, "Did not receive wanted assignments")
|
|
}
|
|
|
|
func TestServer_ListAssignments_FilterPubkeysIndices_NoPagination(t *testing.T) {
|
|
helpers.ClearCache()
|
|
db := dbTest.SetupDB(t)
|
|
|
|
ctx := context.Background()
|
|
count := 100
|
|
validators := make([]*ethpb.Validator, 0, count)
|
|
withdrawCreds := make([]byte, 32)
|
|
for i := 0; i < count; i++ {
|
|
pubKey := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
binary.LittleEndian.PutUint64(pubKey, uint64(i))
|
|
val := ðpb.Validator{
|
|
PublicKey: pubKey,
|
|
WithdrawalCredentials: withdrawCreds,
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
}
|
|
validators = append(validators, val)
|
|
}
|
|
|
|
blk := util.NewBeaconBlock().Block
|
|
blockRoot, err := blk.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
s, err := util.NewBeaconState()
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.SetValidators(validators))
|
|
require.NoError(t, db.SaveState(ctx, s, blockRoot))
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, blockRoot))
|
|
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
FinalizationFetcher: &mock.ChainService{
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
},
|
|
},
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
StateGen: stategen.New(db),
|
|
}
|
|
|
|
pubKey1 := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
binary.LittleEndian.PutUint64(pubKey1, 1)
|
|
pubKey2 := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
binary.LittleEndian.PutUint64(pubKey2, 2)
|
|
req := ðpb.ListValidatorAssignmentsRequest{PublicKeys: [][]byte{pubKey1, pubKey2}, Indices: []types.ValidatorIndex{2, 3}}
|
|
res, err := bs.ListValidatorAssignments(context.Background(), req)
|
|
require.NoError(t, err)
|
|
|
|
// Construct the wanted assignments.
|
|
var wanted []*ethpb.ValidatorAssignments_CommitteeAssignment
|
|
|
|
activeIndices, err := helpers.ActiveValidatorIndices(ctx, s, 0)
|
|
require.NoError(t, err)
|
|
committeeAssignments, proposerIndexToSlots, err := helpers.CommitteeAssignments(context.Background(), s, 0)
|
|
require.NoError(t, err)
|
|
for _, index := range activeIndices[1:4] {
|
|
val, err := s.ValidatorAtIndex(index)
|
|
require.NoError(t, err)
|
|
wanted = append(wanted, ðpb.ValidatorAssignments_CommitteeAssignment{
|
|
BeaconCommittees: committeeAssignments[index].Committee,
|
|
CommitteeIndex: committeeAssignments[index].CommitteeIndex,
|
|
AttesterSlot: committeeAssignments[index].AttesterSlot,
|
|
ProposerSlots: proposerIndexToSlots[index],
|
|
PublicKey: val.PublicKey,
|
|
ValidatorIndex: index,
|
|
})
|
|
}
|
|
|
|
assert.DeepEqual(t, wanted, res.Assignments, "Did not receive wanted assignments")
|
|
}
|
|
|
|
func TestServer_ListAssignments_CanFilterPubkeysIndices_WithPagination(t *testing.T) {
|
|
helpers.ClearCache()
|
|
db := dbTest.SetupDB(t)
|
|
ctx := context.Background()
|
|
count := 100
|
|
validators := make([]*ethpb.Validator, 0, count)
|
|
withdrawCred := make([]byte, 32)
|
|
for i := 0; i < count; i++ {
|
|
pubKey := make([]byte, params.BeaconConfig().BLSPubkeyLength)
|
|
binary.LittleEndian.PutUint64(pubKey, uint64(i))
|
|
val := ðpb.Validator{
|
|
PublicKey: pubKey,
|
|
WithdrawalCredentials: withdrawCred,
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
}
|
|
validators = append(validators, val)
|
|
}
|
|
|
|
blk := util.NewBeaconBlock().Block
|
|
blockRoot, err := blk.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
s, err := util.NewBeaconState()
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.SetValidators(validators))
|
|
require.NoError(t, db.SaveState(ctx, s, blockRoot))
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, blockRoot))
|
|
|
|
bs := &Server{
|
|
BeaconDB: db,
|
|
FinalizationFetcher: &mock.ChainService{
|
|
FinalizedCheckPoint: ðpb.Checkpoint{
|
|
Epoch: 0,
|
|
},
|
|
},
|
|
GenesisTimeFetcher: &mock.ChainService{},
|
|
StateGen: stategen.New(db),
|
|
}
|
|
|
|
req := ðpb.ListValidatorAssignmentsRequest{Indices: []types.ValidatorIndex{1, 2, 3, 4, 5, 6}, PageSize: 2, PageToken: "1"}
|
|
res, err := bs.ListValidatorAssignments(context.Background(), req)
|
|
require.NoError(t, err)
|
|
|
|
// Construct the wanted assignments.
|
|
var assignments []*ethpb.ValidatorAssignments_CommitteeAssignment
|
|
|
|
activeIndices, err := helpers.ActiveValidatorIndices(ctx, s, 0)
|
|
require.NoError(t, err)
|
|
committeeAssignments, proposerIndexToSlots, err := helpers.CommitteeAssignments(context.Background(), s, 0)
|
|
require.NoError(t, err)
|
|
for _, index := range activeIndices[3:5] {
|
|
val, err := s.ValidatorAtIndex(index)
|
|
require.NoError(t, err)
|
|
assignments = append(assignments, ðpb.ValidatorAssignments_CommitteeAssignment{
|
|
BeaconCommittees: committeeAssignments[index].Committee,
|
|
CommitteeIndex: committeeAssignments[index].CommitteeIndex,
|
|
AttesterSlot: committeeAssignments[index].AttesterSlot,
|
|
ProposerSlots: proposerIndexToSlots[index],
|
|
PublicKey: val.PublicKey,
|
|
ValidatorIndex: index,
|
|
})
|
|
}
|
|
|
|
wantedRes := ðpb.ValidatorAssignments{
|
|
Assignments: assignments,
|
|
TotalSize: int32(len(req.Indices)),
|
|
NextPageToken: "2",
|
|
}
|
|
|
|
assert.DeepEqual(t, wantedRes, res, "Did not get wanted assignments")
|
|
|
|
// Test the wrap around scenario.
|
|
assignments = nil
|
|
req = ðpb.ListValidatorAssignmentsRequest{Indices: []types.ValidatorIndex{1, 2, 3, 4, 5, 6}, PageSize: 5, PageToken: "1"}
|
|
res, err = bs.ListValidatorAssignments(context.Background(), req)
|
|
require.NoError(t, err)
|
|
cAssignments, proposerIndexToSlots, err := helpers.CommitteeAssignments(context.Background(), s, 0)
|
|
require.NoError(t, err)
|
|
for _, index := range activeIndices[6:7] {
|
|
val, err := s.ValidatorAtIndex(index)
|
|
require.NoError(t, err)
|
|
assignments = append(assignments, ðpb.ValidatorAssignments_CommitteeAssignment{
|
|
BeaconCommittees: cAssignments[index].Committee,
|
|
CommitteeIndex: cAssignments[index].CommitteeIndex,
|
|
AttesterSlot: cAssignments[index].AttesterSlot,
|
|
ProposerSlots: proposerIndexToSlots[index],
|
|
PublicKey: val.PublicKey,
|
|
ValidatorIndex: index,
|
|
})
|
|
}
|
|
|
|
wantedRes = ðpb.ValidatorAssignments{
|
|
Assignments: assignments,
|
|
TotalSize: int32(len(req.Indices)),
|
|
NextPageToken: "",
|
|
}
|
|
|
|
assert.DeepEqual(t, wantedRes, res, "Did not receive wanted assignments")
|
|
}
|