test: fix: connectivity-check

akiradeveloper · Oct 30, 2024 · 881c593 · 881c593
1 parent ed1de94
commit 881c593
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/env/src/lib.rs b/tests/env/src/lib.rs
@@ -200,10 +200,11 @@ impl Env {
             .or_insert_with(|| port_check::free_local_ipv4_port().unwrap());
         let snap_states = self.penv.as_mut().map(|env| env.get(id));
         let node = Node::new(id, free_port, self.n_shards, snap_states).unwrap();
-        port_check::is_port_reachable_with_timeout(
+        assert!(port_check::is_port_reachable_with_timeout(
             node.address().to_string(),
             Duration::from_secs(5),
-        );
+        ));
+
         self.nodes.insert(id, node);
     }
 

diff --git a/tests/sorock-tests/src/lib.rs b/tests/sorock-tests/src/lib.rs
@@ -34,7 +34,7 @@ impl Builder {
         let mut env = Env::new(p, self.with_persistency, self.with_logging);
         for id in 0..n {
             env.add_node(id);
-            env.check_connectivity(0).await?;
+            env.check_connectivity(id).await?;
         }
         Ok(Cluster { env })
     }