prysm-pulse/beacon-chain/sync/subscriber_beacon_blocks_test.go
Raul Jordan 5aac06f04e
Move EthereumAPIs Into Prysm (#8968)
* begin move

* use same import path

* imports

* regen protos

* regen

* no rename

* generate ssz

* gaz

* fmt

* edit build file

* imports

* modify

* remove generated files

* remove protos

* edit imports in prysm

* beacon chain all builds

* edit script

* add generated pbs

* add replace rules

* license for ethereumapis protos

* change visibility

* fmt

* update build files to gaz ignore

* use proper form

* edit imports

* wrap block

* revert scripts

* revert go mod
2021-06-02 18:49:52 -05:00

111 lines
3.5 KiB
Go

package sync
import (
"context"
"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/eth/v1alpha1"
"github.com/prysmaticlabs/prysm/shared/testutil"
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
"github.com/prysmaticlabs/prysm/shared/testutil/require"
"google.golang.org/protobuf/proto"
)
func TestDeleteAttsInPool(t *testing.T) {
r := &Service{
cfg: &Config{AttPool: attestations.NewPool()},
}
att1 := testutil.HydrateAttestation(&ethpb.Attestation{AggregationBits: bitfield.Bitlist{0b1101}})
att2 := testutil.HydrateAttestation(&ethpb.Attestation{AggregationBits: bitfield.Bitlist{0b1110}})
att3 := testutil.HydrateAttestation(&ethpb.Attestation{AggregationBits: bitfield.Bitlist{0b1011}})
att4 := testutil.HydrateAttestation(&ethpb.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.
testutil.HydrateAttestation(&ethpb.Attestation{AggregationBits: bitfield.Bitlist{0b00011111}}),
// Unaggregated.
testutil.HydrateAttestation(&ethpb.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 := testutil.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(),
},
}
assert.NoError(t, 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)
}
})
}
}