Get tests passing

Author: davidlaprade

test/MockFlexVotingClient.sol

@@ -29,6 +29,8 @@ contract MockFlexVotingClient is FlexVotingClient {
     return deposits[_user];
   }
 
+  // Test hooks
+  // ---------------------------------------------------------------------------
   function exposed_rawBalanceOf(address _user) external view returns (uint208) {
     return _rawBalanceOf(_user);
   }
@@ -56,6 +58,8 @@ contract MockFlexVotingClient is FlexVotingClient {
   function exposed_checkpointRawBalanceOf(address _user) external {
     return _checkpointRawBalanceOf(_user);
   }
+  // End test hooks
+  // ---------------------------------------------------------------------------
 
   /// @notice Allow a holder of the governance token to deposit it into the pool.
   /// @param _amount The amount to be deposited.

test/MockFlexVotingDelegatableClient.sol

@@ -30,6 +30,38 @@ contract MockFlexVotingDelegatableClient is FlexVotingDelegatable {
     return deposits[_user];
   }
 
+  // Test hooks
+  // ---------------------------------------------------------------------------
+  function exposed_rawBalanceOf(address _user) external view returns (uint208) {
+    return _rawBalanceOf(_user);
+  }
+
+  function exposed_latestTotalBalance() external view returns (uint208) {
+    return totalBalanceCheckpoints.latest();
+  }
+
+  function exposed_checkpointTotalBalance(int256 _delta) external {
+    return _checkpointTotalBalance(_delta);
+  }
+
+  function exposed_castVoteReasonString() external returns (string memory) {
+    return _castVoteReasonString();
+  }
+
+  function exposed_selfDelegate() external {
+    return _selfDelegate();
+  }
+
+  function exposed_setDeposits(address _user, uint208 _amount) external {
+    deposits[_user] = _amount;
+  }
+
+  function exposed_checkpointRawBalanceOf(address _user) external {
+    return _checkpointRawBalanceOf(_user);
+  }
+  // End test hooks
+  // ---------------------------------------------------------------------------
+
   /// @notice Allow a holder of the governance token to deposit it into the pool.
   /// @param _amount The amount to be deposited.
   function deposit(uint208 _amount) public {

test/SharedFlexVoting.t.sol

@@ -233,19 +233,19 @@ abstract contract _CheckpointRawBalanceOf is FlexVotingClientTest {
   function testFuzz_StoresTheRawBalanceWithTheTimepoint(
     address _user,
     uint208 _amount,
-    uint48 _timepoint
+    uint48 _future
   ) public {
     vm.assume(_user != address(flexClient));
-    _timepoint = uint48(bound(_timepoint, _now() + 1, type(uint48).max));
+    _future = uint48(bound(_future, _now() + 1, type(uint48).max));
     _amount = uint208(bound(_amount, 1, MAX_VOTES));
+    uint48 _past = _now();
 
+    _advanceTimeTo(_future);
     flexClient.exposed_setDeposits(_user, _amount);
-    assertEq(flexClient.getPastRawBalance(_user, _timepoint), 0);
-
-    _advanceTimeTo(_timepoint);
-
     flexClient.exposed_checkpointRawBalanceOf(_user);
-    assertEq(flexClient.getPastRawBalance(_user, _timepoint), _amount);
+
+    assertEq(flexClient.getPastRawBalance(_user, _past), 0);
+    assertEq(flexClient.getPastRawBalance(_user, _future), _amount);
   }
 }