feat: snapshot asserts

liquity · Oct 11, 2024 · 9e8fd9e · 9e8fd9e
1 parent 7f66112
commit 9e8fd9e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/INTEGRATION.MD b/INTEGRATION.MD
@@ -6,3 +6,7 @@ Votes not meeting the threshold may result in 0 rewards
 
 Claiming more than once will return 0
 
+## INVARIANT: You can only claim for previous epoch
+
+assert(votesSnapshot_.forEpoch == epoch() - 1); /// @audit INVARIANT: You can only claim for previous epoch
+/// All unclaimed rewards are always recycled
diff --git a/src/Governance.sol b/src/Governance.sol
@@ -610,6 +610,8 @@ contract Governance is Multicall, UserProxyFactory, ReentrancyGuard, IGovernance
             return 0;
         }
 
+        assert(votesSnapshot_.forEpoch == epoch() - 1); /// @audit INVARIANT: You can only claim for previous epoch
+        /// All unclaimed rewards are always recycled
 
         initiativeStates[_initiative].lastEpochClaim = epoch() - 1;
         votesForInitiativeSnapshot[_initiative] = votesForInitiativeSnapshot_; // implicitly prevents double claiming