mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-26 05:17:22 +00:00
2586a9e667
* Add and verify context in AddCommitteeShuffledList * Add and verify context in AddCommitteeShuffledList * Change unnecessary arg reordering * fix build, oops * Regression test * Regression test * fix fuzz cache disabled
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
//go:build !fuzz
|
|
// +build !fuzz
|
|
|
|
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(context.Background(), c))
|
|
_, err := cache.Committee(context.Background(), 0, c.Seed, 0)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
assert.Equal(t, maxCommitteesCacheSize, 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(context.Background(), c))
|
|
|
|
indices, err := cache.ActiveIndices(context.Background(), c.Seed)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, c.SortedIndices, indices)
|
|
}
|
|
|
|
assert.Equal(t, maxCommitteesCacheSize, len(cache.CommitteeCache.Keys()), "Incorrect key size")
|
|
}
|