Add missing unit test for autocompounding on pool rewards increase

KiraCore · Oct 11, 2023 · 8e5301b · 8e5301b
1 parent 2ddc9d2
commit 8e5301b
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/x/multistaking/keeper/delegation_test.go b/x/multistaking/keeper/delegation_test.go
@@ -8,6 +8,7 @@ import (
 	"github.com/cometbft/cometbft/crypto/ed25519"
 	simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
 	sdk "github.com/cosmos/cosmos-sdk/types"
+	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
 	minttypes "github.com/cosmos/cosmos-sdk/x/mint/types"
 )
 
@@ -172,6 +173,26 @@ func (suite *KeeperTestSuite) TestIncreasePoolRewards() {
 	suite.Require().Equal(rewards, sdk.Coins{sdk.NewInt64Coin("ukex", 250000)})
 	rewards = suite.app.MultiStakingKeeper.GetDelegatorRewards(suite.ctx, addr2)
 	suite.Require().Equal(rewards, sdk.Coins{sdk.NewInt64Coin("ukex", 250000)})
+
+	// set autocompound info and try adding rewards
+	suite.app.MultiStakingKeeper.SetCompoundInfo(suite.ctx, types.CompoundInfo{
+		Delegator:      addr1.String(),
+		AllDenom:       true,
+		CompoundDenoms: []string{},
+		LastExecBlock:  0,
+	})
+
+	suite.app.BankKeeper.MintCoins(suite.ctx, minttypes.ModuleName, allocation)
+	suite.app.BankKeeper.SendCoinsFromModuleToModule(suite.ctx, minttypes.ModuleName, authtypes.FeeCollectorName, allocation)
+	properties := suite.app.CustomGovKeeper.GetNetworkProperties(suite.ctx)
+	suite.ctx = suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + int64(properties.AutocompoundIntervalNumBlocks) + 1)
+	pool, _ = suite.app.MultiStakingKeeper.GetStakingPoolByValidator(suite.ctx, valAddr.String())
+	suite.app.MultiStakingKeeper.IncreasePoolRewards(suite.ctx, pool, allocation)
+
+	rewards = suite.app.MultiStakingKeeper.GetDelegatorRewards(suite.ctx, addr1)
+	suite.Require().Equal(rewards.String(), "")
+	rewards = suite.app.MultiStakingKeeper.GetDelegatorRewards(suite.ctx, addr2)
+	suite.Require().Equal(rewards.String(), sdk.Coins{sdk.NewInt64Coin("ukex", 500000)}.String())
 }
 
 func (suite *KeeperTestSuite) TestDelegate() {