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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use std::fmt::{Formatter, Result};

use hyper::error::{self, Error};
use hyper::header::{HeaderFormat, Header};

const MAN_HEADER_NAME: &'static str = "MAN";
const MAN_HEADER_VALUE: &'static str = "\"ssdp:discover\"";

/// Represents a header used to specify HTTP extension.
#[derive(Copy, Clone, PartialEq, Eq, Hash, Debug)]
pub struct Man;

impl Header for Man {
    fn header_name() -> &'static str {
        MAN_HEADER_NAME
    }

    fn parse_header(raw: &[Vec<u8>]) -> error::Result<Self> {
        if raw.len() != 1 {
            return Err(Error::Header);
        }

        let man_bytes = MAN_HEADER_VALUE.as_bytes();
        match &raw[0][..] {
            n if n == man_bytes => Ok(Man),
            _ => Err(Error::Header),
        }
    }
}

impl HeaderFormat for Man {
    fn fmt_header(&self, fmt: &mut Formatter) -> Result {
        try!(fmt.write_str(MAN_HEADER_VALUE));

        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use hyper::header::Header;

    use super::Man;

    #[test]
    fn positive_man() {
        let man_header = &[b"\"ssdp:discover\""[..].to_vec()];

        Man::parse_header(man_header).unwrap();
    }

    #[test]
    #[should_panic]
    fn negative_wrong_case() {
        let wrong_case_man_header = &[b"\"SSDP:discover\""[..].to_vec()];

        Man::parse_header(wrong_case_man_header).unwrap();
    }

    #[test]
    #[should_panic]
    fn negative_missing_quotes() {
        let missing_quotes_man_header = &[b"ssdp:discover"[..].to_vec()];

        Man::parse_header(missing_quotes_man_header).unwrap();
    }
}