Add extra check

babylonlabs-io · Oct 30, 2024 · a65782b · a65782b
1 parent 3aa99b0
commit a65782b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/x/babylon/keeper/mint_rewards_test.go b/x/babylon/keeper/mint_rewards_test.go
@@ -68,6 +68,21 @@ func TestMintBlockRewards(t *testing.T) {
 			expectAmount:  sdkmath.ZeroInt(),
 			expectErr:     true,
 		},
+		{
+			name: "zero blocks per year",
+			setup: func(ctx sdk.Context) {
+				initialSupply := sdkmath.NewInt(100_000_000_000) // 100_000 tokens
+				bondDenom, _ := k.StakingKeeper.BondDenom(ctx)
+				coins := sdk.NewCoins(sdk.NewCoin(bondDenom, initialSupply))
+				err := k.BankKeeper.MintCoins(ctx, types.ModuleName, coins)
+				require.NoError(t, err)
+			},
+			recipient:     sdk.AccAddress("test_recipient"),
+			inflationRate: sdkmath.LegacyNewDecWithPrec(-5, 2),
+			blocksPerYear: 0,
+			expectAmount:  sdkmath.ZeroInt(),
+			expectErr:     true,
+		},
 	}
 
 	for _, tc := range tests {