Add config to allow expect revert for internal calls

crates/cheatcodes/src/config.rs

@@ -57,6 +57,8 @@ pub struct CheatsConfig {
     pub assertions_revert: bool,
     /// Optional seed for the RNG algorithm.
     pub seed: Option<U256>,
+    /// Whether to allow `expectRevert` to work for internal calls.
+    pub internal_expect_revert: bool,
 }
 
 impl CheatsConfig {
@@ -98,6 +100,7 @@ impl CheatsConfig {
             running_version,
             assertions_revert: config.assertions_revert,
             seed: config.fuzz.seed,
+            internal_expect_revert: config.allow_internal_expect_revert,
         }
     }
 
@@ -239,6 +242,7 @@ impl Default for CheatsConfig {
             running_version: Default::default(),
             assertions_revert: true,
             seed: None,
+            internal_expect_revert: false,
         }
     }
 }

crates/cheatcodes/src/inspector.rs

@@ -759,6 +759,7 @@ where {
                 let handler_result = expect::handle_expect_revert(
                     false,
                     true,
+                    self.config.internal_expect_revert,
                     &mut expected_revert,
                     outcome.result.result,
                     outcome.result.output.clone(),
@@ -1339,6 +1340,7 @@ impl Inspector<&mut dyn DatabaseExt> for Cheatcodes {
                     let handler_result = expect::handle_expect_revert(
                         cheatcode_call,
                         false,
+                        self.config.internal_expect_revert,
                         &mut expected_revert,
                         outcome.result.result,
                         outcome.result.output.clone(),

crates/cheatcodes/src/test/expect.rs

@@ -786,6 +786,7 @@ fn expect_revert(
 pub(crate) fn handle_expect_revert(
     is_cheatcode: bool,
     is_create: bool,
+    internal_expect_revert: bool,
     expected_revert: &mut ExpectedRevert,
     status: InstructionResult,
     retdata: Bytes,
@@ -809,7 +810,7 @@ pub(crate) fn handle_expect_revert(
         hex::encode_prefixed(data)
     };
 
-    if !is_cheatcode {
+    if !internal_expect_revert {
         ensure!(
             expected_revert.max_depth > expected_revert.depth,
             "call didn't revert at a lower depth than cheatcode call depth"

crates/config/src/lib.rs

@@ -315,6 +315,8 @@ pub struct Config {
     pub invariant: InvariantConfig,
     /// Whether to allow ffi cheatcodes in test
     pub ffi: bool,
+    /// Whether to allow `expectRevert` for internal functions.
+    pub allow_internal_expect_revert: bool,
     /// Use the create 2 factory in all cases including tests and non-broadcasting scripts.
     pub always_use_create_2_factory: bool,
     /// Sets a timeout in seconds for vm.prompt cheatcodes
@@ -2310,6 +2312,7 @@ impl Default for Config {
             invariant: InvariantConfig::new("cache/invariant".into()),
             always_use_create_2_factory: false,
             ffi: false,
+            allow_internal_expect_revert: false,
             prompt_timeout: 120,
             sender: Self::DEFAULT_SENDER,
             tx_origin: Self::DEFAULT_SENDER,

crates/forge/tests/cli/config.rs

@@ -87,6 +87,7 @@ forgetest!(can_extract_config_values, |prj, cmd| {
             ..Default::default()
         },
         ffi: true,
+        allow_internal_expect_revert: false,
         always_use_create_2_factory: false,
         prompt_timeout: 0,
         sender: "00a329c0648769A73afAc7F9381D08FB43dBEA72".parse().unwrap(),

crates/test-utils/src/util.rs

@@ -207,6 +207,7 @@ impl ExtTester {
             test_cmd.env("FOUNDRY_FORK_BLOCK_NUMBER", fork_block.to_string());
         }
         test_cmd.env("FOUNDRY_INVARIANT_DEPTH", "15");
+        test_cmd.env("FOUNDRY_ALLOW_INTERNAL_EXPECT_REVERT", "true");
 
         test_cmd.assert_success();
     }