From a8b9899deeb2a344b903710012c0a5cfb7fc88d7 Mon Sep 17 00:00:00 2001
From: Michael Vines <mvines@gmail.com>
Date: Mon, 7 Jan 2019 19:13:55 -0800
Subject: [PATCH] Add retry, restore ignored tests

---
 src/fullnode.rs         | 2 --
 src/leader_scheduler.rs | 2 +-
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/fullnode.rs b/src/fullnode.rs
index c276416403..9b4f06b327 100644
--- a/src/fullnode.rs
+++ b/src/fullnode.rs
@@ -835,7 +835,6 @@ mod tests {
     }
 
     #[test]
-    #[ignore]
     fn test_wrong_role_transition() {
         // Create the leader node information
         let bootstrap_leader_keypair = Arc::new(Keypair::new());
@@ -983,7 +982,6 @@ mod tests {
     }
 
     #[test]
-    #[ignore]
     fn test_validator_to_leader_transition() {
         // Make a leader identity
         let leader_keypair = Keypair::new();
diff --git a/src/leader_scheduler.rs b/src/leader_scheduler.rs
index 294a3f866f..24eb9dfba4 100644
--- a/src/leader_scheduler.rs
+++ b/src/leader_scheduler.rs
@@ -502,7 +502,7 @@ pub fn make_active_set_entries(
 
     let params = json!([active_keypair.pubkey(), sig, msg.as_bytes()]);
     let resp = RpcRequest::RegisterNode
-        .make_rpc_request(&rpc_client, 1, Some(params))
+        .retry_make_rpc_request(&rpc_client, 1, Some(params), 5)
         .unwrap();
     let vote_account_id: Pubkey = serde_json::from_value(resp).unwrap();