Merge pull request #218 from hallazzang/hallazzang/fix-invariant-check-1

tendermint · Nov 9, 2021 · f137b35 · f137b35
2 parents acfe280 + 92869d2
commit f137b35
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/x/farming/keeper/invariants_test.go b/x/farming/keeper/invariants_test.go
@@ -276,6 +276,22 @@ func (suite *KeeperTestSuite) TestNonNegativeHistoricalRewardsInvariant() {
 }
 
 func (suite *KeeperTestSuite) TestPositiveTotalStakingsAmountInvariant() {
+	k, ctx := suite.keeper, suite.ctx
+
+	// This is normal.
+	k.SetTotalStakings(ctx, denom1, types.TotalStakings{Amount: sdk.NewInt(1000000)})
+	_, broken := farmingkeeper.PositiveTotalStakingsAmountInvariant(k)(ctx)
+	suite.Require().False(broken)
+
+	// Zero-amount total stakings.
+	k.SetTotalStakings(ctx, denom1, types.TotalStakings{Amount: sdk.ZeroInt()})
+	_, broken = farmingkeeper.PositiveTotalStakingsAmountInvariant(k)(ctx)
+	suite.Require().True(broken)
+
+	// Negative-amount total stakings.
+	k.SetTotalStakings(ctx, denom1, types.TotalStakings{Amount: sdk.NewInt(-1)})
+	_, broken = farmingkeeper.PositiveTotalStakingsAmountInvariant(k)(ctx)
+	suite.Require().True(broken)
 }
 
 func (suite *KeeperTestSuite) TestPlanTerminationStatusInvariant() {