Wire up the cleaned up driver implementation

crates/report/src/aggregator.rs

@@ -106,6 +106,7 @@ impl ReportAggregator {
                 RunnerEvent::ContractDeployed(event) => {
                     self.handle_contract_deployed_event(*event);
                 }
+                RunnerEvent::Completion(event) => self.handle_completion(*event),
             }
         }
         debug!("Report aggregation completed");
@@ -382,6 +383,10 @@ impl ReportAggregator {
             .insert(event.contract_instance, event.address);
     }
 
+    fn handle_completion(&mut self, _: CompletionEvent) {
+        self.runner_rx.close();
+    }
+
     fn test_case_report(&mut self, specifier: &TestSpecifier) -> &mut TestCaseReport {
         self.report
             .test_case_information

crates/report/src/runner_event.rs

@@ -613,6 +613,8 @@ define_event! {
             /// The address of the contract.
             address: Address
         },
+        /// Reports the completion of the run.
+        Completion {}
     }
 }