Forráskód Böngészése

Fix a bug in the tests config and another one in backup tests.

Jing Yang 5 éve
szülő
commit
b13b3f0136
2 módosított fájl, 9 hozzáadás és 4 törlés
  1. 1 1
      tests/agreement_tests.rs
  2. 8 3
      tests/config/mod.rs

+ 1 - 1
tests/agreement_tests.rs

@@ -160,7 +160,7 @@ fn rejoin() -> config::Result<()> {
 
 #[test]
 fn backup() -> config::Result<()> {
-    const SERVERS: usize = 3;
+    const SERVERS: usize = 5;
     let cfg = config::make_config(SERVERS, false);
     let _guard = cfg.deferred_cleanup();
 

+ 8 - 3
tests/config/mod.rs

@@ -199,18 +199,23 @@ impl Config {
     }
 
     pub fn set_connect(&self, index: usize, yes: bool) {
-        self.state.lock().connected[index] = yes;
+        let mut state = self.state.lock();
+        state.connected[index] = yes;
 
         let mut network = unlock(&self.network);
 
         // Outgoing clients.
         for j in 0..self.server_count {
-            network.set_enable_client(Self::client_name(index, j), yes)
+            if state.connected[j] {
+                network.set_enable_client(Self::client_name(index, j), yes)
+            }
         }
 
         // Incoming clients.
         for j in 0..self.server_count {
-            network.set_enable_client(Self::client_name(j, index), yes);
+            if state.connected[j] {
+                network.set_enable_client(Self::client_name(j, index), yes);
+            }
         }
     }