From 362a39a9415acf4f64416dec2a21b31ec520ed97 Mon Sep 17 00:00:00 2001
From: sakridge <stephen@solana.com>
Date: Tue, 27 Aug 2019 14:36:48 -0700
Subject: [PATCH] Don't unwrap get_balance immediately in bench-tps move mode
 (#5685)

automerge
---
 bench-tps/src/bench.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/bench-tps/src/bench.rs b/bench-tps/src/bench.rs
index b3ebe6ba9..fea90c879 100644
--- a/bench-tps/src/bench.rs
+++ b/bench-tps/src/bench.rs
@@ -767,12 +767,13 @@ fn fund_move_keys<T: Client>(
     client.send_message(&[funding_key], tx.message).unwrap();
     let mut balance = 0;
     for _ in 0..20 {
-        balance = client.get_balance(&funding_keys[0].pubkey()).unwrap();
-        if balance > 0 {
-            break;
-        } else {
-            sleep(Duration::from_millis(100));
+        if let Ok(balance_) = client.get_balance(&funding_keys[0].pubkey()) {
+            if balance_ > 0 {
+                balance = balance_;
+                break;
+            }
         }
+        sleep(Duration::from_millis(100));
     }
     assert!(balance > 0);
     info!("funded multiple funding accounts.. {:?}", balance);