From 222cd098388512c4e654b32062074e09c5d61d45 Mon Sep 17 00:00:00 2001
From: Antonio Scandurra <me@as-cii.com>
Date: Fri, 8 Apr 2022 14:41:30 +0200
Subject: [PATCH] Allow host to reconnect to the server in randomized test

---
 crates/server/src/rpc.rs | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/crates/server/src/rpc.rs b/crates/server/src/rpc.rs
index 912d5763220287078e4312bd7afcda6b33388b0e..866627ef4885e63b93b870cac71fdede844cb100 100644
--- a/crates/server/src/rpc.rs
+++ b/crates/server/src/rpc.rs
@@ -5209,7 +5209,6 @@ mod tests {
                 log::info!("Guest {} added", guest_id);
             } else if rng.lock().gen_bool(0.05) {
                 host_disconnected.store(true, SeqCst);
-                server.forbid_connections();
                 server.disconnect_client(user_ids[0]);
                 cx.foreground().advance_clock(RECEIVE_TIMEOUT);
                 let mut clients = futures::future::join_all(clients).await;
@@ -5221,11 +5220,14 @@ mod tests {
                     .unwrap()
                     .read_with(&host_cx, |project, _| assert!(!project.is_shared()));
                 for (guest, mut guest_cx) in clients {
-                    assert!(server
+                    let contacts = server
                         .store
                         .read()
-                        .contacts_for_user(guest.current_user_id(&guest_cx))
-                        .is_empty());
+                        .contacts_for_user(guest.current_user_id(&guest_cx));
+                    assert!(!contacts
+                        .iter()
+                        .flat_map(|contact| &contact.projects)
+                        .any(|project| project.id == host_project_id));
                     guest
                         .project
                         .as_ref()