From adf6f748c0f9ee0e7fdae6ee5d912c0102d68c59 Mon Sep 17 00:00:00 2001
From: carllin <wumu727@gmail.com>
Date: Sat, 3 Oct 2020 12:44:03 -0700
Subject: [PATCH] Disable rent in flaky test (#12658)

* Disable rent in test

Co-authored-by: Carl Lin <carl@solana.com>
---
 runtime/src/accounts_db.rs |  8 ++++++++
 runtime/src/bank.rs        | 18 ++++++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/runtime/src/accounts_db.rs b/runtime/src/accounts_db.rs
index 0a19c200fb..8b75b6811e 100644
--- a/runtime/src/accounts_db.rs
+++ b/runtime/src/accounts_db.rs
@@ -2396,6 +2396,14 @@ impl AccountsDB {
             }
         }
     }
+
+    #[cfg(test)]
+    pub fn get_append_vec_id(&self, pubkey: &Pubkey, slot: Slot) -> Option<AppendVecId> {
+        let ancestors = vec![(slot, 1)].into_iter().collect();
+        let accounts_index = self.accounts_index.read().unwrap();
+        let result = accounts_index.get(&pubkey, Some(&ancestors), None);
+        result.map(|(list, index)| list[index].1.store_id)
+    }
 }
 
 #[cfg(test)]
diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index ad0abd2162..55efde3cda 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -9333,7 +9333,25 @@ mod tests {
                 &Pubkey::default(),
             ),
         );
+        assert_ne!(
+            bank1
+                .rc
+                .accounts
+                .accounts_db
+                .get_append_vec_id(&large_account_pubkey, 1)
+                .unwrap(),
+            bank1
+                .rc
+                .accounts
+                .accounts_db
+                .get_append_vec_id(&zero_lamport_pubkey, 1)
+                .unwrap()
+        );
 
+        // Make sure rent collection doesn't overwrite `large_account_pubkey`, which
+        // keeps slot 1 alive in the accounts database. Otherwise, slot 1 and it's bank
+        // hash would be removed from accounts, preventing `rehash()` from succeeding
+        bank1.lazy_rent_collection.store(true, Relaxed);
         bank1.freeze();
         let bank1_hash = bank1.hash();