From 7b3a3b226aac38320b95a4f46971b8a3578c812f Mon Sep 17 00:00:00 2001
From: Titouan Rigoudy <titouan.rigoudy@gmail.com>
Date: Mon, 4 Jan 2021 14:06:32 +0000
Subject: [PATCH] Squash last warning about room.rs.

---
 src/room.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/src/room.rs b/src/room.rs
index 882d293..7cd8fbb 100644
--- a/src/room.rs
+++ b/src/room.rs
@@ -339,3 +339,31 @@ impl RoomMap {
         Ok(())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::proto::server::RoomListResponse;
+
+    use super::{Room, RoomMap, Visibility};
+
+    #[test]
+    fn room_map_new_is_empty() {
+        assert_eq!(RoomMap::new().get_room_list(), vec![]);
+    }
+
+    #[test]
+    fn room_map_get_strict() {
+        let mut rooms = RoomMap::new();
+        rooms.set_room_list(RoomListResponse {
+            rooms: vec![("room a".to_string(), 42), ("room b".to_string(), 1337)],
+            owned_private_rooms: vec![],
+            other_private_rooms: vec![],
+            operated_private_room_names: vec![],
+        });
+
+        assert_eq!(
+            rooms.get_strict("room a").unwrap(),
+            &Room::new(Visibility::Public, 42)
+        );
+    }
+}