mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 23:48:06 -05:00
7f7866ff2a
* Starting a quick PoC
* Rate limit to one epoch worth of blocks in memory
* Proof of concept working
* Quick comment out
* Save previous finalized checkpoint
* Test
* Minor fixes
* More run time fixes
* Remove panic
* Feature flag
* Removed unused methods
* Fixed tests
* E2e test
* comment
* Compatible with current initial sync
* Starting
* New cache
* Cache getters and setters
* It should be part of state gen
* Need to use cache for DB
* Don't have to use finalized state
* Rm unused file
* some changes to memory mgmt when using mempool
* More run time fixes
* Can sync to head
* Feedback
* Revert "some changes to memory mgmt when using mempool"
This reverts commit f5b3e7ff47.
* Fixed sync tests
* Fixed existing tests
* Test for state summary getter
* Gaz
* Fix kafka passthrough
* Fixed inputs
* Gaz
* Fixed build
* Fixed visibility
* Trying without the ignore
* Didn't work..
* Fix kafka
Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com>
100 lines
2.8 KiB
Go
100 lines
2.8 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"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func TestSaveState_ColdStateCanBeSaved(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctx := context.Background()
|
|
db := testDB.SetupDB(t)
|
|
defer testDB.TeardownDB(t, db)
|
|
|
|
service := New(db, cache.NewStateSummaryCache())
|
|
service.slotsPerArchivedPoint = 1
|
|
beaconState, _ := testutil.DeterministicGenesisState(t, 32)
|
|
|
|
// This goes to cold section.
|
|
slot := uint64(1)
|
|
beaconState.SetSlot(slot)
|
|
service.splitInfo.slot = slot + 1
|
|
|
|
r := [32]byte{'a'}
|
|
if err := service.SaveState(ctx, r, beaconState); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !service.beaconDB.HasArchivedPoint(ctx, 1) {
|
|
t.Error("Did not save cold state")
|
|
}
|
|
|
|
if service.beaconDB.ArchivedPointRoot(ctx, 1) != r {
|
|
t.Error("Did not get wanted root")
|
|
}
|
|
|
|
testutil.AssertLogsContain(t, hook, "Saved full state on archived point")
|
|
}
|
|
|
|
func TestSaveState_HotStateCanBeSaved(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctx := context.Background()
|
|
db := testDB.SetupDB(t)
|
|
defer testDB.TeardownDB(t, db)
|
|
|
|
service := New(db, cache.NewStateSummaryCache())
|
|
service.slotsPerArchivedPoint = 1
|
|
beaconState, _ := testutil.DeterministicGenesisState(t, 32)
|
|
// This goes to hot section, verify it can save on epoch boundary.
|
|
beaconState.SetSlot(params.BeaconConfig().SlotsPerEpoch)
|
|
|
|
r := [32]byte{'a'}
|
|
if err := service.SaveState(ctx, r, beaconState); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Should save both state and state summary.
|
|
if !service.beaconDB.HasState(ctx, r) {
|
|
t.Error("Should have saved the state")
|
|
}
|
|
if !service.stateSummaryCache.Has(r) {
|
|
t.Error("Should have saved the state summary")
|
|
}
|
|
testutil.AssertLogsContain(t, hook, "Saved full state on epoch boundary")
|
|
}
|
|
|
|
func TestSaveState_HotStateCached(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctx := context.Background()
|
|
db := testDB.SetupDB(t)
|
|
defer testDB.TeardownDB(t, db)
|
|
|
|
service := New(db, cache.NewStateSummaryCache())
|
|
service.slotsPerArchivedPoint = 1
|
|
beaconState, _ := testutil.DeterministicGenesisState(t, 32)
|
|
beaconState.SetSlot(params.BeaconConfig().SlotsPerEpoch)
|
|
|
|
// Cache the state prior.
|
|
r := [32]byte{'a'}
|
|
service.hotStateCache.Put(r, beaconState)
|
|
if err := service.SaveState(ctx, r, beaconState); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Should not save the state and state summary.
|
|
if service.beaconDB.HasState(ctx, r) {
|
|
t.Error("Should not have saved the state")
|
|
}
|
|
if service.beaconDB.HasStateSummary(ctx, r) {
|
|
t.Error("Should have saved the state summary")
|
|
}
|
|
testutil.AssertLogsDoNotContain(t, hook, "Saved full state on epoch boundary")
|
|
}
|