Ensure the treasury automatically resumes once balance < cap

Author: stronk-dev

contracts/bonding/BondingManager.sol

@@ -470,12 +470,21 @@ contract BondingManager is ManagerProxyTarget, IBondingManager {
     function setCurrentRoundTotalActiveStake() external onlyRoundsManager {
         currentRoundTotalActiveStake = nextRoundTotalActiveStake;
 
+        // Apply any queued changes to the treasury cut
         if (nextRoundTreasuryRewardCutRate != treasuryRewardCutRate) {
             treasuryRewardCutRate = nextRoundTreasuryRewardCutRate;
             // The treasury cut rate changes in a delayed fashion so we want to emit the parameter update event here
             emit ParameterUpdate("treasuryRewardCutRate");
         }
 
+        // Set the treasury cut to 0 while the treasury balance is above the ceiling
+        if (treasuryBalanceCeiling > 0) {
+            uint256 treasuryBalance = livepeerToken().balanceOf(treasury());
+            if (treasuryBalance >= treasuryBalanceCeiling && treasuryRewardCutRate > 0) {
+                treasuryRewardCutRate = 0;
+            }
+        }
+
         bondingVotes().checkpointTotalActiveStake(currentRoundTotalActiveStake, roundsManager().currentRound());
     }
 
@@ -890,14 +899,6 @@ contract BondingManager is ManagerProxyTarget, IBondingManager {
             earningsPool.setStake(t.earningsPoolPerRound[lastUpdateRound].totalStake);
         }
 
-        if (treasuryBalanceCeiling > 0) {
-            uint256 treasuryBalance = livepeerToken().balanceOf(treasury());
-            if (treasuryBalance >= treasuryBalanceCeiling && nextRoundTreasuryRewardCutRate > 0) {
-                // halt treasury contributions until the cut rate param is updated again
-                _setTreasuryRewardCutRate(0);
-            }
-        }
-
         // Create reward based on active transcoder's stake relative to the total active stake
         // rewardTokens = (current mintable tokens for the round * active transcoder stake) / total active stake
         IMinter mtr = minter();