forge-guard 0.1.0

Pre-deployment smart contract auditing framework for Foundry
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

//! `forge invariant` — run invariant tests.

use super::InvariantArgs;
use anyhow::{Context, Result};
use colored::*;

/// Run invariant tests.
pub fn run(args: &InvariantArgs) -> Result<()> {
    eprintln!("{}", "🔬 Forge Guard — Invariant Testing".bold());
    eprintln!("   Runs:       {}", args.runs);
    eprintln!("   Depth:      {}", args.depth);
    eprintln!(
        "   Contract:   {}",
        args.contract.as_deref().unwrap_or("all")
    );

    let mut cmd = std::process::Command::new("forge");
    cmd.arg("test");

    cmd.arg("--invariant-runs").arg(args.runs.to_string());
    cmd.arg("--invariant-depth").arg(args.depth.to_string());

    if args.fail_on_revert {
        cmd.arg("--fail-on-revert");
    }

    if let Some(contract) = &args.contract {
        cmd.arg("--match-contract").arg(contract);
    }

    eprintln!("\n🚀 Running invariant tests...\n");
    let status = cmd.status().context("Failed to run forge test")?;

    if !status.success() {
        anyhow::bail!("Invariant tests failed — invariants broken!");
    }

    eprintln!(
        "\n{}",
        "✅ Invariant tests passed — all invariants hold."
            .green()
            .bold()
    );
    Ok(())
}