62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package simulation
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
|
|
"github.com/evmos/ethermint/x/evm/types"
|
|
)
|
|
|
|
const (
|
|
extraEIPsKey = "extra_eips"
|
|
)
|
|
|
|
// GenExtraEIPs defines a set of extra EIPs with 50% probability
|
|
func GenExtraEIPs(r *rand.Rand) []int64 {
|
|
var extraEIPs []int64
|
|
// 50% chance of having extra EIPs
|
|
if r.Intn(2) == 0 {
|
|
extraEIPs = []int64{1344, 1884, 2200, 2929, 3198, 3529}
|
|
}
|
|
return extraEIPs
|
|
}
|
|
|
|
// GenEnableCreate enables the EnableCreate param with 80% probability
|
|
func GenEnableCreate(r *rand.Rand) bool {
|
|
// 80% chance of enabling create contract
|
|
enableCreate := r.Intn(100) < 80
|
|
return enableCreate
|
|
}
|
|
|
|
// GenEnableCall enables the EnableCall param with 80% probability
|
|
func GenEnableCall(r *rand.Rand) bool {
|
|
// 80% chance of enabling evm account transfer and calling contract
|
|
enableCall := r.Intn(100) < 80
|
|
return enableCall
|
|
}
|
|
|
|
// RandomizedGenState generates a random GenesisState for the EVM module
|
|
func RandomizedGenState(simState *module.SimulationState) {
|
|
// evm params
|
|
var extraEIPs []int64
|
|
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, extraEIPsKey, &extraEIPs, simState.Rand,
|
|
func(r *rand.Rand) { extraEIPs = GenExtraEIPs(r) },
|
|
)
|
|
|
|
params := types.NewParams(types.DefaultEVMDenom, true, true, types.DefaultChainConfig(), extraEIPs...)
|
|
evmGenesis := types.NewGenesisState(params, []types.GenesisAccount{})
|
|
|
|
bz, err := json.MarshalIndent(evmGenesis, "", " ")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Printf("Selected randomly generated %s parameters:\n%s\n", types.ModuleName, bz)
|
|
|
|
simState.GenState[types.ModuleName] = simState.Cdc.MustMarshalJSON(evmGenesis)
|
|
}
|