CNS-694: fix unit test

matsunaga282 · Dec 19, 2023 · f0e86a9 · f0e86a9
1 parent 3c190e6
commit f0e86a9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 11 deletions.
diff --git a/testutil/keeper/mock_keepers.go b/testutil/keeper/mock_keepers.go
@@ -168,13 +168,11 @@ func (k mockBankKeeper) SetBalance(ctx sdk.Context, addr sdk.AccAddress, amounts
 func (k mockBankKeeper) AddToBalance(addr sdk.AccAddress, amounts sdk.Coins) error {
 	if _, ok := balance[addr.String()]; ok {
 		for _, coin := range amounts {
-			fmt.Printf("before balance[addr.String()].String(): %v\n", balance[addr.String()].String())
 			balance[addr.String()] = balance[addr.String()].Add(coin)
 		}
 	} else {
 		balance[addr.String()] = amounts
 	}
-	fmt.Printf("after balance[addr.String()].String(): %v\n", balance[addr.String()].String())
 	return nil
 }
 

diff --git a/x/rewards/keeper/rewards.go b/x/rewards/keeper/rewards.go
@@ -29,16 +29,16 @@ func (k Keeper) DistributeBlockReward(ctx sdk.Context) {
 			utils.Attribute{Key: "distribution_pool_balance", Value: distributionPoolBalance.String()},
 			utils.Attribute{Key: "blocks_to_next_timer_expiry", Value: strconv.FormatInt(blocksToNextTimerExpiry, 10)},
 		)
-	}
-
-	coins := sdk.NewCoins(sdk.NewCoin(k.stakingKeeper.BondDenom(ctx), validatorsRewards))
+	} else {
+		coins := sdk.NewCoins(sdk.NewCoin(k.stakingKeeper.BondDenom(ctx), validatorsRewards))
 
-	// distribute rewards to validators (same as Cosmos mint module)
-	err := k.addCollectedFees(ctx, coins)
-	if err != nil {
-		utils.LavaFormatError("critical - could not send validators rewards to fee collector", err,
-			utils.Attribute{Key: "rewards", Value: coins.String()},
-		)
+		// distribute rewards to validators (same as Cosmos mint module)
+		err := k.addCollectedFees(ctx, coins)
+		if err != nil {
+			utils.LavaFormatError("critical - could not send validators rewards to fee collector", err,
+				utils.Attribute{Key: "rewards", Value: coins.String()},
+			)
+		}
 	}
 }