mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-30 23:16:31 +00:00
4f0bef929f
* change to herumi's bls * change alias * change to better * add benchmark * build * change to bazel fork * fix prefix * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * make it work with library * update to latest * change again * add import * update to latest * add sha commit * new static lib with groups swapped * using herumis new lib * fix dep paths in c headers * update again * new changes * fix commit * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * fix serialization * comment * fix test * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * fix to herumis latest version * fix test * fix benchmarks * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * add new workspace * change commit and remove init * get test to pass * remove parameter * remove reverse byte order * make gazelle happy * set pure to off * fix failing tests * Merge branch 'master' into herumiBLS * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * Merge branch 'herumiBLS' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * remove old ref * use HashWithDomain functions * update to latest version * clean up * gaz * add back removed code * switch off pure * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * use local repo * resolve docker issues * Merge branch 'master' of https://github.com/prysmaticlabs/geth-sharding into herumiBLS * fix build and tests * gaz * Merge branch 'master' into herumiBLS * Merge refs/heads/master into herumiBLS * Merge refs/heads/master into herumiBLS
93 lines
2.7 KiB
Go
93 lines
2.7 KiB
Go
package aggregator
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
mock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
|
|
dbutil "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
mockSync "github.com/prysmaticlabs/prysm/beacon-chain/sync/initial-sync/testing"
|
|
pbp2p "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/rpc/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
)
|
|
|
|
func init() {
|
|
// Use minimal config to reduce test setup time.
|
|
params.OverrideBeaconConfig(params.MinimalSpecConfig())
|
|
}
|
|
|
|
func TestSubmitAggregateAndProof_Syncing(t *testing.T) {
|
|
db := dbutil.SetupDB(t)
|
|
defer dbutil.TeardownDB(t, db)
|
|
ctx := context.Background()
|
|
|
|
s := &pbp2p.BeaconState{}
|
|
|
|
aggregatorServer := &Server{
|
|
HeadFetcher: &mock.ChainService{State: s},
|
|
SyncChecker: &mockSync.Sync{IsSyncing: true},
|
|
BeaconDB: db,
|
|
}
|
|
|
|
req := &pb.AggregationRequest{CommitteeIndex: 1}
|
|
wanted := "Syncing to latest head, not ready to respond"
|
|
if _, err := aggregatorServer.SubmitAggregateAndProof(ctx, req); !strings.Contains(err.Error(), wanted) {
|
|
t.Error("Did not receive wanted error")
|
|
}
|
|
}
|
|
|
|
func TestSubmitAggregateAndProof_CantFindValidatorIndex(t *testing.T) {
|
|
db := dbutil.SetupDB(t)
|
|
defer dbutil.TeardownDB(t, db)
|
|
ctx := context.Background()
|
|
|
|
s := &pbp2p.BeaconState{
|
|
RandaoMixes: make([][]byte, params.BeaconConfig().EpochsPerHistoricalVector),
|
|
}
|
|
|
|
aggregatorServer := &Server{
|
|
HeadFetcher: &mock.ChainService{State: s},
|
|
SyncChecker: &mockSync.Sync{IsSyncing: false},
|
|
BeaconDB: db,
|
|
}
|
|
|
|
priv := bls.RandKey()
|
|
sig := priv.Sign([]byte{'A'}, 0)
|
|
req := &pb.AggregationRequest{CommitteeIndex: 1, SlotSignature: sig.Marshal()}
|
|
wanted := "Could not locate validator index in DB"
|
|
if _, err := aggregatorServer.SubmitAggregateAndProof(ctx, req); !strings.Contains(err.Error(), wanted) {
|
|
t.Error("Did not receive wanted error")
|
|
}
|
|
}
|
|
|
|
func TestSubmitAggregateAndProof_IsAggregator(t *testing.T) {
|
|
db := dbutil.SetupDB(t)
|
|
defer dbutil.TeardownDB(t, db)
|
|
ctx := context.Background()
|
|
|
|
s := &pbp2p.BeaconState{
|
|
RandaoMixes: make([][]byte, params.BeaconConfig().EpochsPerHistoricalVector),
|
|
}
|
|
|
|
aggregatorServer := &Server{
|
|
HeadFetcher: &mock.ChainService{State: s},
|
|
SyncChecker: &mockSync.Sync{IsSyncing: false},
|
|
BeaconDB: db,
|
|
}
|
|
|
|
priv := bls.RandKey()
|
|
sig := priv.Sign([]byte{'A'}, 0)
|
|
pubKey := [48]byte{'A'}
|
|
req := &pb.AggregationRequest{CommitteeIndex: 1, SlotSignature: sig.Marshal(), PublicKey: pubKey[:]}
|
|
if err := aggregatorServer.BeaconDB.SaveValidatorIndex(ctx, pubKey, 100); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if _, err := aggregatorServer.SubmitAggregateAndProof(ctx, req); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|