From 72214b2b68d9acb0837819c89fd0c495ad26c725 Mon Sep 17 00:00:00 2001
From: Stephen Akridge <stephen@solana.com>
Date: Tue, 26 Feb 2019 13:59:39 -0800
Subject: [PATCH] Squash test to test parent bank after squash

---
 runtime/src/bank.rs | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/runtime/src/bank.rs b/runtime/src/bank.rs
index f6695bd35b..c16099d3d0 100644
--- a/runtime/src/bank.rs
+++ b/runtime/src/bank.rs
@@ -1451,4 +1451,20 @@ mod tests {
             bank.squash();
         }
     }
+
+    #[test]
+    fn test_bank_get_account_in_parent_after_squash() {
+        let (genesis_block, mint_keypair) = GenesisBlock::new(500);
+        let parent = Arc::new(Bank::new(&genesis_block));
+
+        let key1 = Keypair::new();
+
+        parent
+            .transfer(1, &mint_keypair, key1.pubkey(), genesis_block.last_id())
+            .unwrap();
+        assert_eq!(parent.get_balance(&key1.pubkey()), 1);
+        let bank = Bank::new_from_parent(&parent);
+        bank.squash();
+        assert_eq!(parent.get_balance(&key1.pubkey()), 1);
+    }
 }