Try to fix CI (Fix harness test's leader node-id to 1)

lablup · Feb 1, 2024 · 3b3a48a · 3b3a48a
1 parent 32ac6f4
commit 3b3a48a
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/harness/src/utils.rs b/harness/src/utils.rs
@@ -57,7 +57,12 @@ pub async fn load_peers(example_filename: &str) -> Result<Peers, Box<dyn std::er
 
     for peer_info in raft_config.raft.peers {
         let addr = SocketAddr::new(peer_info.host.parse().unwrap(), peer_info.port);
-        peers.add_peer(peer_info.node_id, addr, Some(InitialRole::Voter));
+
+        if peer_info.node_id == 1 {
+            peers.add_peer(peer_info.node_id, addr, Some(InitialRole::Leader));
+        } else {
+            peers.add_peer(peer_info.node_id, addr, Some(InitialRole::Voter));
+        }
     }
 
     Ok(peers)

diff --git a/harness/tests/data_replication.rs b/harness/tests/data_replication.rs
@@ -13,7 +13,7 @@ use harness::{
 pub async fn test_data_replication() {
     let peers = load_peers(THREE_NODE_EXAMPLE).await.unwrap();
     let _raft_tasks = tokio::spawn(build_raft_cluster(peers.clone()));
-    sleep(Duration::from_secs(5)).await;
+    sleep(Duration::from_secs(1)).await;
 
     {
         let mut rafts = RAFTS.lock().unwrap();

diff --git a/harness/tests/leader_election.rs b/harness/tests/leader_election.rs
@@ -13,7 +13,7 @@ use harness::{
 pub async fn test_leader_election_in_three_node_example() {
     let peers = load_peers(THREE_NODE_EXAMPLE).await.unwrap();
     let _raft_tasks = tokio::spawn(build_raft_cluster(peers.clone()));
-    sleep(Duration::from_secs(5)).await;
+    sleep(Duration::from_secs(1)).await;
 
     let mut rafts = RAFTS.lock().unwrap();
     let raft_1 = rafts.get_mut(&1).unwrap();