fluencelabs/rust-libp2p
1
0
Fork 0
mirror of https://github.com/fluencelabs/rust-libp2p synced 2025-05-28 10:11:19 +00:00
Code Issues Projects Releases Wiki Activity

extend full_bucket_discard_pending test

Browse Source
This commit is contained in:
folex 2020-03-25 17:33:20 +03:00
parent e53e4b9059
commit c895ec386a
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
protocols/kad/src/kbucket/bucket.rs
View File

@ -479,14 +479,22 @@ mod tests {
Some((&first_disconnected, NodeStatus::Connected)),
bucket.iter().last()
);
/*
assert_eq!(
bucket.position(&first_disconnected.key).map(|p| p.0),
bucket.first_connected_pos
);
assert_eq!(1, bucket.num_connected());
assert_eq!(K_VALUE.get() - 1, bucket.num_disconnected());
*/
let num_connected = bucket
.iter()
.filter(|(_, s)| *s == NodeStatus::Connected)
.count();
let num_disconnected = bucket
.iter()
.filter(|(_, s)| *s == NodeStatus::Disconnected)
.count();
let position = bucket
.iter()
.position(|(n, _)| n.key.as_ref() == first_disconnected.key.as_ref());
assert_eq!(1, num_connected);
assert_eq!(K_VALUE.get() - 1, num_disconnected);
assert_eq!(position, Some(num_disconnected));
}
#[test]