cosmos-sdk/cmd/gaia/app/sim_test.go

101 lines
2.7 KiB
Go
Raw Normal View History

package app
2018-07-11 15:14:37 -07:00
import (
2018-07-17 15:04:10 -07:00
"encoding/json"
2018-07-18 23:40:46 -07:00
"flag"
2018-07-17 15:04:10 -07:00
"math/rand"
2018-07-11 15:14:37 -07:00
"testing"
"github.com/stretchr/testify/require"
2018-07-12 13:01:43 -07:00
dbm "github.com/tendermint/tendermint/libs/db"
"github.com/tendermint/tendermint/libs/log"
2018-07-17 15:04:10 -07:00
sdk "github.com/cosmos/cosmos-sdk/types"
2018-07-18 00:05:48 -07:00
banksim "github.com/cosmos/cosmos-sdk/x/bank/simulation"
2018-07-16 18:25:15 -07:00
"github.com/cosmos/cosmos-sdk/x/mock/simulation"
2018-07-17 11:33:53 -07:00
stake "github.com/cosmos/cosmos-sdk/x/stake"
2018-07-17 15:04:10 -07:00
stakesim "github.com/cosmos/cosmos-sdk/x/stake/simulation"
2018-07-16 18:15:50 -07:00
)
2018-07-18 23:40:46 -07:00
var (
seed int64
numKeys int
numBlocks int
blockSize int
enabled bool
2018-07-11 15:14:37 -07:00
)
2018-07-18 23:40:46 -07:00
func init() {
flag.Int64Var(&seed, "SimulationSeed", 42, "Simulation random seed")
flag.IntVar(&numKeys, "SimulationNumKeys", 10, "Number of keys (accounts)")
flag.IntVar(&numBlocks, "SimulationNumBlocks", 100, "Number of blocks")
flag.IntVar(&blockSize, "SimulationBlockSize", 100, "Operations per block")
flag.BoolVar(&enabled, "SimulationEnabled", false, "Enable the simulation")
}
2018-07-17 15:04:10 -07:00
func appStateFn(r *rand.Rand, accs []sdk.AccAddress) json.RawMessage {
var genesisAccounts []GenesisAccount
2018-07-17 11:33:53 -07:00
2018-07-17 15:04:10 -07:00
// Randomly generate some genesis accounts
for _, addr := range accs {
coins := sdk.Coins{sdk.Coin{"steak", sdk.NewInt(100)}}
genesisAccounts = append(genesisAccounts, GenesisAccount{
Address: addr,
Coins: coins,
})
}
2018-07-16 18:15:50 -07:00
2018-07-17 11:33:53 -07:00
// Default genesis state
2018-07-17 16:01:36 -07:00
stakeGenesis := stake.DefaultGenesisState()
stakeGenesis.Pool.LooseTokens = sdk.NewRat(1000)
genesis := GenesisState{
2018-07-17 15:04:10 -07:00
Accounts: genesisAccounts,
2018-07-17 16:01:36 -07:00
StakeData: stakeGenesis,
2018-07-17 11:33:53 -07:00
}
// Marshal genesis
appState, err := MakeCodec().MarshalJSON(genesis)
2018-07-17 11:33:53 -07:00
if err != nil {
panic(err)
}
2018-07-17 15:04:10 -07:00
return appState
}
func TestFullGaiaSimulation(t *testing.T) {
2018-07-18 23:40:46 -07:00
if !enabled {
2018-07-17 15:04:10 -07:00
t.Skip("Skipping Gaia simulation")
}
// Setup Gaia application
logger := log.NewNopLogger()
db := dbm.NewMemDB()
app := NewGaiaApp(logger, db, nil)
require.Equal(t, "GaiaApp", app.Name())
2018-07-16 18:15:50 -07:00
// Run randomized simulation
2018-07-17 22:37:38 -07:00
simulation.SimulateFromSeed(
t, app.BaseApp, appStateFn, seed,
2018-07-17 15:04:10 -07:00
[]simulation.TestAndRunTx{
2018-07-18 00:05:48 -07:00
banksim.TestAndRunSingleInputMsgSend(app.accountMapper),
2018-07-17 15:04:10 -07:00
stakesim.SimulateMsgCreateValidator(app.accountMapper, app.stakeKeeper),
2018-07-17 22:44:40 -07:00
stakesim.SimulateMsgEditValidator(app.stakeKeeper),
2018-07-17 22:37:38 -07:00
stakesim.SimulateMsgDelegate(app.accountMapper, app.stakeKeeper),
stakesim.SimulateMsgBeginUnbonding(app.accountMapper, app.stakeKeeper),
stakesim.SimulateMsgCompleteUnbonding(app.stakeKeeper),
stakesim.SimulateMsgBeginRedelegate(app.accountMapper, app.stakeKeeper),
stakesim.SimulateMsgCompleteRedelegate(app.stakeKeeper),
2018-07-17 15:04:10 -07:00
},
2018-07-16 18:25:15 -07:00
[]simulation.RandSetup{},
2018-07-17 15:04:10 -07:00
[]simulation.Invariant{
2018-07-18 00:05:48 -07:00
banksim.NonnegativeBalanceInvariant(app.accountMapper),
2018-07-17 15:04:10 -07:00
stakesim.AllInvariants(app.coinKeeper, app.stakeKeeper, app.accountMapper),
},
2018-07-18 23:40:46 -07:00
numKeys,
numBlocks,
2018-07-18 00:54:41 -07:00
blockSize,
2018-07-16 18:15:50 -07:00
)
2018-07-11 15:14:37 -07:00
}