From 685c4bf1ed66c152fb379af48d66f3b9213f32ed Mon Sep 17 00:00:00 2001
From: Titouan Rigoudy <titouan.rigoudy@gmail.com>
Date: Fri, 2 Dec 2022 00:55:04 +0000
Subject: [PATCH] Add tests for insert_member() and remove_member().

---
 client/src/room/map.rs | 52 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/client/src/room/map.rs b/client/src/room/map.rs
index fac2ed7..0fb6b50 100644
--- a/client/src/room/map.rs
+++ b/client/src/room/map.rs
@@ -386,6 +386,58 @@ mod tests {
     );
   }
 
+  #[test]
+  fn entry_insert_member() {
+    let mut room = RoomEntry::new("bleep".to_string(), RoomState::default());
+
+    room.insert_member("shruti".to_string());
+
+    assert_eq!(
+      room.into_state(),
+      RoomState {
+        members: ["shruti".to_string()]
+          .into_iter()
+          .collect::<HashSet<String>>(),
+        ..RoomState::default()
+      }
+    );
+  }
+
+  #[test]
+  fn entry_insert_member_twice() {
+    let mut room = RoomEntry::new("bleep".to_string(), RoomState::default());
+
+    room.insert_member("shruti".to_string());
+    room.insert_member("shruti".to_string());
+
+    assert_eq!(
+      room.into_state(),
+      RoomState {
+        members: ["shruti".to_string()]
+          .into_iter()
+          .collect::<HashSet<String>>(),
+        ..RoomState::default()
+      }
+    );
+  }
+
+  #[test]
+  fn entry_remove_member() {
+    let mut room = RoomEntry::new(
+      "bleep".to_string(),
+      RoomState {
+        members: ["shruti".to_string()]
+          .into_iter()
+          .collect::<HashSet<String>>(),
+        ..RoomState::default()
+      },
+    );
+
+    room.remove_member("shruti");
+
+    assert_eq!(room.into_state(), RoomState::default());
+  }
+
   #[test]
   fn map_new_is_empty() {
     assert_eq!(RoomMap::new().get_room_list(), vec![]);