6606007297
* Move distr params to simulation param generator * Cleanup sim params output * Print params when possible instead of entire genesis struct * Update module param simulator keys * Rename module params simulator keys * Add/fix missing module params and implement SimParams * UPdate sim test to accept params file * Allow weights to be provided from file * Create doc.go * Remove TODO * Implement MustMarshalJSONIndent * Use mustMarshalJSONIndent for non-codec * Remove TODO * Fix doc.go * Update points * Update simapp/doc.go Co-Authored-By: Alessio Treglia <quadrispro@ubuntu.com> * Update simapp/doc.go Co-Authored-By: Alessio Treglia <quadrispro@ubuntu.com> * Cleanup appStateFn * Use constants for simulation parameter keys * Update msgs.go * minor formatting
88 lines
2.4 KiB
Go
88 lines
2.4 KiB
Go
package simulation
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// assertAll asserts the all invariants against application state
|
|
func assertAllInvariants(t *testing.T, app *baseapp.BaseApp, invs sdk.Invariants,
|
|
event string, logWriter LogWriter) {
|
|
|
|
ctx := app.NewContext(false, abci.Header{Height: app.LastBlockHeight() + 1})
|
|
|
|
for i := 0; i < len(invs); i++ {
|
|
if err := invs[i](ctx); err != nil {
|
|
fmt.Printf("Invariants broken after %s\n%s\n", event, err.Error())
|
|
logWriter.PrintLogs()
|
|
t.Fatal()
|
|
}
|
|
}
|
|
}
|
|
|
|
func getTestingMode(tb testing.TB) (testingMode bool, t *testing.T, b *testing.B) {
|
|
testingMode = false
|
|
if _t, ok := tb.(*testing.T); ok {
|
|
t = _t
|
|
testingMode = true
|
|
} else {
|
|
b = tb.(*testing.B)
|
|
}
|
|
return testingMode, t, b
|
|
}
|
|
|
|
// getBlockSize returns a block size as determined from the transition matrix.
|
|
// It targets making average block size the provided parameter. The three
|
|
// states it moves between are:
|
|
// - "over stuffed" blocks with average size of 2 * avgblocksize,
|
|
// - normal sized blocks, hitting avgBlocksize on average,
|
|
// - and empty blocks, with no txs / only txs scheduled from the past.
|
|
func getBlockSize(r *rand.Rand, params Params,
|
|
lastBlockSizeState, avgBlockSize int) (state, blocksize int) {
|
|
|
|
// TODO: Make default blocksize transition matrix actually make the average
|
|
// blocksize equal to avgBlockSize.
|
|
state = params.BlockSizeTransitionMatrix.NextState(r, lastBlockSizeState)
|
|
switch state {
|
|
case 0:
|
|
blocksize = r.Intn(avgBlockSize * 4)
|
|
case 1:
|
|
blocksize = r.Intn(avgBlockSize * 2)
|
|
default:
|
|
blocksize = 0
|
|
}
|
|
return state, blocksize
|
|
}
|
|
|
|
// PeriodicInvariants returns an array of wrapped Invariants. Where each
|
|
// invariant function is only executed periodically defined by period and offset.
|
|
func PeriodicInvariants(invariants []sdk.Invariant, period int, offset int) []sdk.Invariant {
|
|
var outInvariants []sdk.Invariant
|
|
for _, invariant := range invariants {
|
|
outInvariant := func(ctx sdk.Context) error {
|
|
if int(ctx.BlockHeight())%period == offset {
|
|
return invariant(ctx)
|
|
}
|
|
return nil
|
|
}
|
|
outInvariants = append(outInvariants, outInvariant)
|
|
}
|
|
return outInvariants
|
|
}
|
|
|
|
func mustMarshalJSONIndent(o interface{}) []byte {
|
|
bz, err := json.MarshalIndent(o, "", " ")
|
|
if err != nil {
|
|
panic(fmt.Sprintf("failed to JSON encode: %s", err))
|
|
}
|
|
|
|
return bz
|
|
}
|