mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
b7175b3482
* Revert "Revert "Update fastssz" (#7100)"
This reverts commit b954db9704
.
* Preston's patch
* Merge branch 'master' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Update fssz, add regression test case
* more HTR with fssz
* fix some tests
* only one test left
* Make it so that HTR will work
* gofmt, imports
* gofmt, imports
* fix
* Merge branch 'master' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* fix
* Merge branch 'master' into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
* gaz
* Merge branch 'revert-7100-revert-6760-update-fssz' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
* fix test
* Merge branch 'revert-7100-revert-6760-update-fssz' of github.com:prysmaticlabs/prysm into revert-7100-revert-6760-update-fssz
* Merge refs/heads/master into revert-7100-revert-6760-update-fssz
182 lines
5.7 KiB
Go
182 lines
5.7 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
testDB "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
"github.com/prysmaticlabs/prysm/shared/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
// Ensure Service implements chain info interface.
|
|
var _ = ChainInfoFetcher(&Service{})
|
|
var _ = TimeFetcher(&Service{})
|
|
var _ = ForkFetcher(&Service{})
|
|
|
|
func TestFinalizedCheckpt_Nil(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
c := setupBeaconChain(t, db, sc)
|
|
assert.DeepEqual(t, params.BeaconConfig().ZeroHash[:], c.FinalizedCheckpt().Root, "Incorrect pre chain start value")
|
|
}
|
|
|
|
func TestHeadRoot_Nil(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
c := setupBeaconChain(t, db, sc)
|
|
headRoot, err := c.HeadRoot(context.Background())
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, params.BeaconConfig().ZeroHash[:], headRoot, "Incorrect pre chain start value")
|
|
}
|
|
|
|
func TestFinalizedCheckpt_CanRetrieve(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
cp := ðpb.Checkpoint{Epoch: 5, Root: bytesutil.PadTo([]byte("foo"), 32)}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.finalizedCheckpt = cp
|
|
|
|
assert.Equal(t, cp.Epoch, c.FinalizedCheckpt().Epoch, "Unexpected finalized epoch")
|
|
}
|
|
|
|
func TestFinalizedCheckpt_GenesisRootOk(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
genesisRoot := [32]byte{'A'}
|
|
cp := ðpb.Checkpoint{Root: genesisRoot[:]}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.finalizedCheckpt = cp
|
|
c.genesisRoot = genesisRoot
|
|
assert.DeepEqual(t, c.genesisRoot[:], c.FinalizedCheckpt().Root)
|
|
}
|
|
|
|
func TestCurrentJustifiedCheckpt_CanRetrieve(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
cp := ðpb.Checkpoint{Epoch: 6, Root: bytesutil.PadTo([]byte("foo"), 32)}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.justifiedCheckpt = cp
|
|
|
|
assert.Equal(t, cp.Epoch, c.CurrentJustifiedCheckpt().Epoch, "Unexpected justified epoch")
|
|
}
|
|
|
|
func TestJustifiedCheckpt_GenesisRootOk(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
genesisRoot := [32]byte{'B'}
|
|
cp := ðpb.Checkpoint{Root: genesisRoot[:]}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.justifiedCheckpt = cp
|
|
c.genesisRoot = genesisRoot
|
|
assert.DeepEqual(t, c.genesisRoot[:], c.CurrentJustifiedCheckpt().Root)
|
|
}
|
|
|
|
func TestPreviousJustifiedCheckpt_CanRetrieve(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
cp := ðpb.Checkpoint{Epoch: 7, Root: bytesutil.PadTo([]byte("foo"), 32)}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.prevJustifiedCheckpt = cp
|
|
assert.Equal(t, cp.Epoch, c.PreviousJustifiedCheckpt().Epoch, "Unexpected previous justified epoch")
|
|
}
|
|
|
|
func TestPrevJustifiedCheckpt_GenesisRootOk(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
|
|
genesisRoot := [32]byte{'C'}
|
|
cp := ðpb.Checkpoint{Root: genesisRoot[:]}
|
|
c := setupBeaconChain(t, db, sc)
|
|
c.prevJustifiedCheckpt = cp
|
|
c.genesisRoot = genesisRoot
|
|
assert.DeepEqual(t, c.genesisRoot[:], c.PreviousJustifiedCheckpt().Root)
|
|
}
|
|
|
|
func TestHeadSlot_CanRetrieve(t *testing.T) {
|
|
c := &Service{}
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{})
|
|
require.NoError(t, err)
|
|
c.head = &head{slot: 100, state: s}
|
|
assert.Equal(t, uint64(100), c.headSlot())
|
|
}
|
|
|
|
func TestHeadRoot_CanRetrieve(t *testing.T) {
|
|
c := &Service{}
|
|
c.head = &head{root: [32]byte{'A'}}
|
|
assert.Equal(t, [32]byte{'A'}, c.headRoot())
|
|
}
|
|
|
|
func TestHeadBlock_CanRetrieve(t *testing.T) {
|
|
b := testutil.NewBeaconBlock()
|
|
b.Block.Slot = 1
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{})
|
|
require.NoError(t, err)
|
|
c := &Service{}
|
|
c.head = &head{block: b, state: s}
|
|
|
|
recevied, err := c.HeadBlock(context.Background())
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, b, recevied, "Incorrect head block received")
|
|
}
|
|
|
|
func TestHeadState_CanRetrieve(t *testing.T) {
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{Slot: 2, GenesisValidatorsRoot: params.BeaconConfig().ZeroHash[:]})
|
|
require.NoError(t, err)
|
|
c := &Service{}
|
|
c.head = &head{state: s}
|
|
headState, err := c.HeadState(context.Background())
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, headState.InnerStateUnsafe(), s.InnerStateUnsafe(), "Incorrect head state received")
|
|
}
|
|
|
|
func TestGenesisTime_CanRetrieve(t *testing.T) {
|
|
c := &Service{genesisTime: time.Unix(999, 0)}
|
|
wanted := time.Unix(999, 0)
|
|
assert.Equal(t, wanted, c.GenesisTime(), "Did not get wanted genesis time")
|
|
}
|
|
|
|
func TestCurrentFork_CanRetrieve(t *testing.T) {
|
|
f := &pb.Fork{Epoch: 999}
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{Fork: f})
|
|
require.NoError(t, err)
|
|
c := &Service{}
|
|
c.head = &head{state: s}
|
|
if !proto.Equal(c.CurrentFork(), f) {
|
|
t.Error("Received incorrect fork version")
|
|
}
|
|
}
|
|
|
|
func TestGenesisValidatorRoot_CanRetrieve(t *testing.T) {
|
|
// Should not panic if head state is nil.
|
|
c := &Service{}
|
|
assert.Equal(t, [32]byte{}, c.GenesisValidatorRoot(), "Did not get correct genesis validator root")
|
|
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{GenesisValidatorsRoot: []byte{'a'}})
|
|
require.NoError(t, err)
|
|
c.head = &head{state: s}
|
|
assert.Equal(t, [32]byte{'a'}, c.GenesisValidatorRoot(), "Did not get correct genesis validator root")
|
|
}
|
|
|
|
func TestHeadETH1Data_Nil(t *testing.T) {
|
|
db, sc := testDB.SetupDB(t)
|
|
c := setupBeaconChain(t, db, sc)
|
|
assert.DeepEqual(t, ðpb.Eth1Data{}, c.HeadETH1Data(), "Incorrect pre chain start value")
|
|
}
|
|
|
|
func TestHeadETH1Data_CanRetrieve(t *testing.T) {
|
|
d := ðpb.Eth1Data{DepositCount: 999}
|
|
s, err := state.InitializeFromProto(&pb.BeaconState{Eth1Data: d})
|
|
require.NoError(t, err)
|
|
c := &Service{}
|
|
c.head = &head{state: s}
|
|
if !proto.Equal(c.HeadETH1Data(), d) {
|
|
t.Error("Received incorrect eth1 data")
|
|
}
|
|
}
|