Skip to content

Commit fe80744

Browse files
wadealexcypatil12
authored andcommitted
test: implement random withdrawal generator (#349)
see PR for changes/notes
1 parent 950eee3 commit fe80744

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/test/integration/IntegrationBase.t.sol

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -355,13 +355,9 @@ abstract contract IntegrationBase is IntegrationDeployer {
355355
// Use timewarp to get previous token balances
356356
uint[] memory prevTokenBalances = _getPrevTokenBalances(staker, tokens);
357357

358-
for (uint i = 0; i < strategies.length; i++) {
359-
// Ignore BeaconChainETH strategy since it doesn't keep track of global strategy shares
360-
if (strategies[i] == BEACONCHAIN_ETH_STRAT) {
361-
continue;
362-
}
363-
uint256 prevShare = prevShares[i];
364-
uint256 curShare = curShares[i];
358+
for (uint i = 0; i < tokens.length; i++) {
359+
uint prevBalance = prevTokenBalances[i];
360+
uint curBalance = curTokenBalances[i];
365361

366362
assertEq(prevBalance - removedTokens[i], curBalance, err);
367363
}

0 commit comments

Comments
 (0)