use minidom::Element; use error::Error; use ns; #[derive(Debug, Clone)] pub enum Attention { Attention, } pub fn parse_attention(root: &Element) -> Result { if !root.is("attention", ns::ATTENTION) { return Err(Error::ParseError("This is not an attention element.")); } for _ in root.children() { return Err(Error::ParseError("Unknown child in attention element.")); } Ok(Attention::Attention) } #[cfg(test)] mod tests { use minidom::Element; use error::Error; use attention; #[test] fn test_simple() { let elem: Element = "".parse().unwrap(); attention::parse_attention(&elem).unwrap(); } #[test] fn test_invalid_child() { let elem: Element = "".parse().unwrap(); let error = attention::parse_attention(&elem).unwrap_err(); let message = match error { Error::ParseError(string) => string, _ => panic!(), }; assert_eq!(message, "Unknown child in attention element."); } }