mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 20:07:17 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
166 lines
6.2 KiB
Go
166 lines
6.2 KiB
Go
package client
|
|
|
|
import (
|
|
"testing"
|
|
|
|
fieldparams "github.com/prysmaticlabs/prysm/v4/config/fieldparams"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func TestUpdateLogAggregateStats(t *testing.T) {
|
|
v := &validator{
|
|
logValidatorBalances: true,
|
|
startBalances: make(map[[fieldparams.BLSPubkeyLength]byte]uint64),
|
|
prevBalance: make(map[[fieldparams.BLSPubkeyLength]byte]uint64),
|
|
voteStats: voteStats{
|
|
startEpoch: 0, // this would otherwise have been previously set in LogValidatorGainsAndLosses()
|
|
},
|
|
}
|
|
|
|
pubKeyBytes := [][fieldparams.BLSPubkeyLength]byte{
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000012345678")),
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000099999999")),
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000055555555")),
|
|
}
|
|
|
|
v.startBalances[pubKeyBytes[0]] = uint64(32100000000)
|
|
v.startBalances[pubKeyBytes[1]] = uint64(32200000000)
|
|
v.startBalances[pubKeyBytes[2]] = uint64(33000000000)
|
|
|
|
responses := []*ethpb.ValidatorPerformanceResponse{
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{false, true, false},
|
|
CorrectlyVotedSource: []bool{false, true, true},
|
|
CorrectlyVotedTarget: []bool{false, true, true},
|
|
},
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{true, true, true},
|
|
CorrectlyVotedSource: []bool{true, true, true},
|
|
CorrectlyVotedTarget: []bool{true, true, true},
|
|
},
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{true, false, true},
|
|
CorrectlyVotedSource: []bool{true, false, true},
|
|
CorrectlyVotedTarget: []bool{false, false, true},
|
|
},
|
|
}
|
|
|
|
v.prevBalance[pubKeyBytes[0]] = uint64(33200000000)
|
|
v.prevBalance[pubKeyBytes[1]] = uint64(33300000000)
|
|
v.prevBalance[pubKeyBytes[2]] = uint64(31000000000)
|
|
|
|
var hook *logTest.Hook
|
|
|
|
for i, val := range responses {
|
|
if i == len(responses)-1 { // Handle last log.
|
|
hook = logTest.NewGlobal()
|
|
}
|
|
v.UpdateLogAggregateStats(val, params.BeaconConfig().SlotsPerEpoch*primitives.Slot(i+1))
|
|
}
|
|
|
|
require.LogsContain(t, hook, "msg=\"Previous epoch aggregated voting summary\" attestationInclusionPct=\"67%\" "+
|
|
"correctlyVotedHeadPct=\"100%\" correctlyVotedSourcePct=\"100%\" correctlyVotedTargetPct=\"50%\" epoch=2")
|
|
require.LogsContain(t, hook, "msg=\"Vote summary since launch\" attestationsInclusionPct=\"78%\" "+
|
|
"averageInclusionDistance=\"0.00 slots\" correctlyVotedHeadPct=\"86%\" correctlyVotedSourcePct=\"100%\" "+
|
|
"correctlyVotedTargetPct=\"86%\" numberOfEpochs=3 pctChangeCombinedBalance=\"0.20555%\"")
|
|
|
|
}
|
|
|
|
func TestUpdateLogAltairAggregateStats(t *testing.T) {
|
|
v := &validator{
|
|
logValidatorBalances: true,
|
|
startBalances: make(map[[fieldparams.BLSPubkeyLength]byte]uint64),
|
|
prevBalance: make(map[[fieldparams.BLSPubkeyLength]byte]uint64),
|
|
voteStats: voteStats{
|
|
startEpoch: params.BeaconConfig().AltairForkEpoch, // this would otherwise have been previously set in LogValidatorGainsAndLosses()
|
|
},
|
|
}
|
|
|
|
pubKeyBytes := [][fieldparams.BLSPubkeyLength]byte{
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000012345678")),
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000099999999")),
|
|
bytesutil.ToBytes48([]byte("000000000000000000000000000000000000000055555555")),
|
|
}
|
|
|
|
v.startBalances[pubKeyBytes[0]] = uint64(32100000000)
|
|
v.startBalances[pubKeyBytes[1]] = uint64(32200000000)
|
|
v.startBalances[pubKeyBytes[2]] = uint64(33000000000)
|
|
|
|
// 7 attestations included
|
|
responses := []*ethpb.ValidatorPerformanceResponse{
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{false, true, false},
|
|
CorrectlyVotedSource: []bool{false, true, true},
|
|
CorrectlyVotedTarget: []bool{false, false, true}, // test one fast target that is included
|
|
},
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{true, true, true},
|
|
CorrectlyVotedSource: []bool{true, true, true},
|
|
CorrectlyVotedTarget: []bool{true, true, true},
|
|
},
|
|
{
|
|
PublicKeys: [][]byte{
|
|
bytesutil.FromBytes48(pubKeyBytes[0]),
|
|
bytesutil.FromBytes48(pubKeyBytes[1]),
|
|
bytesutil.FromBytes48(pubKeyBytes[2]),
|
|
},
|
|
CorrectlyVotedHead: []bool{true, false, true},
|
|
CorrectlyVotedSource: []bool{true, false, true},
|
|
CorrectlyVotedTarget: []bool{false, false, true},
|
|
},
|
|
}
|
|
|
|
v.prevBalance[pubKeyBytes[0]] = uint64(33200000000)
|
|
v.prevBalance[pubKeyBytes[1]] = uint64(33300000000)
|
|
v.prevBalance[pubKeyBytes[2]] = uint64(31000000000)
|
|
|
|
var hook *logTest.Hook
|
|
|
|
for i, val := range responses {
|
|
if i == len(responses)-1 { // Handle last log.
|
|
hook = logTest.NewGlobal()
|
|
}
|
|
altairStart, err := slots.EpochStart(params.BeaconConfig().AltairForkEpoch)
|
|
require.NoError(t, err)
|
|
|
|
v.UpdateLogAggregateStats(val, altairStart+params.BeaconConfig().SlotsPerEpoch*primitives.Slot(i+1))
|
|
}
|
|
|
|
require.LogsContain(t, hook, "msg=\"Previous epoch aggregated voting summary\" attestationInclusionPct=\"67%\" "+
|
|
"averageInactivityScore=0 correctlyVotedHeadPct=\"100%\" correctlyVotedSourcePct=\"100%\" correctlyVotedTargetPct=\"50%\" epoch=74242")
|
|
require.LogsContain(t, hook, "msg=\"Vote summary since launch\" attestationsInclusionPct=\"78%\" "+
|
|
"correctlyVotedHeadPct=\"86%\" correctlyVotedSourcePct=\"100%\" "+
|
|
"correctlyVotedTargetPct=\"71%\" numberOfEpochs=3 pctChangeCombinedBalance=\"0.20555%\"")
|
|
}
|