Open up bound range for simple test with single stake

uniswapfoundation · Dec 20, 2023 · 02b5b71 · 02b5b71
1 parent 7f903aa
commit 02b5b71
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/UniStaker.t.sol b/test/UniStaker.t.sol
@@ -64,7 +64,7 @@ contract Stake is UniStakerTest {
     uint256 _amount,
     address _delegatee
   ) public {
-    _amount = _boundMintAmount(_amount);
+    _amount = bound(_amount, 1, type(uint224).max);
     _mintGovToken(_depositor, _amount);
     _stake(_depositor, _amount, _delegatee);
 
@@ -95,7 +95,7 @@ contract Stake is UniStakerTest {
     // Perform the second stake with this delegatee
     _stake(_depositor2, _amount2, _delegatee);
 
-    // Ensure surrogate for this delegatee hasn't changed and has 2x the balance
+    // Ensure surrogate for this delegatee hasn't changed and has summed stake balance
     assertEq(address(uniStaker.surrogates(_delegatee)), address(_surrogate));
     assertEq(govToken.delegates(address(_surrogate)), _delegatee);
     assertEq(govToken.balanceOf(address(_surrogate)), _amount1 + _amount2);