From e759cdf061a38fd4fccdb4545655f4dbd1a27ab3 Mon Sep 17 00:00:00 2001
From: Anatoly Yakovenko <aeyakovenko@gmail.com>
Date: Sat, 23 Jun 2018 16:30:40 -0700
Subject: [PATCH] tests

---
 src/crdt.rs | 26 +++++++++++++++++++++-----
 1 file changed, 21 insertions(+), 5 deletions(-)

diff --git a/src/crdt.rs b/src/crdt.rs
index 97ceda7ead..0a57c549b4 100644
--- a/src/crdt.rs
+++ b/src/crdt.rs
@@ -857,8 +857,10 @@ mod tests {
     use crdt::{parse_port_or_addr, Crdt, ReplicatedData, GOSSIP_SLEEP_MILLIS, MIN_TABLE_SIZE};
     use packet::BlobRecycler;
     use result::Error;
+    use signature::PublicKey;
     use signature::{KeyPair, KeyPairUtil};
     use std::collections::HashMap;
+    use std::net::SocketAddr;
     use std::sync::atomic::{AtomicBool, Ordering};
     use std::sync::mpsc::channel;
     use std::sync::{Arc, RwLock};
@@ -1224,14 +1226,28 @@ mod tests {
         crdt.update_leader();
         assert_eq!(crdt.my_data().current_leader_id, lead2.id);
     }
+    struct TestArrays {
+        pub v1: SocketAddr,
+        pub key2: PublicKey,
+        pub v3: SocketAddr,
+    }
+
     #[test]
-    fn test_update_leader_pubkeys() {
-        let key1 = ReplicatedData::new_entry_point("127.0.0.1:1234".parse().unwrap());
-        let key2 = ReplicatedData::new_entry_point("127.0.0.1:1234".parse().unwrap());
+    fn test_update_leader_keys() {
+        let key1 = TestArrays {
+            v1: "127.0.0.1:4124".parse().unwrap(),
+            key2: KeyPair::new().pubkey(),
+            v3: "224.245.124.012:4124".parse().unwrap(),
+        };
+        let key2 = TestArrays {
+            v1: "134.245.124.012:4000".parse().unwrap(),
+            key2: KeyPair::new().pubkey(),
+            v3: "244.245.124.124:4124".parse().unwrap(),
+        };
         let mut table = HashMap::new();
-        table.entry(&key1.current_leader_id).or_insert(0);
+        table.entry(&key1.key2).or_insert(0);
         for _ in 0..1000 {
-            let a = table.entry(&key2.current_leader_id).or_insert(0);
+            let a = table.entry(&key2.key2).or_insert(0);
             *a += 1;
         }
         assert_eq!(table.len(), 2);