mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 03:55:19 +00:00
117 lines
4.6 KiB
Go
117 lines
4.6 KiB
Go
package incentive_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/incentive"
|
|
"github.com/kava-labs/kava/x/incentive/types"
|
|
"github.com/kava-labs/kava/x/kavadist"
|
|
)
|
|
|
|
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
|
|
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
|
|
|
|
type HandlerTestSuite struct {
|
|
suite.Suite
|
|
|
|
ctx sdk.Context
|
|
app app.TestApp
|
|
handler sdk.Handler
|
|
keeper incentive.Keeper
|
|
addrs []sdk.AccAddress
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) SetupTest() {
|
|
tApp := app.NewTestApp()
|
|
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()})
|
|
keeper := tApp.GetIncentiveKeeper()
|
|
|
|
// Set up genesis state and initialize
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(3)
|
|
coins := []sdk.Coins{}
|
|
for j := 0; j < 3; j++ {
|
|
coins = append(coins, cs(c("bnb", 10000000000), c("ukava", 10000000000)))
|
|
}
|
|
authGS := app.NewAuthGenState(addrs, coins)
|
|
incentiveGS := incentive.NewGenesisState(
|
|
incentive.NewParams(
|
|
incentive.RewardPeriods{incentive.NewRewardPeriod(true, "bnb-a", time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC), time.Date(2024, 12, 15, 14, 0, 0, 0, time.UTC), c("ukava", 122354))},
|
|
incentive.MultiRewardPeriods{incentive.NewMultiRewardPeriod(true, "bnb-a", time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC), time.Date(2024, 12, 15, 14, 0, 0, 0, time.UTC), cs(c("ukava", 122354)))},
|
|
incentive.MultiRewardPeriods{incentive.NewMultiRewardPeriod(true, "bnb-a", time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC), time.Date(2024, 12, 15, 14, 0, 0, 0, time.UTC), cs(c("ukava", 122354)))},
|
|
incentive.RewardPeriods{incentive.NewRewardPeriod(true, "bnb-a", time.Date(2020, 12, 15, 14, 0, 0, 0, time.UTC), time.Date(2024, 12, 15, 14, 0, 0, 0, time.UTC), c("ukava", 122354))},
|
|
incentive.Multipliers{incentive.NewMultiplier(incentive.MultiplierName("small"), 1, d("0.25")), incentive.NewMultiplier(incentive.MultiplierName("large"), 12, d("1.0"))},
|
|
time.Date(2025, 12, 15, 14, 0, 0, 0, time.UTC),
|
|
),
|
|
incentive.DefaultGenesisAccumulationTimes,
|
|
incentive.DefaultGenesisAccumulationTimes,
|
|
incentive.DefaultGenesisAccumulationTimes,
|
|
incentive.DefaultGenesisAccumulationTimes,
|
|
incentive.DefaultUSDXClaims,
|
|
incentive.DefaultHardClaims,
|
|
)
|
|
tApp.InitializeFromGenesisStates(authGS, app.GenesisState{incentive.ModuleName: incentive.ModuleCdc.MustMarshalJSON(incentiveGS)}, NewCDPGenStateMulti(), NewPricefeedGenStateMulti())
|
|
|
|
suite.addrs = addrs
|
|
suite.handler = incentive.NewHandler(keeper)
|
|
suite.keeper = keeper
|
|
suite.app = tApp
|
|
suite.ctx = ctx
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) TestMsgUSDXMintingClaimReward() {
|
|
suite.addUSDXMintingClaim()
|
|
msg := incentive.NewMsgClaimUSDXMintingReward(suite.addrs[0], "small")
|
|
res, err := suite.handler(suite.ctx, msg)
|
|
suite.NoError(err)
|
|
suite.Require().NotNil(res)
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) TestMsgHardClaimReward() {
|
|
suite.addHardLiquidityProviderClaim()
|
|
msg := incentive.NewMsgClaimHardReward(suite.addrs[0], "small")
|
|
res, err := suite.handler(suite.ctx, msg)
|
|
suite.NoError(err)
|
|
suite.Require().NotNil(res)
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) addHardLiquidityProviderClaim() {
|
|
sk := suite.app.GetSupplyKeeper()
|
|
err := sk.MintCoins(suite.ctx, kavadist.ModuleName, cs(c("ukava", 1000000000000)))
|
|
suite.Require().NoError(err)
|
|
rewardPeriod := types.RewardIndexes{types.NewRewardIndex("bnb-s", sdk.ZeroDec())}
|
|
|
|
multiRewardIndex := types.NewMultiRewardIndex("bnb-s", rewardPeriod)
|
|
multiRewardIndexes := types.MultiRewardIndexes{multiRewardIndex}
|
|
c1 := incentive.NewHardLiquidityProviderClaim(suite.addrs[0], cs(c("ukava", 1000000)), multiRewardIndexes, multiRewardIndexes, rewardPeriod)
|
|
suite.NotPanics(func() {
|
|
suite.keeper.SetHardLiquidityProviderClaim(suite.ctx, c1)
|
|
})
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) addUSDXMintingClaim() {
|
|
sk := suite.app.GetSupplyKeeper()
|
|
err := sk.MintCoins(suite.ctx, kavadist.ModuleName, cs(c("ukava", 1000000000000)))
|
|
suite.Require().NoError(err)
|
|
c1 := incentive.NewUSDXMintingClaim(suite.addrs[0], c("ukava", 1000000), types.RewardIndexes{types.NewRewardIndex("bnb-s", sdk.ZeroDec())})
|
|
suite.NotPanics(func() {
|
|
suite.keeper.SetUSDXMintingClaim(suite.ctx, c1)
|
|
})
|
|
}
|
|
|
|
func TestHandlerTestSuite(t *testing.T) {
|
|
suite.Run(t, new(HandlerTestSuite))
|
|
}
|
|
|
|
// Avoid cluttering test cases with long function names
|
|
func i(in int64) sdk.Int { return sdk.NewInt(in) }
|
|
func d(str string) sdk.Dec { return sdk.MustNewDecFromStr(str) }
|