migrate tests to get_slot_storage_entry (solana-labs#29515)

gnapoli23 · Jan 10, 2023 · faf0cb8 · faf0cb8
1 parent f287fd4
commit faf0cb8
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/runtime/src/accounts_db.rs b/runtime/src/accounts_db.rs
@@ -4249,17 +4249,14 @@ impl AccountsDb {
 
  #[cfg(test)]
  pub(crate) fn sizes_of_accounts_in_storage_for_tests(&self, slot: Slot) -> Vec<usize> {
- self.get_storages_for_slot(slot)
- .map(|storages| {
- storages
- .iter()
- .flat_map(|storage| {
- storage
- .accounts
- .account_iter()
- .map(|account| account.stored_size)
- })
- .collect::<Vec<_>>()
+ self.storage
+ .get_slot_storage_entry(slot)
+ .map(|storage| {
+ storage
+ .accounts
+ .account_iter()
+ .map(|account| account.stored_size)
+ .collect()
  })
  .unwrap_or_default()
  }