fix tests

eqlabs · Jan 18, 2024 · 1f332f9 · 1f332f9
1 parent 6db5639
commit 1f332f9
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/crates/p2p/src/tests.rs b/crates/p2p/src/tests.rs
@@ -404,7 +404,8 @@ async fn duplicate_connection() {
             // Bootstrapping can cause redials, so set the offset to a high value.
             start_offset: Duration::from_secs(10),
         },
-        connection_timeout: CONNECTION_TIMEOUT,
+        direct_connection_timeout: CONNECTION_TIMEOUT,
+        relay_connection_timeout: Duration::from_millis(500),
     };
     let keypair = Keypair::generate_ed25519();
     let mut peer1 = TestPeer::new(periodic_cfg, keypair.clone());
@@ -421,15 +422,15 @@ async fn duplicate_connection() {
         .await
         .unwrap();
 
-    wait_event(&mut peer1.event_receiver, |event| match event {
+    wait_for_event(&mut peer1.event_receiver, |event| match event {
         Event::Test(TestEvent::ConnectionEstablished { remote, .. }) if remote == peer2.peer_id => {
             Some(())
         }
         _ => None,
     })
     .await;
 
-    wait_event(&mut peer2.event_receiver, |event| match event {
+    wait_for_event(&mut peer2.event_receiver, |event| match event {
         Event::Test(TestEvent::ConnectionEstablished { remote, .. }) if remote == peer1.peer_id => {
             Some(())
         }
@@ -449,15 +450,15 @@ async fn duplicate_connection() {
         .await
         .unwrap();
 
-    wait_event(&mut peer1_copy.event_receiver, |event| match event {
+    wait_for_event(&mut peer1_copy.event_receiver, |event| match event {
         Event::Test(TestEvent::ConnectionEstablished { remote, .. }) if remote == peer2.peer_id => {
             Some(())
         }
         _ => None,
     })
     .await;
 
-    wait_event(&mut peer1_copy.event_receiver, |event| match event {
+    wait_for_event(&mut peer1_copy.event_receiver, |event| match event {
         Event::Test(TestEvent::ConnectionClosed { remote, .. }) if remote == peer2.peer_id => {
             Some(())
         }