1// Copyright (c) 2017 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
2//
3// This Source Code Form is subject to the terms of the Mozilla Public
4// License, v. 2.0. If a copy of the MPL was not distributed with this
5// file, You can obtain one at http://mozilla.org/MPL/2.0/.
6
7use minidom::{Element, IntoAttributeValue};
8
9use error::Error;
10
11use ns;
12use helpers::Base64;
13
14generate_attribute!(Stanza, "stanza", {
15 Iq => "iq",
16 Message => "message",
17}, Default = Iq);
18
19generate_element_with_only_attributes!(Open, "open", ns::IBB, [
20 block_size: u16 = "block-size" => required,
21 sid: String = "sid" => required,
22 stanza: Stanza = "stanza" => default,
23]);
24
25generate_element_with_text!(Data, "data", ns::IBB,
26 [
27 seq: u16 = "seq" => required,
28 sid: String = "sid" => required
29 ],
30 data: Base64<Vec<u8>>
31);
32
33generate_element_with_only_attributes!(Close, "close", ns::IBB, [
34 sid: String = "sid" => required,
35]);
36
37#[cfg(test)]
38mod tests {
39 use super::*;
40 use try_from::TryFrom;
41 use std::error::Error as StdError;
42
43 #[test]
44 fn test_simple() {
45 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='3' sid='coucou'/>".parse().unwrap();
46 let open = Open::try_from(elem).unwrap();
47 assert_eq!(open.block_size, 3);
48 assert_eq!(open.sid, "coucou");
49 assert_eq!(open.stanza, Stanza::Iq);
50
51 let elem: Element = "<data xmlns='http://jabber.org/protocol/ibb' seq='0' sid='coucou'>AAAA</data>".parse().unwrap();
52 let data = Data::try_from(elem).unwrap();
53 assert_eq!(data.seq, 0);
54 assert_eq!(data.sid, "coucou");
55 assert_eq!(data.data, vec!(0, 0, 0));
56
57 let elem: Element = "<close xmlns='http://jabber.org/protocol/ibb' sid='coucou'/>".parse().unwrap();
58 let close = Close::try_from(elem).unwrap();
59 assert_eq!(close.sid, "coucou");
60 }
61
62 #[test]
63 fn test_invalid() {
64 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb'/>".parse().unwrap();
65 let error = Open::try_from(elem).unwrap_err();
66 let message = match error {
67 Error::ParseError(string) => string,
68 _ => panic!(),
69 };
70 assert_eq!(message, "Required attribute 'block-size' missing.");
71
72 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='-5'/>".parse().unwrap();
73 let error = Open::try_from(elem).unwrap_err();
74 let message = match error {
75 Error::ParseIntError(error) => error,
76 _ => panic!(),
77 };
78 assert_eq!(message.description(), "invalid digit found in string");
79
80 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='128'/>".parse().unwrap();
81 let error = Open::try_from(elem).unwrap_err();
82 let message = match error {
83 Error::ParseError(error) => error,
84 _ => panic!(),
85 };
86 assert_eq!(message, "Required attribute 'sid' missing.");
87 }
88
89 #[test]
90 fn test_invalid_stanza() {
91 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='128' sid='coucou' stanza='fdsq'/>".parse().unwrap();
92 let error = Open::try_from(elem).unwrap_err();
93 let message = match error {
94 Error::ParseError(string) => string,
95 _ => panic!(),
96 };
97 assert_eq!(message, "Unknown value for 'stanza' attribute.");
98 }
99}