mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
5828278807
* add fix and reg test * nogo * nogo
223 lines
4.9 KiB
Go
223 lines
4.9 KiB
Go
package kv
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/prysmaticlabs/go-ssz"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/db/filters"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
)
|
|
|
|
func TestStore_AttestationCRUD(t *testing.T) {
|
|
db := setupDB(t)
|
|
defer teardownDB(t, db)
|
|
att := ðpb.Attestation{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: []byte("parent"),
|
|
StartEpoch: 1,
|
|
EndEpoch: 2,
|
|
},
|
|
},
|
|
}
|
|
ctx := context.Background()
|
|
attDataRoot, err := ssz.HashTreeRoot(att.Data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
retrievedAtt, err := db.Attestation(ctx, attDataRoot)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if retrievedAtt != nil {
|
|
t.Errorf("Expected nil attestation, received %v", retrievedAtt)
|
|
}
|
|
if err := db.SaveAttestation(ctx, att); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !db.HasAttestation(ctx, attDataRoot) {
|
|
t.Error("Expected attestation to exist in the db")
|
|
}
|
|
retrievedAtt, err = db.Attestation(ctx, attDataRoot)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !proto.Equal(att, retrievedAtt) {
|
|
t.Errorf("Wanted %v, received %v", att, retrievedAtt)
|
|
}
|
|
if err := db.DeleteAttestation(ctx, attDataRoot); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if db.HasAttestation(ctx, attDataRoot) {
|
|
t.Error("Expected attestation to have been deleted from the db")
|
|
}
|
|
}
|
|
|
|
func TestStore_BoltDontPanic(t *testing.T) {
|
|
db := setupDB(t)
|
|
defer teardownDB(t, db)
|
|
var wg sync.WaitGroup
|
|
|
|
for i := 0; i <= 100; i++ {
|
|
att := ðpb.Attestation{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: []byte("parent"),
|
|
StartEpoch: uint64(i + 1),
|
|
EndEpoch: 2,
|
|
},
|
|
},
|
|
}
|
|
ctx := context.Background()
|
|
attDataRoot, err := ssz.HashTreeRoot(att.Data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
retrievedAtt, err := db.Attestation(ctx, attDataRoot)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if retrievedAtt != nil {
|
|
t.Errorf("Expected nil attestation, received %v", retrievedAtt)
|
|
}
|
|
if err := db.SaveAttestation(ctx, att); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
// if indices are improperly deleted this test will then panic.
|
|
for i := 0; i <= 100; i++ {
|
|
startEpoch := i + 1
|
|
wg.Add(1)
|
|
go func() {
|
|
att := ðpb.Attestation{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: []byte("parent"),
|
|
StartEpoch: uint64(startEpoch),
|
|
EndEpoch: 2,
|
|
},
|
|
},
|
|
}
|
|
ctx := context.Background()
|
|
attDataRoot, err := ssz.HashTreeRoot(att.Data)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.DeleteAttestation(ctx, attDataRoot); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if db.HasAttestation(ctx, attDataRoot) {
|
|
t.Error("Expected attestation to have been deleted from the db")
|
|
}
|
|
wg.Done()
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
}
|
|
|
|
func TestStore_Attestations_FiltersCorrectly(t *testing.T) {
|
|
db := setupDB(t)
|
|
defer teardownDB(t, db)
|
|
sameParentRoot := [32]byte{1, 2, 3}
|
|
otherParentRoot := [32]byte{4, 5, 6}
|
|
atts := []*ethpb.Attestation{
|
|
{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: sameParentRoot[:],
|
|
StartEpoch: 1,
|
|
EndEpoch: 2,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: sameParentRoot[:],
|
|
StartEpoch: 10,
|
|
EndEpoch: 11,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Data: ðpb.AttestationData{
|
|
Crosslink: ðpb.Crosslink{
|
|
Shard: 5,
|
|
ParentRoot: otherParentRoot[:],
|
|
StartEpoch: 1,
|
|
EndEpoch: 20,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
ctx := context.Background()
|
|
if err := db.SaveAttestations(ctx, atts); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
tests := []struct {
|
|
filter *filters.QueryFilter
|
|
expectedNumAtt int
|
|
}{
|
|
{
|
|
filter: filters.NewFilter().
|
|
SetShard(5),
|
|
expectedNumAtt: 3,
|
|
},
|
|
{
|
|
filter: filters.NewFilter().
|
|
SetShard(5).
|
|
SetParentRoot(otherParentRoot[:]),
|
|
expectedNumAtt: 1,
|
|
},
|
|
{
|
|
filter: filters.NewFilter().
|
|
SetShard(5).
|
|
SetParentRoot(sameParentRoot[:]),
|
|
expectedNumAtt: 2,
|
|
},
|
|
{
|
|
filter: filters.NewFilter().SetStartEpoch(1),
|
|
expectedNumAtt: 2,
|
|
},
|
|
{
|
|
filter: filters.NewFilter().
|
|
SetParentRoot(otherParentRoot[:]),
|
|
expectedNumAtt: 1,
|
|
},
|
|
{
|
|
// Only two attestation in the list meet the composite filter criteria above.
|
|
filter: filters.NewFilter().SetShard(5).SetStartEpoch(1),
|
|
expectedNumAtt: 2,
|
|
},
|
|
{
|
|
// No specified filter should return all attestations.
|
|
filter: nil,
|
|
expectedNumAtt: 3,
|
|
},
|
|
{
|
|
// No attestation meets the criteria below.
|
|
filter: filters.NewFilter().
|
|
SetShard(1000),
|
|
expectedNumAtt: 0,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
retrievedAtts, err := db.Attestations(ctx, tt.filter)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(retrievedAtts) != tt.expectedNumAtt {
|
|
t.Errorf("Expected %d attestations, received %d", tt.expectedNumAtt, len(retrievedAtts))
|
|
}
|
|
}
|
|
}
|