mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-27 00:35:18 +00:00
101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
|
package keeper_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"time"
|
||
|
|
|
||
|
|
vestingtypes "github.com/cosmos/cosmos-sdk/x/auth/vesting/types"
|
||
|
|
|
||
|
|
"github.com/kava-labs/kava/x/incentive/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func (suite *HandlerTestSuite) TestPayoutHardClaimMultiDenom() {
|
||
|
|
userAddr, receiverAddr := suite.addrs[0], suite.addrs[1]
|
||
|
|
|
||
|
|
authBulder := suite.authBuilder().
|
||
|
|
WithSimpleAccount(userAddr, cs(c("bnb", 1e12))).
|
||
|
|
WithSimpleAccount(receiverAddr, nil)
|
||
|
|
|
||
|
|
incentBuilder := suite.incentiveBuilder().
|
||
|
|
WithSimpleSupplyRewardPeriod("bnb", cs(c("hard", 1e6), c("swap", 1e6))).
|
||
|
|
WithSimpleBorrowRewardPeriod("bnb", cs(c("hard", 1e6), c("swap", 1e6)))
|
||
|
|
|
||
|
|
suite.SetupWithGenState(authBulder, incentBuilder)
|
||
|
|
|
||
|
|
// create a deposit and borrow
|
||
|
|
suite.NoError(suite.DeliverHardMsgDeposit(userAddr, cs(c("bnb", 1e11))))
|
||
|
|
suite.NoError(suite.DeliverHardMsgBorrow(userAddr, cs(c("bnb", 1e10))))
|
||
|
|
|
||
|
|
// accumulate some rewards
|
||
|
|
suite.NextBlockAfter(7 * time.Second)
|
||
|
|
|
||
|
|
preClaimBal := suite.GetBalance(userAddr)
|
||
|
|
|
||
|
|
msg := types.NewMsgClaimHardReward(
|
||
|
|
userAddr.String(),
|
||
|
|
types.Selections{
|
||
|
|
types.NewSelection("hard", "small"),
|
||
|
|
types.NewSelection("swap", "medium"),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
// Claim denoms
|
||
|
|
err := suite.DeliverIncentiveMsg(&msg)
|
||
|
|
suite.NoError(err)
|
||
|
|
|
||
|
|
// Check rewards were paid out
|
||
|
|
expectedRewardsHard := c("hard", int64(0.2*float64(2*7*1e6)))
|
||
|
|
expectedRewardsSwap := c("swap", int64(0.5*float64(2*7*1e6)))
|
||
|
|
suite.BalanceEquals(userAddr, preClaimBal.Add(expectedRewardsHard, expectedRewardsSwap))
|
||
|
|
|
||
|
|
suite.VestingPeriodsEqual(userAddr, []vestingtypes.Period{
|
||
|
|
{Length: (17+31)*secondsPerDay - 7, Amount: cs(expectedRewardsHard)},
|
||
|
|
{Length: (28 + 31 + 30 + 31 + 30) * secondsPerDay, Amount: cs(expectedRewardsSwap)}, // second length is stacked on top of the first
|
||
|
|
})
|
||
|
|
// Check that claimed coins have been removed from a claim's reward
|
||
|
|
suite.HardRewardEquals(userAddr, nil)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *HandlerTestSuite) TestPayoutHardClaimSingleDenom() {
|
||
|
|
userAddr := suite.addrs[0]
|
||
|
|
|
||
|
|
authBulder := suite.authBuilder().
|
||
|
|
WithSimpleAccount(userAddr, cs(c("bnb", 1e12)))
|
||
|
|
|
||
|
|
incentBuilder := suite.incentiveBuilder().
|
||
|
|
WithSimpleSupplyRewardPeriod("bnb", cs(c("hard", 1e6), c("swap", 1e6))).
|
||
|
|
WithSimpleBorrowRewardPeriod("bnb", cs(c("hard", 1e6), c("swap", 1e6)))
|
||
|
|
|
||
|
|
suite.SetupWithGenState(authBulder, incentBuilder)
|
||
|
|
|
||
|
|
// create a deposit and borrow
|
||
|
|
suite.NoError(suite.DeliverHardMsgDeposit(userAddr, cs(c("bnb", 1e11))))
|
||
|
|
suite.NoError(suite.DeliverHardMsgBorrow(userAddr, cs(c("bnb", 1e10))))
|
||
|
|
|
||
|
|
// accumulate some rewards
|
||
|
|
suite.NextBlockAfter(7 * time.Second)
|
||
|
|
|
||
|
|
preClaimBal := suite.GetBalance(userAddr)
|
||
|
|
|
||
|
|
msg := types.NewMsgClaimHardReward(
|
||
|
|
userAddr.String(),
|
||
|
|
types.Selections{
|
||
|
|
types.NewSelection("swap", "large"),
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
// Claim rewards
|
||
|
|
err := suite.DeliverIncentiveMsg(&msg)
|
||
|
|
suite.NoError(err)
|
||
|
|
|
||
|
|
// Check rewards were paid out
|
||
|
|
expectedRewards := c("swap", 2*7*1e6)
|
||
|
|
suite.BalanceEquals(userAddr, preClaimBal.Add(expectedRewards))
|
||
|
|
|
||
|
|
suite.VestingPeriodsEqual(userAddr, []vestingtypes.Period{
|
||
|
|
{Length: (17+31+28+31+30+31+30+31+31+30+31+30+31)*secondsPerDay - 7, Amount: cs(expectedRewards)},
|
||
|
|
})
|
||
|
|
|
||
|
|
// Check that claimed coins have been removed from a claim's reward
|
||
|
|
suite.HardRewardEquals(userAddr, cs(c("hard", 2*7*1e6)))
|
||
|
|
}
|