mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
d62e989a67
* Remove unused fields and function * Rename splitInfo to finalizedInfo * Merge branch 'master' into stategen-cleanup * Fmt * Merge branch 'stategen-cleanup' of github.com:prysmaticlabs/prysm into stategen-cleanup * Merge refs/heads/master into stategen-cleanup
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
package stategen
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/cache"
|
|
testDB "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
)
|
|
|
|
func TestResume(t *testing.T) {
|
|
ctx := context.Background()
|
|
db, _ := testDB.SetupDB(t)
|
|
|
|
service := New(db, cache.NewStateSummaryCache())
|
|
root := [32]byte{'A'}
|
|
beaconState, _ := testutil.DeterministicGenesisState(t, 32)
|
|
if err := beaconState.SetSlot(params.BeaconConfig().SlotsPerEpoch); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.beaconDB.SaveState(ctx, beaconState, root); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.beaconDB.SaveArchivedPointRoot(ctx, root, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.beaconDB.SaveLastArchivedIndex(ctx, 1); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
resumeState, err := service.Resume(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !proto.Equal(beaconState.InnerStateUnsafe(), resumeState.InnerStateUnsafe()) {
|
|
t.Error("Diff saved state")
|
|
}
|
|
if service.finalizedInfo.slot != params.BeaconConfig().SlotsPerEpoch {
|
|
t.Errorf("Did not get watned slot")
|
|
}
|
|
if root != service.finalizedInfo.root {
|
|
t.Errorf("Did not get wanted root")
|
|
}
|
|
}
|