1306 lines
45 KiB
Go
1306 lines
45 KiB
Go
package simulation
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"math/rand"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
cdctypes "github.com/cosmos/cosmos-sdk/codec/types"
|
|
"github.com/cosmos/cosmos-sdk/simapp/helpers"
|
|
simappparams "github.com/cosmos/cosmos-sdk/simapp/params"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
"github.com/cosmos/cosmos-sdk/x/group/keeper"
|
|
"github.com/cosmos/cosmos-sdk/x/simulation"
|
|
|
|
"github.com/cosmos/cosmos-sdk/x/group"
|
|
)
|
|
|
|
var initialGroupID = uint64(100000000000000)
|
|
|
|
// group message types
|
|
var (
|
|
TypeMsgCreateGroup = sdk.MsgTypeURL(&group.MsgCreateGroup{})
|
|
TypeMsgUpdateGroupMembers = sdk.MsgTypeURL(&group.MsgUpdateGroupMembers{})
|
|
TypeMsgUpdateGroupAdmin = sdk.MsgTypeURL(&group.MsgUpdateGroupAdmin{})
|
|
TypeMsgUpdateGroupMetadata = sdk.MsgTypeURL(&group.MsgUpdateGroupMetadata{})
|
|
TypeMsgCreateGroupWithPolicy = sdk.MsgTypeURL(&group.MsgCreateGroupWithPolicy{})
|
|
TypeMsgCreateGroupPolicy = sdk.MsgTypeURL(&group.MsgCreateGroupPolicy{})
|
|
TypeMsgUpdateGroupPolicyAdmin = sdk.MsgTypeURL(&group.MsgUpdateGroupPolicyAdmin{})
|
|
TypeMsgUpdateGroupPolicyDecisionPolicy = sdk.MsgTypeURL(&group.MsgUpdateGroupPolicyDecisionPolicy{})
|
|
TypeMsgUpdateGroupPolicyMetadata = sdk.MsgTypeURL(&group.MsgUpdateGroupPolicyMetadata{})
|
|
TypeMsgSubmitProposal = sdk.MsgTypeURL(&group.MsgSubmitProposal{})
|
|
TypeMsgWithdrawProposal = sdk.MsgTypeURL(&group.MsgWithdrawProposal{})
|
|
TypeMsgVote = sdk.MsgTypeURL(&group.MsgVote{})
|
|
TypeMsgExec = sdk.MsgTypeURL(&group.MsgExec{})
|
|
TypeMsgLeaveGroup = sdk.MsgTypeURL(&group.MsgLeaveGroup{})
|
|
)
|
|
|
|
// Simulation operation weights constants
|
|
const (
|
|
OpMsgCreateGroup = "op_weight_msg_create_group"
|
|
OpMsgUpdateGroupAdmin = "op_weight_msg_update_group_admin"
|
|
OpMsgUpdateGroupMetadata = "op_wieght_msg_update_group_metadata"
|
|
OpMsgUpdateGroupMembers = "op_weight_msg_update_group_members"
|
|
OpMsgCreateGroupPolicy = "op_weight_msg_create_group_account"
|
|
OpMsgCreateGroupWithPolicy = "op_weight_msg_create_group_with_policy"
|
|
OpMsgUpdateGroupPolicyAdmin = "op_weight_msg_update_group_account_admin"
|
|
OpMsgUpdateGroupPolicyDecisionPolicy = "op_weight_msg_update_group_account_decision_policy"
|
|
OpMsgUpdateGroupPolicyMetaData = "op_weight_msg_update_group_account_metadata"
|
|
OpMsgSubmitProposal = "op_weight_msg_submit_proposal"
|
|
OpMsgWithdrawProposal = "op_weight_msg_withdraw_proposal"
|
|
OpMsgVote = "op_weight_msg_vote"
|
|
OpMsgExec = "ops_weight_msg_exec"
|
|
OpMsgLeaveGroup = "ops_weight_msg_leave_group"
|
|
)
|
|
|
|
// If update group or group policy txn's executed, `SimulateMsgVote` & `SimulateMsgExec` txn's returns `noOp`.
|
|
// That's why we have less weight for update group & group-policy txn's.
|
|
const (
|
|
WeightMsgCreateGroup = 100
|
|
WeightMsgCreateGroupPolicy = 50
|
|
WeightMsgSubmitProposal = 90
|
|
WeightMsgVote = 90
|
|
WeightMsgExec = 90
|
|
WeightMsgLeaveGroup = 5
|
|
WeightMsgUpdateGroupMetadata = 5
|
|
WeightMsgUpdateGroupAdmin = 5
|
|
WeightMsgUpdateGroupMembers = 5
|
|
WeightMsgUpdateGroupPolicyAdmin = 5
|
|
WeightMsgUpdateGroupPolicyDecisionPolicy = 5
|
|
WeightMsgUpdateGroupPolicyMetadata = 5
|
|
WeightMsgWithdrawProposal = 20
|
|
WeightMsgCreateGroupWithPolicy = 50
|
|
)
|
|
|
|
// WeightedOperations returns all the operations from the module with their respective weights
|
|
func WeightedOperations(
|
|
appParams simtypes.AppParams, cdc codec.JSONCodec, ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper, appCdc cdctypes.AnyUnpacker) simulation.WeightedOperations {
|
|
var (
|
|
weightMsgCreateGroup int
|
|
weightMsgUpdateGroupAdmin int
|
|
weightMsgUpdateGroupMetadata int
|
|
weightMsgUpdateGroupMembers int
|
|
weightMsgCreateGroupPolicy int
|
|
weightMsgUpdateGroupPolicyAdmin int
|
|
weightMsgUpdateGroupPolicyDecisionPolicy int
|
|
weightMsgUpdateGroupPolicyMetadata int
|
|
weightMsgSubmitProposal int
|
|
weightMsgVote int
|
|
weightMsgExec int
|
|
weightMsgLeaveGroup int
|
|
weightMsgWithdrawProposal int
|
|
weightMsgCreateGroupWithPolicy int
|
|
)
|
|
|
|
appParams.GetOrGenerate(cdc, OpMsgCreateGroup, &weightMsgCreateGroup, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgCreateGroup = WeightMsgCreateGroup
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgCreateGroupPolicy, &weightMsgCreateGroupPolicy, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgCreateGroupPolicy = WeightMsgCreateGroupPolicy
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgLeaveGroup, &weightMsgLeaveGroup, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgLeaveGroup = WeightMsgLeaveGroup
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgCreateGroupWithPolicy, &weightMsgCreateGroupWithPolicy, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgCreateGroupWithPolicy = WeightMsgCreateGroupWithPolicy
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgSubmitProposal, &weightMsgSubmitProposal, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgSubmitProposal = WeightMsgSubmitProposal
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgVote, &weightMsgVote, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgVote = WeightMsgVote
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgExec, &weightMsgExec, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgExec = WeightMsgExec
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupMetadata, &weightMsgUpdateGroupMetadata, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupMetadata = WeightMsgUpdateGroupMetadata
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupAdmin, &weightMsgUpdateGroupAdmin, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupAdmin = WeightMsgUpdateGroupAdmin
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupMembers, &weightMsgUpdateGroupMembers, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupMembers = WeightMsgUpdateGroupMembers
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupPolicyAdmin, &weightMsgUpdateGroupPolicyAdmin, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupPolicyAdmin = WeightMsgUpdateGroupPolicyAdmin
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupPolicyDecisionPolicy, &weightMsgUpdateGroupPolicyDecisionPolicy, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupPolicyDecisionPolicy = WeightMsgUpdateGroupPolicyDecisionPolicy
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgUpdateGroupPolicyMetaData, &weightMsgUpdateGroupPolicyMetadata, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgUpdateGroupPolicyMetadata = WeightMsgUpdateGroupPolicyMetadata
|
|
},
|
|
)
|
|
appParams.GetOrGenerate(cdc, OpMsgWithdrawProposal, &weightMsgWithdrawProposal, nil,
|
|
func(_ *rand.Rand) {
|
|
weightMsgWithdrawProposal = WeightMsgWithdrawProposal
|
|
},
|
|
)
|
|
|
|
// create two proposals for weightedOperations
|
|
var createProposalOps simulation.WeightedOperations
|
|
for i := 0; i < 2; i++ {
|
|
createProposalOps = append(createProposalOps, simulation.NewWeightedOperation(
|
|
weightMsgSubmitProposal,
|
|
SimulateMsgSubmitProposal(ak, bk, k),
|
|
))
|
|
}
|
|
|
|
wPreCreateProposalOps := simulation.WeightedOperations{
|
|
simulation.NewWeightedOperation(
|
|
weightMsgCreateGroup,
|
|
SimulateMsgCreateGroup(ak, bk),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgCreateGroupPolicy,
|
|
SimulateMsgCreateGroupPolicy(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgCreateGroupWithPolicy,
|
|
SimulateMsgCreateGroupWithPolicy(ak, bk),
|
|
),
|
|
}
|
|
|
|
wPostCreateProposalOps := simulation.WeightedOperations{
|
|
simulation.NewWeightedOperation(
|
|
WeightMsgWithdrawProposal,
|
|
SimulateMsgWithdrawProposal(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgVote,
|
|
SimulateMsgVote(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgExec,
|
|
SimulateMsgExec(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupMetadata,
|
|
SimulateMsgUpdateGroupMetadata(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupAdmin,
|
|
SimulateMsgUpdateGroupAdmin(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupMembers,
|
|
SimulateMsgUpdateGroupMembers(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupPolicyAdmin,
|
|
SimulateMsgUpdateGroupPolicyAdmin(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupPolicyDecisionPolicy,
|
|
SimulateMsgUpdateGroupPolicyDecisionPolicy(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgUpdateGroupPolicyMetadata,
|
|
SimulateMsgUpdateGroupPolicyMetadata(ak, bk, k),
|
|
),
|
|
simulation.NewWeightedOperation(
|
|
weightMsgLeaveGroup,
|
|
SimulateMsgLeaveGroup(k, ak, bk),
|
|
),
|
|
}
|
|
|
|
return append(wPreCreateProposalOps, append(createProposalOps, wPostCreateProposalOps...)...)
|
|
}
|
|
|
|
// SimulateMsgCreateGroup generates a MsgCreateGroup with random values
|
|
func SimulateMsgCreateGroup(ak group.AccountKeeper, bk group.BankKeeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, ctx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
acc, _ := simtypes.RandomAcc(r, accounts)
|
|
account := ak.GetAccount(ctx, acc.Address)
|
|
accAddr := acc.Address.String()
|
|
|
|
spendableCoins := bk.SpendableCoins(ctx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, ctx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroup, "fee error"), nil, err
|
|
}
|
|
|
|
members := genGroupMembers(r, accounts)
|
|
msg := &group.MsgCreateGroup{Admin: accAddr, Members: members, Metadata: simtypes.RandStringOfLength(r, 10)}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroup, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgCreateGroupWithPolicy generates a MsgCreateGroupWithPolicy with random values
|
|
func SimulateMsgCreateGroupWithPolicy(ak group.AccountKeeper, bk group.BankKeeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, ctx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
acc, _ := simtypes.RandomAcc(r, accounts)
|
|
account := ak.GetAccount(ctx, acc.Address)
|
|
accAddr := acc.Address.String()
|
|
|
|
spendableCoins := bk.SpendableCoins(ctx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, ctx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroup, "fee error"), nil, err
|
|
}
|
|
|
|
members := genGroupMembers(r, accounts)
|
|
decisionPolicy := &group.ThresholdDecisionPolicy{
|
|
Threshold: fmt.Sprintf("%d", simtypes.RandIntBetween(r, 1, 10)),
|
|
Windows: &group.DecisionPolicyWindows{
|
|
VotingPeriod: time.Second * time.Duration(30*24*60*60),
|
|
},
|
|
}
|
|
|
|
msg := &group.MsgCreateGroupWithPolicy{
|
|
Admin: accAddr,
|
|
Members: members,
|
|
GroupMetadata: simtypes.RandStringOfLength(r, 10),
|
|
GroupPolicyMetadata: simtypes.RandStringOfLength(r, 10),
|
|
GroupPolicyAsAdmin: r.Float32() < 0.5,
|
|
}
|
|
msg.SetDecisionPolicy(decisionPolicy)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to set decision policy"), nil, err
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupWithPolicy, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(msg, true, "", nil), nil, nil
|
|
}
|
|
}
|
|
|
|
// SimulateMsgCreateGroupPolicy generates a NewMsgCreateGroupPolicy with random values
|
|
func SimulateMsgCreateGroupPolicy(ak group.AccountKeeper, bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
groupInfo, acc, account, err := randomGroup(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupPolicy, ""), nil, err
|
|
}
|
|
if groupInfo == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupPolicy, ""), nil, nil
|
|
}
|
|
groupID := groupInfo.Id
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupPolicy, "fee error"), nil, err
|
|
}
|
|
|
|
msg, err := group.NewMsgCreateGroupPolicy(
|
|
acc.Address,
|
|
groupID,
|
|
simtypes.RandStringOfLength(r, 10),
|
|
&group.ThresholdDecisionPolicy{
|
|
Threshold: fmt.Sprintf("%d", simtypes.RandIntBetween(r, 1, 10)),
|
|
Windows: &group.DecisionPolicyWindows{
|
|
VotingPeriod: time.Second * time.Duration(30*24*60*60),
|
|
},
|
|
},
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupPolicy, err.Error()), nil, err
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgCreateGroupPolicy, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
fmt.Printf("ERR DELIVER %v\n", err)
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgSubmitProposal generates a NewMsgSubmitProposal with random values
|
|
func SimulateMsgSubmitProposal(ak group.AccountKeeper, bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
g, groupPolicy, _, _, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, ""), nil, err
|
|
}
|
|
if g == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, "no group found"), nil, nil
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, "no group policy found"), nil, nil
|
|
}
|
|
groupID := g.Id
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
// Return a no-op if we know the proposal cannot be created
|
|
policy, err := groupPolicy.GetDecisionPolicy()
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, ""), nil, nil
|
|
}
|
|
err = policy.Validate(*g, group.DefaultConfig())
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, ""), nil, nil
|
|
}
|
|
|
|
// Pick a random member from the group
|
|
ctx := sdk.WrapSDKContext(sdkCtx)
|
|
acc, account, err := randomMember(r, k, ak, ctx, accounts, groupID)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, ""), nil, err
|
|
}
|
|
if account == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, "no group member found"), nil, nil
|
|
}
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, "fee error"), nil, err
|
|
}
|
|
|
|
msg := group.MsgSubmitProposal{
|
|
GroupPolicyAddress: groupPolicyAddr,
|
|
Proposers: []string{acc.Address.String()},
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgSubmitProposal, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgUpdateGroupAdmin generates a MsgUpdateGroupAdmin with random values
|
|
func SimulateMsgUpdateGroupAdmin(ak group.AccountKeeper, bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
groupInfo, acc, account, err := randomGroup(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupAdmin, ""), nil, err
|
|
}
|
|
if groupInfo == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupAdmin, ""), nil, nil
|
|
}
|
|
groupID := groupInfo.Id
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupAdmin, "fee error"), nil, err
|
|
}
|
|
|
|
if len(accounts) == 1 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupAdmin, "can't set a new admin with only one account"), nil, nil
|
|
}
|
|
newAdmin, _ := simtypes.RandomAcc(r, accounts)
|
|
// disallow setting current admin as new admin
|
|
for acc.PubKey.Equals(newAdmin.PubKey) {
|
|
newAdmin, _ = simtypes.RandomAcc(r, accounts)
|
|
}
|
|
|
|
msg := group.MsgUpdateGroupAdmin{
|
|
GroupId: groupID,
|
|
Admin: account.GetAddress().String(),
|
|
NewAdmin: newAdmin.Address.String(),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupAdmin, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgUpdateGroupMetadata generates a MsgUpdateGroupMetadata with random values
|
|
func SimulateMsgUpdateGroupMetadata(ak group.AccountKeeper, bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
groupInfo, acc, account, err := randomGroup(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMetadata, ""), nil, err
|
|
}
|
|
if groupInfo == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMetadata, ""), nil, nil
|
|
}
|
|
groupID := groupInfo.Id
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMetadata, "fee error"), nil, err
|
|
}
|
|
|
|
msg := group.MsgUpdateGroupMetadata{
|
|
GroupId: groupID,
|
|
Admin: account.GetAddress().String(),
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMetadata, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgUpdateGroupMembers generates a MsgUpdateGroupMembers with random values
|
|
func SimulateMsgUpdateGroupMembers(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
groupInfo, acc, account, err := randomGroup(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMembers, ""), nil, err
|
|
}
|
|
if groupInfo == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMembers, ""), nil, nil
|
|
}
|
|
groupID := groupInfo.Id
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMembers, "fee error"), nil, err
|
|
}
|
|
|
|
members := genGroupMembers(r, accounts)
|
|
ctx := sdk.UnwrapSDKContext(sdkCtx)
|
|
res, err := k.GroupMembers(ctx, &group.QueryGroupMembersRequest{GroupId: groupID})
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMembers, "group members"), nil, err
|
|
}
|
|
|
|
// set existing radnom group member weight to zero to remove from the group
|
|
existigMembers := res.Members
|
|
if len(existigMembers) > 0 {
|
|
memberToRemove := existigMembers[r.Intn(len(existigMembers))]
|
|
var isDuplicateMember bool
|
|
for idx, m := range members {
|
|
if m.Address == memberToRemove.Member.Address {
|
|
members[idx].Weight = "0"
|
|
isDuplicateMember = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if !isDuplicateMember {
|
|
m := memberToRemove.Member
|
|
m.Weight = "0"
|
|
members = append(members, *m)
|
|
}
|
|
}
|
|
|
|
msg := group.MsgUpdateGroupMembers{
|
|
GroupId: groupID,
|
|
Admin: acc.Address.String(),
|
|
MemberUpdates: members,
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupMembers, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgUpdateGroupPolicyAdmin generates a MsgUpdateGroupPolicyAdmin with random values
|
|
func SimulateMsgUpdateGroupPolicyAdmin(ak group.AccountKeeper, bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
_, groupPolicy, acc, account, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyAdmin, ""), nil, err
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyAdmin, "no group policy found"), nil, nil
|
|
}
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyAdmin, "fee error"), nil, err
|
|
}
|
|
|
|
if len(accounts) == 1 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyAdmin, "can't set a new admin with only one account"), nil, nil
|
|
}
|
|
newAdmin, _ := simtypes.RandomAcc(r, accounts)
|
|
// disallow setting current admin as new admin
|
|
for acc.PubKey.Equals(newAdmin.PubKey) {
|
|
newAdmin, _ = simtypes.RandomAcc(r, accounts)
|
|
}
|
|
|
|
msg := group.MsgUpdateGroupPolicyAdmin{
|
|
Admin: acc.Address.String(),
|
|
GroupPolicyAddress: groupPolicyAddr,
|
|
NewAdmin: newAdmin.Address.String(),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyAdmin, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// // SimulateMsgUpdateGroupPolicyDecisionPolicy generates a NewMsgUpdateGroupPolicyDecisionPolicy with random values
|
|
func SimulateMsgUpdateGroupPolicyDecisionPolicy(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
_, groupPolicy, acc, account, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, ""), nil, err
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, "no group policy found"), nil, nil
|
|
}
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, "fee error"), nil, err
|
|
}
|
|
|
|
groupPolicyBech32, err := sdk.AccAddressFromBech32(groupPolicyAddr)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, fmt.Sprintf("fail to decide bech32 address: %s", err.Error())), nil, nil
|
|
}
|
|
|
|
msg, err := group.NewMsgUpdateGroupPolicyDecisionPolicy(acc.Address, groupPolicyBech32, &group.ThresholdDecisionPolicy{
|
|
Threshold: fmt.Sprintf("%d", simtypes.RandIntBetween(r, 1, 10)),
|
|
Windows: &group.DecisionPolicyWindows{
|
|
VotingPeriod: time.Second * time.Duration(simtypes.RandIntBetween(r, 100, 1000)),
|
|
},
|
|
})
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, err.Error()), nil, err
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyDecisionPolicy, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
return simtypes.NewOperationMsg(msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// // SimulateMsgUpdateGroupPolicyMetadata generates a MsgUpdateGroupPolicyMetadata with random values
|
|
func SimulateMsgUpdateGroupPolicyMetadata(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
_, groupPolicy, acc, account, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, ""), nil, err
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "no group policy found"), nil, nil
|
|
}
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "fee error"), nil, err
|
|
}
|
|
|
|
msg := group.MsgUpdateGroupPolicyMetadata{
|
|
Admin: acc.Address.String(),
|
|
GroupPolicyAddress: groupPolicyAddr,
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgWithdrawProposal generates a MsgWithdrawProposal with random values
|
|
func SimulateMsgWithdrawProposal(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
g, groupPolicy, _, _, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, ""), nil, err
|
|
}
|
|
if g == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "no group found"), nil, nil
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "no group policy found"), nil, nil
|
|
}
|
|
|
|
groupPolicyAddr := groupPolicy.Address
|
|
ctx := sdk.WrapSDKContext(sdkCtx)
|
|
|
|
policy, err := groupPolicy.GetDecisionPolicy()
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, err.Error()), nil, nil
|
|
}
|
|
err = policy.Validate(*g, group.DefaultConfig())
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, err.Error()), nil, nil
|
|
}
|
|
|
|
proposalsResult, err := k.ProposalsByGroupPolicy(ctx, &group.QueryProposalsByGroupPolicyRequest{Address: groupPolicyAddr})
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "fail to query group info"), nil, err
|
|
}
|
|
|
|
proposals := proposalsResult.GetProposals()
|
|
if len(proposals) == 0 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "no proposals found"), nil, nil
|
|
}
|
|
|
|
var proposal *group.Proposal
|
|
proposalID := -1
|
|
|
|
for _, p := range proposals {
|
|
if p.Status == group.PROPOSAL_STATUS_SUBMITTED {
|
|
timeout := p.VotingPeriodEnd
|
|
proposal = p
|
|
proposalID = int(p.Id)
|
|
if timeout.Before(sdkCtx.BlockTime()) || timeout.Equal(sdkCtx.BlockTime()) {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "voting period ended: skipping"), nil, nil
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
// return no-op if no proposal found
|
|
if proposalID == -1 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "no proposals found"), nil, nil
|
|
}
|
|
|
|
// select a random proposer
|
|
proposers := proposal.Proposers
|
|
n := randIntInRange(r, len(proposers))
|
|
proposerIdx := findAccount(accounts, proposers[n])
|
|
proposer := accounts[proposerIdx]
|
|
proposerAcc := ak.GetAccount(sdkCtx, proposer.Address)
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, proposer.Address)
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgWithdrawProposal, "fee error"), nil, err
|
|
}
|
|
|
|
msg := group.MsgWithdrawProposal{
|
|
ProposalId: uint64(proposalID),
|
|
Address: proposer.Address.String(),
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{proposerAcc.GetAccountNumber()},
|
|
[]uint64{proposerAcc.GetSequence()},
|
|
proposer.PrivKey,
|
|
)
|
|
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "group was modified") || strings.Contains(err.Error(), "group policy was modified") {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "no-op:group/group-policy was modified"), nil, nil
|
|
}
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgVote generates a MsgVote with random values
|
|
func SimulateMsgVote(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
g, groupPolicy, _, _, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, ""), nil, err
|
|
}
|
|
if g == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "no group found"), nil, nil
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "no group policy found"), nil, nil
|
|
}
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
// Pick a random member from the group
|
|
ctx := sdk.WrapSDKContext(sdkCtx)
|
|
acc, account, err := randomMember(r, k, ak, ctx, accounts, g.Id)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, ""), nil, err
|
|
}
|
|
if account == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "no group member found"), nil, nil
|
|
}
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "fee error"), nil, err
|
|
}
|
|
|
|
proposalsResult, err := k.ProposalsByGroupPolicy(ctx, &group.QueryProposalsByGroupPolicyRequest{Address: groupPolicyAddr})
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "fail to query group info"), nil, err
|
|
}
|
|
proposals := proposalsResult.GetProposals()
|
|
if len(proposals) == 0 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "no proposals found"), nil, nil
|
|
}
|
|
|
|
proposalID := -1
|
|
|
|
for _, p := range proposals {
|
|
if p.Status == group.PROPOSAL_STATUS_SUBMITTED {
|
|
timeout := p.VotingPeriodEnd
|
|
proposalID = int(p.Id)
|
|
if timeout.Before(sdkCtx.BlockTime()) || timeout.Equal(sdkCtx.BlockTime()) {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "voting period ended: skipping"), nil, nil
|
|
}
|
|
break
|
|
}
|
|
}
|
|
|
|
// return no-op if no proposal found
|
|
if proposalID == -1 {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "no proposals found"), nil, nil
|
|
}
|
|
|
|
// Ensure member hasn't already voted
|
|
res, _ := k.VoteByProposalVoter(ctx, &group.QueryVoteByProposalVoterRequest{
|
|
Voter: acc.Address.String(),
|
|
ProposalId: uint64(proposalID),
|
|
})
|
|
if res != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgVote, "member has already voted"), nil, nil
|
|
}
|
|
|
|
msg := group.MsgVote{
|
|
ProposalId: uint64(proposalID),
|
|
Voter: acc.Address.String(),
|
|
Option: group.VOTE_OPTION_YES,
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
}
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "group was modified") || strings.Contains(err.Error(), "group policy was modified") {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "no-op:group/group-policy was modified"), nil, nil
|
|
}
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// // SimulateMsgExec generates a MsgExec with random values
|
|
func SimulateMsgExec(ak group.AccountKeeper,
|
|
bk group.BankKeeper, k keeper.Keeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
_, groupPolicy, acc, account, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, ""), nil, err
|
|
}
|
|
if groupPolicy == nil {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, "no group policy found"), nil, nil
|
|
}
|
|
groupPolicyAddr := groupPolicy.Address
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, account.GetAddress())
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, "fee error"), nil, err
|
|
}
|
|
|
|
ctx := sdk.WrapSDKContext(sdkCtx)
|
|
proposalsResult, err := k.ProposalsByGroupPolicy(ctx, &group.QueryProposalsByGroupPolicyRequest{Address: groupPolicyAddr})
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, "fail to query group info"), nil, err
|
|
}
|
|
proposals := proposalsResult.GetProposals()
|
|
if len(proposals) == 0 {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, "no proposals found"), nil, nil
|
|
}
|
|
|
|
proposalID := -1
|
|
|
|
for _, proposal := range proposals {
|
|
if proposal.Status == group.PROPOSAL_STATUS_ACCEPTED {
|
|
proposalID = int(proposal.Id)
|
|
break
|
|
}
|
|
}
|
|
|
|
// return no-op if no proposal found
|
|
if proposalID == -1 {
|
|
return simtypes.NoOpMsg(TypeMsgExec, TypeMsgExec, "no proposals found"), nil, nil
|
|
}
|
|
|
|
msg := group.MsgExec{
|
|
ProposalId: uint64(proposalID),
|
|
Executor: acc.Address.String(),
|
|
}
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{&msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgUpdateGroupPolicyMetadata, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
if strings.Contains(err.Error(), "group was modified") || strings.Contains(err.Error(), "group policy was modified") {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "no-op:group/group-policy was modified"), nil, nil
|
|
}
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), "unable to deliver tx"), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(&msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
// SimulateMsgLeaveGroup generates a MsgLeaveGroup with random values
|
|
func SimulateMsgLeaveGroup(k keeper.Keeper, ak group.AccountKeeper, bk group.BankKeeper) simtypes.Operation {
|
|
return func(
|
|
r *rand.Rand, app *baseapp.BaseApp, sdkCtx sdk.Context, accounts []simtypes.Account, chainID string) (simtypes.OperationMsg, []simtypes.FutureOperation, error) {
|
|
|
|
ctx := sdk.WrapSDKContext(sdkCtx)
|
|
groupInfo, policyInfo, _, _, err := randomGroupPolicy(r, k, ak, sdkCtx, accounts)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, ""), nil, err
|
|
}
|
|
|
|
if policyInfo == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, "no policy found"), nil, nil
|
|
}
|
|
|
|
// Pick a random member from the group
|
|
acc, account, err := randomMember(r, k, ak, ctx, accounts, groupInfo.Id)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, ""), nil, err
|
|
}
|
|
if account == nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, "no group member found"), nil, nil
|
|
}
|
|
|
|
spendableCoins := bk.SpendableCoins(sdkCtx, acc.Address)
|
|
fees, err := simtypes.RandomFees(r, sdkCtx, spendableCoins)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, "fee error"), nil, err
|
|
}
|
|
|
|
msg := &group.MsgLeaveGroup{
|
|
Address: acc.Address.String(),
|
|
GroupId: groupInfo.Id,
|
|
}
|
|
|
|
txGen := simappparams.MakeTestEncodingConfig().TxConfig
|
|
tx, err := helpers.GenTx(
|
|
txGen,
|
|
[]sdk.Msg{msg},
|
|
fees,
|
|
helpers.DefaultGenTxGas,
|
|
chainID,
|
|
[]uint64{account.GetAccountNumber()},
|
|
[]uint64{account.GetSequence()},
|
|
acc.PrivKey,
|
|
)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, TypeMsgLeaveGroup, "unable to generate mock tx"), nil, err
|
|
}
|
|
|
|
_, _, err = app.SimDeliver(txGen.TxEncoder(), tx)
|
|
if err != nil {
|
|
return simtypes.NoOpMsg(group.ModuleName, msg.Type(), err.Error()), nil, err
|
|
}
|
|
|
|
return simtypes.NewOperationMsg(msg, true, "", nil), nil, err
|
|
}
|
|
}
|
|
|
|
func randomGroup(r *rand.Rand, k keeper.Keeper, ak group.AccountKeeper,
|
|
ctx sdk.Context, accounts []simtypes.Account) (groupInfo *group.GroupInfo, acc simtypes.Account, account authtypes.AccountI, err error) {
|
|
groupID := k.GetGroupSequence(ctx)
|
|
|
|
switch {
|
|
case groupID > initialGroupID:
|
|
// select a random ID between [initialGroupID, groupID]
|
|
groupID = uint64(simtypes.RandIntBetween(r, int(initialGroupID), int(groupID)))
|
|
|
|
default:
|
|
// This is called on the first call to this function
|
|
// in order to update the global variable
|
|
initialGroupID = groupID
|
|
}
|
|
|
|
res, err := k.GroupInfo(sdk.WrapSDKContext(ctx), &group.QueryGroupInfoRequest{GroupId: groupID})
|
|
if err != nil {
|
|
return nil, simtypes.Account{}, nil, err
|
|
}
|
|
|
|
groupInfo = res.Info
|
|
groupAdmin := groupInfo.Admin
|
|
found := -1
|
|
for i := range accounts {
|
|
if accounts[i].Address.String() == groupAdmin {
|
|
found = i
|
|
break
|
|
}
|
|
}
|
|
if found < 0 {
|
|
return nil, simtypes.Account{}, nil, nil
|
|
}
|
|
acc = accounts[found]
|
|
account = ak.GetAccount(ctx, acc.Address)
|
|
return groupInfo, acc, account, nil
|
|
}
|
|
|
|
func randomGroupPolicy(r *rand.Rand, k keeper.Keeper, ak group.AccountKeeper,
|
|
ctx sdk.Context, accounts []simtypes.Account) (groupInfo *group.GroupInfo, groupPolicyInfo *group.GroupPolicyInfo, acc simtypes.Account, account authtypes.AccountI, err error) {
|
|
groupInfo, _, _, err = randomGroup(r, k, ak, ctx, accounts)
|
|
if err != nil {
|
|
return nil, nil, simtypes.Account{}, nil, err
|
|
}
|
|
if groupInfo == nil {
|
|
return nil, nil, simtypes.Account{}, nil, nil
|
|
}
|
|
groupID := groupInfo.Id
|
|
|
|
result, err := k.GroupPoliciesByGroup(sdk.WrapSDKContext(ctx), &group.QueryGroupPoliciesByGroupRequest{GroupId: groupID})
|
|
if err != nil {
|
|
return groupInfo, nil, simtypes.Account{}, nil, err
|
|
}
|
|
|
|
n := randIntInRange(r, len(result.GroupPolicies))
|
|
if n < 0 {
|
|
return groupInfo, nil, simtypes.Account{}, nil, nil
|
|
}
|
|
groupPolicyInfo = result.GroupPolicies[n]
|
|
|
|
idx := findAccount(accounts, groupPolicyInfo.Admin)
|
|
if idx < 0 {
|
|
return groupInfo, nil, simtypes.Account{}, nil, nil
|
|
}
|
|
acc = accounts[idx]
|
|
account = ak.GetAccount(ctx, acc.Address)
|
|
return groupInfo, groupPolicyInfo, acc, account, nil
|
|
}
|
|
|
|
func randomMember(r *rand.Rand, k keeper.Keeper, ak group.AccountKeeper,
|
|
ctx context.Context, accounts []simtypes.Account, groupID uint64) (acc simtypes.Account, account authtypes.AccountI, err error) {
|
|
res, err := k.GroupMembers(ctx, &group.QueryGroupMembersRequest{
|
|
GroupId: groupID,
|
|
})
|
|
if err != nil {
|
|
return simtypes.Account{}, nil, err
|
|
}
|
|
n := randIntInRange(r, len(res.Members))
|
|
if n < 0 {
|
|
return simtypes.Account{}, nil, err
|
|
}
|
|
idx := findAccount(accounts, res.Members[n].Member.Address)
|
|
if idx < 0 {
|
|
return simtypes.Account{}, nil, err
|
|
}
|
|
acc = accounts[idx]
|
|
account = ak.GetAccount(sdk.UnwrapSDKContext(ctx), acc.Address)
|
|
return acc, account, nil
|
|
}
|
|
|
|
func randIntInRange(r *rand.Rand, l int) int {
|
|
if l == 0 {
|
|
return -1
|
|
}
|
|
if l == 1 {
|
|
return 0
|
|
} else {
|
|
return simtypes.RandIntBetween(r, 0, l-1)
|
|
}
|
|
}
|
|
|
|
func findAccount(accounts []simtypes.Account, addr string) (idx int) {
|
|
idx = -1
|
|
for i := range accounts {
|
|
if accounts[i].Address.String() == addr {
|
|
idx = i
|
|
break
|
|
}
|
|
}
|
|
return idx
|
|
}
|
|
|
|
func genGroupMembers(r *rand.Rand, accounts []simtypes.Account) []group.Member {
|
|
if len(accounts) == 1 {
|
|
return []group.Member{
|
|
{
|
|
Address: accounts[0].Address.String(),
|
|
Weight: fmt.Sprintf("%d", simtypes.RandIntBetween(r, 1, 10)),
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
},
|
|
}
|
|
}
|
|
|
|
max := 5
|
|
if len(accounts) < max {
|
|
max = len(accounts)
|
|
}
|
|
|
|
membersLen := simtypes.RandIntBetween(r, 1, max)
|
|
members := make([]group.Member, membersLen)
|
|
|
|
for i := 0; i < membersLen; i++ {
|
|
members[i] = group.Member{
|
|
Address: accounts[i].Address.String(),
|
|
Weight: fmt.Sprintf("%d", simtypes.RandIntBetween(r, 1, 10)),
|
|
Metadata: simtypes.RandStringOfLength(r, 10),
|
|
}
|
|
}
|
|
|
|
return members
|
|
}
|