rebuilt to use message router.
This commit is contained in:
93
src/message.rs
Normal file
93
src/message.rs
Normal file
@ -0,0 +1,93 @@
|
||||
use super::{ClientMsg, Msg, SessionMsg};
|
||||
use uuid::Uuid;
|
||||
|
||||
/// Message Types
|
||||
#[derive(Clone)]
|
||||
pub enum MsgData {
|
||||
Client(ClientMsg),
|
||||
Session(SessionMsg),
|
||||
Test1,
|
||||
Test2,
|
||||
}
|
||||
|
||||
/// MoreThanText Message Structure
|
||||
#[derive(Clone)]
|
||||
pub struct Message {
|
||||
id: Uuid,
|
||||
msg: MsgData,
|
||||
}
|
||||
|
||||
impl Message {
|
||||
pub fn new<D>(data: &D) -> Self
|
||||
where
|
||||
D: Msg,
|
||||
{
|
||||
Self {
|
||||
id: Uuid::new_v4(),
|
||||
msg: data.to_msgdata(),
|
||||
}
|
||||
}
|
||||
|
||||
pub fn get_message(&self) -> &MsgData {
|
||||
&self.msg
|
||||
}
|
||||
|
||||
pub fn get_id(&self) -> Uuid {
|
||||
self.id.clone()
|
||||
}
|
||||
|
||||
pub fn reply<D>(&self, data: &D) -> Self
|
||||
where
|
||||
D: Msg,
|
||||
{
|
||||
Self {
|
||||
id: self.id.clone(),
|
||||
msg: data.to_msgdata(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod messages {
|
||||
use super::{super::test_message::Tester, *};
|
||||
|
||||
#[test]
|
||||
fn new_messagees() {
|
||||
let data = Tester::Test1;
|
||||
let msg = Message::new(&data);
|
||||
match msg.get_message() {
|
||||
MsgData::Test1 => {}
|
||||
_ => unreachable!("Should have received Test1"),
|
||||
}
|
||||
let data = Tester::Test2;
|
||||
let msg = Message::new(&data);
|
||||
match msg.get_message() {
|
||||
MsgData::Test2 => {}
|
||||
_ => unreachable!("Should have received Test1"),
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn message_ids_are_unique() {
|
||||
let mut ids: Vec<Uuid> = Vec::new();
|
||||
let data = Tester::Test1;
|
||||
for _ in 1..10 {
|
||||
let msg = Message::new(&data);
|
||||
assert!(!ids.contains(&msg.get_id()), "Had a duplicate id");
|
||||
ids.push(msg.get_id());
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn create_replies() {
|
||||
let data1 = Tester::Test1;
|
||||
let data2 = Tester::Test2;
|
||||
let msg = Message::new(&data1);
|
||||
let reply = msg.reply(&data2);
|
||||
assert_eq!(reply.get_id(), msg.get_id());
|
||||
match reply.get_message() {
|
||||
MsgData::Test2 => {}
|
||||
_ => unreachable!("Should have been a Test1"),
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user