foundry_config/
invariant.rs

1//! Configuration for invariant testing
2
3use crate::fuzz::{FuzzCorpusConfig, FuzzDictionaryConfig};
4use serde::{Deserialize, Serialize};
5use std::path::PathBuf;
6
7/// Contains for invariant testing
8#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
9pub struct InvariantConfig {
10    /// The number of runs that must execute for each invariant test group.
11    pub runs: u32,
12    /// The number of calls executed to attempt to break invariants in one run.
13    pub depth: u32,
14    /// Fails the invariant fuzzing if a revert occurs
15    pub fail_on_revert: bool,
16    /// Allows overriding an unsafe external call when running invariant tests. eg. reentrancy
17    /// checks
18    pub call_override: bool,
19    /// The fuzz dictionary configuration
20    #[serde(flatten)]
21    pub dictionary: FuzzDictionaryConfig,
22    /// The maximum number of attempts to shrink the sequence
23    pub shrink_run_limit: u32,
24    /// The maximum number of rejects via `vm.assume` which can be encountered during a single
25    /// invariant run.
26    pub max_assume_rejects: u32,
27    /// Number of runs to execute and include in the gas report.
28    pub gas_report_samples: u32,
29    /// The fuzz corpus configuration.
30    #[serde(flatten)]
31    pub corpus: FuzzCorpusConfig,
32    /// Path where invariant failures are recorded and replayed.
33    pub failure_persist_dir: Option<PathBuf>,
34    /// Whether to collect and display fuzzed selectors metrics.
35    pub show_metrics: bool,
36    /// Optional timeout (in seconds) for each invariant test.
37    pub timeout: Option<u32>,
38    /// Display counterexample as solidity calls.
39    pub show_solidity: bool,
40    /// Maximum time (in seconds) between generated txs.
41    pub max_time_delay: Option<u32>,
42    /// Maximum number of blocks elapsed between generated txs.
43    pub max_block_delay: Option<u32>,
44}
45
46impl Default for InvariantConfig {
47    fn default() -> Self {
48        Self {
49            runs: 256,
50            depth: 500,
51            fail_on_revert: false,
52            call_override: false,
53            dictionary: FuzzDictionaryConfig { dictionary_weight: 80, ..Default::default() },
54            shrink_run_limit: 5000,
55            max_assume_rejects: 65536,
56            gas_report_samples: 256,
57            corpus: FuzzCorpusConfig::default(),
58            failure_persist_dir: None,
59            show_metrics: true,
60            timeout: None,
61            show_solidity: false,
62            max_time_delay: None,
63            max_block_delay: None,
64        }
65    }
66}
67
68impl InvariantConfig {
69    /// Creates invariant configuration to write failures in `{PROJECT_ROOT}/cache/fuzz` dir.
70    pub fn new(cache_dir: PathBuf) -> Self {
71        Self { failure_persist_dir: Some(cache_dir), ..Default::default() }
72    }
73}