mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-15 04:25:27 +00:00
240 lines
7.6 KiB
Go
240 lines
7.6 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"time"
|
|
|
|
abci "github.com/cometbft/cometbft/abci/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
earntypes "github.com/0glabs/0g-chain/x/earn/types"
|
|
"github.com/0glabs/0g-chain/x/incentive"
|
|
"github.com/0glabs/0g-chain/x/incentive/testutil"
|
|
"github.com/0glabs/0g-chain/x/incentive/types"
|
|
liquidtypes "github.com/0glabs/0g-chain/x/liquid/types"
|
|
"github.com/cosmos/cosmos-sdk/x/distribution"
|
|
distrtypes "github.com/cosmos/cosmos-sdk/x/distribution/types"
|
|
"github.com/cosmos/cosmos-sdk/x/mint"
|
|
minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
|
|
)
|
|
|
|
func (suite *HandlerTestSuite) TestEarnLiquidClaim() {
|
|
userAddr1, userAddr2, validatorAddr1, validatorAddr2 := suite.addrs[0], suite.addrs[1], suite.addrs[2], suite.addrs[3]
|
|
|
|
valAddr1 := sdk.ValAddress(validatorAddr1)
|
|
valAddr2 := sdk.ValAddress(validatorAddr2)
|
|
|
|
authBuilder := suite.authBuilder().
|
|
WithSimpleAccount(userAddr1, cs(c("ukava", 1e12))).
|
|
WithSimpleAccount(userAddr2, cs(c("ukava", 1e12))).
|
|
WithSimpleAccount(validatorAddr1, cs(c("ukava", 1e12))).
|
|
WithSimpleAccount(validatorAddr2, cs(c("ukava", 1e12)))
|
|
|
|
incentBuilder := suite.incentiveBuilder().
|
|
WithSimpleEarnRewardPeriod("bkava", cs())
|
|
|
|
savingsBuilder := testutil.NewSavingsGenesisBuilder().
|
|
WithSupportedDenoms("bkava")
|
|
|
|
earnBuilder := testutil.NewEarnGenesisBuilder().
|
|
WithAllowedVaults(earntypes.AllowedVault{
|
|
Denom: "bkava",
|
|
Strategies: earntypes.StrategyTypes{earntypes.STRATEGY_TYPE_SAVINGS},
|
|
IsPrivateVault: false,
|
|
AllowedDepositors: nil,
|
|
})
|
|
|
|
suite.SetupWithGenState(
|
|
authBuilder,
|
|
incentBuilder,
|
|
earnBuilder,
|
|
savingsBuilder,
|
|
)
|
|
|
|
// ak := suite.App.GetAccountKeeper()
|
|
// bk := suite.App.GetBankKeeper()
|
|
sk := suite.App.GetStakingKeeper()
|
|
lq := suite.App.GetLiquidKeeper()
|
|
mk := suite.App.GetMintKeeper()
|
|
dk := suite.App.GetDistrKeeper()
|
|
ik := suite.App.GetIncentiveKeeper()
|
|
|
|
iParams := ik.GetParams(suite.Ctx)
|
|
period, found := iParams.EarnRewardPeriods.GetMultiRewardPeriod("bkava")
|
|
suite.Require().True(found)
|
|
suite.Require().Equal("bkava", period.CollateralType)
|
|
|
|
// Use ukava for mint denom
|
|
mParams := mk.GetParams(suite.Ctx)
|
|
mParams.MintDenom = "ukava"
|
|
mk.SetParams(suite.Ctx, mParams)
|
|
|
|
bkavaDenom1 := lq.GetLiquidStakingTokenDenom(valAddr1)
|
|
bkavaDenom2 := lq.GetLiquidStakingTokenDenom(valAddr2)
|
|
|
|
err := suite.App.FundModuleAccount(suite.Ctx, distrtypes.ModuleName, cs(c("ukava", 1e12)))
|
|
suite.NoError(err)
|
|
|
|
// Create validators
|
|
err = suite.DeliverMsgCreateValidator(valAddr1, c("ukava", 1e9))
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.DeliverMsgCreateValidator(valAddr2, c("ukava", 1e9))
|
|
suite.Require().NoError(err)
|
|
|
|
// new block required to bond validator
|
|
suite.NextBlockAfter(7 * time.Second)
|
|
// Now the delegation is bonded, accumulate some delegator rewards
|
|
suite.NextBlockAfter(7 * time.Second)
|
|
|
|
// Create delegations from users
|
|
// User 1: 1e9 ukava to validator 1
|
|
// User 2: 99e9 ukava to validator 1 AND 2
|
|
err = suite.DeliverMsgDelegate(userAddr1, valAddr1, c("ukava", 1e9))
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.DeliverMsgDelegate(userAddr2, valAddr1, c("ukava", 99e9))
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.DeliverMsgDelegate(userAddr2, valAddr2, c("ukava", 99e9))
|
|
suite.Require().NoError(err)
|
|
|
|
// Mint liquid tokens
|
|
_, err = suite.DeliverMsgMintDerivative(userAddr1, valAddr1, c("ukava", 1e9))
|
|
suite.Require().NoError(err)
|
|
|
|
_, err = suite.DeliverMsgMintDerivative(userAddr2, valAddr1, c("ukava", 99e9))
|
|
suite.Require().NoError(err)
|
|
|
|
_, err = suite.DeliverMsgMintDerivative(userAddr2, valAddr2, c("ukava", 99e9))
|
|
suite.Require().NoError(err)
|
|
|
|
// Deposit liquid tokens to earn
|
|
err = suite.DeliverEarnMsgDeposit(userAddr1, c(bkavaDenom1, 1e9), earntypes.STRATEGY_TYPE_SAVINGS)
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.DeliverEarnMsgDeposit(userAddr2, c(bkavaDenom1, 99e9), earntypes.STRATEGY_TYPE_SAVINGS)
|
|
suite.Require().NoError(err)
|
|
err = suite.DeliverEarnMsgDeposit(userAddr2, c(bkavaDenom2, 99e9), earntypes.STRATEGY_TYPE_SAVINGS)
|
|
suite.Require().NoError(err)
|
|
|
|
// BeginBlocker to update minter annual provisions as it starts at 0 which results in no minted coins
|
|
_ = suite.App.BeginBlocker(suite.Ctx, abci.RequestBeginBlock{})
|
|
|
|
// DeliverMsgCreateValidator uses a generated pubkey, so we need to fetch
|
|
// the validator to get the correct pubkey
|
|
validator1, found := sk.GetValidator(suite.Ctx, valAddr1)
|
|
suite.Require().True(found)
|
|
|
|
pk, err := validator1.ConsPubKey()
|
|
suite.Require().NoError(err)
|
|
|
|
val := abci.Validator{
|
|
Address: pk.Address(),
|
|
Power: 100,
|
|
}
|
|
|
|
// Query for next block to get staking rewards
|
|
suite.Ctx = suite.Ctx.
|
|
WithBlockHeight(suite.Ctx.BlockHeight() + 1).
|
|
WithBlockTime(suite.Ctx.BlockTime().Add(7 * time.Second))
|
|
|
|
// Mint tokens
|
|
mint.BeginBlocker(
|
|
suite.Ctx,
|
|
suite.App.GetMintKeeper(),
|
|
minttypes.DefaultInflationCalculationFn,
|
|
)
|
|
// Distribute to validators, block needs votes
|
|
distribution.BeginBlocker(
|
|
suite.Ctx,
|
|
abci.RequestBeginBlock{
|
|
LastCommitInfo: abci.CommitInfo{
|
|
Votes: []abci.VoteInfo{{
|
|
Validator: val,
|
|
SignedLastBlock: true,
|
|
}},
|
|
},
|
|
},
|
|
dk,
|
|
)
|
|
|
|
liquidMacc := suite.App.GetAccountKeeper().GetModuleAccount(suite.Ctx, liquidtypes.ModuleAccountName)
|
|
delegation, found := sk.GetDelegation(suite.Ctx, liquidMacc.GetAddress(), valAddr1)
|
|
suite.Require().True(found)
|
|
|
|
// Get amount of rewards
|
|
endingPeriod := dk.IncrementValidatorPeriod(suite.Ctx, validator1)
|
|
|
|
// Zero rewards since this block is the same as the block it was last claimed
|
|
|
|
// This needs to run **after** staking rewards are minted/distributed in
|
|
// x/mint + x/distribution but **before** the x/incentive BeginBlocker.
|
|
|
|
// Order of operations:
|
|
// 1. x/mint + x/distribution BeginBlocker
|
|
// 2. CalculateDelegationRewards
|
|
// 3. x/incentive BeginBlocker to claim staking rewards
|
|
delegationRewards := dk.CalculateDelegationRewards(suite.Ctx, validator1, delegation, endingPeriod)
|
|
suite.Require().False(delegationRewards.IsZero(), "expected non-zero delegation rewards")
|
|
|
|
// Claim staking rewards via incentive.
|
|
// Block height was updated earlier.
|
|
incentive.BeginBlocker(
|
|
suite.Ctx,
|
|
ik,
|
|
)
|
|
|
|
preClaimBal1 := suite.GetBalance(userAddr1)
|
|
preClaimBal2 := suite.GetBalance(userAddr2)
|
|
|
|
// Claim ukava staking rewards
|
|
denomsToClaim := map[string]string{"ukava": "large"}
|
|
selections := types.NewSelectionsFromMap(denomsToClaim)
|
|
|
|
msg1 := types.NewMsgClaimEarnReward(userAddr1.String(), selections)
|
|
msg2 := types.NewMsgClaimEarnReward(userAddr2.String(), selections)
|
|
|
|
err = suite.DeliverIncentiveMsg(&msg1)
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.DeliverIncentiveMsg(&msg2)
|
|
suite.Require().NoError(err)
|
|
|
|
// Check rewards were paid out
|
|
// User 1 gets 1% of rewards
|
|
// User 2 gets 99% of rewards
|
|
stakingRewards1 := delegationRewards.
|
|
AmountOf("ukava").
|
|
Quo(sdk.NewDec(100)).
|
|
RoundInt()
|
|
suite.BalanceEquals(userAddr1, preClaimBal1.Add(sdk.NewCoin("ukava", stakingRewards1)))
|
|
|
|
// Total * 99 / 100
|
|
stakingRewards2 := delegationRewards.
|
|
AmountOf("ukava").
|
|
Mul(sdk.NewDec(99)).
|
|
Quo(sdk.NewDec(100)).
|
|
RoundInt()
|
|
|
|
suite.BalanceInEpsilon(
|
|
userAddr2,
|
|
preClaimBal2.Add(sdk.NewCoin("ukava", stakingRewards2)),
|
|
// Highest precision to allow 1ukava margin of error
|
|
// 820778117815 vs 820778117814
|
|
1e-11,
|
|
)
|
|
|
|
suite.InEpsilonf(
|
|
delegationRewards.AmountOf("ukava").RoundInt().Int64(),
|
|
stakingRewards1.Add(stakingRewards2).Int64(),
|
|
1e-11,
|
|
"expected rewards should add up to staking rewards within a margin of error (%v vs %v)",
|
|
delegationRewards.AmountOf("ukava").RoundInt().Int64(),
|
|
stakingRewards1.Add(stakingRewards2).Int64(),
|
|
)
|
|
|
|
// Check that claimed coins have been removed from a claim's reward
|
|
suite.EarnRewardEquals(userAddr1, cs())
|
|
suite.EarnRewardEquals(userAddr2, cs())
|
|
}
|