mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 23:48:06 -05:00
288b33750d
* always init service through NewService * move head state cache to service struct * changelog --------- Co-authored-by: Kasey Kirkham <kasey@users.noreply.github.com>
82 lines
2.3 KiB
Go
82 lines
2.3 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/OffchainLabs/prysm/v6/consensus-types/blocks"
|
|
"github.com/OffchainLabs/prysm/v6/encoding/bytesutil"
|
|
ethpb "github.com/OffchainLabs/prysm/v6/proto/prysm/v1alpha1"
|
|
"github.com/OffchainLabs/prysm/v6/testing/require"
|
|
"github.com/OffchainLabs/prysm/v6/testing/util"
|
|
)
|
|
|
|
func TestHeadSlot_DataRace(t *testing.T) {
|
|
s := testServiceWithDB(t)
|
|
b, err := blocks.NewSignedBeaconBlock(util.NewBeaconBlock())
|
|
require.NoError(t, err)
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
wait := make(chan struct{})
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(t.Context(), [32]byte{}, b, st))
|
|
}()
|
|
s.HeadSlot()
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadRoot_DataRace(t *testing.T) {
|
|
s := testServiceWithDB(t)
|
|
s.head = &head{root: [32]byte{'A'}}
|
|
b, err := blocks.NewSignedBeaconBlock(util.NewBeaconBlock())
|
|
require.NoError(t, err)
|
|
wait := make(chan struct{})
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(t.Context(), [32]byte{}, b, st))
|
|
|
|
}()
|
|
_, err = s.HeadRoot(t.Context())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadBlock_DataRace(t *testing.T) {
|
|
wsb, err := blocks.NewSignedBeaconBlock(ðpb.SignedBeaconBlock{Block: ðpb.BeaconBlock{Body: ðpb.BeaconBlockBody{}}})
|
|
require.NoError(t, err)
|
|
s := testServiceWithDB(t)
|
|
s.head = &head{block: wsb}
|
|
b, err := blocks.NewSignedBeaconBlock(util.NewBeaconBlock())
|
|
require.NoError(t, err)
|
|
wait := make(chan struct{})
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(t.Context(), [32]byte{}, b, st))
|
|
|
|
}()
|
|
_, err = s.HeadBlock(t.Context())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadState_DataRace(t *testing.T) {
|
|
s := testServiceWithDB(t)
|
|
beaconDB := s.cfg.BeaconDB
|
|
b, err := blocks.NewSignedBeaconBlock(util.NewBeaconBlock())
|
|
require.NoError(t, err)
|
|
wait := make(chan struct{})
|
|
st, _ := util.DeterministicGenesisState(t, 1)
|
|
root := bytesutil.ToBytes32(bytesutil.PadTo([]byte{'s'}, 32))
|
|
require.NoError(t, beaconDB.SaveGenesisBlockRoot(t.Context(), root))
|
|
require.NoError(t, beaconDB.SaveState(t.Context(), st, root))
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(t.Context(), [32]byte{}, b, st))
|
|
|
|
}()
|
|
_, err = s.HeadState(t.Context())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|