feat(forge): Add call tracing support (#192)

* first pass * fixes * fmt * better fmting * updates colored prints, better dev ux, verbosity > 2 trace printing * fmt * updates * fmt * fix after master merge * fix tests post master merge * warning fixes * fmt * lots of fixes * fmt * fix * cyan color * fixes * prettier raw logs + parse setup contracts * update diff_score threshold * better printing * remove integration tests * improvements * improvements + fmt + clippy * fixes * more cleanup * cleanup and verbosity > 3 setup print * refactor printing * documentation + cleanup * fix negative number printing * fix tests to match master and fix tracing_enabled * fix unnecessary trace_index set * refactor runner tracing + tracing_enabled * nits + value printing * last nits
foundry-rs · Dec 23, 2021 · 8602e1b · 8602e1b
1 parent c5f18b6
commit 8602e1b
Show file tree

Hide file tree

Showing 15 changed files with 1,200 additions and 60 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/cli/src/cmd/test.rs b/cli/src/cmd/test.rs
@@ -158,8 +158,14 @@ impl Cmd for TestArgs {
                 let backend = Arc::new(backend);
 
                 let precompiles = PRECOMPILES_MAP.clone();
-                let evm =
-                    Executor::new_with_cheatcodes(backend, env.gas_limit, &cfg, &precompiles, ffi);
+                let evm = Executor::new_with_cheatcodes(
+                    backend,
+                    env.gas_limit,
+                    &cfg,
+                    &precompiles,
+                    ffi,
+                    verbosity > 2,
+                );
 
                 test(builder, project, evm, pattern, json, verbosity, allow_failure)
             }
@@ -318,6 +324,36 @@ fn test<A: ArtifactOutput + 'static, S: Clone, E: evm_adapters::Evm<S>>(
                     }
 
                     println!();
+
+                    if verbosity > 2 {
+                        if let (Some(traces), Some(identified_contracts)) =
+                            (&result.traces, &result.identified_contracts)
+                        {
+                            let mut ident = identified_contracts.clone();
+                            if verbosity > 3 {
+                                // print setup calls as well
+                                traces.iter().for_each(|trace| {
+                                    trace.pretty_print(
+                                        0,
+                                        &runner.known_contracts,
+                                        &mut ident,
+                                        &runner.evm,
+                                        "",
+                                    );
+                                });
+                            } else if !traces.is_empty() {
+                                traces.last().expect("no last but not empty").pretty_print(
+                                    0,
+                                    &runner.known_contracts,
+                                    &mut ident,
+                                    &runner.evm,
+                                    "",
+                                );
+                            }
+
+                            println!();
+                        }
+                    }
                 }
             }
         }

diff --git a/evm-adapters/Cargo.toml b/evm-adapters/Cargo.toml
@@ -27,6 +27,7 @@ futures = "0.3.17"
 revm_precompiles = "0.1.0"
 serde_json = "1.0.72"
 serde = "1.0.130"
+ansi_term = "0.12.1"
 
 [dev-dependencies]
 evmodin = { git = "https://github.com/vorot93/evmodin", features = ["util"] }