fix: connection_limit test change condition

libp2p · Jul 20, 2023 · e180673 · e180673
1 parent d854d71
commit e180673
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/misc/connection-limits/src/lib.rs b/misc/connection-limits/src/lib.rs
@@ -364,7 +364,10 @@ impl NetworkBehaviour for Behaviour {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use libp2p_swarm::{dial_opts::DialOpts, DialError, ListenError, Swarm, SwarmEvent};
+    use libp2p_swarm::{
+        dial_opts::{DialOpts, PeerCondition},
+        DialError, ListenError, Swarm, SwarmEvent,
+    };
     use libp2p_swarm_test::SwarmExt;
     use quickcheck::*;
 
@@ -387,14 +390,20 @@ mod tests {
             network
                 .dial(
                     DialOpts::peer_id(target)
+                        .condition(PeerCondition::Disconnected)
                         .addresses(vec![addr.clone()])
                         .build(),
                 )
                 .expect("Unexpected connection limit.");
         }
 
         match network
-            .dial(DialOpts::peer_id(target).addresses(vec![addr]).build())
+            .dial(
+                DialOpts::peer_id(target)
+                    .condition(PeerCondition::Disconnected)
+                    .addresses(vec![addr])
+                    .build(),
+            )
             .expect_err("Unexpected dialing success.")
         {
             DialError::Denied { cause } => {