Merge branch 'main' into revert-3557-revert-7bcc41c67b74cd97668e169705d9af365f5c1297

Author: nharper285

src/agent/onefuzz-task/src/tasks/coverage/generic.rs

@@ -146,11 +146,11 @@ impl CoverageTask {
             bail!("input is not specified on the command line or arguments for the target");
         }
 
-        context.heartbeat.alive();
-
         info!("report initial coverage");
         context.report_coverage_stats().await;
 
+        context.heartbeat.alive();
+
         for dir in &self.config.readonly_inputs {
             debug!("recording coverage for {}", dir.local_path.display());
 
@@ -174,6 +174,9 @@ impl CoverageTask {
             context.save_and_sync_coverage().await?;
         }
 
+        info!("report coverage");
+        context.report_coverage_stats().await;
+
         context.heartbeat.alive();
 
         if let Some(queue) = &self.config.input_queue {