mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 18:15:19 +00:00
d56bb77231
* organise testing committee gen state * remove repeated test app initialization * minor fixes from linter in tests * move more setup to SetupApp * split up KeeperTestSuite for each reward type * simplify KeeperTestSuite * simplify PayoutKeeperSuite * simplify DelegatorRewardSuite * simplify SupplyRewardsSuite * simplify BorrowRewardsSuite * simplify USDXRewardsSuite * add auth genesis builder for easier test setup * migrate all incentive tests to auth builder * add incentive genesis builder for easier setup migrate hard incentive tests * migrate all tests to incentive builder * add hard genesis builder * small tidy ups * deduplicate initialTime from borrow tests * deduplicate initialtTime from supply tests * deduplicate initialTime from usdx and keeper tests * deduplicate initialTime in delgator tests * deduplicate genesis time in payout test * deduplicate test app initialization * make authGenesisBuilder available for all modules * remove unused pricefeed setup * export incentive genesis builder * remove commented out test cases * migrate cdp test to new test state builders * migrate vv payout tests to use new builders
70 lines
2.8 KiB
Go
70 lines
2.8 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/incentive/types"
|
|
)
|
|
|
|
func TestRiskyCDPsAccumulateRewards(t *testing.T) {
|
|
genesisTime := time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC)
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(5)
|
|
|
|
initialCollateral := c("bnb", 1_000_000_000)
|
|
user := addrs[0]
|
|
authBuilder := app.NewAuthGenesisBuilder().
|
|
WithSimpleAccount(user, cs(initialCollateral))
|
|
|
|
collateralType := "bnb-a"
|
|
rewardsPerSecond := c(types.USDXMintingRewardDenom, 1_000_000)
|
|
|
|
incentBuilder := NewIncentiveGenesisBuilder().
|
|
WithGenesisTime(genesisTime).
|
|
WithSimpleUSDXRewardPeriod(collateralType, rewardsPerSecond)
|
|
|
|
tApp := app.NewTestApp()
|
|
tApp.InitializeFromGenesisStates(
|
|
authBuilder.BuildMarshalled(),
|
|
NewPricefeedGenStateMultiFromTime(genesisTime),
|
|
NewCDPGenStateMulti(),
|
|
incentBuilder.BuildMarshalled(),
|
|
)
|
|
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: genesisTime})
|
|
|
|
// Setup cdp state containing one CDP
|
|
cdpKeeper := tApp.GetCDPKeeper()
|
|
err := cdpKeeper.AddCdp(ctx, user, initialCollateral, c("usdx", 100_000_000), collateralType)
|
|
require.NoError(t, err)
|
|
|
|
// Skip ahead two blocks to accumulate both interest and usdx reward for the cdp
|
|
// Two blocks are required because the cdp begin blocker runs before incentive begin blocker.
|
|
// In the first begin block the cdp is synced, which triggers its claim to sync. But no global rewards have accumulated yet so the sync does nothing.
|
|
// Global rewards accumulate immediately after during the incentive begin blocker.
|
|
// Rewards are added to the cdp's claim in the next block when the cdp is synced.
|
|
_ = tApp.EndBlocker(ctx, abci.RequestEndBlock{})
|
|
ctx = ctx.WithBlockTime(ctx.BlockTime().Add(10 * time.Minute))
|
|
_ = tApp.BeginBlocker(ctx, abci.RequestBeginBlock{}) // height and time in header are ignored by module begin blockers
|
|
|
|
_ = tApp.EndBlocker(ctx, abci.RequestEndBlock{})
|
|
ctx = ctx.WithBlockTime(ctx.BlockTime().Add(10 * time.Minute))
|
|
_ = tApp.BeginBlocker(ctx, abci.RequestBeginBlock{})
|
|
|
|
// check cdp rewards
|
|
cdp, found := cdpKeeper.GetCdpByOwnerAndCollateralType(ctx, user, collateralType)
|
|
require.True(t, found)
|
|
// This additional sync adds the rewards accumulated at the end of the last begin block.
|
|
// They weren't added during the begin blocker as the incentive BB runs after the CDP BB.
|
|
incentiveKeeper := tApp.GetIncentiveKeeper()
|
|
incentiveKeeper.SynchronizeUSDXMintingReward(ctx, cdp)
|
|
claim, found := incentiveKeeper.GetUSDXMintingClaim(ctx, user)
|
|
require.True(t, found)
|
|
|
|
// rewards are roughly rewardsPerSecond * secondsElapsed (10mins) * num blocks (2)
|
|
require.Equal(t, c(types.USDXMintingRewardDenom, 1_200_000_557), claim.Reward)
|
|
}
|