prysm-pulse/testing/slasher/simulator/simulator_test.go
terencechain d17996f8b0
Update to V4 🚀 (#12134)
* Update V3 from V4

* Fix build v3 -> v4

* Update ssz

* Update beacon_chain.pb.go

* Fix formatter import

* Update update-mockgen.sh comment to v4

* Fix conflicts. Pass build and tests

* Fix test
2023-03-17 18:52:56 +00:00

49 lines
1.7 KiB
Go

package simulator
import (
"testing"
mock "github.com/prysmaticlabs/prysm/v4/beacon-chain/blockchain/testing"
dbtest "github.com/prysmaticlabs/prysm/v4/beacon-chain/db/testing"
mockstategen "github.com/prysmaticlabs/prysm/v4/beacon-chain/state/stategen/mock"
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
"github.com/prysmaticlabs/prysm/v4/crypto/bls"
ethpb "github.com/prysmaticlabs/prysm/v4/proto/prysm/v1alpha1"
"github.com/prysmaticlabs/prysm/v4/testing/require"
"github.com/prysmaticlabs/prysm/v4/testing/util"
)
func setupService(t *testing.T, params *Parameters) *Simulator {
slasherDB := dbtest.SetupSlasherDB(t)
beaconState, err := util.NewBeaconState()
require.NoError(t, err)
// We setup validators in the beacon state along with their
// private keys used to generate valid signatures in generated objects.
validators := make([]*ethpb.Validator, params.NumValidators)
privKeys := make(map[primitives.ValidatorIndex]bls.SecretKey)
for valIdx := range validators {
privKey, err := bls.RandKey()
require.NoError(t, err)
privKeys[primitives.ValidatorIndex(valIdx)] = privKey
validators[valIdx] = &ethpb.Validator{
PublicKey: privKey.PublicKey().Marshal(),
WithdrawalCredentials: make([]byte, 32),
}
}
err = beaconState.SetValidators(validators)
require.NoError(t, err)
gen := mockstategen.NewMockService()
gen.AddStateForRoot(beaconState, [32]byte{})
return &Simulator{
srvConfig: &ServiceConfig{
Params: params,
Database: slasherDB,
AttestationStateFetcher: &mock.ChainService{State: beaconState},
PrivateKeysByValidatorIndex: privKeys,
StateGen: gen,
},
}
}