From 812a436d9b5d3a468f411a61aa37cc1ba8d55f5c Mon Sep 17 00:00:00 2001 From: Svyatoslav Nikolsky Date: Mon, 31 Oct 2016 17:20:08 +0300 Subject: [PATCH] added couple of sync peers tests --- sync/src/synchronization_peers.rs | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/sync/src/synchronization_peers.rs b/sync/src/synchronization_peers.rs index 16ab6662..2986414c 100644 --- a/sync/src/synchronization_peers.rs +++ b/sync/src/synchronization_peers.rs @@ -118,6 +118,29 @@ mod tests { assert_eq!(info.active, 0); } + #[test] + fn peers_all_idle_after_reset() { + let mut peers = Peers::new(); + peers.on_blocks_requested(7, &vec![H256::default()]); + peers.on_blocks_requested(8, &vec![H256::default()]); + assert_eq!(peers.information().idle, 0); + assert_eq!(peers.information().active, 2); + peers.reset(); + assert_eq!(peers.information().idle, 2); + assert_eq!(peers.information().active, 0); + } + + #[test] + fn peers_removed_after_inventory_request() { + let mut peers = Peers::new(); + peers.insert(5); + peers.insert(7); + assert_eq!(peers.information().idle, 2); + assert_eq!(peers.information().active, 0); + peers.on_inventory_requested(5); + assert_eq!(peers.information().idle, 1); + } + #[test] fn peers_insert_remove_idle() { let mut peers = Peers::new();