mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-08 21:08:10 -05:00
2fd6bd8150
* Ran gopls modernize to fix everything go run golang.org/x/tools/gopls/internal/analysis/modernize/cmd/modernize@latest -fix -test ./... * Override rules_go provided dependency for golang.org/x/tools to v0.38.0. To update this, checked out rules_go, then ran `bazel run //go/tools/releaser -- upgrade-dep -mirror=false org_golang_x_tools` and copied the patches. * Fix buildtag violations and ignore buildtag violations in external * Introduce modernize analyzer package. * Add modernize "any" analyzer. * Fix violations of any analyzer * Add modernize "appendclipped" analyzer. * Fix violations of appendclipped * Add modernize "bloop" analyzer. * Add modernize "fmtappendf" analyzer. * Add modernize "forvar" analyzer. * Add modernize "mapsloop" analyzer. * Add modernize "minmax" analyzer. * Fix violations of minmax analyzer * Add modernize "omitzero" analyzer. * Add modernize "rangeint" analyzer. * Fix violations of rangeint. * Add modernize "reflecttypefor" analyzer. * Fix violations of reflecttypefor analyzer. * Add modernize "slicescontains" analyzer. * Add modernize "slicessort" analyzer. * Add modernize "slicesdelete" analyzer. This is disabled by default for now. See https://go.dev/issue/73686. * Add modernize "stringscutprefix" analyzer. * Add modernize "stringsbuilder" analyzer. * Fix violations of stringsbuilder analyzer. * Add modernize "stringsseq" analyzer. * Add modernize "testingcontext" analyzer. * Add modernize "waitgroup" analyzer. * Changelog fragment * gofmt * gazelle * Add modernize "newexpr" analyzer. * Disable newexpr until go1.26 * Add more details in WORKSPACE on how to update the override * @nalepae feedback on min() * gofmt * Fix violations of forvar
197 lines
9.8 KiB
Go
197 lines
9.8 KiB
Go
package stateutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/OffchainLabs/prysm/v7/config/params"
|
|
multi_value_slice "github.com/OffchainLabs/prysm/v7/container/multi-value-slice"
|
|
ethpb "github.com/OffchainLabs/prysm/v7/proto/prysm/v1alpha1"
|
|
"github.com/OffchainLabs/prysm/v7/testing/require"
|
|
)
|
|
|
|
func TestState_UnrealizedCheckpointBalances(t *testing.T) {
|
|
validators := make([]*ethpb.Validator, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
targetFlag := params.BeaconConfig().TimelyTargetFlagIndex
|
|
expectedActive := params.BeaconConfig().MinGenesisActiveValidatorCount * params.BeaconConfig().MaxEffectiveBalance
|
|
|
|
balances := make([]uint64, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
for i := range validators {
|
|
validators[i] = ðpb.Validator{
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
}
|
|
balances[i] = params.BeaconConfig().MaxEffectiveBalance
|
|
}
|
|
cp := make([]byte, len(validators))
|
|
pp := make([]byte, len(validators))
|
|
|
|
t.Run("No one voted last two epochs", func(tt *testing.T) {
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 0)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("bad votes in last two epochs", func(tt *testing.T) {
|
|
copy(cp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00})
|
|
copy(pp, []byte{0x00, 0x00, 0x00, 0x00})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("two votes in last epoch", func(tt *testing.T) {
|
|
copy(cp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 1 << targetFlag, 1 << targetFlag})
|
|
copy(pp, []byte{0x00, 0x00, 0x00, 0x00, 0xFF ^ (1 << targetFlag)})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("two votes in previous epoch", func(tt *testing.T) {
|
|
copy(cp, []byte{0x00, 0x00, 0x00, 0x00, 0xFF ^ (1 << targetFlag), 0x00})
|
|
copy(pp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 1 << targetFlag, 1 << targetFlag})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
|
|
t.Run("votes in both epochs, decreased balance in first validator", func(tt *testing.T) {
|
|
validators[0].EffectiveBalance = params.BeaconConfig().MaxEffectiveBalance - params.BeaconConfig().MinDepositAmount
|
|
copy(cp, []byte{0xFF, 0xFF, 0x00, 0x00, 0xFF ^ (1 << targetFlag), 0})
|
|
copy(pp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 0xFF, 0xFF})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 1)
|
|
require.NoError(tt, err)
|
|
expectedActive -= params.BeaconConfig().MinDepositAmount
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
|
|
t.Run("slash a validator", func(tt *testing.T) {
|
|
validators[1].Slashed = true
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
t.Run("Exit a validator", func(tt *testing.T) {
|
|
validators[4].ExitEpoch = 1
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValSliceReader(validators), 2)
|
|
require.NoError(tt, err)
|
|
expectedActive -= params.BeaconConfig().MaxEffectiveBalance
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
}
|
|
|
|
func TestState_MVSlice_UnrealizedCheckpointBalances(t *testing.T) {
|
|
validators := make([]*ethpb.Validator, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
targetFlag := params.BeaconConfig().TimelyTargetFlagIndex
|
|
expectedActive := params.BeaconConfig().MinGenesisActiveValidatorCount * params.BeaconConfig().MaxEffectiveBalance
|
|
|
|
balances := make([]uint64, params.BeaconConfig().MinGenesisActiveValidatorCount)
|
|
for i := range validators {
|
|
validators[i] = ðpb.Validator{
|
|
ExitEpoch: params.BeaconConfig().FarFutureEpoch,
|
|
EffectiveBalance: params.BeaconConfig().MaxEffectiveBalance,
|
|
}
|
|
balances[i] = params.BeaconConfig().MaxEffectiveBalance
|
|
}
|
|
|
|
mv := &multi_value_slice.Slice[*ethpb.Validator]{}
|
|
mv.Init(validators)
|
|
|
|
cp := make([]byte, len(validators))
|
|
pp := make([]byte, len(validators))
|
|
|
|
t.Run("No one voted last two epochs", func(tt *testing.T) {
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 0)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("bad votes in last two epochs", func(tt *testing.T) {
|
|
copy(cp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00})
|
|
copy(pp, []byte{0x00, 0x00, 0x00, 0x00})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("two votes in last epoch", func(tt *testing.T) {
|
|
copy(cp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 1 << targetFlag, 1 << targetFlag})
|
|
copy(pp, []byte{0x00, 0x00, 0x00, 0x00, 0xFF ^ (1 << targetFlag)})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, current)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, previous)
|
|
})
|
|
|
|
t.Run("two votes in previous epoch", func(tt *testing.T) {
|
|
copy(cp, []byte{0x00, 0x00, 0x00, 0x00, 0xFF ^ (1 << targetFlag), 0x00})
|
|
copy(pp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 1 << targetFlag, 1 << targetFlag})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().EffectiveBalanceIncrement, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
|
|
t.Run("votes in both epochs, decreased balance in first validator", func(tt *testing.T) {
|
|
validators[0].EffectiveBalance = params.BeaconConfig().MaxEffectiveBalance - params.BeaconConfig().MinDepositAmount
|
|
copy(cp, []byte{0xFF, 0xFF, 0x00, 0x00, 0xFF ^ (1 << targetFlag), 0})
|
|
copy(pp, []byte{0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0xFF ^ (1 << targetFlag), 0x00, 0xFF, 0xFF})
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 1)
|
|
require.NoError(tt, err)
|
|
expectedActive -= params.BeaconConfig().MinDepositAmount
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
|
|
t.Run("slash a validator", func(tt *testing.T) {
|
|
validators[1].Slashed = true
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 1)
|
|
require.NoError(tt, err)
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, 2*params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
t.Run("Exit a validator", func(tt *testing.T) {
|
|
validators[4].ExitEpoch = 1
|
|
active, previous, current, err := UnrealizedCheckpointBalances(cp, pp, NewValMultiValueSliceReader(mv, &testObject{id: 0}), 2)
|
|
require.NoError(tt, err)
|
|
expectedActive -= params.BeaconConfig().MaxEffectiveBalance
|
|
require.Equal(tt, expectedActive, active)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance-params.BeaconConfig().MinDepositAmount, current)
|
|
require.Equal(tt, params.BeaconConfig().MaxEffectiveBalance, previous)
|
|
})
|
|
}
|
|
|
|
type testObject struct {
|
|
id uint64
|
|
}
|
|
|
|
func (o *testObject) Id() uint64 {
|
|
return o.id
|
|
}
|
|
|
|
func (o *testObject) SetId(id uint64) {
|
|
o.id = id
|
|
}
|