allow resetting hasPranked for tests

Layr-Labs · Dec 11, 2024 · a484bd7 · a484bd7
1 parent a7da452
commit a484bd7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/templates/MultisigBuilder.sol b/src/templates/MultisigBuilder.sol
@@ -8,7 +8,6 @@ import "../utils/ZeusScript.sol";
  * @dev Abstract contract for building arbitrary multisig scripts.
  */
 abstract contract MultisigBuilder is ZeusScript {
-
     bool private hasPranked;
 
     modifier prank(address caller) {
@@ -45,4 +44,9 @@ abstract contract MultisigBuilder is ZeusScript {
         require(hasPranked, "MultisigBuilder._stopPrank: _startPrank not called");
         vm.stopPrank();
     }
+
+    /// @dev Only meant for use with tests. Please ensure you know what you are doing if you call this!
+    function _unsafeResetHasPranked() internal {
+        hasPranked = false;
+    }
 }