mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
eef1730f9e
* Add bellatrix sync changes * Fix tests * Update validate_beacon_blocks.go * check state and block versions * Sync with develop * Update rpc_chunked_response_test.go Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
176 lines
5.5 KiB
Go
176 lines
5.5 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/go-bitfield"
|
|
chainMock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/core/helpers"
|
|
dbtest "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/operations/attestations"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/block"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/wrapper"
|
|
"github.com/prysmaticlabs/prysm/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
"github.com/prysmaticlabs/prysm/testing/util"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func TestDeleteAttsInPool(t *testing.T) {
|
|
r := &Service{
|
|
cfg: &config{attPool: attestations.NewPool()},
|
|
}
|
|
|
|
att1 := util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1101}})
|
|
att2 := util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1110}})
|
|
att3 := util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1011}})
|
|
att4 := util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1001}})
|
|
require.NoError(t, r.cfg.attPool.SaveAggregatedAttestation(att1))
|
|
require.NoError(t, r.cfg.attPool.SaveAggregatedAttestation(att2))
|
|
require.NoError(t, r.cfg.attPool.SaveAggregatedAttestation(att3))
|
|
require.NoError(t, r.cfg.attPool.SaveUnaggregatedAttestation(att4))
|
|
|
|
// Seen 1, 3 and 4 in block.
|
|
require.NoError(t, r.deleteAttsInPool([]*ethpb.Attestation{att1, att3, att4}))
|
|
|
|
// Only 2 should remain.
|
|
assert.DeepEqual(t, []*ethpb.Attestation{att2}, r.cfg.attPool.AggregatedAttestations(), "Did not get wanted attestations")
|
|
}
|
|
|
|
func TestService_beaconBlockSubscriber(t *testing.T) {
|
|
pooledAttestations := []*ethpb.Attestation{
|
|
// Aggregated.
|
|
util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b00011111}}),
|
|
// Unaggregated.
|
|
util.HydrateAttestation(ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b00010001}}),
|
|
}
|
|
|
|
type args struct {
|
|
msg proto.Message
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantedErr string
|
|
check func(*testing.T, *Service)
|
|
}{
|
|
{
|
|
name: "invalid block does not remove attestations",
|
|
args: args{
|
|
msg: func() *ethpb.SignedBeaconBlock {
|
|
b := util.NewBeaconBlock()
|
|
b.Block.Body.Attestations = pooledAttestations
|
|
return b
|
|
}(),
|
|
},
|
|
wantedErr: "nil inner state",
|
|
check: func(t *testing.T, s *Service) {
|
|
if s.cfg.attPool.AggregatedAttestationCount() == 0 {
|
|
t.Error("Expected at least 1 aggregated attestation in the pool")
|
|
}
|
|
if s.cfg.attPool.UnaggregatedAttestationCount() == 0 {
|
|
t.Error("Expected at least 1 unaggregated attestation in the pool")
|
|
}
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
db := dbtest.SetupDB(t)
|
|
s := &Service{
|
|
cfg: &config{
|
|
chain: &chainMock.ChainService{
|
|
DB: db,
|
|
Root: make([]byte, 32),
|
|
},
|
|
attPool: attestations.NewPool(),
|
|
},
|
|
}
|
|
s.initCaches()
|
|
// Set up attestation pool.
|
|
for _, att := range pooledAttestations {
|
|
if helpers.IsAggregated(att) {
|
|
assert.NoError(t, s.cfg.attPool.SaveAggregatedAttestation(att))
|
|
} else {
|
|
assert.NoError(t, s.cfg.attPool.SaveUnaggregatedAttestation(att))
|
|
}
|
|
}
|
|
// Perform method under test call.
|
|
err := s.beaconBlockSubscriber(context.Background(), tt.args.msg)
|
|
if tt.wantedErr != "" {
|
|
assert.ErrorContains(t, tt.wantedErr, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
}
|
|
if tt.check != nil {
|
|
tt.check(t, s)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestBlockFromProto(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
msgCreator func(t *testing.T) proto.Message
|
|
want block.SignedBeaconBlock
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "invalid type provided",
|
|
msgCreator: func(t *testing.T) proto.Message {
|
|
return ðpb.SignedAggregateAttestationAndProof{}
|
|
},
|
|
want: nil,
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "phase 0 type provided",
|
|
msgCreator: func(t *testing.T) proto.Message {
|
|
return ðpb.SignedBeaconBlock{Block: ðpb.BeaconBlock{Slot: 100}}
|
|
},
|
|
want: wrapper.WrappedPhase0SignedBeaconBlock(ðpb.SignedBeaconBlock{Block: ðpb.BeaconBlock{Slot: 100}}),
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "altair type provided",
|
|
msgCreator: func(t *testing.T) proto.Message {
|
|
return ðpb.SignedBeaconBlockAltair{Block: ðpb.BeaconBlockAltair{Slot: 100}}
|
|
},
|
|
want: func() block.SignedBeaconBlock {
|
|
wsb, err := wrapper.WrappedAltairSignedBeaconBlock(ðpb.SignedBeaconBlockAltair{Block: ðpb.BeaconBlockAltair{Slot: 100}})
|
|
require.NoError(t, err)
|
|
return wsb
|
|
}(),
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "bellatrix type provided",
|
|
msgCreator: func(t *testing.T) proto.Message {
|
|
return ðpb.SignedBeaconBlockBellatrix{Block: ðpb.BeaconBlockBellatrix{Slot: 100}}
|
|
},
|
|
want: func() block.SignedBeaconBlock {
|
|
wsb, err := wrapper.WrappedBellatrixSignedBeaconBlock(ðpb.SignedBeaconBlockBellatrix{Block: ðpb.BeaconBlockBellatrix{Slot: 100}})
|
|
require.NoError(t, err)
|
|
return wsb
|
|
}(),
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := blockFromProto(tt.msgCreator(t))
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("blockFromProto() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("blockFromProto() got = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|