92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
package simapp
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
flag "github.com/spf13/pflag"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
|
|
"github.com/cosmos/cosmos-sdk/server"
|
|
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
|
|
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
|
"github.com/cosmos/cosmos-sdk/x/simulation"
|
|
simcli "github.com/cosmos/cosmos-sdk/x/simulation/client/cli"
|
|
)
|
|
|
|
var FlagEnableBenchStreamingValue bool
|
|
|
|
// Get flags every time the simulator is run
|
|
func init() {
|
|
flag.BoolVar(&FlagEnableBenchStreamingValue, "EnableStreaming", false, "Enable streaming service")
|
|
}
|
|
|
|
// Profile with:
|
|
// /usr/local/go/bin/go test -benchmem -run=^$ cosmossdk.io/simapp -bench ^BenchmarkFullAppSimulation$ -Commit=true -cpuprofile cpu.out
|
|
func BenchmarkFullAppSimulation(b *testing.B) {
|
|
b.ReportAllocs()
|
|
|
|
config := simcli.NewConfigFromFlags()
|
|
config.ChainID = SimAppChainID
|
|
|
|
db, dir, logger, skip, err := simtestutil.SetupSimulation(config, "goleveldb-app-sim", "Simulation", simcli.FlagVerboseValue, simcli.FlagEnabledValue)
|
|
if err != nil {
|
|
b.Fatalf("simulation setup failed: %s", err.Error())
|
|
}
|
|
|
|
if skip {
|
|
b.Skip("skipping benchmark application simulation")
|
|
}
|
|
|
|
defer func() {
|
|
require.NoError(b, db.Close())
|
|
require.NoError(b, os.RemoveAll(dir))
|
|
}()
|
|
|
|
appOptions := viper.New()
|
|
if FlagEnableStreamingValue {
|
|
m := make(map[string]interface{})
|
|
m["streaming.abci.keys"] = []string{"*"}
|
|
m["streaming.abci.plugin"] = "abci_v1"
|
|
m["streaming.abci.stop-node-on-err"] = true
|
|
for key, value := range m {
|
|
appOptions.SetDefault(key, value)
|
|
}
|
|
}
|
|
appOptions.SetDefault(flags.FlagHome, DefaultNodeHome)
|
|
appOptions.SetDefault(server.FlagInvCheckPeriod, simcli.FlagPeriodValue)
|
|
|
|
app := NewSimApp(logger, db, nil, true, appOptions, interBlockCacheOpt(), baseapp.SetChainID(SimAppChainID))
|
|
|
|
// run randomized simulation
|
|
_, simParams, simErr := simulation.SimulateFromSeed(
|
|
b,
|
|
os.Stdout,
|
|
app.BaseApp,
|
|
simtestutil.AppStateFn(app.AppCodec(), app.AuthKeeper.AddressCodec(), app.StakingKeeper.ValidatorAddressCodec(), app.SimulationManager(), app.DefaultGenesis()),
|
|
simtypes.RandomAccounts, // Replace with own random account function if using keys other than secp256k1
|
|
simtestutil.SimulationOperations(app, app.AppCodec(), config),
|
|
BlockedAddresses(),
|
|
config,
|
|
app.AppCodec(),
|
|
codectestutil.CodecOptions{}.GetAddressCodec(),
|
|
)
|
|
|
|
// export state and simParams before the simulation error is checked
|
|
if err = simtestutil.CheckExportSimulation(app, config, simParams); err != nil {
|
|
b.Fatal(err)
|
|
}
|
|
|
|
if simErr != nil {
|
|
b.Fatal(simErr)
|
|
}
|
|
|
|
if config.Commit {
|
|
simtestutil.PrintStats(db)
|
|
}
|
|
}
|