Wen restart fix flaky test (#3525)

* Fix flaky test, make sure bank is frozen before reading bankhash. * Add more comments here.
anza-xyz · Nov 7, 2024 · 5abb627 · 5abb627
1 parent 7688d91
commit 5abb627
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/wen-restart/src/wen_restart.rs b/wen-restart/src/wen_restart.rs
@@ -1817,8 +1817,11 @@ mod tests {
                 .unwrap()
                 .get(my_heaviest_fork_slot)
             {
-                my_heaviest_fork_bankhash = bank.hash();
-                break;
+                // When deciding the local heaviest fork, we will freeze the bank.
+                if bank.is_frozen() {
+                    my_heaviest_fork_bankhash = bank.hash();
+                    break;
+                }
             }
             sleep(Duration::from_millis(100));
         }