Skip to content

Commit 6b12118

Browse files
wadealexcypatil12
authored andcommitted
test: enable shared setups for integration tests (#1036)
* test: improve integration invariants * also removes unneeded fork logic * adds checks to some invariants * fixes some broken tests * test(integration): enable shared setups
1 parent bc06414 commit 6b12118

File tree

3 files changed

+328
-475
lines changed

3 files changed

+328
-475
lines changed

src/test/integration/IntegrationBase.t.sol

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,9 @@ abstract contract IntegrationBase is IntegrationDeployer, TypeImporter {
142142
assert_Snap_Added_OperatorShares(operator, strategies, addedShares, "_newRandomOperator: failed to award shares to operator");
143143
assertTrue(delegationManager.isOperator(address(operator)), "_newRandomOperator: operator should be registered");
144144

145-
assert_Snap_Added_Staker_DepositShares(operator, strategies, addedShares, "_newRandomOperator: failed to add delegatable shares");
145+
assert_Snap_Added_Staker_DepositShares(operator, strategies, addedShares, "_newRandomOperator: failed to add delegatable shares");
146+
}
147+
146148
assert_Snap_Added_OperatorShares(operator, strategies, addedShares, "_newRandomOperator: failed to award shares to operator");
147149
assertTrue(delegationManager.isOperator(address(operator)), "_newRandomOperator: operator should be registered");
148150

@@ -2863,13 +2865,8 @@ abstract contract IntegrationBase is IntegrationDeployer, TypeImporter {
28632865
}
28642866

28652867
function _getCheckpointPodBalanceGwei(User staker) internal view returns (uint64) {
2866-
if (forkType != LOCAL && !isUpgraded) {
2867-
IEigenPod_DeprecatedM2 pod = IEigenPod_DeprecatedM2(address(staker.pod()));
2868-
return uint64(pod.currentCheckpoint().podBalanceGwei);
2869-
} else {
2870-
EigenPod pod = staker.pod();
2871-
return uint64(pod.currentCheckpoint().podBalanceGwei);
2872-
}
2868+
EigenPod pod = staker.pod();
2869+
return uint64(pod.currentCheckpoint().podBalanceGwei);
28732870
}
28742871

28752872
function _getPrevCheckpointPodBalanceGwei(User staker) internal timewarp() returns (uint64) {

0 commit comments

Comments
 (0)