mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 21:27:19 +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
76 lines
2.8 KiB
Go
76 lines
2.8 KiB
Go
package altair_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/core/altair"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/util"
|
|
)
|
|
|
|
func TestProcessEpoch_CanProcess(t *testing.T) {
|
|
st, _ := util.DeterministicGenesisStateAltair(t, params.BeaconConfig().MaxValidatorsPerCommittee)
|
|
require.NoError(t, st.SetSlot(10*params.BeaconConfig().SlotsPerEpoch))
|
|
newState, err := altair.ProcessEpoch(context.Background(), st)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(0), newState.Slashings()[2], "Unexpected slashed balance")
|
|
|
|
b := st.Balances()
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(b)))
|
|
require.Equal(t, uint64(31999841265), b[0])
|
|
|
|
s, err := st.InactivityScores()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(s)))
|
|
|
|
p, err := st.PreviousEpochParticipation()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(p)))
|
|
|
|
p, err = st.CurrentEpochParticipation()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(p)))
|
|
|
|
sc, err := st.CurrentSyncCommittee()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().SyncCommitteeSize, uint64(len(sc.Pubkeys)))
|
|
|
|
sc, err = st.NextSyncCommittee()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().SyncCommitteeSize, uint64(len(sc.Pubkeys)))
|
|
}
|
|
|
|
func TestProcessEpoch_CanProcessBellatrix(t *testing.T) {
|
|
st, _ := util.DeterministicGenesisStateBellatrix(t, params.BeaconConfig().MaxValidatorsPerCommittee)
|
|
require.NoError(t, st.SetSlot(10*params.BeaconConfig().SlotsPerEpoch))
|
|
newState, err := altair.ProcessEpoch(context.Background(), st)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(0), newState.Slashings()[2], "Unexpected slashed balance")
|
|
|
|
b := st.Balances()
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(b)))
|
|
require.Equal(t, uint64(31999839993), b[0])
|
|
|
|
s, err := st.InactivityScores()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(s)))
|
|
|
|
p, err := st.PreviousEpochParticipation()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(p)))
|
|
|
|
p, err = st.CurrentEpochParticipation()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().MaxValidatorsPerCommittee, uint64(len(p)))
|
|
|
|
sc, err := st.CurrentSyncCommittee()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().SyncCommitteeSize, uint64(len(sc.Pubkeys)))
|
|
|
|
sc, err = st.NextSyncCommittee()
|
|
require.NoError(t, err)
|
|
require.Equal(t, params.BeaconConfig().SyncCommitteeSize, uint64(len(sc.Pubkeys)))
|
|
}
|