Skip to content

Commit 3398d4c

Browse files
committed
add a third node to non advertising node test
1 parent 56c0e0e commit 3398d4c

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

iroh/src/discovery/mdns.rs

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -562,16 +562,24 @@ mod tests {
562562
let (_, discovery_a) = make_discoverer(false)?;
563563
let (node_id_b, discovery_b) = make_discoverer(false)?;
564564

565-
let node_data = NodeData::new(None, BTreeSet::from(["0.0.0.0:11111".parse().unwrap()]));
565+
let (node_id_c, discovery_c) = make_discoverer(true)?;
566+
let node_data_c =
567+
NodeData::new(None, BTreeSet::from(["0.0.0.0:22222".parse().unwrap()]));
568+
discovery_c.publish(&node_data_c);
566569

567-
discovery_b.publish(&node_data);
570+
let node_data_b =
571+
NodeData::new(None, BTreeSet::from(["0.0.0.0:11111".parse().unwrap()]));
572+
discovery_b.publish(&node_data_b);
568573

569-
let mut stream = discovery_a.resolve(node_id_b).unwrap();
574+
let mut stream_c = discovery_a.resolve(node_id_c).unwrap();
575+
let result_c = tokio::time::timeout(Duration::from_secs(2), stream_c.next()).await;
576+
assert!(result_c.is_ok(), "Advertising node should be discoverable");
570577

571-
let result = tokio::time::timeout(Duration::from_secs(2), stream.next()).await;
578+
let mut stream_b = discovery_a.resolve(node_id_b).unwrap();
579+
let result_b = tokio::time::timeout(Duration::from_secs(2), stream_b.next()).await;
572580
assert!(
573-
result.is_err(),
574-
"Expected timeout since node isn't advertising"
581+
result_b.is_err(),
582+
"Expected timeout since node b isn't advertising"
575583
);
576584

577585
Ok(())

0 commit comments

Comments
 (0)