From ce2a7521252cd51f2de685e36e10354d743ffaa9 Mon Sep 17 00:00:00 2001
From: Noah McGregor Harper <74685766+nharper285@users.noreply.github.com>
Date: Fri, 27 Oct 2023 18:39:59 +0000
Subject: [PATCH] Adding coverage output to every iteration of coverage task.

---
 src/agent/onefuzz-task/src/tasks/coverage/generic.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/agent/onefuzz-task/src/tasks/coverage/generic.rs b/src/agent/onefuzz-task/src/tasks/coverage/generic.rs
index 8043deee94..36d161a6b2 100644
--- a/src/agent/onefuzz-task/src/tasks/coverage/generic.rs
+++ b/src/agent/onefuzz-task/src/tasks/coverage/generic.rs
@@ -148,7 +148,6 @@ impl CoverageTask {
 
         info!("report initial coverage");
         context.report_coverage_stats().await;
-
         context.heartbeat.alive();
 
         for dir in &self.config.readonly_inputs {
@@ -174,7 +173,7 @@ impl CoverageTask {
             context.save_and_sync_coverage().await?;
         }
 
-        info!("report coverage");
+        info!("report final coverage");
         context.report_coverage_stats().await;
 
         context.heartbeat.alive();
@@ -447,6 +446,8 @@ impl<'a> TaskContext<'a> {
                             if count % 10 == 0 {
                                 self.save_and_sync_coverage().await?;
                             }
+                            info!("report coverage");
+                            self.report_coverage_stats().await;
                         }
                     } else {
                         warn!("skipping non-file dir entry: {}", entry.path().display());