mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
190d862552
* Return status.Errorf instead of plain errors from gRPC functions * return plain errors from helper functions * change errors to lowercase in node * correct test expectations * extracted StateFetcher * StateFetcher tests * extract beacon state creation option and fix state tests * add comment to StateFetcher * register the server * implement grpc function * test ToProto * gRPC function test with mock state fetcher * reduce visibility of packages * add missing error assertion * removed unused code * overwrite config name * gzl * Fix service fields * rename StateFetcher to Provider * Update beacon-chain/state/stateV0/state_trie.go Co-authored-by: Raul Jordan <raul@prysmaticlabs.com> * adjust code to new v0 interfaces * interface/struct naming changes Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com> Co-authored-by: Raul Jordan <raul@prysmaticlabs.com>
185 lines
5.1 KiB
Go
185 lines
5.1 KiB
Go
package statefetcher
|
|
|
|
import (
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
|
types "github.com/prysmaticlabs/eth2-types"
|
|
eth "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
chainMock "github.com/prysmaticlabs/prysm/beacon-chain/blockchain/testing"
|
|
testDB "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
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"
|
|
)
|
|
|
|
func TestGetStateRoot(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
headSlot := types.Slot(123)
|
|
fillSlot := func(state *pb.BeaconState) error {
|
|
state.Slot = headSlot
|
|
return nil
|
|
}
|
|
state, err := testutil.NewBeaconState(testutil.FillRootsNaturalOpt, fillSlot)
|
|
require.NoError(t, err)
|
|
stateRoot, err := state.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("Head", func(t *testing.T) {
|
|
p := StateProvider{
|
|
ChainInfoFetcher: &chainMock.ChainService{State: state},
|
|
}
|
|
|
|
s, err := p.State(ctx, []byte("head"))
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Genesis", func(t *testing.T) {
|
|
params.SetupTestConfigCleanup(t)
|
|
cfg := params.BeaconConfig()
|
|
cfg.ConfigName = "test"
|
|
params.OverrideBeaconConfig(cfg)
|
|
|
|
db := testDB.SetupDB(t)
|
|
b := testutil.NewBeaconBlock()
|
|
b.Block.StateRoot = bytesutil.PadTo([]byte("foo"), 32)
|
|
require.NoError(t, db.SaveBlock(ctx, b))
|
|
r, err := b.Block.HashTreeRoot()
|
|
require.NoError(t, err)
|
|
|
|
state, err := testutil.NewBeaconState(func(state *pb.BeaconState) error {
|
|
state.BlockRoots[0] = r[:]
|
|
return nil
|
|
})
|
|
require.NoError(t, err)
|
|
stateRoot, err := state.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
|
|
require.NoError(t, db.SaveStateSummary(ctx, &pb.StateSummary{Root: r[:]}))
|
|
require.NoError(t, db.SaveGenesisBlockRoot(ctx, r))
|
|
require.NoError(t, db.SaveState(ctx, state, r))
|
|
|
|
p := StateProvider{
|
|
BeaconDB: db,
|
|
}
|
|
|
|
s, err := p.State(ctx, []byte("genesis"))
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Finalized", func(t *testing.T) {
|
|
stateGen := stategen.NewMockService()
|
|
stateGen.StatesByRoot[stateRoot] = state
|
|
|
|
p := StateProvider{
|
|
ChainInfoFetcher: &chainMock.ChainService{
|
|
FinalizedCheckPoint: ð.Checkpoint{
|
|
Root: stateRoot[:],
|
|
},
|
|
},
|
|
StateGenService: stateGen,
|
|
}
|
|
|
|
s, err := p.State(ctx, []byte("finalized"))
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Justified", func(t *testing.T) {
|
|
stateGen := stategen.NewMockService()
|
|
stateGen.StatesByRoot[stateRoot] = state
|
|
|
|
p := StateProvider{
|
|
ChainInfoFetcher: &chainMock.ChainService{
|
|
CurrentJustifiedCheckPoint: ð.Checkpoint{
|
|
Root: stateRoot[:],
|
|
},
|
|
},
|
|
StateGenService: stateGen,
|
|
}
|
|
|
|
s, err := p.State(ctx, []byte("justified"))
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Hex root", func(t *testing.T) {
|
|
stateId, err := hexutil.Decode("0x" + strings.Repeat("0", 63) + "1")
|
|
require.NoError(t, err)
|
|
stateGen := stategen.NewMockService()
|
|
stateGen.StatesByRoot[bytesutil.ToBytes32(stateId)] = state
|
|
|
|
p := StateProvider{
|
|
ChainInfoFetcher: &chainMock.ChainService{State: state},
|
|
StateGenService: stateGen,
|
|
}
|
|
|
|
s, err := p.State(ctx, stateId)
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Hex root not found", func(t *testing.T) {
|
|
p := StateProvider{
|
|
ChainInfoFetcher: &chainMock.ChainService{State: state},
|
|
}
|
|
stateId, err := hexutil.Decode("0x" + strings.Repeat("f", 64))
|
|
require.NoError(t, err)
|
|
_, err = p.State(ctx, stateId)
|
|
require.ErrorContains(t, "state not found in the last 8192 state roots in head state", err)
|
|
})
|
|
|
|
t.Run("Slot", func(t *testing.T) {
|
|
stateGen := stategen.NewMockService()
|
|
stateGen.StatesBySlot[headSlot] = state
|
|
|
|
p := StateProvider{
|
|
GenesisTimeFetcher: &chainMock.ChainService{Slot: &headSlot},
|
|
StateGenService: stateGen,
|
|
}
|
|
|
|
s, err := p.State(ctx, []byte(strconv.FormatUint(uint64(headSlot), 10)))
|
|
require.NoError(t, err)
|
|
sRoot, err := s.HashTreeRoot(ctx)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, stateRoot, sRoot)
|
|
})
|
|
|
|
t.Run("Slot too big", func(t *testing.T) {
|
|
p := StateProvider{
|
|
GenesisTimeFetcher: &chainMock.ChainService{
|
|
Genesis: time.Now(),
|
|
},
|
|
}
|
|
_, err := p.State(ctx, []byte(strconv.FormatUint(1, 10)))
|
|
assert.ErrorContains(t, "slot cannot be in the future", err)
|
|
})
|
|
|
|
t.Run("Invalid state", func(t *testing.T) {
|
|
p := StateProvider{}
|
|
_, err := p.State(ctx, []byte("foo"))
|
|
require.ErrorContains(t, "invalid state ID: foo", err)
|
|
})
|
|
}
|