@@ -15,7 +15,9 @@ contract Pause is MultisigBuilder, EigenPodPausingConstants {
1515
1616 function _runAsMultisig () prank (Env.pauserMultisig ()) internal virtual override {
1717 uint mask = 2 ** PAUSED_START_CHECKPOINT |
18- 2 ** PAUSED_EIGENPODS_VERIFY_CREDENTIALS;
18+ 2 ** PAUSED_EIGENPODS_VERIFY_CREDENTIALS |
19+ 2 ** PAUSED_VERIFY_STALE_BALANCE |
20+ 2 ** PAUSED_EIGENPODS_VERIFY_CHECKPOINT_PROOFS;
1921
2022 Env.proxy.eigenPodManager ().pause (mask);
2123 }
@@ -25,10 +27,13 @@ contract Pause is MultisigBuilder, EigenPodPausingConstants {
2527
2628 assertTrue (Env.proxy.eigenPodManager ().paused (PAUSED_START_CHECKPOINT), "Not paused! " );
2729 assertTrue (Env.proxy.eigenPodManager ().paused (PAUSED_EIGENPODS_VERIFY_CREDENTIALS), "Not paused! " );
30+ assertTrue (Env.proxy.eigenPodManager ().paused (PAUSED_VERIFY_STALE_BALANCE), "Not paused! " );
31+ assertTrue (Env.proxy.eigenPodManager ().paused (PAUSED_EIGENPODS_VERIFY_CHECKPOINT_PROOFS), "Not paused! " );
2832
2933 // Create a new pod and try to verify credentials + start checkpoint
3034 EigenPod pod = EigenPod (payable (Env.proxy.eigenPodManager ().createPod ()));
3135
36+ // Revert verifying withdrawal credentials
3237 vm.expectRevert (IEigenPodErrors.CurrentlyPaused.selector );
3338 BeaconChainProofs.StateRootProof memory emptyProof;
3439 pod.verifyWithdrawalCredentials (
@@ -38,7 +43,24 @@ contract Pause is MultisigBuilder, EigenPodPausingConstants {
3843 new bytes [](0 ),
3944 new bytes32 [][](0 )
4045 );
46+
47+ // Revert starting checkpoint
4148 vm.expectRevert (IEigenPodErrors.CurrentlyPaused.selector );
4249 pod.startCheckpoint (false );
50+
51+ // Revert verifying stale balance
52+ BeaconChainProofs.ValidatorProof memory validatorProof;
53+ vm.expectRevert (IEigenPodErrors.CurrentlyPaused.selector );
54+ pod.verifyStaleBalance (
55+ 0 ,
56+ emptyProof,
57+ validatorProof
58+ );
59+
60+ // Revert completing checkpoint
61+ BeaconChainProofs.BalanceContainerProof memory balanceContainerProof;
62+ BeaconChainProofs.BalanceProof[] memory proofs;
63+ vm.expectRevert (IEigenPodErrors.CurrentlyPaused.selector );
64+ pod.verifyCheckpointProofs (balanceContainerProof, proofs);
4365 }
4466}
0 commit comments