wen_restart: correctly handle HeaviestFork received from the coordinator.

core/src/validator.rs

@@ -1424,7 +1424,7 @@ impl Validator {
         .map_err(ValidatorError::Other)?;
 
         if in_wen_restart {
-            info!("Waiting for wen_restart phase one to finish");
+            info!("Waiting for wen_restart to finish");
             wait_for_wen_restart(WenRestartConfig {
                 wen_restart_path: config.wen_restart_proto_path.clone().unwrap(),
                 wen_restart_coordinator: config.wen_restart_coordinator.unwrap(),

wen-restart/src/heaviest_fork_aggregate.rs

@@ -161,8 +161,10 @@ impl HeaviestForkAggregate {
         })
     }
 
-    pub(crate) fn block_stake_map(self) -> HashMap<(Slot, Hash), u64> {
-        self.block_stake_map
+    pub(crate) fn print_block_stake_map(&self) {
+        for ((slot, hash), stake) in self.block_stake_map.iter() {
+            info!("Slot: {}, Hash: {}, Stake: {}", slot, hash, stake,);
+        }
     }
 }