diff --git a/protocols/relay/tests/lib.rs b/protocols/relay/tests/lib.rs index da485748..92538549 100644 --- a/protocols/relay/tests/lib.rs +++ b/protocols/relay/tests/lib.rs @@ -210,12 +210,16 @@ fn connect() { src.dial(dst_addr).unwrap(); pool.run_until(futures::future::join( - connection_established_to(src, relay_peer_id, dst_peer_id), - connection_established_to(dst, relay_peer_id, src_peer_id), + connection_established_to(&mut src, relay_peer_id, dst_peer_id), + connection_established_to(&mut dst, relay_peer_id, src_peer_id), )); } -async fn connection_established_to(mut swarm: Swarm, relay_peer_id: PeerId, other: PeerId) { +async fn connection_established_to( + swarm: &mut Swarm, + relay_peer_id: PeerId, + other: PeerId, +) { loop { match swarm.select_next_some().await { SwarmEvent::Dialing(peer_id) if peer_id == relay_peer_id => {}