mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-24 12:27:18 +00:00
3cf385fe91
* unrealized justification API * Add time elapse logging * add unrealized justification checkpoint * Use UnrealizedJustificationCheckpoint * Refactor unrealized checkpoints * Move logic to state package * do not use ctx on a sum * fix ctx * add tests * fix conflicts * unhandled error * Fix ordering in computing checkpoints * gaz * keep finalized checkpoint if nothing justified * gaz * copy checkpoint * fix check for nil * Add state package tests * Add tests * Radek's review * add more tests * Update beacon-chain/core/epoch/precompute/justification_finalization.go Co-authored-by: terencechain <terence@prysmaticlabs.com> * deduplicate to stateutil * missing file * Add stateutil test * Minor refactor, don't export certain things * Fix exports in tests * remove unused error Co-authored-by: terence tsao <terence@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
65 lines
2.4 KiB
Go
65 lines
2.4 KiB
Go
package v2
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/config/params"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
)
|
|
|
|
func TestState_UnrealizedCheckpointBalances(t *testing.T) {
|
|
validators := make([]*ethpb.Validator, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
balances := make([]uint64, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
for i := 0; i < len(validators); i++ {
|
|
validators[i] = ðpb.Validator{
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
}
|
|
balances[i] = params.BeaconConfig().MaxEffectiveBalance
|
|
}
|
|
base := ðpb.BeaconStateAltair{
|
|
Slot: 2,
|
|
RandaoMixes: make([][]byte, params.BeaconConfig().EpochsPerHistoricalVector),
|
|
|
|
Validators: validators,
|
|
CurrentEpochParticipation: make([]byte, params.BeaconConfig().MinGenesisActiveValidatorCount),
|
|
PreviousEpochParticipation: make([]byte, params.BeaconConfig().MinGenesisActiveValidatorCount),
|
|
Balances: balances,
|
|
}
|
|
state, err := InitializeFromProto(base)
|
|
require.NoError(t, err)
|
|
|
|
// No one voted in the last two epochs
|
|
allActive := params.BeaconConfig().MinGenesisActiveValidatorCount * params.BeaconConfig().MaxEffectiveBalance
|
|
active, previous, current, err := state.UnrealizedCheckpointBalances()
|
|
require.NoError(t, err)
|
|
require.Equal(t, allActive, active)
|
|
require.Equal(t, uint64(0), current)
|
|
require.Equal(t, uint64(0), previous)
|
|
|
|
// Add some votes in the last two epochs:
|
|
base.CurrentEpochParticipation[0] = 0xFF
|
|
base.PreviousEpochParticipation[0] = 0xFF
|
|
base.PreviousEpochParticipation[1] = 0xFF
|
|
|
|
state, err = InitializeFromProto(base)
|
|
require.NoError(t, err)
|
|
active, previous, current, err = state.UnrealizedCheckpointBalances()
|
|
require.NoError(t, err)
|
|
require.Equal(t, allActive, active)
|
|
require.Equal(t, params.BeaconConfig().MaxEffectiveBalance, current)
|
|
require.Equal(t, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
|
|
// Slash some validators
|
|
validators[0].Slashed = true
|
|
state, err = InitializeFromProto(base)
|
|
require.NoError(t, err)
|
|
active, previous, current, err = state.UnrealizedCheckpointBalances()
|
|
require.NoError(t, err)
|
|
require.Equal(t, allActive-params.BeaconConfig().MaxEffectiveBalance, active)
|
|
require.Equal(t, uint64(0), current)
|
|
require.Equal(t, params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
|
|
}
|