39 lines
976 B
Go
39 lines
976 B
Go
package simulation
|
|
|
|
import (
|
|
"math/rand"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
"github.com/cosmos/cosmos-sdk/x/epochs/types"
|
|
)
|
|
|
|
// GenDuration randomized GenDuration
|
|
func GenDuration(r *rand.Rand) time.Duration {
|
|
return time.Hour * time.Duration(r.Intn(168)+1) // between 1 hour to 1 week
|
|
}
|
|
|
|
func RandomizedEpochs(r *rand.Rand) []types.EpochInfo {
|
|
// Gen max 10 epoch
|
|
n := r.Intn(11)
|
|
var epochs []types.EpochInfo
|
|
for i := range n {
|
|
identifier := "identifier-" + strconv.Itoa(i)
|
|
duration := GenDuration(r)
|
|
epoch := types.NewGenesisEpochInfo(identifier, duration)
|
|
epochs = append(epochs, epoch)
|
|
}
|
|
return epochs
|
|
}
|
|
|
|
// RandomizedGenState generates a random GenesisState for distribution
|
|
func RandomizedGenState(simState *module.SimulationState) {
|
|
epochs := RandomizedEpochs(simState.Rand)
|
|
epochsGenesis := types.GenesisState{
|
|
Epochs: epochs,
|
|
}
|
|
|
|
simState.GenState[types.ModuleName] = simState.Cdc.MustMarshalJSON(&epochsGenesis)
|
|
}
|