mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
918129cf36
* refactor initialization to blocking startup method * require genesisSetter in blockchain, fix tests * work-around gazelle weirdness * fix dep gazelle ignores * only call SetGenesis once * fix typo * validator test setup and fix to return right error * move waitForChainStart to Start * wire up sync Service.genesisWaiter * fix p2p genesisWaiter plumbing * remove extra clock type, integrate into genesis and rename * use time.Now when no Nower is specified * remove unused ClockSetter * simplify rpc context checking * fix typo * use clock everywhere in sync; [32]byte val root * don't use DeepEqual to compare [32]byte and []byte * don't use clock in init sync, not wired up yet * use clock waiter in blockchain as well * use cancelable contexts in tests with goroutines * missed a reference to WithClockSetter * Update beacon-chain/startup/genesis.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * Update beacon-chain/blockchain/service_test.go Co-authored-by: Radosław Kapka <rkapka@wp.pl> * more clear docs * doc for NewClock * move clock typedef to more logical file name * adding documentation * gaz * fixes for capella * reducing test raciness * fix races in committee cache tests * lint * add tests on Duration slot math helper * startup package test coverage * fix bad merge * set non-zero genesis time in tests that call Start * happy deepsource, happy me-epsource * replace Synced event with channel * remove unused error * remove accidental wip commit * gaz! * remove unused event constants * remove sync statefeed subscription to fix deadlock * remove state notifier * fix build --------- Co-authored-by: Kasey Kirkham <kasey@users.noreply.github.com> Co-authored-by: Radosław Kapka <rkapka@wp.pl> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> Co-authored-by: nisdas <nishdas93@gmail.com>
159 lines
5.6 KiB
Go
159 lines
5.6 KiB
Go
package sync
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
mockChain "github.com/prysmaticlabs/prysm/v4/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/core/signing"
|
|
testingdb "github.com/prysmaticlabs/prysm/v4/beacon-chain/db/testing"
|
|
doublylinkedtree "github.com/prysmaticlabs/prysm/v4/beacon-chain/forkchoice/doubly-linked-tree"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/operations/blstoexec"
|
|
mockp2p "github.com/prysmaticlabs/prysm/v4/beacon-chain/p2p/testing"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/state/stategen"
|
|
mockSync "github.com/prysmaticlabs/prysm/v4/beacon-chain/sync/initial-sync/testing"
|
|
"github.com/prysmaticlabs/prysm/v4/config/params"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/util"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func TestBroadcastBLSChanges(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
c := params.BeaconConfig()
|
|
c.CapellaForkEpoch = c.BellatrixForkEpoch.Add(2)
|
|
params.OverrideBeaconConfig(c)
|
|
chainService := &mockChain.ChainService{
|
|
Genesis: time.Now(),
|
|
ValidatorsRoot: [32]byte{'A'},
|
|
}
|
|
s := NewService(context.Background(),
|
|
WithP2P(mockp2p.NewTestP2P(t)),
|
|
WithInitialSync(&mockSync.Sync{IsSyncing: false}),
|
|
WithChainService(chainService),
|
|
WithOperationNotifier(chainService.OperationNotifier()),
|
|
WithBlsToExecPool(blstoexec.NewPool()),
|
|
)
|
|
var emptySig [96]byte
|
|
s.cfg.blsToExecPool.InsertBLSToExecChange(ðpb.SignedBLSToExecutionChange{
|
|
Message: ðpb.BLSToExecutionChange{
|
|
ValidatorIndex: 10,
|
|
FromBlsPubkey: make([]byte, 48),
|
|
ToExecutionAddress: make([]byte, 20),
|
|
},
|
|
Signature: emptySig[:],
|
|
})
|
|
|
|
capellaStart, err := slots.EpochStart(params.BeaconConfig().CapellaForkEpoch)
|
|
require.NoError(t, err)
|
|
s.broadcastBLSChanges(capellaStart + 1)
|
|
}
|
|
|
|
func TestRateBLSChanges(t *testing.T) {
|
|
logHook := logTest.NewGlobal()
|
|
params.SetupTestConfigCleanup(t)
|
|
c := params.BeaconConfig()
|
|
c.CapellaForkEpoch = c.BellatrixForkEpoch.Add(2)
|
|
params.OverrideBeaconConfig(c)
|
|
chainService := &mockChain.ChainService{
|
|
Genesis: time.Now(),
|
|
ValidatorsRoot: [32]byte{'A'},
|
|
}
|
|
p1 := mockp2p.NewTestP2P(t)
|
|
s := NewService(context.Background(),
|
|
WithP2P(p1),
|
|
WithInitialSync(&mockSync.Sync{IsSyncing: false}),
|
|
WithChainService(chainService),
|
|
WithOperationNotifier(chainService.OperationNotifier()),
|
|
WithBlsToExecPool(blstoexec.NewPool()),
|
|
)
|
|
beaconDB := testingdb.SetupDB(t)
|
|
s.cfg.stateGen = stategen.New(beaconDB, doublylinkedtree.New())
|
|
s.cfg.beaconDB = beaconDB
|
|
s.initCaches()
|
|
st, keys := util.DeterministicGenesisStateCapella(t, 256)
|
|
s.cfg.chain = &mockChain.ChainService{
|
|
ValidatorsRoot: [32]byte{'A'},
|
|
Genesis: time.Now().Add(-time.Second * time.Duration(params.BeaconConfig().SecondsPerSlot) * time.Duration(10)),
|
|
State: st,
|
|
}
|
|
|
|
for i := 0; i < 200; i++ {
|
|
message := ðpb.BLSToExecutionChange{
|
|
ValidatorIndex: primitives.ValidatorIndex(i),
|
|
FromBlsPubkey: keys[i+1].PublicKey().Marshal(),
|
|
ToExecutionAddress: bytesutil.PadTo([]byte("address"), 20),
|
|
}
|
|
epoch := params.BeaconConfig().CapellaForkEpoch + 1
|
|
domain, err := signing.Domain(st.Fork(), epoch, params.BeaconConfig().DomainBLSToExecutionChange, st.GenesisValidatorsRoot())
|
|
assert.NoError(t, err)
|
|
htr, err := signing.SigningData(message.HashTreeRoot, domain)
|
|
assert.NoError(t, err)
|
|
signed := ðpb.SignedBLSToExecutionChange{
|
|
Message: message,
|
|
Signature: keys[i+1].Sign(htr[:]).Marshal(),
|
|
}
|
|
|
|
s.cfg.blsToExecPool.InsertBLSToExecChange(signed)
|
|
}
|
|
|
|
require.Equal(t, false, p1.BroadcastCalled)
|
|
slot, err := slots.EpochStart(params.BeaconConfig().CapellaForkEpoch)
|
|
require.NoError(t, err)
|
|
s.broadcastBLSChanges(slot)
|
|
time.Sleep(100 * time.Millisecond) // Need a sleep for the go routine to be ready
|
|
require.Equal(t, true, p1.BroadcastCalled)
|
|
require.LogsDoNotContain(t, logHook, "could not")
|
|
|
|
p1.BroadcastCalled = false
|
|
time.Sleep(500 * time.Millisecond) // Need a sleep for the second batch to be broadcast
|
|
require.Equal(t, true, p1.BroadcastCalled)
|
|
require.LogsDoNotContain(t, logHook, "could not")
|
|
}
|
|
|
|
func TestBroadcastBLSBatch_changes_slice(t *testing.T) {
|
|
message := ðpb.BLSToExecutionChange{
|
|
FromBlsPubkey: make([]byte, 48),
|
|
ToExecutionAddress: make([]byte, 20),
|
|
}
|
|
signed := ðpb.SignedBLSToExecutionChange{
|
|
Message: message,
|
|
Signature: make([]byte, 96),
|
|
}
|
|
changes := make([]*ethpb.SignedBLSToExecutionChange, 200)
|
|
for i := 0; i < len(changes); i++ {
|
|
changes[i] = signed
|
|
}
|
|
p1 := mockp2p.NewTestP2P(t)
|
|
chainService := &mockChain.ChainService{
|
|
Genesis: time.Now(),
|
|
ValidatorsRoot: [32]byte{'A'},
|
|
}
|
|
s := NewService(context.Background(),
|
|
WithP2P(p1),
|
|
WithInitialSync(&mockSync.Sync{IsSyncing: false}),
|
|
WithChainService(chainService),
|
|
WithOperationNotifier(chainService.OperationNotifier()),
|
|
WithBlsToExecPool(blstoexec.NewPool()),
|
|
)
|
|
beaconDB := testingdb.SetupDB(t)
|
|
s.cfg.stateGen = stategen.New(beaconDB, doublylinkedtree.New())
|
|
s.cfg.beaconDB = beaconDB
|
|
s.initCaches()
|
|
st, _ := util.DeterministicGenesisStateCapella(t, 32)
|
|
s.cfg.chain = &mockChain.ChainService{
|
|
ValidatorsRoot: [32]byte{'A'},
|
|
Genesis: time.Now().Add(-time.Second * time.Duration(params.BeaconConfig().SecondsPerSlot) * time.Duration(10)),
|
|
State: st,
|
|
}
|
|
|
|
s.broadcastBLSBatch(s.ctx, &changes)
|
|
require.Equal(t, 200-128, len(changes))
|
|
}
|