-
Notifications
You must be signed in to change notification settings - Fork 83
/
gov.go
132 lines (121 loc) · 3.88 KB
/
gov.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
package app
import (
"bytes"
"testing"
"time"
abcitypes "github.com/cometbft/cometbft/abci/types"
sdk "github.com/cosmos/cosmos-sdk/types"
govtypesv1 "github.com/cosmos/cosmos-sdk/x/gov/types/v1"
"github.com/dydxprotocol/v4-chain/protocol/testutil/constants"
"github.com/stretchr/testify/require"
)
const (
TestMetadata = "test metadata"
TestTitle = "test title"
TestSummary = "test summary"
TestSubmitProposalTxHeight = uint32(2)
TestProposalTallyHeight = uint32(10)
)
var (
TestVotingPeriod = 1 * time.Minute
TestDeposit = sdk.Coins{sdk.NewInt64Coin(constants.TestNativeTokenDenom, 10_000_000)}
)
// SubmitAndTallyProposal simulates the following:
// - A proposal with the given messages is submitted. Check proposal submission succeeds or fails as expected.
// - If proposal successfully submitted:
// -- All validators vote for the proposal.
// -- The proposal is tallied after voting period ends.
func SubmitAndTallyProposal(
t *testing.T,
ctx sdk.Context,
tApp *TestApp,
messages []sdk.Msg,
expectSubmitProposalFails bool,
expectedProposalStatus govtypesv1.ProposalStatus,
) sdk.Context {
// Create a MsgSubmitProposal
msgSubmitProposal, err := govtypesv1.NewMsgSubmitProposal(
messages,
TestDeposit,
constants.AliceAccAddress.String(),
TestMetadata,
TestTitle,
TestSummary,
)
require.NoError(t, err)
// Create a signed transaction with MsgSubmitProposal
submitProposalCheckTx := MustMakeCheckTxWithPrivKeySupplier(
ctx,
tApp.App,
MustMakeCheckTxOptions{
AccAddressForSigning: constants.AliceAccAddress.String(),
Gas: constants.TestGasLimit,
FeeAmt: constants.TestFeeCoins_5Cents,
},
constants.GetPrivateKeyFromAddress,
msgSubmitProposal,
)
require.True(t, tApp.CheckTx(submitProposalCheckTx).IsOK())
if expectSubmitProposalFails {
ctx = tApp.AdvanceToBlock(TestSubmitProposalTxHeight, AdvanceToBlockOptions{
ValidateDeliverTxs: func(
context sdk.Context,
request abcitypes.RequestDeliverTx,
response abcitypes.ResponseDeliverTx,
_ int,
) (haltChain bool) {
if bytes.Equal(request.Tx, submitProposalCheckTx.Tx) {
require.True(t, response.IsErr())
} else {
require.True(t, response.IsOK())
}
return false
},
})
// Proposal submission failed. Return early.
return ctx
} else {
ctx = tApp.AdvanceToBlock(TestSubmitProposalTxHeight, AdvanceToBlockOptions{})
}
proposals := tApp.App.GovKeeper.GetProposals(ctx)
require.Len(t, proposals, 1)
// Have all 4 validators vote for the proposal.
for _, validator := range []sdk.AccAddress{
constants.AliceAccAddress,
constants.BobAccAddress,
constants.CarlAccAddress,
constants.DaveAccAddress,
} {
// Create a signed vote transaction
msgVote := govtypesv1.NewMsgVote(
validator,
1, // proposal ID
govtypesv1.VoteOption_VOTE_OPTION_YES,
"", // metdata
)
voteCheckTx := MustMakeCheckTxWithPrivKeySupplier(
ctx,
tApp.App,
MustMakeCheckTxOptions{
AccAddressForSigning: validator.String(),
Gas: constants.TestGasLimit,
FeeAmt: constants.TestFeeCoins_5Cents,
},
constants.GetPrivateKeyFromAddress,
msgVote,
)
require.True(t, tApp.CheckTx(voteCheckTx).IsOK())
}
// Advance to the height right before voting period ends.
ctx = tApp.AdvanceToBlock(TestProposalTallyHeight-1, AdvanceToBlockOptions{})
// Advance height to TestProposalTallyHeight; advance block time to right after voting period.
ctx = tApp.AdvanceToBlock(TestProposalTallyHeight, AdvanceToBlockOptions{
// Ensure the voting period has passed.
BlockTime: ctx.BlockTime().Add(TestVotingPeriod).Add(1 * time.Second),
})
proposals = tApp.App.GovKeeper.GetProposals(ctx)
require.Len(t, proposals, 1)
// Check that proposal was executed or failed as expected.
require.Equal(t, expectedProposalStatus, proposals[0].Status)
return ctx
}