diff --git a/src/relay/appstate.rs b/src/relay/appstate.rs index 088b50c..ac76b01 100644 --- a/src/relay/appstate.rs +++ b/src/relay/appstate.rs @@ -46,4 +46,18 @@ impl AppState { rooms: HashMap::new(), })) } + } + +#[cfg(test)] +mod tests { + use super::*; + use std::sync::{Arc}; + + #[test] + fn test_new() { + let app_state = AppState::new(); + + assert!(Arc::ptr_eq(&app_state, &app_state.clone())); + } +} \ No newline at end of file diff --git a/src/relay/client.rs b/src/relay/client.rs index 58a6064..52c20f8 100644 --- a/src/relay/client.rs +++ b/src/relay/client.rs @@ -488,3 +488,9 @@ impl Client { self.handle_leave_room(server).await } } +// TODO: Add tests +#[cfg(test)] +mod tests { + use super::*; + +} \ No newline at end of file diff --git a/src/relay/room.rs b/src/relay/room.rs index cea60f3..bbb9fb6 100644 --- a/src/relay/room.rs +++ b/src/relay/room.rs @@ -71,3 +71,16 @@ impl Room { } } } +#[cfg(test)] +mod tests { + use super::*; + + #[tokio::test] + async fn test_room_new() { + let room = Room::new(5); + + assert_eq!(room.size, 5); + + assert!(room.senders.is_empty()); + } +}