add House struct

smart-house/src/house.rs

@@ -0,0 +1,91 @@
+use crate::Room;
+
+pub struct House {
+    address: String,
+    rooms: Box<[Room]>,
+}
+
+impl House {
+    pub fn new(address: impl AsRef<str>, rooms: Box<[Room]>) -> Self {
+        Self {
+            address: address.as_ref().to_string(),
+            rooms,
+        }
+    }
+
+    fn check_bounds(&self, idx: usize) {
+        if idx >= self.rooms.len() {
+            panic!("Index is out of bounds")
+        }
+    }
+
+    pub fn get_room(&self, idx: usize) -> &Room {
+        self.check_bounds(idx);
+        &self.rooms[idx]
+    }
+
+    pub fn get_room_mut(&mut self, idx: usize) -> &mut Room {
+        self.check_bounds(idx);
+        &mut self.rooms[idx]
+    }
+
+    pub fn print_status(&self) {
+        println!("HOUSE '{}':", self.address);
+        println!("{}", "=".repeat(32));
+        for d in self.rooms.iter() {
+            d.print_status();
+            println!();
+        }
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::{Device, PowerSocket, Thermometer};
+
+    fn create_test_house() -> House {
+        House::new(
+            "Best street, 777",
+            Box::new([
+                Room::new(
+                    "main",
+                    Box::new([
+                        Device::Thermometer(Thermometer::new(20.0)),
+                        Device::PowerSocket(PowerSocket::new(12.34, false)),
+                        Device::PowerSocket(PowerSocket::new(10.01, true)),
+                    ]),
+                ),
+                Room::new(
+                    "bedroom",
+                    Box::new([Device::PowerSocket(PowerSocket::new(11.11, true)), Device::Thermometer(Thermometer::new(17.99))]),
+                ),
+            ]),
+        )
+    }
+
+    #[test]
+    fn smoke_test() {
+        let mut house = create_test_house();
+        house.print_status();
+        assert_eq!(house.address, "Best street, 777");
+
+        assert_eq!(format!("{}", house.get_room(0).get_device(0).display()), "DEV:Thermometer[ 20.0 ]");
+        assert_eq!(format!("{}", house.get_room(0).get_device(1).display()), "DEV:PowerSocket[ OFF : 0.0 ]");
+        assert_eq!(format!("{}", house.get_room(1).get_device(0).display()), "DEV:PowerSocket[ ON : 11.1 ]");
+
+        let Device::PowerSocket(powers_socket) = house.get_room_mut(0).get_device_mut(1) else {
+            unreachable!()
+        };
+        powers_socket.set_on(true);
+
+        assert_eq!(format!("{}", house.get_room(0).get_device(1).display()), "DEV:PowerSocket[ ON : 12.3 ]");
+    }
+
+    #[test]
+    #[should_panic(expected = "Index is out of bounds")]
+    fn panic_test() {
+        let house = create_test_house();
+        house.check_bounds(2);
+    }
+}