2021-08-09 21:27:51 +00:00
|
|
|
package altair_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
|
|
"github.com/prysmaticlabs/go-bitfield"
|
2021-09-03 20:10:31 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core"
|
2021-08-09 21:27:51 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/altair"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
|
|
stateAltair "github.com/prysmaticlabs/prysm/beacon-chain/state/v2"
|
|
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
2021-09-20 16:17:03 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/attestation"
|
2021-08-09 21:27:51 +00:00
|
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestTranslateParticipation(t *testing.T) {
|
|
|
|
s, _ := testutil.DeterministicGenesisStateAltair(t, 64)
|
|
|
|
st, ok := s.(*stateAltair.BeaconState)
|
|
|
|
require.Equal(t, true, ok)
|
|
|
|
require.NoError(t, st.SetSlot(st.Slot()+params.BeaconConfig().MinAttestationInclusionDelay))
|
|
|
|
|
|
|
|
var err error
|
|
|
|
newState, err := altair.TranslateParticipation(st, nil)
|
|
|
|
require.NoError(t, err)
|
|
|
|
participation, err := newState.PreviousEpochParticipation()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepSSZEqual(t, make([]byte, 64), participation)
|
|
|
|
|
|
|
|
aggBits := bitfield.NewBitlist(2)
|
|
|
|
aggBits.SetBitAt(0, true)
|
|
|
|
aggBits.SetBitAt(1, true)
|
|
|
|
r, err := helpers.BlockRootAtSlot(s, 0)
|
|
|
|
require.NoError(t, err)
|
|
|
|
var pendingAtts []*ethpb.PendingAttestation
|
|
|
|
for i := 0; i < 3; i++ {
|
|
|
|
pendingAtts = append(pendingAtts, ðpb.PendingAttestation{
|
|
|
|
Data: ðpb.AttestationData{
|
|
|
|
CommitteeIndex: types.CommitteeIndex(i),
|
|
|
|
BeaconBlockRoot: r,
|
|
|
|
Source: ðpb.Checkpoint{Epoch: 0, Root: make([]byte, 32)},
|
|
|
|
Target: ðpb.Checkpoint{Epoch: 0, Root: make([]byte, 32)},
|
|
|
|
},
|
|
|
|
AggregationBits: aggBits,
|
|
|
|
InclusionDelay: 1,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
newState, err = altair.TranslateParticipation(newState, pendingAtts)
|
|
|
|
require.NoError(t, err)
|
|
|
|
participation, err = newState.PreviousEpochParticipation()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepNotSSZEqual(t, make([]byte, 64), participation)
|
|
|
|
|
|
|
|
committee, err := helpers.BeaconCommitteeFromState(st, pendingAtts[0].Data.Slot, pendingAtts[0].Data.CommitteeIndex)
|
|
|
|
require.NoError(t, err)
|
2021-09-20 16:17:03 +00:00
|
|
|
indices, err := attestation.AttestingIndices(pendingAtts[0].AggregationBits, committee)
|
2021-08-09 21:27:51 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
for _, index := range indices {
|
|
|
|
require.Equal(t, true, altair.HasValidatorFlag(participation[index], params.BeaconConfig().TimelyHeadFlagIndex))
|
|
|
|
require.Equal(t, true, altair.HasValidatorFlag(participation[index], params.BeaconConfig().TimelyTargetFlagIndex))
|
|
|
|
require.Equal(t, true, altair.HasValidatorFlag(participation[index], params.BeaconConfig().TimelySourceFlagIndex))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestUpgradeToAltair(t *testing.T) {
|
|
|
|
st, _ := testutil.DeterministicGenesisState(t, params.BeaconConfig().MaxValidatorsPerCommittee)
|
|
|
|
preForkState := st.Copy()
|
|
|
|
aState, err := altair.UpgradeToAltair(context.Background(), st)
|
|
|
|
require.NoError(t, err)
|
|
|
|
_, ok := aState.(state.BeaconStateAltair)
|
|
|
|
require.Equal(t, true, ok)
|
|
|
|
|
|
|
|
require.Equal(t, preForkState.GenesisTime(), aState.GenesisTime())
|
|
|
|
require.DeepSSZEqual(t, preForkState.GenesisValidatorRoot(), aState.GenesisValidatorRoot())
|
|
|
|
require.Equal(t, preForkState.Slot(), aState.Slot())
|
|
|
|
require.DeepSSZEqual(t, preForkState.LatestBlockHeader(), aState.LatestBlockHeader())
|
|
|
|
require.DeepSSZEqual(t, preForkState.BlockRoots(), aState.BlockRoots())
|
|
|
|
require.DeepSSZEqual(t, preForkState.StateRoots(), aState.StateRoots())
|
|
|
|
require.DeepSSZEqual(t, preForkState.HistoricalRoots(), aState.HistoricalRoots())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Eth1Data(), aState.Eth1Data())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Eth1DataVotes(), aState.Eth1DataVotes())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Eth1DepositIndex(), aState.Eth1DepositIndex())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Validators(), aState.Validators())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Balances(), aState.Balances())
|
|
|
|
require.DeepSSZEqual(t, preForkState.RandaoMixes(), aState.RandaoMixes())
|
|
|
|
require.DeepSSZEqual(t, preForkState.Slashings(), aState.Slashings())
|
|
|
|
require.DeepSSZEqual(t, preForkState.JustificationBits(), aState.JustificationBits())
|
|
|
|
require.DeepSSZEqual(t, preForkState.PreviousJustifiedCheckpoint(), aState.PreviousJustifiedCheckpoint())
|
|
|
|
require.DeepSSZEqual(t, preForkState.CurrentJustifiedCheckpoint(), aState.CurrentJustifiedCheckpoint())
|
|
|
|
require.DeepSSZEqual(t, preForkState.FinalizedCheckpoint(), aState.FinalizedCheckpoint())
|
|
|
|
numValidators := aState.NumValidators()
|
|
|
|
p, err := aState.PreviousEpochParticipation()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepSSZEqual(t, make([]byte, numValidators), p)
|
|
|
|
p, err = aState.CurrentEpochParticipation()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepSSZEqual(t, make([]byte, numValidators), p)
|
|
|
|
s, err := aState.InactivityScores()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepSSZEqual(t, make([]uint64, numValidators), s)
|
|
|
|
|
|
|
|
f := aState.Fork()
|
|
|
|
require.DeepSSZEqual(t, ðpb.Fork{
|
|
|
|
PreviousVersion: st.Fork().CurrentVersion,
|
|
|
|
CurrentVersion: params.BeaconConfig().AltairForkVersion,
|
2021-09-03 20:10:31 +00:00
|
|
|
Epoch: core.CurrentEpoch(st),
|
2021-08-09 21:27:51 +00:00
|
|
|
}, f)
|
|
|
|
csc, err := aState.CurrentSyncCommittee()
|
|
|
|
require.NoError(t, err)
|
|
|
|
nsc, err := aState.NextSyncCommittee()
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.DeepSSZEqual(t, nsc, csc)
|
|
|
|
}
|