mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-26 08:15:19 +00:00
6ef9bab67d
* wip Add claim * Add distr keeper and claiming * Add claim test * Update claim test with failures * wip Add staking rewards * -S Fix savings to earn incentive methods * Use a single accural time for all earn incentives * Add additional required liquid methods * Update genesis to only include 1 accrual time for earn * Revert "Update genesis to only include 1 accrual time for earn" This reverts commit cc7e35347298681c0c8a4a0b9bf9b9b296c25531. * Revert "Use a single accural time for all earn incentives" This reverts commit aeb49c4622d4e3d99dc6421c8830932b1b546be9. * Update tests with incentive distribution * Add earn to incentive rewards query * add earn cli tx * Update claim example to use ukava large * Use underlying ukava to determine proportional reward amount * Rename liquid methods to reflect derivative value * Add tests for derivative values * Return error to panic in BeginBlocker Co-authored-by: karzak <kjydavis3@gmail.com>
89 lines
3.2 KiB
Go
89 lines
3.2 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/staking"
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/liquid/types"
|
|
|
|
distrtypes "github.com/cosmos/cosmos-sdk/x/distribution/types"
|
|
)
|
|
|
|
func (suite *KeeperTestSuite) TestCollectStakingRewards() {
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(5)
|
|
valAccAddr1, delegator := addrs[0], addrs[1]
|
|
valAddr1 := sdk.ValAddress(valAccAddr1)
|
|
|
|
initialBalance := i(1e9)
|
|
delegateAmount := i(100e6)
|
|
|
|
suite.NoError(suite.App.FundModuleAccount(
|
|
suite.Ctx,
|
|
distrtypes.ModuleName,
|
|
sdk.NewCoins(
|
|
sdk.NewCoin("ukava", initialBalance),
|
|
),
|
|
))
|
|
|
|
suite.CreateAccountWithAddress(valAccAddr1, suite.NewBondCoins(initialBalance))
|
|
suite.CreateAccountWithAddress(delegator, suite.NewBondCoins(initialBalance))
|
|
|
|
suite.CreateNewUnbondedValidator(valAddr1, initialBalance)
|
|
suite.CreateDelegation(valAddr1, delegator, delegateAmount)
|
|
staking.EndBlocker(suite.Ctx, suite.StakingKeeper)
|
|
|
|
// Transfers delegation to module account
|
|
_, err := suite.Keeper.MintDerivative(suite.Ctx, delegator, valAddr1, suite.NewBondCoin(delegateAmount))
|
|
suite.Require().NoError(err)
|
|
|
|
validator, found := suite.StakingKeeper.GetValidator(suite.Ctx, valAddr1)
|
|
suite.Require().True(found)
|
|
|
|
suite.Ctx = suite.Ctx.WithBlockHeight(2)
|
|
|
|
distrKeeper := suite.App.GetDistrKeeper()
|
|
stakingKeeper := suite.App.GetStakingKeeper()
|
|
accKeeper := suite.App.GetAccountKeeper()
|
|
liquidMacc := accKeeper.GetModuleAccount(suite.Ctx, types.ModuleAccountName)
|
|
|
|
// Add rewards
|
|
rewardCoins := sdk.NewDecCoins(sdk.NewDecCoin("ukava", sdk.NewInt(500e6)))
|
|
distrKeeper.AllocateTokensToValidator(suite.Ctx, validator, rewardCoins)
|
|
|
|
delegation, found := stakingKeeper.GetDelegation(suite.Ctx, liquidMacc.GetAddress(), valAddr1)
|
|
suite.Require().True(found)
|
|
|
|
// Get amount of rewards
|
|
endingPeriod := distrKeeper.IncrementValidatorPeriod(suite.Ctx, validator)
|
|
delegationRewards := distrKeeper.CalculateDelegationRewards(suite.Ctx, validator, delegation, endingPeriod)
|
|
truncatedRewards, _ := delegationRewards.TruncateDecimal()
|
|
|
|
suite.Run("collect staking rewards", func() {
|
|
// Collect rewards
|
|
derivativeDenom := suite.Keeper.GetLiquidStakingTokenDenom(valAddr1)
|
|
rewards, err := suite.Keeper.CollectStakingRewardsByDenom(suite.Ctx, derivativeDenom, types.ModuleName)
|
|
suite.Require().NoError(err)
|
|
suite.Require().Equal(truncatedRewards, rewards)
|
|
|
|
suite.True(rewards.AmountOf("ukava").IsPositive())
|
|
|
|
// Check balances
|
|
suite.AccountBalanceEqual(liquidMacc.GetAddress(), rewards)
|
|
})
|
|
|
|
suite.Run("collect staking rewards with non-validator", func() {
|
|
// acc2 not a validator
|
|
derivativeDenom := suite.Keeper.GetLiquidStakingTokenDenom(sdk.ValAddress(addrs[2]))
|
|
_, err := suite.Keeper.CollectStakingRewardsByDenom(suite.Ctx, derivativeDenom, types.ModuleName)
|
|
suite.Require().Error(err)
|
|
suite.Require().Equal("no validator distribution info", err.Error())
|
|
})
|
|
|
|
suite.Run("collect staking rewards with invalid denom", func() {
|
|
derivativeDenom := "bkava"
|
|
_, err := suite.Keeper.CollectStakingRewardsByDenom(suite.Ctx, derivativeDenom, types.ModuleName)
|
|
suite.Require().Error(err)
|
|
suite.Require().Equal("cannot parse denom bkava", err.Error())
|
|
})
|
|
}
|