132 lines
3.7 KiB
Go
132 lines
3.7 KiB
Go
package simapp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
dbm "github.com/tendermint/tm-db"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/simapp/helpers"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/kv"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
|
)
|
|
|
|
// SetupSimulation creates the config, db (levelDB), temporary directory and logger for
|
|
// the simulation tests. If `FlagEnabledValue` is false it skips the current test.
|
|
// Returns error on an invalid db intantiation or temp dir creation.
|
|
func SetupSimulation(dirPrefix, dbName string) (simtypes.Config, dbm.DB, string, log.Logger, bool, error) {
|
|
if !FlagEnabledValue {
|
|
return simtypes.Config{}, nil, "", nil, true, nil
|
|
}
|
|
|
|
config := NewConfigFromFlags()
|
|
config.ChainID = helpers.SimAppChainID
|
|
|
|
var logger log.Logger
|
|
if FlagVerboseValue {
|
|
logger = log.TestingLogger()
|
|
} else {
|
|
logger = log.NewNopLogger()
|
|
}
|
|
|
|
dir, err := ioutil.TempDir("", dirPrefix)
|
|
if err != nil {
|
|
return simtypes.Config{}, nil, "", nil, false, err
|
|
}
|
|
|
|
db, err := sdk.NewLevelDB(dbName, dir)
|
|
if err != nil {
|
|
return simtypes.Config{}, nil, "", nil, false, err
|
|
}
|
|
|
|
return config, db, dir, logger, false, nil
|
|
}
|
|
|
|
// SimulationOperations retrieves the simulation params from the provided file path
|
|
// and returns all the modules weighted operations
|
|
func SimulationOperations(app App, cdc codec.JSONCodec, config simtypes.Config) []simtypes.WeightedOperation {
|
|
simState := module.SimulationState{
|
|
AppParams: make(simtypes.AppParams),
|
|
Cdc: cdc,
|
|
}
|
|
|
|
if config.ParamsFile != "" {
|
|
bz, err := ioutil.ReadFile(config.ParamsFile)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
err = json.Unmarshal(bz, &simState.AppParams)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
simState.ParamChanges = app.SimulationManager().GenerateParamChanges(config.Seed)
|
|
simState.Contents = app.SimulationManager().GetProposalContents(simState)
|
|
return app.SimulationManager().WeightedOperations(simState)
|
|
}
|
|
|
|
// CheckExportSimulation exports the app state and simulation parameters to JSON
|
|
// if the export paths are defined.
|
|
func CheckExportSimulation(
|
|
app App, config simtypes.Config, params simtypes.Params,
|
|
) error {
|
|
if config.ExportStatePath != "" {
|
|
fmt.Println("exporting app state...")
|
|
exported, err := app.ExportAppStateAndValidators(false, nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := ioutil.WriteFile(config.ExportStatePath, []byte(exported.AppState), 0600); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if config.ExportParamsPath != "" {
|
|
fmt.Println("exporting simulation params...")
|
|
paramsBz, err := json.MarshalIndent(params, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := ioutil.WriteFile(config.ExportParamsPath, paramsBz, 0600); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// PrintStats prints the corresponding statistics from the app DB.
|
|
func PrintStats(db dbm.DB) {
|
|
fmt.Println("\nLevelDB Stats")
|
|
fmt.Println(db.Stats()["leveldb.stats"])
|
|
fmt.Println("LevelDB cached block size", db.Stats()["leveldb.cachedblock"])
|
|
}
|
|
|
|
// GetSimulationLog unmarshals the KVPair's Value to the corresponding type based on the
|
|
// each's module store key and the prefix bytes of the KVPair's key.
|
|
func GetSimulationLog(storeName string, sdr sdk.StoreDecoderRegistry, kvAs, kvBs []kv.Pair) (log string) {
|
|
for i := 0; i < len(kvAs); i++ {
|
|
if len(kvAs[i].Value) == 0 && len(kvBs[i].Value) == 0 {
|
|
// skip if the value doesn't have any bytes
|
|
continue
|
|
}
|
|
|
|
decoder, ok := sdr[storeName]
|
|
if ok {
|
|
log += decoder(kvAs[i], kvBs[i])
|
|
} else {
|
|
log += fmt.Sprintf("store A %X => %X\nstore B %X => %X\n", kvAs[i].Key, kvAs[i].Value, kvBs[i].Key, kvBs[i].Value)
|
|
}
|
|
}
|
|
|
|
return log
|
|
}
|