mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 21:07:18 +00:00
376d248c22
* Add in progress handler for committee cache * Remove debug print * Update validators.go * Fix all the tests * More tests * Update committee_disabled.go * Update committee_disabled.go * Update testing util * Update main.go Co-authored-by: Nishant Das <nishdas93@gmail.com>
55 lines
1.3 KiB
Go
55 lines
1.3 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
fuzz "github.com/google/gofuzz"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
)
|
|
|
|
func TestCommitteeKeyFuzz_OK(t *testing.T) {
|
|
fuzzer := fuzz.NewWithSeed(0)
|
|
c := &Committees{}
|
|
|
|
for i := 0; i < 100000; i++ {
|
|
fuzzer.Fuzz(c)
|
|
k, err := committeeKeyFn(c)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, key(c.Seed), k)
|
|
}
|
|
}
|
|
|
|
func TestCommitteeCache_FuzzCommitteesByEpoch(t *testing.T) {
|
|
cache := NewCommitteesCache()
|
|
fuzzer := fuzz.NewWithSeed(0)
|
|
c := &Committees{}
|
|
|
|
for i := 0; i < 100000; i++ {
|
|
fuzzer.Fuzz(c)
|
|
require.NoError(t, cache.AddCommitteeShuffledList(c))
|
|
_, err := cache.Committee(context.Background(), 0, c.Seed, 0)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
assert.Equal(t, maxCommitteesCacheSize, uint64(len(cache.CommitteeCache.Keys())), "Incorrect key size")
|
|
}
|
|
|
|
func TestCommitteeCache_FuzzActiveIndices(t *testing.T) {
|
|
cache := NewCommitteesCache()
|
|
fuzzer := fuzz.NewWithSeed(0)
|
|
c := &Committees{}
|
|
|
|
for i := 0; i < 100000; i++ {
|
|
fuzzer.Fuzz(c)
|
|
require.NoError(t, cache.AddCommitteeShuffledList(c))
|
|
|
|
indices, err := cache.ActiveIndices(context.Background(), c.Seed)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, c.SortedIndices, indices)
|
|
}
|
|
|
|
assert.Equal(t, maxCommitteesCacheSize, uint64(len(cache.CommitteeCache.Keys())), "Incorrect key size")
|
|
}
|