diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs index 166492a1db..8482d2dd4b 100644 --- a/runtime/src/bank.rs +++ b/runtime/src/bank.rs @@ -6811,6 +6811,14 @@ impl Bank { .check_complete() } + pub fn wait_for_initial_accounts_hash_verification_completed_for_tests(&self) { + self.rc + .accounts + .accounts_db + .verify_accounts_hash_in_bg + .wait_for_complete() + } + /// Get this bank's storages to use for snapshots. /// /// If a base slot is provided, return only the storages that are *higher* than this slot. diff --git a/runtime/src/snapshot_utils.rs b/runtime/src/snapshot_utils.rs index 4455536dbd..a26b976cac 100644 --- a/runtime/src/snapshot_utils.rs +++ b/runtime/src/snapshot_utils.rs @@ -3362,7 +3362,7 @@ mod tests { &Arc::default(), ) .unwrap(); - + roundtrip_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!(original_bank, roundtrip_bank); } @@ -3474,7 +3474,7 @@ mod tests { &Arc::default(), ) .unwrap(); - + roundtrip_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!(*bank4, roundtrip_bank); } @@ -3606,7 +3606,7 @@ mod tests { &Arc::default(), ) .unwrap(); - + roundtrip_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!(*bank4, roundtrip_bank); } @@ -3728,7 +3728,7 @@ mod tests { &Arc::default(), ) .unwrap(); - + deserialized_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!(deserialized_bank, *bank4); } @@ -3867,6 +3867,7 @@ mod tests { &Arc::default(), ) .unwrap(); + deserialized_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!( deserialized_bank, *bank2, "Ensure rebuilding from an incremental snapshot works" @@ -3931,6 +3932,7 @@ mod tests { &Arc::default(), ) .unwrap(); + deserialized_bank.wait_for_initial_accounts_hash_verification_completed_for_tests(); assert_eq!( deserialized_bank, *bank4, "Ensure rebuilding from an incremental snapshot works",