98 lines
3.3 KiB
Go
98 lines
3.3 KiB
Go
package simulation
|
|
|
|
import (
|
|
"errors"
|
|
"math/rand"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/simapp/helpers"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/simulation"
|
|
"github.com/cosmos/cosmos-sdk/x/slashing/internal/keeper"
|
|
"github.com/cosmos/cosmos-sdk/x/slashing/internal/types"
|
|
stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
|
|
)
|
|
|
|
// SimulateMsgUnjail generates a MsgUnjail with random values
|
|
// nolint: funlen
|
|
func SimulateMsgUnjail(ak types.AccountKeeper, k keeper.Keeper, sk stakingkeeper.Keeper) simulation.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, ctx sdk.Context,
|
|
accs []simulation.Account, chainID string,
|
|
) (simulation.OperationMsg, []simulation.FutureOperation, error) {
|
|
|
|
validator, ok := stakingkeeper.RandomValidator(r, sk, ctx)
|
|
if !ok {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, nil // skip
|
|
}
|
|
|
|
simAccount, found := simulation.FindAccount(accs, sdk.AccAddress(validator.GetOperator()))
|
|
if !found {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, nil // skip
|
|
}
|
|
|
|
if !validator.IsJailed() {
|
|
// TODO: due to this condition this message is almost, if not always, skipped !
|
|
return simulation.NoOpMsg(types.ModuleName), nil, nil
|
|
}
|
|
|
|
consAddr := sdk.ConsAddress(validator.GetConsPubKey().Address())
|
|
info, found := k.GetValidatorSigningInfo(ctx, consAddr)
|
|
if !found {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, nil // skip
|
|
}
|
|
|
|
selfDel := sk.Delegation(ctx, simAccount.Address, validator.GetOperator())
|
|
if selfDel == nil {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, nil // skip
|
|
}
|
|
|
|
account := ak.GetAccount(ctx, sdk.AccAddress(validator.GetOperator()))
|
|
fees, err := simulation.RandomFees(r, ctx, account.SpendableCoins(ctx.BlockTime()))
|
|
if err != nil {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, err
|
|
}
|
|
|
|
msg := types.NewMsgUnjail(validator.GetOperator())
|
|
|
|
tx := helpers.GenTx(
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
simAccount.PrivKey,
|
|
)
|
|
|
|
res := app.Deliver(tx)
|
|
|
|
// result should fail if:
|
|
// - validator cannot be unjailed due to tombstone
|
|
// - validator is still in jailed period
|
|
// - self delegation too low
|
|
if info.Tombstoned ||
|
|
ctx.BlockHeader().Time.Before(info.JailedUntil) ||
|
|
validator.TokensFromShares(selfDel.GetShares()).TruncateInt().LT(validator.GetMinSelfDelegation()) {
|
|
if res.IsOK() {
|
|
if info.Tombstoned {
|
|
return simulation.NewOperationMsg(msg, true, ""), nil, errors.New("validator should not have been unjailed if validator tombstoned")
|
|
}
|
|
if ctx.BlockHeader().Time.Before(info.JailedUntil) {
|
|
return simulation.NewOperationMsg(msg, true, ""), nil, errors.New("validator unjailed while validator still in jail period")
|
|
}
|
|
if validator.TokensFromShares(selfDel.GetShares()).TruncateInt().LT(validator.GetMinSelfDelegation()) {
|
|
return simulation.NewOperationMsg(msg, true, ""), nil, errors.New("validator unjailed even though self-delegation too low")
|
|
}
|
|
}
|
|
// msg failed as expected
|
|
return simulation.NewOperationMsg(msg, false, ""), nil, nil
|
|
}
|
|
|
|
if !res.IsOK() {
|
|
return simulation.NoOpMsg(types.ModuleName), nil, errors.New(res.Log)
|
|
}
|
|
|
|
return simulation.NewOperationMsg(msg, true, ""), nil, nil
|
|
}
|
|
}
|