81 lines
2.2 KiB
Rust
81 lines
2.2 KiB
Rust
use crate::Device;
|
|
|
|
pub struct Room {
|
|
name: String,
|
|
devices: Box<[Device]>,
|
|
}
|
|
|
|
impl Room {
|
|
pub fn new(name: impl AsRef<str>, devices: Box<[Device]>) -> Self {
|
|
Self {
|
|
name: name.as_ref().to_string(),
|
|
devices,
|
|
}
|
|
}
|
|
|
|
fn is_valid_idx(&self, idx: usize) -> bool {
|
|
idx < self.devices.len()
|
|
}
|
|
|
|
pub fn get_device(&self, idx: usize) -> Option<&Device> {
|
|
if self.is_valid_idx(idx) {
|
|
Some(&self.devices[idx])
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
pub fn get_device_mut(&mut self, idx: usize) -> Option<&mut Device> {
|
|
self.is_valid_idx(idx);
|
|
Some(&mut self.devices[idx])
|
|
}
|
|
|
|
pub fn print_status(&self) {
|
|
println!("ROOM '{}':", self.name);
|
|
println!("{}", "-".repeat(24));
|
|
for d in self.devices.iter() {
|
|
d.print_status();
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use crate::{PowerSocket, Thermometer};
|
|
|
|
#[test]
|
|
fn smoke_test() {
|
|
let devices = Box::new([
|
|
Device::PowerSocket(PowerSocket::new(12.34, false)),
|
|
Device::Thermometer(Thermometer::new(21.56)),
|
|
]);
|
|
let mut room = Room::new("test_room", devices);
|
|
assert_eq!(room.name, "test_room");
|
|
room.print_status();
|
|
|
|
assert_eq!(format!("{}", room.get_device(0).unwrap().display()), "DEV:PowerSocket[ OFF : 0.0 ]");
|
|
assert_eq!(format!("{}", room.get_device(1).unwrap().display()), "DEV:Thermometer[ 21.6 ]");
|
|
|
|
let Device::PowerSocket(power_socket) = room.get_device_mut(0).unwrap() else {
|
|
unreachable!()
|
|
};
|
|
power_socket.set_on(true);
|
|
|
|
assert_eq!(format!("{}", room.get_device(0).unwrap().display()), "DEV:PowerSocket[ ON : 12.3 ]");
|
|
assert_eq!(format!("{}", room.get_device(1).unwrap().display()), "DEV:Thermometer[ 21.6 ]");
|
|
}
|
|
|
|
#[test]
|
|
fn check_out_of_bounds() {
|
|
let room = Room::new(
|
|
"test_room",
|
|
Box::new([
|
|
Device::PowerSocket(PowerSocket::new(12.34, false)),
|
|
Device::Thermometer(Thermometer::new(21.56)),
|
|
]),
|
|
);
|
|
assert!(room.get_device(2).is_none());
|
|
}
|
|
}
|