1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

//! Helper type for writing unit tests.

#[derive(Clone, Debug, PartialEq)]
pub struct TestMsg(pub String);

use super::serialize;
use std::io::prelude::*;
impl serialize::AsRawMsg for TestMsg {
    fn get_hdr(&self) -> (u8, u32, u32) {
        (0xff, serialize::HDR_LENGTH + self.0.len() as u32, 0)
    }

    fn get_bytes<W: Write>(&self, w: &mut W) -> super::Result<()> {
        w.write_all(self.0.as_bytes())?;
        Ok(())
    }

    fn from_raw_msg(msg: serialize::RawMsg) -> super::Result<Self> {
        let b = msg.get_bytes()?;
        let got: String = std::str::from_utf8(b)
            .expect("parse message to str")
            .chars()
            .take_while(|b| *b != '\0')
            .collect();
        Ok(TestMsg(got))
    }
}