mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
0b1a777d62
* ran gofmt @ v1.17, these are the resulting changes * fix some flaky tests that are tripping up this PR * gofmt Co-authored-by: Kasey Kirkham <kasey@users.noreply.github.com>
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package blockchain
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
testDB "github.com/prysmaticlabs/prysm/beacon-chain/db/testing"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/state/stategen"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/proto/prysm/v1alpha1/wrapper"
|
|
"github.com/prysmaticlabs/prysm/testing/require"
|
|
)
|
|
|
|
func TestHeadSlot_DataRace(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
s := &Service{
|
|
cfg: &config{BeaconDB: beaconDB},
|
|
}
|
|
wait := make(chan struct{})
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(context.Background(), [32]byte{}))
|
|
}()
|
|
s.HeadSlot()
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadRoot_DataRace(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
s := &Service{
|
|
cfg: &config{BeaconDB: beaconDB, StateGen: stategen.New(beaconDB)},
|
|
head: &head{root: [32]byte{'A'}},
|
|
}
|
|
wait := make(chan struct{})
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(context.Background(), [32]byte{}))
|
|
}()
|
|
_, err := s.HeadRoot(context.Background())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadBlock_DataRace(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
s := &Service{
|
|
cfg: &config{BeaconDB: beaconDB, StateGen: stategen.New(beaconDB)},
|
|
head: &head{block: wrapper.WrappedPhase0SignedBeaconBlock(ðpb.SignedBeaconBlock{})},
|
|
}
|
|
wait := make(chan struct{})
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(context.Background(), [32]byte{}))
|
|
}()
|
|
_, err := s.HeadBlock(context.Background())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|
|
|
|
func TestHeadState_DataRace(t *testing.T) {
|
|
beaconDB := testDB.SetupDB(t)
|
|
s := &Service{
|
|
cfg: &config{BeaconDB: beaconDB, StateGen: stategen.New(beaconDB)},
|
|
}
|
|
wait := make(chan struct{})
|
|
go func() {
|
|
defer close(wait)
|
|
require.NoError(t, s.saveHead(context.Background(), [32]byte{}))
|
|
}()
|
|
_, err := s.HeadState(context.Background())
|
|
require.NoError(t, err)
|
|
<-wait
|
|
}
|