238 lines
12 KiB
Go
238 lines
12 KiB
Go
package simulation_test
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
simappparams "github.com/cosmos/cosmos-sdk/simapp/params"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
simtypes "github.com/cosmos/cosmos-sdk/types/simulation"
|
|
"github.com/cosmos/cosmos-sdk/x/gov/simulation"
|
|
"github.com/cosmos/cosmos-sdk/x/gov/types"
|
|
minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
|
|
)
|
|
|
|
type MockWeightedProposalContent struct {
|
|
n int
|
|
}
|
|
|
|
func (m MockWeightedProposalContent) AppParamsKey() string {
|
|
return fmt.Sprintf("AppParamsKey-%d", m.n)
|
|
}
|
|
|
|
func (m MockWeightedProposalContent) DefaultWeight() int {
|
|
return m.n
|
|
}
|
|
|
|
func (m MockWeightedProposalContent) ContentSimulatorFn() simtypes.ContentSimulatorFn {
|
|
return func(r *rand.Rand, _ sdk.Context, _ []simtypes.Account) simtypes.Content {
|
|
return types.NewTextProposal(
|
|
fmt.Sprintf("title-%d: %s", m.n, simtypes.RandStringOfLength(r, 100)),
|
|
fmt.Sprintf("description-%d: %s", m.n, simtypes.RandStringOfLength(r, 4000)),
|
|
)
|
|
}
|
|
}
|
|
|
|
// make sure the MockWeightedProposalContent satisfied the WeightedProposalContent interface
|
|
var _ simtypes.WeightedProposalContent = MockWeightedProposalContent{}
|
|
|
|
func mockWeightedProposalContent(n int) []simtypes.WeightedProposalContent {
|
|
wpc := make([]simtypes.WeightedProposalContent, n)
|
|
for i := 0; i < n; i++ {
|
|
wpc[i] = MockWeightedProposalContent{i}
|
|
|
|
}
|
|
return wpc
|
|
|
|
}
|
|
|
|
// TestWeightedOperations tests the weights of the operations.
|
|
func TestWeightedOperations(t *testing.T) {
|
|
app, ctx := createTestApp(false)
|
|
ctx.WithChainID("test-chain")
|
|
|
|
cdc := app.AppCodec()
|
|
appParams := make(simtypes.AppParams)
|
|
|
|
weightesOps := simulation.WeightedOperations(appParams, cdc, app.AccountKeeper,
|
|
app.BankKeeper, app.GovKeeper, mockWeightedProposalContent(3),
|
|
)
|
|
|
|
// setup 3 accounts
|
|
s := rand.NewSource(1)
|
|
r := rand.New(s)
|
|
accs := getTestingAccounts(t, r, app, ctx, 3)
|
|
|
|
expected := []struct {
|
|
weight int
|
|
opMsgRoute string
|
|
opMsgName string
|
|
}{
|
|
{0, types.ModuleName, "submit_proposal"},
|
|
{1, types.ModuleName, "submit_proposal"},
|
|
{2, types.ModuleName, "submit_proposal"},
|
|
{simappparams.DefaultWeightMsgDeposit, types.ModuleName, types.TypeMsgDeposit},
|
|
{simappparams.DefaultWeightMsgVote, types.ModuleName, types.TypeMsgVote},
|
|
}
|
|
|
|
for i, w := range weightesOps {
|
|
operationMsg, _, _ := w.Op()(r, app.BaseApp, ctx, accs, ctx.ChainID())
|
|
// the following checks are very much dependent from the ordering of the output given
|
|
// by WeightedOperations. if the ordering in WeightedOperations changes some tests
|
|
// will fail
|
|
require.Equal(t, expected[i].weight, w.Weight(), "weight should be the same")
|
|
require.Equal(t, expected[i].opMsgRoute, operationMsg.Route, "route should be the same")
|
|
require.Equal(t, expected[i].opMsgName, operationMsg.Name, "operation Msg name should be the same")
|
|
}
|
|
}
|
|
|
|
// TestSimulateMsgSubmitProposal tests the normal scenario of a valid message of type TypeMsgSubmitProposal.
|
|
// Abonormal scenarios, where the message is created by an errors are not tested here.
|
|
func TestSimulateMsgSubmitProposal(t *testing.T) {
|
|
app, ctx := createTestApp(false)
|
|
|
|
// setup 3 accounts
|
|
s := rand.NewSource(1)
|
|
r := rand.New(s)
|
|
accounts := getTestingAccounts(t, r, app, ctx, 3)
|
|
|
|
// begin a new block
|
|
app.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: app.LastBlockHeight() + 1, AppHash: app.LastCommitID().Hash}})
|
|
|
|
// execute operation
|
|
op := simulation.SimulateMsgSubmitProposal(app.AccountKeeper, app.BankKeeper, app.GovKeeper, MockWeightedProposalContent{3}.ContentSimulatorFn())
|
|
operationMsg, _, err := op(r, app.BaseApp, ctx, accounts, "")
|
|
require.NoError(t, err)
|
|
|
|
var msg types.MsgSubmitProposal
|
|
types.ModuleCdc.UnmarshalJSON(operationMsg.Msg, &msg)
|
|
|
|
require.True(t, operationMsg.OK)
|
|
require.Equal(t, "cosmos1p8wcgrjr4pjju90xg6u9cgq55dxwq8j7u4x9a0", msg.Proposer.String())
|
|
require.Equal(t, "2686011stake", msg.InitialDeposit.String())
|
|
require.Equal(t, "title-3: ZBSpYuLyYggwexjxusrBqDOTtGTOWeLrQKjLxzIivHSlcxgdXhhuTSkuxKGLwQvuyNhYFmBZHeAerqyNEUzXPFGkqEGqiQWIXnku", msg.GetContent().GetTitle())
|
|
require.Equal(t, "description-3: NJWzHdBNpAXKJPHWQdrGYcAHSctgVlqwqHoLfHsXUdStwfefwzqLuKEhmMyYLdbZrcPgYqjNHxPexsruwEGStAneKbWkQDDIlCWBLSiAASNhZqNFlPtfqPJoxKsgMdzjWqLWdqKQuJqWPMvwPQWZUtVMOTMYKJbfdlZsjdsomuScvDmbDkgRualsxDvRJuCAmPOXitIbcyWsKGSdrEunFAOdmXnsuyFVgJqEjbklvmwrUlsxjRSfKZxGcpayDdgoFcnVSutxjRgOSFzPwidAjubMncNweqpbxhXGchpZUxuFDOtpnhNUycJICRYqsPhPSCjPTWZFLkstHWJxvdPEAyEIxXgLwbNOjrgzmaujiBABBIXvcXpLrbcEWNNQsbjvgJFgJkflpRohHUutvnaUqoopuKjTDaemDeSdqbnOzcfJpcTuAQtZoiLZOoAIlboFDAeGmSNwkvObPRvRWQgWkGkxwtPauYgdkmypLjbqhlHJIQTntgWjXwZdOyYEdQRRLfMSdnxqppqUofqLbLQDUjwKVKfZJUJQPsWIPwIVaSTrmKskoAhvmZyJgeRpkaTfGgrJzAigcxtfshmiDCFkuiluqtMOkidknnTBtumyJYlIsWLnCQclqdVmikUoMOPdPWwYbJxXyqUVicNxFxyqJTenNblyyKSdlCbiXxUiYUiMwXZASYfvMDPFgxniSjWaZTjHkqlJvtBsXqwPpyVxnJVGFWhfSxgOcduoxkiopJvFjMmFabrGYeVtTXLhxVUEiGwYUvndjFGzDVntUvibiyZhfMQdMhgsiuysLMiePBNXifRLMsSmXPkwlPloUbJveCvUlaalhZHuvdkCnkSHbMbmOnrfEGPwQiACiPlnihiaOdbjPqPiTXaHDoJXjSlZmltGqNHHNrcKdlFSCdmVOuvDcBLdSklyGJmcLTbSFtALdGlPkqqecJrpLCXNPWefoTJNgEJlyMEPneVaxxduAAEqQpHWZodWyRkDAxzyMnFMcjSVqeRXLqsNyNtQBbuRvunZflWSbbvXXdkyLikYqutQhLPONXbvhcQZJPSWnOulqQaXmbfFxAkqfYeseSHOQidHwbcsOaMnSrrmGjjRmEMQNuknupMxJiIeVjmgZvbmjPIQTEhQFULQLBMPrxcFPvBinaOPYWGvYGRKxLZdwamfRQQFngcdSlvwjfaPbURasIsGJVHtcEAxnIIrhSriiXLOlbEBLXFElXJFGxHJczRBIxAuPKtBisjKBwfzZFagdNmjdwIRvwzLkFKWRTDPxJCmpzHUcrPiiXXHnOIlqNVoGSXZewdnCRhuxeYGPVTfrNTQNOxZmxInOazUYNTNDgzsxlgiVEHPKMfbesvPHUqpNkUqbzeuzfdrsuLDpKHMUbBMKczKKWOdYoIXoPYtEjfOnlQLoGnbQUCuERdEFaptwnsHzTJDsuZkKtzMpFaZobynZdzNydEeJJHDYaQcwUxcqvwfWwNUsCiLvkZQiSfzAHftYgAmVsXgtmcYgTqJIawstRYJrZdSxlfRiqTufgEQVambeZZmaAyRQbcmdjVUZZCgqDrSeltJGXPMgZnGDZqISrGDOClxXCxMjmKqEPwKHoOfOeyGmqWqihqjINXLqnyTesZePQRqaWDQNqpLgNrAUKulklmckTijUltQKuWQDwpLmDyxLppPVMwsmBIpOwQttYFMjgJQZLYFPmxWFLIeZihkRNnkzoypBICIxgEuYsVWGIGRbbxqVasYnstWomJnHwmtOhAFSpttRYYzBmyEtZXiCthvKvWszTXDbiJbGXMcrYpKAgvUVFtdKUfvdMfhAryctklUCEdjetjuGNfJjajZtvzdYaqInKtFPPLYmRaXPdQzxdSQfmZDEVHlHGEGNSPRFJuIfKLLfUmnHxHnRjmzQPNlqrXgifUdzAGKVabYqvcDeYoTYgPsBUqehrBhmQUgTvDnsdpuhUoxskDdppTsYMcnDIPSwKIqhXDCIxOuXrywahvVavvHkPuaenjLmEbMgrkrQLHEAwrhHkPRNvonNQKqprqOFVZKAtpRSpvQUxMoXCMZLSSbnLEFsjVfANdQNQVwTmGxqVjVqRuxREAhuaDrFgEZpYKhwWPEKBevBfsOIcaZKyykQafzmGPLRAKDtTcJxJVgiiuUkmyMYuDUNEUhBEdoBLJnamtLmMJQgmLiUELIhLpiEvpOXOvXCPUeldLFqkKOwfacqIaRcnnZvERKRMCKUkMABbDHytQqQblrvoxOZkwzosQfDKGtIdfcXRJNqlBNwOCWoQBcEWyqrMlYZIAXYJmLfnjoJepgSFvrgajaBAIksoyeHqgqbGvpAstMIGmIhRYGGNPRIfOQKsGoKgxtsidhTaAePRCBFqZgPDWCIkqOJezGVkjfYUCZTlInbxBXwUAVRsxHTQtJFnnpmMvXDYCVlEmnZBKhmmxQOIQzxFWpJQkQoSAYzTEiDWEOsVLNrbfzeHFRyeYATakQQWmFDLPbVMCJcWjFGJjfqCoVzlbNNEsqxdSmNPjTjHYOkuEMFLkXYGaoJlraLqayMeCsTjWNRDPBywBJLAPVkGQqTwApVVwYAetlwSbzsdHWsTwSIcctkyKDuRWYDQikRqsKTMJchrliONJeaZIzwPQrNbTwxsGdwuduvibtYndRwpdsvyCktRHFalvUuEKMqXbItfGcNGWsGzubdPMYayOUOINjpcFBeESdwpdlTYmrPsLsVDhpTzoMegKrytNVZkfJRPuDCUXxSlSthOohmsuxmIZUedzxKmowKOdXTMcEtdpHaPWgIsIjrViKrQOCONlSuazmLuCUjLltOGXeNgJKedTVrrVCpWYWHyVrdXpKgNaMJVjbXxnVMSChdWKuZdqpisvrkBJPoURDYxWOtpjzZoOpWzyUuYNhCzRoHsMjmmWDcXzQiHIyjwdhPNwiPqFxeUfMVFQGImhykFgMIlQEoZCaRoqSBXTSWAeDumdbsOGtATwEdZlLfoBKiTvodQBGOEcuATWXfiinSjPmJKcWgQrTVYVrwlyMWhxqNbCMpIQNoSMGTiWfPTCezUjYcdWppnsYJihLQCqbNLRGgqrwHuIvsazapTpoPZIyZyeeSueJuTIhpHMEJfJpScshJubJGfkusuVBgfTWQoywSSliQQSfbvaHKiLnyjdSbpMkdBgXepoSsHnCQaYuHQqZsoEOmJCiuQUpJkmfyfbIShzlZpHFmLCsbknEAkKXKfRTRnuwdBeuOGgFbJLbDksHVapaRayWzwoYBEpmrlAxrUxYMUekKbpjPNfjUCjhbdMAnJmYQVZBQZkFVweHDAlaqJjRqoQPoOMLhyvYCzqEuQsAFoxWrzRnTVjStPadhsESlERnKhpEPsfDxNvxqcOyIulaCkmPdambLHvGhTZzysvqFauEgkFRItPfvisehFmoBhQqmkfbHVsgfHXDPJVyhwPllQpuYLRYvGodxKjkarnSNgsXoKEMlaSKxKdcVgvOkuLcfLFfdtXGTclqfPOfeoVLbqcjcXCUEBgAGplrkgsmIEhWRZLlGPGCwKWRaCKMkBHTAcypUrYjWwCLtOPVygMwMANGoQwFnCqFrUGMCRZUGJKTZIGPyldsifauoMnJPLTcDHmilcmahlqOELaAUYDBuzsVywnDQfwRLGIWozYaOAilMBcObErwgTDNGWnwQMUgFFSKtPDMEoEQCTKVREqrXZSGLqwTMcxHfWotDllNkIJPMbXzjDVjPOOjCFuIvTyhXKLyhUScOXvYthRXpPfKwMhptXaxIxgqBoUqzrWbaoLTVpQoottZyPFfNOoMioXHRuFwMRYUiKvcWPkrayyTLOCFJlAyslDameIuqVAuxErqFPEWIScKpBORIuZqoXlZuTvAjEdlEWDODFRregDTqGNoFBIHxvimmIZwLfFyKUfEWAnNBdtdzDmTPXtpHRGdIbuucfTjOygZsTxPjfweXhSUkMhPjMaxKlMIJMOXcnQfyzeOcbWwNbeH", msg.GetContent().GetDescription())
|
|
require.Equal(t, "gov", msg.Route())
|
|
require.Equal(t, types.TypeMsgSubmitProposal, msg.Type())
|
|
}
|
|
|
|
// TestSimulateMsgDeposit tests the normal scenario of a valid message of type TypeMsgDeposit.
|
|
// Abonormal scenarios, where the message is created by an errors are not tested here.
|
|
func TestSimulateMsgDeposit(t *testing.T) {
|
|
app, ctx := createTestApp(false)
|
|
blockTime := time.Now().UTC()
|
|
ctx = ctx.WithBlockTime(blockTime)
|
|
|
|
// setup 3 accounts
|
|
s := rand.NewSource(1)
|
|
r := rand.New(s)
|
|
accounts := getTestingAccounts(t, r, app, ctx, 3)
|
|
|
|
// setup a proposal
|
|
content := types.NewTextProposal("Test", "description")
|
|
|
|
submitTime := ctx.BlockHeader().Time
|
|
depositPeriod := app.GovKeeper.GetDepositParams(ctx).MaxDepositPeriod
|
|
|
|
proposal, err := types.NewProposal(content, 1, submitTime, submitTime.Add(depositPeriod))
|
|
require.NoError(t, err)
|
|
|
|
app.GovKeeper.SetProposal(ctx, proposal)
|
|
|
|
// begin a new block
|
|
app.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: app.LastBlockHeight() + 1, AppHash: app.LastCommitID().Hash, Time: blockTime}})
|
|
|
|
// execute operation
|
|
op := simulation.SimulateMsgDeposit(app.AccountKeeper, app.BankKeeper, app.GovKeeper)
|
|
operationMsg, _, err := op(r, app.BaseApp, ctx, accounts, "")
|
|
require.NoError(t, err)
|
|
|
|
var msg types.MsgDeposit
|
|
types.ModuleCdc.UnmarshalJSON(operationMsg.Msg, &msg)
|
|
|
|
require.True(t, operationMsg.OK)
|
|
require.Equal(t, uint64(1), msg.ProposalId)
|
|
require.Equal(t, "cosmos1ghekyjucln7y67ntx7cf27m9dpuxxemn4c8g4r", msg.Depositor.String())
|
|
require.Equal(t, "560969stake", msg.Amount.String())
|
|
require.Equal(t, "gov", msg.Route())
|
|
require.Equal(t, types.TypeMsgDeposit, msg.Type())
|
|
}
|
|
|
|
// TestSimulateMsgVote tests the normal scenario of a valid message of type TypeMsgVote.
|
|
// Abonormal scenarios, where the message is created by an errors are not tested here.
|
|
func TestSimulateMsgVote(t *testing.T) {
|
|
app, ctx := createTestApp(false)
|
|
blockTime := time.Now().UTC()
|
|
ctx = ctx.WithBlockTime(blockTime)
|
|
|
|
// setup 3 accounts
|
|
s := rand.NewSource(1)
|
|
r := rand.New(s)
|
|
accounts := getTestingAccounts(t, r, app, ctx, 3)
|
|
|
|
// setup a proposal
|
|
content := types.NewTextProposal("Test", "description")
|
|
|
|
submitTime := ctx.BlockHeader().Time
|
|
depositPeriod := app.GovKeeper.GetDepositParams(ctx).MaxDepositPeriod
|
|
|
|
proposal, err := types.NewProposal(content, 1, submitTime, submitTime.Add(depositPeriod))
|
|
require.NoError(t, err)
|
|
|
|
app.GovKeeper.ActivateVotingPeriod(ctx, proposal)
|
|
|
|
// begin a new block
|
|
app.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: app.LastBlockHeight() + 1, AppHash: app.LastCommitID().Hash, Time: blockTime}})
|
|
|
|
// execute operation
|
|
op := simulation.SimulateMsgVote(app.AccountKeeper, app.BankKeeper, app.GovKeeper)
|
|
operationMsg, _, err := op(r, app.BaseApp, ctx, accounts, "")
|
|
require.NoError(t, err)
|
|
|
|
var msg types.MsgVote
|
|
types.ModuleCdc.UnmarshalJSON(operationMsg.Msg, &msg)
|
|
|
|
require.True(t, operationMsg.OK)
|
|
require.Equal(t, uint64(1), msg.ProposalId)
|
|
require.Equal(t, "cosmos1ghekyjucln7y67ntx7cf27m9dpuxxemn4c8g4r", msg.Voter.String())
|
|
require.Equal(t, types.OptionYes, msg.Option)
|
|
require.Equal(t, "gov", msg.Route())
|
|
require.Equal(t, types.TypeMsgVote, msg.Type())
|
|
|
|
}
|
|
|
|
// returns context and an app with updated mint keeper
|
|
func createTestApp(isCheckTx bool) (*simapp.SimApp, sdk.Context) {
|
|
app := simapp.Setup(isCheckTx)
|
|
|
|
ctx := app.BaseApp.NewContext(isCheckTx, tmproto.Header{})
|
|
app.MintKeeper.SetParams(ctx, minttypes.DefaultParams())
|
|
app.MintKeeper.SetMinter(ctx, minttypes.DefaultInitialMinter())
|
|
|
|
return app, ctx
|
|
}
|
|
|
|
func getTestingAccounts(t *testing.T, r *rand.Rand, app *simapp.SimApp, ctx sdk.Context, n int) []simtypes.Account {
|
|
accounts := simtypes.RandomAccounts(r, n)
|
|
|
|
initAmt := sdk.TokensFromConsensusPower(200)
|
|
initCoins := sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, initAmt))
|
|
|
|
// add coins to the accounts
|
|
for _, account := range accounts {
|
|
acc := app.AccountKeeper.NewAccountWithAddress(ctx, account.Address)
|
|
app.AccountKeeper.SetAccount(ctx, acc)
|
|
err := app.BankKeeper.SetBalances(ctx, account.Address, initCoins)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
return accounts
|
|
}
|