test: implement random withdrawal generator (#349)

see PR for changes/notes
Layr-Labs · Dec 4, 2023 · fe80744 · fe80744
1 parent 950eee3
commit fe80744
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/test/integration/IntegrationBase.t.sol b/src/test/integration/IntegrationBase.t.sol
@@ -355,13 +355,9 @@ abstract contract IntegrationBase is IntegrationDeployer {
         // Use timewarp to get previous token balances
         uint[] memory prevTokenBalances = _getPrevTokenBalances(staker, tokens);
 
-        for (uint i = 0; i < strategies.length; i++) {
-            // Ignore BeaconChainETH strategy since it doesn't keep track of global strategy shares
-            if (strategies[i] == BEACONCHAIN_ETH_STRAT) {
-                continue;
-            }
-            uint256 prevShare = prevShares[i];
-            uint256 curShare = curShares[i];
+        for (uint i = 0; i < tokens.length; i++) {
+            uint prevBalance = prevTokenBalances[i];
+            uint curBalance = curTokenBalances[i];
 
             assertEq(prevBalance - removedTokens[i], curBalance, err);
         }