mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
77d3ccb9ad
* Clean up state types * rename package
205 lines
5.9 KiB
Go
205 lines
5.9 KiB
Go
package cache_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v3/beacon-chain/cache"
|
|
"github.com/prysmaticlabs/prysm/v3/consensus-types/primitives"
|
|
ethpb "github.com/prysmaticlabs/prysm/v3/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/util"
|
|
)
|
|
|
|
func TestSyncCommitteeCache_CanUpdateAndRetrieve(t *testing.T) {
|
|
numValidators := 101
|
|
deterministicState, _ := util.DeterministicGenesisStateAltair(t, uint64(numValidators))
|
|
pubKeys := make([][]byte, deterministicState.NumValidators())
|
|
for i, val := range deterministicState.Validators() {
|
|
pubKeys[i] = val.PublicKey
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
currentSyncCommittee *ethpb.SyncCommittee
|
|
nextSyncCommittee *ethpb.SyncCommittee
|
|
currentSyncMap map[primitives.ValidatorIndex][]primitives.CommitteeIndex
|
|
nextSyncMap map[primitives.ValidatorIndex][]primitives.CommitteeIndex
|
|
}{
|
|
{
|
|
name: "only current epoch",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[1], pubKeys[2], pubKeys[3], pubKeys[2], pubKeys[2],
|
|
}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {0},
|
|
2: {1, 3, 4},
|
|
3: {2},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {},
|
|
2: {},
|
|
3: {},
|
|
},
|
|
},
|
|
{
|
|
name: "only next epoch",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[1], pubKeys[2], pubKeys[3], pubKeys[2], pubKeys[2],
|
|
}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {},
|
|
2: {},
|
|
3: {},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {0},
|
|
2: {1, 3, 4},
|
|
3: {2},
|
|
},
|
|
},
|
|
{
|
|
name: "some current epoch and some next epoch",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[1],
|
|
pubKeys[2],
|
|
pubKeys[3],
|
|
pubKeys[2],
|
|
pubKeys[2],
|
|
}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[7],
|
|
pubKeys[6],
|
|
pubKeys[5],
|
|
pubKeys[4],
|
|
pubKeys[7],
|
|
}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {0},
|
|
2: {1, 3, 4},
|
|
3: {2},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
7: {0, 4},
|
|
6: {1},
|
|
5: {2},
|
|
4: {3},
|
|
},
|
|
},
|
|
{
|
|
name: "some current epoch and some next epoch duplicated across",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[1],
|
|
pubKeys[2],
|
|
pubKeys[3],
|
|
pubKeys[2],
|
|
pubKeys[2],
|
|
}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[2],
|
|
pubKeys[1],
|
|
pubKeys[3],
|
|
pubKeys[2],
|
|
pubKeys[1],
|
|
}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {0},
|
|
2: {1, 3, 4},
|
|
3: {2},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {1, 4},
|
|
2: {0, 3},
|
|
3: {2},
|
|
},
|
|
},
|
|
{
|
|
name: "all duplicated",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
100: {0, 1, 2, 3},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
100: {0, 1, 2, 3},
|
|
},
|
|
},
|
|
{
|
|
name: "unknown keys",
|
|
currentSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
}),
|
|
nextSyncCommittee: util.ConvertToCommittee([][]byte{
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
pubKeys[100],
|
|
}),
|
|
currentSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {},
|
|
},
|
|
nextSyncMap: map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s, _ := util.DeterministicGenesisStateAltair(t, uint64(numValidators))
|
|
require.NoError(t, s.SetCurrentSyncCommittee(tt.currentSyncCommittee))
|
|
require.NoError(t, s.SetNextSyncCommittee(tt.nextSyncCommittee))
|
|
c := cache.NewSyncCommittee()
|
|
r := [32]byte{'a'}
|
|
require.NoError(t, c.UpdatePositionsInCommittee(r, s))
|
|
for key, indices := range tt.currentSyncMap {
|
|
pos, err := c.CurrentPeriodIndexPosition(r, key)
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, indices, pos)
|
|
}
|
|
for key, indices := range tt.nextSyncMap {
|
|
pos, err := c.NextPeriodIndexPosition(r, key)
|
|
require.NoError(t, err)
|
|
require.DeepEqual(t, indices, pos)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSyncCommitteeCache_RootDoesNotExist(t *testing.T) {
|
|
c := cache.NewSyncCommittee()
|
|
_, err := c.CurrentPeriodIndexPosition([32]byte{}, 0)
|
|
require.Equal(t, cache.ErrNonExistingSyncCommitteeKey, err)
|
|
}
|
|
|
|
func TestSyncCommitteeCache_CanRotate(t *testing.T) {
|
|
c := cache.NewSyncCommittee()
|
|
s, _ := util.DeterministicGenesisStateAltair(t, 64)
|
|
require.NoError(t, s.SetCurrentSyncCommittee(util.ConvertToCommittee([][]byte{{1}})))
|
|
require.NoError(t, c.UpdatePositionsInCommittee([32]byte{'a'}, s))
|
|
require.NoError(t, s.SetCurrentSyncCommittee(util.ConvertToCommittee([][]byte{{2}})))
|
|
require.NoError(t, c.UpdatePositionsInCommittee([32]byte{'b'}, s))
|
|
require.NoError(t, s.SetCurrentSyncCommittee(util.ConvertToCommittee([][]byte{{3}})))
|
|
require.NoError(t, c.UpdatePositionsInCommittee([32]byte{'c'}, s))
|
|
require.NoError(t, s.SetCurrentSyncCommittee(util.ConvertToCommittee([][]byte{{4}})))
|
|
require.NoError(t, c.UpdatePositionsInCommittee([32]byte{'d'}, s))
|
|
|
|
_, err := c.CurrentPeriodIndexPosition([32]byte{'a'}, 0)
|
|
require.Equal(t, cache.ErrNonExistingSyncCommitteeKey, err)
|
|
|
|
_, err = c.CurrentPeriodIndexPosition([32]byte{'c'}, 0)
|
|
require.NoError(t, err)
|
|
}
|