mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-22 03:30:35 +00:00
5a66807989
* First take at updating everything to v5 * Patch gRPC gateway to use prysm v5 Fix patch * Update go ssz --------- Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"testing"
|
|
|
|
fieldparams "github.com/prysmaticlabs/prysm/v5/config/fieldparams"
|
|
"github.com/prysmaticlabs/prysm/v5/config/params"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v5/testing/require"
|
|
)
|
|
|
|
func TestSyncSubnetIDsCache_Roundtrip(t *testing.T) {
|
|
c := newSyncSubnetIDs()
|
|
|
|
for i := 0; i < 20; i++ {
|
|
pubkey := [fieldparams.BLSPubkeyLength]byte{byte(i)}
|
|
c.AddSyncCommitteeSubnets(pubkey[:], 100, []uint64{uint64(i)}, 0)
|
|
}
|
|
|
|
for i := uint64(0); i < 20; i++ {
|
|
pubkey := [fieldparams.BLSPubkeyLength]byte{byte(i)}
|
|
|
|
idxs, _, ok, _ := c.GetSyncCommitteeSubnets(pubkey[:], 100)
|
|
if !ok {
|
|
t.Errorf("Couldn't find entry in cache for pubkey %#x", pubkey)
|
|
continue
|
|
}
|
|
require.Equal(t, i, idxs[0])
|
|
}
|
|
coms := c.GetAllSubnets(100)
|
|
assert.Equal(t, 20, len(coms))
|
|
}
|
|
|
|
func TestSyncSubnetIDsCache_ValidateCurrentEpoch(t *testing.T) {
|
|
c := newSyncSubnetIDs()
|
|
|
|
for i := 0; i < 20; i++ {
|
|
pubkey := [fieldparams.BLSPubkeyLength]byte{byte(i)}
|
|
c.AddSyncCommitteeSubnets(pubkey[:], 100, []uint64{uint64(i)}, 0)
|
|
}
|
|
|
|
coms := c.GetAllSubnets(50)
|
|
assert.Equal(t, 0, len(coms))
|
|
|
|
for i := uint64(0); i < 20; i++ {
|
|
pubkey := [fieldparams.BLSPubkeyLength]byte{byte(i)}
|
|
|
|
_, jEpoch, ok, _ := c.GetSyncCommitteeSubnets(pubkey[:], 100)
|
|
if !ok {
|
|
t.Errorf("Couldn't find entry in cache for pubkey %#x", pubkey)
|
|
continue
|
|
}
|
|
require.Equal(t, true, uint64(jEpoch) >= 100-params.BeaconConfig().SyncCommitteeSubnetCount)
|
|
}
|
|
|
|
coms = c.GetAllSubnets(99)
|
|
assert.Equal(t, 20, len(coms))
|
|
}
|