chore: clear only when not coverage

crates/forge/bin/cmd/test/mod.rs

@@ -453,10 +453,6 @@ impl TestArgs {
                 // processing the remaining tests and print the suite summary.
                 any_test_failed |= result.status == TestStatus::Failure;
 
-                if result.traces.is_empty() {
-                    continue;
-                }
-
                 // Clear the addresses and labels from previous runs.
                 decoder.clear_addresses();
                 decoder
@@ -524,8 +520,8 @@ impl TestArgs {
             // Print suite summary.
             shell::println(suite_result.summary())?;
 
-            // Free memory.
-            suite_result.keep_only_results();
+            // Free memory if it's not needed.
+            suite_result.clear_unneeded();
 
             // Add the suite result to the outcome.
             outcome.results.insert(contract_name, suite_result);

crates/forge/src/result.rs

@@ -215,8 +215,10 @@ impl SuiteResult {
     }
 
     /// Frees memory that is not used for the final output.
-    pub fn keep_only_results(&mut self) {
-        ContractsByArtifact::clear(&mut self.known_contracts);
+    pub fn clear_unneeded(&mut self) {
+        if !self.test_results.values().any(|r| r.coverage.is_some()) {
+            ContractsByArtifact::clear(&mut self.known_contracts);
+        }
     }
 
     /// Returns an iterator over all individual succeeding tests and their names.

crates/forge/tests/cli/coverage.rs

@@ -70,11 +70,9 @@ contract AContractTest is DSTest {
     let lcov_data = std::fs::read_to_string(lcov_info).unwrap();
     // AContract.init must be hit at least once
     let re = Regex::new(r"FNDA:(\d+),AContract\.init").unwrap();
-    assert!(lcov_data.lines().any(|line| re.captures(line).map_or(false, |caps| caps
-        .get(1)
-        .unwrap()
-        .as_str()
-        .parse::<i32>()
-        .unwrap() >
-        0)));
+    let valid_line = |line| {
+        re.captures(line)
+            .map_or(false, |caps| caps.get(1).unwrap().as_str().parse::<i32>().unwrap() > 0)
+    };
+    assert!(lcov_data.lines().any(valid_line), "{lcov_data}");
 });