Invariants feat: add config option to turn off shrinking

config/README.md

@@ -187,6 +187,7 @@ call_override = false
 dictionary_weight = 80
 include_storage = true
 include_push_bytes = true
+shrink_sequence = true
 
 [fmt]
 line_length = 100

config/src/invariant.rs

@@ -18,6 +18,8 @@ pub struct InvariantConfig {
     /// The fuzz dictionary configuration
     #[serde(flatten)]
     pub dictionary: FuzzDictionaryConfig,
+    /// Attempt to shrink the failure case to its smallest sequence of calls
+    pub shrink_sequence: bool,
 }
 
 impl Default for InvariantConfig {
@@ -28,6 +30,7 @@ impl Default for InvariantConfig {
             fail_on_revert: false,
             call_override: false,
             dictionary: FuzzDictionaryConfig { dictionary_weight: 80, ..Default::default() },
+            shrink_sequence: true,
         }
     }
 }

config/src/lib.rs

@@ -3374,6 +3374,7 @@ mod tests {
                 depth = 15
                 fail_on_revert = false
                 call_override = false
+                shrink_sequence = true
             "#,
             )?;
 

evm/src/fuzz/invariant/error.rs

@@ -28,6 +28,8 @@ pub struct InvariantFuzzError {
     pub func: Option<ethers::prelude::Bytes>,
     /// Inner fuzzing Sequence coming from overriding calls.
     pub inner_sequence: Vec<Option<BasicTxDetails>>,
+    /// Shrink the failed test case to the smallest sequence.
+    pub shrink: bool,
 }
 
 impl InvariantFuzzError {
@@ -37,6 +39,7 @@ impl InvariantFuzzError {
         calldata: &[BasicTxDetails],
         call_result: RawCallResult,
         inner_sequence: &[Option<BasicTxDetails>],
+        shrink_sequence: bool,
     ) -> Self {
         let mut func = None;
         let origin: String;
@@ -77,6 +80,7 @@ impl InvariantFuzzError {
             addr: invariant_contract.address,
             func,
             inner_sequence: inner_sequence.to_vec(),
+            shrink: shrink_sequence,
         }
     }
 
@@ -96,7 +100,11 @@ impl InvariantFuzzError {
             TestError::Fail(_, ref calls) => calls,
         };
 
-        let calls = self.try_shrinking(calls, &executor);
+        if self.shrink {
+            let _ = self.try_shrinking(calls, &executor);
+        } else {
+            trace!(target: "forge::test", "Shrinking disabled.");
+        }
 
         // We want traces for a failed case.
         executor.set_tracing(true);

evm/src/fuzz/invariant/executor.rs

@@ -187,6 +187,7 @@ impl<'a> InvariantExecutor<'a> {
                         &inputs,
                         &mut failures.borrow_mut(),
                         self.config.fail_on_revert,
+                        self.config.shrink_sequence,
                     );
 
                     if !can_continue {
@@ -558,6 +559,7 @@ fn can_continue(
     calldata: &[BasicTxDetails],
     failures: &mut InvariantFailures,
     fail_on_revert: bool,
+    shrink_sequence: bool,
 ) -> (bool, Option<BTreeMap<String, RawCallResult>>) {
     let mut call_results = None;
     if !call_result.reverted {
@@ -571,8 +573,14 @@ fn can_continue(
         // The user might want to stop all execution if a revert happens to
         // better bound their testing space.
         if fail_on_revert {
-            let error =
-                InvariantFuzzError::new(invariant_contract, None, calldata, call_result, &[]);
+            let error = InvariantFuzzError::new(
+                invariant_contract,
+                None,
+                calldata,
+                call_result,
+                &[],
+                shrink_sequence,
+            );
 
             failures.revert_reason = Some(error.revert_reason.clone());
 

evm/src/fuzz/invariant/mod.rs

@@ -95,6 +95,7 @@ pub fn assert_invariants(
                         calldata,
                         call_result,
                         &inner_sequence,
+                        true,
                     )),
                 );
                 found_case = true;

forge/tests/it/config.rs

@@ -118,6 +118,7 @@ pub static TEST_OPTS: TestOptions = TestOptions {
             max_fuzz_dictionary_addresses: 10_000,
             max_fuzz_dictionary_values: 10_000,
         },
+        shrink_sequence: true,
     },
 };
 

testdata/foundry.toml

@@ -16,6 +16,7 @@ ffi = false
 force = false
 invariant_fail_on_revert = false
 invariant_call_override = false
+invariant_shrink_sequence = true
 gas_limit = 9223372036854775807
 gas_price = 0
 gas_reports = ['*']