100 lines
2.7 KiB
Go
100 lines
2.7 KiB
Go
package mock
|
|
|
|
import (
|
|
"math/big"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto"
|
|
)
|
|
|
|
// BigInterval is a representation of the interval [lo, hi), where
|
|
// lo and hi are both of type sdk.Int
|
|
type BigInterval struct {
|
|
lo sdk.Int
|
|
hi sdk.Int
|
|
}
|
|
|
|
// RandFromBigInterval chooses an interval uniformly from the provided list of
|
|
// BigIntervals, and then chooses an element from an interval uniformly at random.
|
|
func RandFromBigInterval(r *rand.Rand, intervals []BigInterval) sdk.Int {
|
|
if len(intervals) == 0 {
|
|
return sdk.ZeroInt()
|
|
}
|
|
|
|
interval := intervals[r.Intn(len(intervals))]
|
|
|
|
lo := interval.lo
|
|
hi := interval.hi
|
|
|
|
diff := hi.Sub(lo)
|
|
result := sdk.NewIntFromBigInt(new(big.Int).Rand(r, diff.BigInt()))
|
|
result = result.Add(lo)
|
|
|
|
return result
|
|
}
|
|
|
|
// CheckBalance checks the balance of an account.
|
|
func CheckBalance(t *testing.T, app *App, addr sdk.AccAddress, exp sdk.Coins) {
|
|
ctxCheck := app.BaseApp.NewContext(true, abci.Header{})
|
|
res := app.AccountMapper.GetAccount(ctxCheck, addr)
|
|
|
|
require.Equal(t, exp, res.GetCoins())
|
|
}
|
|
|
|
// CheckGenTx checks a generated signed transaction. The result of the check is
|
|
// compared against the parameter 'expPass'. A test assertion is made using the
|
|
// parameter 'expPass' against the result. A corresponding result is returned.
|
|
func CheckGenTx(
|
|
t *testing.T, app *baseapp.BaseApp, msgs []sdk.Msg, accNums []int64,
|
|
seq []int64, expPass bool, priv ...crypto.PrivKey,
|
|
) sdk.Result {
|
|
tx := GenTx(msgs, accNums, seq, priv...)
|
|
res := app.Check(tx)
|
|
|
|
if expPass {
|
|
require.Equal(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
} else {
|
|
require.NotEqual(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
// SignCheckDeliver checks a generated signed transaction and simulates a
|
|
// block commitment with the given transaction. A test assertion is made using
|
|
// the parameter 'expPass' against the result. A corresponding result is
|
|
// returned.
|
|
func SignCheckDeliver(
|
|
t *testing.T, app *baseapp.BaseApp, msgs []sdk.Msg, accNums []int64,
|
|
seq []int64, expPass bool, priv ...crypto.PrivKey,
|
|
) sdk.Result {
|
|
tx := GenTx(msgs, accNums, seq, priv...)
|
|
res := app.Check(tx)
|
|
|
|
if expPass {
|
|
require.Equal(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
} else {
|
|
require.NotEqual(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
}
|
|
|
|
// Simulate a sending a transaction and committing a block
|
|
app.BeginBlock(abci.RequestBeginBlock{})
|
|
res = app.Deliver(tx)
|
|
|
|
if expPass {
|
|
require.Equal(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
} else {
|
|
require.NotEqual(t, sdk.ABCICodeOK, res.Code, res.Log)
|
|
}
|
|
|
|
app.EndBlock(abci.RequestEndBlock{})
|
|
app.Commit()
|
|
|
|
return res
|
|
}
|