diff --git a/crates/radicle-node/src/test/node.rs b/crates/radicle-node/src/test/node.rs index dad7371e..268f4284 100644 --- a/crates/radicle-node/src/test/node.rs +++ b/crates/radicle-node/src/test/node.rs @@ -99,6 +99,9 @@ impl Drop for NodeHandle { impl + cyphernet::Ecdh> NodeHandle { /// Connect this node to another node, and wait for the connection to be established both ways. + /// + /// If the remote has blocked this node, then the remote event will be + /// [`Event::PeerDisconnected`]. pub fn connect(&mut self, remote: &NodeHandle) -> &mut Self { let local_events = self.handle.events(); let remote_events = remote.handle.events(); @@ -119,7 +122,9 @@ impl + cyphernet::Ecdh> NodeHandle { .iter() .find(|e| { matches!( - e, Event::PeerConnected { nid } if nid == &self.id + e, + Event::PeerConnected { nid } | Event::PeerDisconnected { nid, .. } + if nid == &self.id ) }) .unwrap();