mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-11 04:00:05 +00:00
879e310332
* panic in SizeSSZ * moving slowly * adapt old code to new interfaces * return interfaces from factory functions * replace the rest of WrappedSignedBeaconBlock * WrappedBeaconBlock * WrappedBeaconBlockBody * miscellaneous * Test_BeaconBlockIsNil * replace usages of BeaconBlockIsNil * replace usages of mutator * fix all build errors * fix some more issues * mutator changes * relax assertions when initializing * revert changes in object_mapping.go * allow calling Proto on nil * Revert "allow calling Proto on nil" This reverts commit ecc84e455381b03d24aec2fa0fa17bddbec71705. * modify Copy and Proto methods * remove unused var * fix block batch tests * correct BUILD file * Error when initializing nil objects * one more error fix * add missing comma * rename alias to blocktest * add logging * error when SignedBeaconBlock is nil * fix last test * import fix * broken * working * test fixes * reduce complexity of processPendingBlocks * simplified
86 lines
2.5 KiB
Go
86 lines
2.5 KiB
Go
package sync
|
|
|
|
import (
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/consensus-types/blocks"
|
|
"github.com/prysmaticlabs/prysm/consensus-types/interfaces"
|
|
types "github.com/prysmaticlabs/prysm/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/encoding/bytesutil"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
)
|
|
|
|
func TestSortedObj_SortBlocksRoots(t *testing.T) {
|
|
source := rand.NewSource(33)
|
|
randGen := rand.New(source)
|
|
var blks []interfaces.SignedBeaconBlock
|
|
var roots [][32]byte
|
|
randFunc := func() int64 {
|
|
return randGen.Int63n(50)
|
|
}
|
|
|
|
for i := 0; i < 10; i++ {
|
|
slot := types.Slot(randFunc())
|
|
newBlk, err := blocks.NewSignedBeaconBlock(ðpb.SignedBeaconBlock{Block: ðpb.BeaconBlock{Slot: slot, Body: ðpb.BeaconBlockBody{}}})
|
|
require.NoError(t, err)
|
|
blks = append(blks, newBlk)
|
|
root := bytesutil.ToBytes32(bytesutil.Bytes32(uint64(slot)))
|
|
roots = append(roots, root)
|
|
}
|
|
|
|
r := &Service{}
|
|
|
|
newBlks, newRoots := r.sortBlocksAndRoots(blks, roots)
|
|
|
|
previousSlot := types.Slot(0)
|
|
for i, b := range newBlks {
|
|
if b.Block().Slot() < previousSlot {
|
|
t.Errorf("Block list is not sorted as %d is smaller than previousSlot %d", b.Block().Slot(), previousSlot)
|
|
}
|
|
if bytesutil.FromBytes8(newRoots[i][:]) != uint64(b.Block().Slot()) {
|
|
t.Errorf("root doesn't match stored slot in block: wanted %d but got %d", b.Block().Slot(), bytesutil.FromBytes8(newRoots[i][:]))
|
|
}
|
|
previousSlot = b.Block().Slot()
|
|
}
|
|
}
|
|
|
|
func TestSortedObj_NoDuplicates(t *testing.T) {
|
|
source := rand.NewSource(33)
|
|
randGen := rand.New(source)
|
|
var blks []interfaces.SignedBeaconBlock
|
|
var roots [][32]byte
|
|
randFunc := func() int64 {
|
|
return randGen.Int63n(50)
|
|
}
|
|
|
|
for i := 0; i < 10; i++ {
|
|
slot := types.Slot(randFunc())
|
|
newBlk := ðpb.SignedBeaconBlock{Block: ðpb.BeaconBlock{Slot: slot, Body: ðpb.BeaconBlockBody{}}}
|
|
// append twice
|
|
wsb, err := blocks.NewSignedBeaconBlock(newBlk)
|
|
require.NoError(t, err)
|
|
wsbCopy, err := wsb.Copy()
|
|
require.NoError(t, err)
|
|
blks = append(blks, wsb, wsbCopy)
|
|
|
|
// append twice
|
|
root := bytesutil.ToBytes32(bytesutil.Bytes32(uint64(slot)))
|
|
roots = append(roots, root, root)
|
|
}
|
|
|
|
r := &Service{}
|
|
|
|
newBlks, newRoots, err := r.dedupBlocksAndRoots(blks, roots)
|
|
require.NoError(t, err)
|
|
|
|
rootMap := make(map[[32]byte]bool)
|
|
for i, b := range newBlks {
|
|
if rootMap[newRoots[i]] {
|
|
t.Errorf("Duplicated root exists %#x with block %v", newRoots[i], b)
|
|
}
|
|
rootMap[newRoots[i]] = true
|
|
}
|
|
}
|