From c5ba2e08837982fb82adc1a894eba5ef8a53b462 Mon Sep 17 00:00:00 2001
From: Trent Nelson <trent@solana.com>
Date: Mon, 8 Jul 2019 16:55:50 -0600
Subject: [PATCH] bank_forks test stability (#4959)

automerge
---
 core/src/bank_forks.rs | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/core/src/bank_forks.rs b/core/src/bank_forks.rs
index 10645da98e..ee8aa6f590 100644
--- a/core/src/bank_forks.rs
+++ b/core/src/bank_forks.rs
@@ -442,6 +442,15 @@ mod tests {
         pub paths: String,
     }
 
+    impl TempPaths {
+        fn remove_all(&self) {
+            let paths: Vec<String> = self.paths.split(',').map(|s| s.to_string()).collect();
+            paths.iter().for_each(|p| {
+                let _ignored = remove_dir_all(p);
+            });
+        }
+    }
+
     #[macro_export]
     macro_rules! tmp_bank_accounts_name {
         () => {
@@ -458,10 +467,7 @@ mod tests {
 
     impl Drop for TempPaths {
         fn drop(&mut self) {
-            let paths: Vec<String> = self.paths.split(',').map(|s| s.to_string()).collect();
-            paths.iter().for_each(|p| {
-                let _ignored = remove_dir_all(p);
-            });
+            self.remove_all()
         }
     }
 
@@ -521,6 +527,8 @@ mod tests {
             mint_keypair,
             ..
         } = create_genesis_block(10_000);
+        path.remove_all();
+        spath.remove_all();
         for index in 0..10 {
             let bank0 = Bank::new_with_paths(&genesis_block, Some(path.paths.clone()));
             bank0.freeze();