jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 10:59:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: extend test_pending_ping() to respect limit of pending_pings (#7520)

Browse Source
This commit is contained in:
int88
2024-04-09 20:18:04 +08:00
committed by GitHub
parent 2e87b2a8d5
commit 5efe17370f
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
crates/net/discv4/src/lib.rs
View File

@ -2268,12 +2268,27 @@ mod tests {
let local_addr = service.local_addr();
let mut num_inserted = 0;
for _ in 0..MAX_NODES_PING {
loop {
let node = NodeRecord::new(local_addr, PeerId::random());
if service.add_node(node) {
num_inserted += 1;
assert!(service.pending_pings.contains_key(&node.id));
assert_eq!(service.pending_pings.len(), num_inserted);
if num_inserted == MAX_NODES_PING {
break;
}
}
}
// `pending_pings` is full, insert into `queued_pings`.
num_inserted = 0;
for _ in 0..MAX_NODES_PING {
let node = NodeRecord::new(local_addr, PeerId::random());
if service.add_node(node) {
num_inserted += 1;
assert!(!service.pending_pings.contains_key(&node.id));
assert_eq!(service.pending_pings.len(), MAX_NODES_PING);
assert_eq!(service.queued_pings.len(), num_inserted);
}
}
}