mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
ba07ccb484
* slasher/beaconclient tests * slasher/db/kv tests * Merge branch 'master' into apply-testutils-assertions-to-slasher * fix build * slasher/detection tests * rest of the tests * misc tests * tools tests * Merge refs/heads/master into apply-testutils-assertions-misc * Merge branch 'master' into apply-testutils-assertions-misc * agg tests * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge branch 'master' into apply-testutils-assertions-misc * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc * updates aggregated_test * beacon-chain/operations/attestations/kv/* tests updated * beacon-chain/operations/attestations tests updated * beacon-chain/operations/slashings tests updated * Merge branch 'master' into apply-testutils-assertions-misc * gazelle * beacon-chain/core tests updated * fixes test * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc * beacon-chain/rpc tests updated * beacon-chain/sync/initial-sync tests * misc tests * optimizes error message parsing in testutils * Merge branch 'assertutils-optimize-processing' into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * endtoend tests * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc * gazelle * Merge refs/heads/master into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * beacon-chain/blockchain tests updated * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc * beacon-chain/state/stategen tests updated * beacon-chain all left-over tests are done * Merge refs/heads/master into apply-testutils-assertions-misc * validator tests updated * slasher tests * Merge branch 'master' into apply-testutils-assertions-misc * gofmt * gazelle * Merge refs/heads/master into apply-testutils-assertions-misc * shared upd * end2end tests deps fixed * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc * Merge refs/heads/master into apply-testutils-assertions-misc * misc * all tests are updated * Merge branch 'apply-testutils-assertions-misc' of github.com:prysmaticlabs/prysm into apply-testutils-assertions-misc
114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"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"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func TestDeleteAttsInPool(t *testing.T) {
|
|
r := &Service{
|
|
attPool: attestations.NewPool(),
|
|
}
|
|
data := ðpb.AttestationData{}
|
|
att1 := ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1101}, Data: data}
|
|
att2 := ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1110}, Data: data}
|
|
att3 := ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1011}, Data: data}
|
|
att4 := ðpb.Attestation{AggregationBits: bitfield.Bitlist{0b1001}, Data: data}
|
|
require.NoError(t, r.attPool.SaveAggregatedAttestation(att1))
|
|
require.NoError(t, r.attPool.SaveAggregatedAttestation(att2))
|
|
require.NoError(t, r.attPool.SaveAggregatedAttestation(att3))
|
|
require.NoError(t, r.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.attPool.AggregatedAttestations(), "Did not get wanted attestations")
|
|
}
|
|
|
|
func TestService_beaconBlockSubscriber(t *testing.T) {
|
|
pooledAttestations := []*ethpb.Attestation{
|
|
// Aggregated.
|
|
{
|
|
AggregationBits: bitfield.Bitlist{0b00011111},
|
|
Data: ðpb.AttestationData{},
|
|
},
|
|
// Unaggregated.
|
|
{
|
|
AggregationBits: bitfield.Bitlist{0b00010001},
|
|
Data: ðpb.AttestationData{},
|
|
},
|
|
}
|
|
|
|
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: ðpb.SignedBeaconBlock{
|
|
Block: ðpb.BeaconBlock{
|
|
// An empty block will return an err in mocked chainService.ReceiveBlock.
|
|
Body: ðpb.BeaconBlockBody{Attestations: pooledAttestations},
|
|
},
|
|
},
|
|
},
|
|
wantedErr: "nil inner state",
|
|
check: func(t *testing.T, s *Service) {
|
|
if s.attPool.AggregatedAttestationCount() == 0 {
|
|
t.Error("Expected at least 1 aggregated attestation in the pool")
|
|
}
|
|
if s.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{
|
|
chain: &chainMock.ChainService{
|
|
DB: db,
|
|
},
|
|
attPool: attestations.NewPool(),
|
|
}
|
|
assert.NoError(t, s.initCaches())
|
|
// Set up attestation pool.
|
|
for _, att := range pooledAttestations {
|
|
if helpers.IsAggregated(att) {
|
|
assert.NoError(t, s.attPool.SaveAggregatedAttestation(att))
|
|
} else {
|
|
assert.NoError(t, s.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)
|
|
}
|
|
})
|
|
}
|
|
}
|