1use minidom::Element;
 2
 3use error::Error;
 4
 5use ns;
 6
 7#[derive(Debug, Clone)]
 8pub enum MessageCorrect {
 9    Replace(String),
10}
11
12pub fn parse_message_correct(root: &Element) -> Result<MessageCorrect, Error> {
13    if !root.is("replace", ns::MESSAGE_CORRECT) {
14        return Err(Error::ParseError("This is not a replace element."));
15    }
16    for _ in root.children() {
17        return Err(Error::ParseError("Unknown child in replace element."));
18    }
19    let id = root.attr("id").unwrap_or("").to_owned();
20    Ok(MessageCorrect::Replace(id))
21}
22
23#[cfg(test)]
24mod tests {
25    use minidom::Element;
26    use error::Error;
27    use message_correct;
28
29    #[test]
30    fn test_simple() {
31        let elem: Element = "<replace xmlns='urn:xmpp:message-correct:0'/>".parse().unwrap();
32        message_correct::parse_message_correct(&elem).unwrap();
33    }
34
35    #[test]
36    fn test_invalid_child() {
37        let elem: Element = "<replace xmlns='urn:xmpp:message-correct:0'><coucou/></replace>".parse().unwrap();
38        let error = message_correct::parse_message_correct(&elem).unwrap_err();
39        let message = match error {
40            Error::ParseError(string) => string,
41            _ => panic!(),
42        };
43        assert_eq!(message, "Unknown child in replace element.");
44    }
45}