mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-25 15:55:18 +00:00
411 lines
14 KiB
Go
411 lines
14 KiB
Go
package app
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
|
|
"github.com/cosmos/cosmos-sdk/crypto/keys/ed25519"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
govkeeper "github.com/cosmos/cosmos-sdk/x/gov/keeper"
|
|
govv1 "github.com/cosmos/cosmos-sdk/x/gov/types/v1"
|
|
govv1beta1 "github.com/cosmos/cosmos-sdk/x/gov/types/v1beta1"
|
|
"github.com/cosmos/cosmos-sdk/x/staking"
|
|
stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
|
|
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
// d is an alias for sdk.MustNewDecFromStr
|
|
var d = sdk.MustNewDecFromStr
|
|
|
|
type tallyHandlerSuite struct {
|
|
suite.Suite
|
|
app TestApp
|
|
ctx sdk.Context
|
|
|
|
staking stakingHelper
|
|
|
|
tallier TallyHandler
|
|
}
|
|
|
|
func TestTallyHandlerSuite(t *testing.T) {
|
|
suite.Run(t, new(tallyHandlerSuite))
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) SetupTest() {
|
|
suite.app = NewTestApp()
|
|
suite.app.InitializeFromGenesisStates()
|
|
genesisTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.app.NewContext(false, tmproto.Header{Height: 1, Time: genesisTime})
|
|
|
|
stakingKeeper := *suite.app.GetStakingKeeper()
|
|
suite.staking = stakingHelper{stakingKeeper}
|
|
suite.staking.setBondDenom(suite.ctx, "ukava")
|
|
|
|
suite.tallier = NewTallyHandler(
|
|
suite.app.GetGovKeeper(),
|
|
stakingKeeper,
|
|
suite.app.GetBankKeeper(),
|
|
)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) TestVotePower_AllSourcesCounted() {
|
|
user := suite.createAccount(suite.newBondCoin(sdkmath.NewInt(1e9)))
|
|
|
|
validator := suite.delegateToNewBondedValidator(user.GetAddress(), sdkmath.NewInt(1e9))
|
|
|
|
derivatives := suite.mintDerivative(user.GetAddress(), validator.GetOperator(), sdkmath.NewInt(500e6))
|
|
|
|
suite.allowBKavaEarnDeposits()
|
|
suite.earnDeposit(
|
|
user.GetAddress(),
|
|
sdk.NewCoin(derivatives.Denom, sdkmath.NewInt(250e6)),
|
|
)
|
|
|
|
proposal := suite.createProposal()
|
|
suite.voteOnProposal(user.GetAddress(), proposal.Id, govv1beta1.OptionYes)
|
|
|
|
_, _, results := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Equal(sdkmath.NewInt(500e6+250e6+250e6).String(), results.YesCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.NoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.NoWithVetoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.AbstainCount)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) TestVotePower_UserOverridesValidator() {
|
|
user := suite.createAccount(suite.newBondCoin(sdkmath.NewInt(1e9)))
|
|
|
|
delegated := sdkmath.NewInt(1e9)
|
|
validator := suite.delegateToNewBondedValidator(user.GetAddress(), delegated)
|
|
selfDelegated := validator.GetTokens().Sub(delegated)
|
|
|
|
derivatives := suite.mintDerivative(user.GetAddress(), validator.GetOperator(), sdkmath.NewInt(500e6))
|
|
|
|
suite.allowBKavaEarnDeposits()
|
|
suite.earnDeposit(
|
|
user.GetAddress(),
|
|
sdk.NewCoin(derivatives.Denom, sdkmath.NewInt(250e6)),
|
|
)
|
|
|
|
proposal := suite.createProposal()
|
|
|
|
// Validator votes, inheriting user's stake and bkava.
|
|
suite.voteOnProposal(validator.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionYes)
|
|
|
|
// use wrapped context to discard the state changes
|
|
readOnlyCtx, _ := suite.ctx.CacheContext()
|
|
_, _, results := suite.tallier.Tally(readOnlyCtx, proposal)
|
|
userPower := sdkmath.NewInt(500e6 + 250e6 + 250e6)
|
|
suite.Equal(
|
|
selfDelegated.Add(userPower).String(),
|
|
results.YesCount,
|
|
)
|
|
suite.Equal(sdk.ZeroInt().String(), results.NoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.NoWithVetoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.AbstainCount)
|
|
|
|
// User votes, taking power away from validator.
|
|
suite.voteOnProposal(user.GetAddress(), proposal.Id, govv1beta1.OptionNo)
|
|
|
|
_, _, results = suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Equal(selfDelegated.String(), results.YesCount)
|
|
suite.Equal(userPower.String(), results.NoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.NoWithVetoCount)
|
|
suite.Equal(sdk.ZeroInt().String(), results.AbstainCount)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) TestTallyOutcomes() {
|
|
suite.Run("VotedPowerBelowQuorumFails", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
v1 := suite.createNewBondedValidator(sdkmath.NewInt(399_999_999))
|
|
suite.createNewBondedValidator(sdkmath.NewInt(600_000_001))
|
|
|
|
suite.voteOnProposal(v1.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionYes)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Falsef(passes, "expected proposal to fail, tally: %v", tally)
|
|
suite.Truef(burns, "expected deposit to be burned, tally: %v", tally)
|
|
})
|
|
suite.Run("VetoedFails", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
v1 := suite.createNewBondedValidator(sdkmath.NewInt(334_000_001))
|
|
v2 := suite.createNewBondedValidator(sdkmath.NewInt(665_999_999))
|
|
|
|
suite.voteOnProposal(v1.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionNoWithVeto)
|
|
suite.voteOnProposal(v2.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionYes)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Falsef(passes, "expected proposal to fail, tally: %v", tally)
|
|
suite.Truef(burns, "expected deposit to be burned, tally: %v", tally)
|
|
})
|
|
suite.Run("UnvetoedAndYesAboveThresholdPasses", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
v1 := suite.createNewBondedValidator(sdkmath.NewInt(900_000_000))
|
|
v2 := suite.createNewBondedValidator(sdkmath.NewInt(50_000_001))
|
|
v3 := suite.createNewBondedValidator(sdkmath.NewInt(49_999_999))
|
|
|
|
suite.voteOnProposal(v1.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionAbstain)
|
|
suite.voteOnProposal(v2.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionYes)
|
|
suite.voteOnProposal(v3.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionNo)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Truef(passes, "expected proposal to pass, tally: %v", tally)
|
|
suite.Falsef(burns, "expected deposit to not burn, tally: %v", tally)
|
|
})
|
|
suite.Run("UnvetoedAndYesBelowThresholdFails", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
v1 := suite.createNewBondedValidator(sdkmath.NewInt(900_000_000))
|
|
v2 := suite.createNewBondedValidator(sdkmath.NewInt(49_999_999))
|
|
v3 := suite.createNewBondedValidator(sdkmath.NewInt(50_000_001))
|
|
|
|
suite.voteOnProposal(v1.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionAbstain)
|
|
suite.voteOnProposal(v2.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionYes)
|
|
suite.voteOnProposal(v3.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionNo)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Falsef(passes, "expected proposal to pass, tally: %v", tally)
|
|
suite.Falsef(burns, "expected deposit to not burn, tally: %v", tally)
|
|
})
|
|
suite.Run("NotEnoughStakeFails", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
// no stake
|
|
suite.app.DeleteGenesisValidator(suite.T(), suite.ctx)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Falsef(passes, "expected proposal to pass, tally: %v", tally)
|
|
suite.Falsef(burns, "expected deposit to not burn, tally: %v", tally)
|
|
})
|
|
suite.Run("UnvetoedAndAllAbstainedFails", func() {
|
|
suite.SetupTest()
|
|
suite.setTallyParams(d("0.4"), d("0.5"), d("0.334"))
|
|
proposal := suite.createProposal()
|
|
|
|
v1 := suite.createNewBondedValidator(sdkmath.NewInt(1e9))
|
|
|
|
suite.voteOnProposal(v1.GetOperator().Bytes(), proposal.Id, govv1beta1.OptionAbstain)
|
|
|
|
passes, burns, tally := suite.tallier.Tally(suite.ctx, proposal)
|
|
suite.Falsef(passes, "expected proposal to pass, tally: %v", tally)
|
|
suite.Falsef(burns, "expected deposit to not burn, tally: %v", tally)
|
|
})
|
|
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) setTallyParams(quorum, threshold, veto sdk.Dec) {
|
|
params := suite.app.GetGovKeeper().GetParams(suite.ctx)
|
|
params.Quorum = quorum.String()
|
|
params.Threshold = threshold.String()
|
|
params.VetoThreshold = veto.String()
|
|
params.BurnVoteQuorum = true
|
|
suite.app.GetGovKeeper().SetParams(suite.ctx, params)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) voteOnProposal(
|
|
voter sdk.AccAddress,
|
|
proposalID uint64,
|
|
option govv1beta1.VoteOption,
|
|
) {
|
|
gk := suite.app.GetGovKeeper()
|
|
|
|
err := gk.AddVote(suite.ctx,
|
|
proposalID,
|
|
voter,
|
|
govv1.NewNonSplitVoteOption(govv1.VoteOption(option)),
|
|
"",
|
|
)
|
|
suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) createProposal() govv1.Proposal {
|
|
gk := suite.app.GetGovKeeper()
|
|
deposit := gk.GetParams(suite.ctx).MinDeposit
|
|
proposer := suite.createAccount(deposit...)
|
|
|
|
msg, err := govv1beta1.NewMsgSubmitProposal(
|
|
govv1beta1.NewTextProposal("a title", "a description"),
|
|
deposit,
|
|
proposer.GetAddress(),
|
|
)
|
|
suite.Require().NoError(err)
|
|
|
|
msgServerv1 := govkeeper.NewMsgServerImpl(&gk)
|
|
|
|
govAcct := gk.GetGovernanceAccount(suite.ctx).GetAddress()
|
|
msgServer := govkeeper.NewLegacyMsgServerImpl(govAcct.String(), msgServerv1)
|
|
res, err := msgServer.SubmitProposal(sdk.WrapSDKContext(suite.ctx), msg)
|
|
suite.Require().NoError(err)
|
|
|
|
proposal, found := gk.GetProposal(suite.ctx, res.ProposalId)
|
|
if !found {
|
|
panic("proposal not found")
|
|
}
|
|
return proposal
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) newBondCoin(amount sdkmath.Int) sdk.Coin {
|
|
return suite.staking.newBondCoin(suite.ctx, amount)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) allowBKavaEarnDeposits() {
|
|
// ek := suite.app.GetEarnKeeper()
|
|
// earnParams := ek.GetParams(suite.ctx)
|
|
|
|
// vault := earntypes.NewAllowedVault(
|
|
// liquidtypes.DefaultDerivativeDenom,
|
|
// earntypes.StrategyTypes{earntypes.STRATEGY_TYPE_SAVINGS},
|
|
// false,
|
|
// nil,
|
|
// )
|
|
|
|
// earnParams.AllowedVaults = append(earnParams.AllowedVaults, vault)
|
|
// ek.SetParams(suite.ctx, earnParams)
|
|
|
|
// sk := suite.app.GetSavingsKeeper()
|
|
// savingsParams := sk.GetParams(suite.ctx)
|
|
// savingsParams.SupportedDenoms = append(savingsParams.SupportedDenoms, liquidtypes.DefaultDerivativeDenom)
|
|
// sk.SetParams(suite.ctx, savingsParams)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) earnDeposit(owner sdk.AccAddress, derivative sdk.Coin) {
|
|
// ek := suite.app.GetEarnKeeper()
|
|
|
|
// err := ek.Deposit(suite.ctx, owner, derivative, earntypes.STRATEGY_TYPE_SAVINGS)
|
|
// suite.Require().NoError(err)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) mintDerivative(owner sdk.AccAddress, validator sdk.ValAddress, amount sdkmath.Int) sdk.Coin {
|
|
// lk := suite.app.GetLiquidKeeper()
|
|
|
|
// minted, err := lk.MintDerivative(suite.ctx, owner, validator, suite.newBondCoin(amount))
|
|
// suite.Require().NoError(err)
|
|
|
|
// return minted
|
|
return sdk.NewCoin("ukava", amount)
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) delegateToNewBondedValidator(delegator sdk.AccAddress, amount sdkmath.Int) stakingtypes.ValidatorI {
|
|
valAcc := suite.createAccount(suite.newBondCoin(sdkmath.NewInt(1e9)))
|
|
validator, err := suite.staking.createUnbondedValidator(suite.ctx, valAcc.GetAddress().Bytes(), sdkmath.NewInt(1e9))
|
|
suite.Require().NoError(err)
|
|
|
|
_, err = suite.staking.delegate(suite.ctx, delegator, validator.GetOperator(), amount)
|
|
suite.Require().NoError(err)
|
|
|
|
// bond the validator
|
|
sk := suite.app.GetStakingKeeper()
|
|
staking.EndBlocker(suite.ctx, sk)
|
|
|
|
validator, found := sk.GetValidator(suite.ctx, validator.GetOperator())
|
|
if !found {
|
|
panic("validator not found")
|
|
}
|
|
return validator
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) createNewBondedValidator(selfDelegation sdkmath.Int) stakingtypes.ValidatorI {
|
|
valAcc := suite.createAccount(suite.newBondCoin(selfDelegation))
|
|
validator, err := suite.staking.createUnbondedValidator(suite.ctx, valAcc.GetAddress().Bytes(), selfDelegation)
|
|
suite.Require().NoError(err)
|
|
|
|
// bond the validator
|
|
sk := suite.app.GetStakingKeeper()
|
|
staking.EndBlocker(suite.ctx, sk)
|
|
|
|
validator, found := sk.GetValidator(suite.ctx, validator.GetOperator())
|
|
if !found {
|
|
panic("validator not found")
|
|
}
|
|
return validator
|
|
}
|
|
|
|
func (suite *tallyHandlerSuite) createAccount(initialBalance ...sdk.Coin) authtypes.AccountI {
|
|
ak := suite.app.GetAccountKeeper()
|
|
|
|
acc := ak.NewAccountWithAddress(suite.ctx, RandomAddress())
|
|
ak.SetAccount(suite.ctx, acc)
|
|
|
|
err := suite.app.FundAccount(suite.ctx, acc.GetAddress(), initialBalance)
|
|
suite.Require().NoError(err)
|
|
|
|
return acc
|
|
}
|
|
|
|
// stakingHelper wraps the staking keeper with helper functions for testing.
|
|
type stakingHelper struct {
|
|
keeper stakingkeeper.Keeper
|
|
}
|
|
|
|
func (h stakingHelper) createUnbondedValidator(ctx sdk.Context, address sdk.ValAddress, selfDelegation sdkmath.Int) (stakingtypes.ValidatorI, error) {
|
|
msg, err := stakingtypes.NewMsgCreateValidator(
|
|
address,
|
|
ed25519.GenPrivKey().PubKey(),
|
|
h.newBondCoin(ctx, selfDelegation),
|
|
stakingtypes.Description{},
|
|
stakingtypes.NewCommissionRates(sdk.ZeroDec(), sdk.ZeroDec(), sdk.ZeroDec()),
|
|
sdkmath.NewInt(1e6),
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
msgServer := stakingkeeper.NewMsgServerImpl(&h.keeper)
|
|
_, err = msgServer.CreateValidator(sdk.WrapSDKContext(ctx), msg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
validator, found := h.keeper.GetValidator(ctx, address)
|
|
if !found {
|
|
panic("validator not found")
|
|
}
|
|
return validator, nil
|
|
}
|
|
|
|
func (h stakingHelper) delegate(ctx sdk.Context, delegator sdk.AccAddress, validator sdk.ValAddress, amount sdkmath.Int) (sdk.Dec, error) {
|
|
msg := stakingtypes.NewMsgDelegate(
|
|
delegator,
|
|
validator,
|
|
h.newBondCoin(ctx, amount),
|
|
)
|
|
|
|
msgServer := stakingkeeper.NewMsgServerImpl(&h.keeper)
|
|
_, err := msgServer.Delegate(sdk.WrapSDKContext(ctx), msg)
|
|
if err != nil {
|
|
return sdk.Dec{}, err
|
|
}
|
|
|
|
del, found := h.keeper.GetDelegation(ctx, delegator, validator)
|
|
if !found {
|
|
panic("delegation not found")
|
|
}
|
|
return del.Shares, nil
|
|
}
|
|
|
|
func (h stakingHelper) newBondCoin(ctx sdk.Context, amount sdkmath.Int) sdk.Coin {
|
|
return sdk.NewCoin(h.keeper.BondDenom(ctx), amount)
|
|
}
|
|
|
|
func (h stakingHelper) setBondDenom(ctx sdk.Context, denom string) {
|
|
params := h.keeper.GetParams(ctx)
|
|
params.BondDenom = denom
|
|
h.keeper.SetParams(ctx, params)
|
|
}
|