1use minidom::Element;
 2
 3use error::Error;
 4
 5use ns;
 6
 7#[derive(Debug, Clone)]
 8pub enum ChatState {
 9    Active,
10    Composing,
11    Gone,
12    Inactive,
13    Paused,
14}
15
16pub fn parse_chatstate(root: &Element) -> Result<ChatState, Error> {
17    for _ in root.children() {
18        return Err(Error::ParseError("Unknown child in chatstate element."));
19    }
20    if root.is("active", ns::CHATSTATES) {
21        Ok(ChatState::Active)
22    } else if root.is("composing", ns::CHATSTATES) {
23        Ok(ChatState::Composing)
24    } else if root.is("gone", ns::CHATSTATES) {
25        Ok(ChatState::Gone)
26    } else if root.is("inactive", ns::CHATSTATES) {
27        Ok(ChatState::Inactive)
28    } else if root.is("paused", ns::CHATSTATES) {
29        Ok(ChatState::Paused)
30    } else {
31        Err(Error::ParseError("This is not a chatstate element."))
32    }
33}
34
35pub fn serialise(chatstate: &ChatState) -> Element {
36    Element::builder(match *chatstate {
37        ChatState::Active => "active",
38        ChatState::Composing => "composing",
39        ChatState::Gone => "gone",
40        ChatState::Inactive => "inactive",
41        ChatState::Paused => "paused",
42    }).ns(ns::CHATSTATES)
43      .build()
44}
45
46#[cfg(test)]
47mod tests {
48    use minidom::Element;
49    use error::Error;
50    use chatstates;
51    use ns;
52
53    #[test]
54    fn test_simple() {
55        let elem: Element = "<active xmlns='http://jabber.org/protocol/chatstates'/>".parse().unwrap();
56        chatstates::parse_chatstate(&elem).unwrap();
57    }
58
59    #[test]
60    fn test_invalid() {
61        let elem: Element = "<coucou xmlns='http://jabber.org/protocol/chatstates'/>".parse().unwrap();
62        let error = chatstates::parse_chatstate(&elem).unwrap_err();
63        let message = match error {
64            Error::ParseError(string) => string,
65            _ => panic!(),
66        };
67        assert_eq!(message, "This is not a chatstate element.");
68    }
69
70    #[test]
71    fn test_invalid_child() {
72        let elem: Element = "<gone xmlns='http://jabber.org/protocol/chatstates'><coucou/></gone>".parse().unwrap();
73        let error = chatstates::parse_chatstate(&elem).unwrap_err();
74        let message = match error {
75            Error::ParseError(string) => string,
76            _ => panic!(),
77        };
78        assert_eq!(message, "Unknown child in chatstate element.");
79    }
80
81    #[test]
82    #[ignore]
83    fn test_invalid_attribute() {
84        let elem: Element = "<inactive xmlns='http://jabber.org/protocol/chatstates' coucou=''/>".parse().unwrap();
85        let error = chatstates::parse_chatstate(&elem).unwrap_err();
86        let message = match error {
87            Error::ParseError(string) => string,
88            _ => panic!(),
89        };
90        assert_eq!(message, "Unknown attribute in chatstate element.");
91    }
92
93    #[test]
94    fn test_serialise() {
95        let chatstate = chatstates::ChatState::Active;
96        let elem = chatstates::serialise(&chatstate);
97        assert!(elem.is("active", ns::CHATSTATES));
98    }
99}