135 lines
4.6 KiB
Go
135 lines
4.6 KiB
Go
package simulation
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
"github.com/cosmos/cosmos-sdk/types/simulation"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
vestingtypes "github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
|
|
)
|
|
|
|
// Simulation parameter constants
|
|
const (
|
|
MaxMemoChars = "max_memo_characters"
|
|
TxSigLimit = "tx_sig_limit"
|
|
TxSizeCostPerByte = "tx_size_cost_per_byte"
|
|
SigVerifyCostED25519 = "sig_verify_cost_ed25519"
|
|
SigVerifyCostSECP256K1 = "sig_verify_cost_secp256k1"
|
|
)
|
|
|
|
// RandomGenesisAccounts defines the default RandomGenesisAccountsFn used on the SDK.
|
|
// It creates a slice of BaseAccount, ContinuousVestingAccount and DelayedVestingAccount.
|
|
func RandomGenesisAccounts(simState *module.SimulationState) types.GenesisAccounts {
|
|
genesisAccs := make(types.GenesisAccounts, len(simState.Accounts))
|
|
for i, acc := range simState.Accounts {
|
|
bacc := types.NewBaseAccountWithAddress(acc.Address)
|
|
|
|
// Only consider making a vesting account once the initial bonded validator
|
|
// set is exhausted due to needing to track DelegatedVesting.
|
|
if !(int64(i) > simState.NumBonded && simState.Rand.Intn(100) < 50) {
|
|
genesisAccs[i] = bacc
|
|
continue
|
|
}
|
|
|
|
initialVesting := sdk.NewCoins(sdk.NewInt64Coin(sdk.DefaultBondDenom, simState.Rand.Int63n(simState.InitialStake)))
|
|
var endTime int64
|
|
|
|
startTime := simState.GenTimestamp.Unix()
|
|
|
|
// Allow for some vesting accounts to vest very quickly while others very slowly.
|
|
if simState.Rand.Intn(100) < 50 {
|
|
endTime = int64(simulation.RandIntBetween(simState.Rand, int(startTime)+1, int(startTime+(60*60*24*30))))
|
|
} else {
|
|
endTime = int64(simulation.RandIntBetween(simState.Rand, int(startTime)+1, int(startTime+(60*60*12))))
|
|
}
|
|
|
|
bva := vestingtypes.NewBaseVestingAccount(bacc, initialVesting, endTime)
|
|
|
|
if simState.Rand.Intn(100) < 50 {
|
|
genesisAccs[i] = vestingtypes.NewContinuousVestingAccountRaw(bva, startTime)
|
|
} else {
|
|
genesisAccs[i] = vestingtypes.NewDelayedVestingAccountRaw(bva)
|
|
}
|
|
}
|
|
|
|
return genesisAccs
|
|
}
|
|
|
|
// GenMaxMemoChars randomized MaxMemoChars
|
|
func GenMaxMemoChars(r *rand.Rand) uint64 {
|
|
return uint64(simulation.RandIntBetween(r, 100, 200))
|
|
}
|
|
|
|
// GenTxSigLimit randomized TxSigLimit
|
|
// make sure that sigLimit is always high
|
|
// so that arbitrarily simulated messages from other
|
|
// modules can still create valid transactions
|
|
func GenTxSigLimit(r *rand.Rand) uint64 {
|
|
return uint64(r.Intn(7) + 5)
|
|
}
|
|
|
|
// GenTxSizeCostPerByte randomized TxSizeCostPerByte
|
|
func GenTxSizeCostPerByte(r *rand.Rand) uint64 {
|
|
return uint64(simulation.RandIntBetween(r, 5, 15))
|
|
}
|
|
|
|
// GenSigVerifyCostED25519 randomized SigVerifyCostED25519
|
|
func GenSigVerifyCostED25519(r *rand.Rand) uint64 {
|
|
return uint64(simulation.RandIntBetween(r, 500, 1000))
|
|
}
|
|
|
|
// GenSigVerifyCostSECP256K1 randomized SigVerifyCostSECP256K1
|
|
func GenSigVerifyCostSECP256K1(r *rand.Rand) uint64 {
|
|
return uint64(simulation.RandIntBetween(r, 500, 1000))
|
|
}
|
|
|
|
// RandomizedGenState generates a random GenesisState for auth
|
|
func RandomizedGenState(simState *module.SimulationState, randGenAccountsFn types.RandomGenesisAccountsFn) {
|
|
var maxMemoChars uint64
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, MaxMemoChars, &maxMemoChars, simState.Rand,
|
|
func(r *rand.Rand) { maxMemoChars = GenMaxMemoChars(r) },
|
|
)
|
|
|
|
var txSigLimit uint64
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, TxSigLimit, &txSigLimit, simState.Rand,
|
|
func(r *rand.Rand) { txSigLimit = GenTxSigLimit(r) },
|
|
)
|
|
|
|
var txSizeCostPerByte uint64
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, TxSizeCostPerByte, &txSizeCostPerByte, simState.Rand,
|
|
func(r *rand.Rand) { txSizeCostPerByte = GenTxSizeCostPerByte(r) },
|
|
)
|
|
|
|
var sigVerifyCostED25519 uint64
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, SigVerifyCostED25519, &sigVerifyCostED25519, simState.Rand,
|
|
func(r *rand.Rand) { sigVerifyCostED25519 = GenSigVerifyCostED25519(r) },
|
|
)
|
|
|
|
var sigVerifyCostSECP256K1 uint64
|
|
simState.AppParams.GetOrGenerate(
|
|
simState.Cdc, SigVerifyCostSECP256K1, &sigVerifyCostSECP256K1, simState.Rand,
|
|
func(r *rand.Rand) { sigVerifyCostSECP256K1 = GenSigVerifyCostSECP256K1(r) },
|
|
)
|
|
|
|
params := types.NewParams(maxMemoChars, txSigLimit, txSizeCostPerByte,
|
|
sigVerifyCostED25519, sigVerifyCostSECP256K1)
|
|
genesisAccs := randGenAccountsFn(simState)
|
|
|
|
authGenesis := types.NewGenesisState(params, genesisAccs)
|
|
|
|
bz, err := json.MarshalIndent(&authGenesis.Params, "", " ")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
fmt.Printf("Selected randomly generated auth parameters:\n%s\n", bz)
|
|
simState.GenState[types.ModuleName] = simState.Cdc.MustMarshalJSON(authGenesis)
|
|
}
|