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 helpers::Base64;
8
9generate_attribute!(Stanza, "stanza", {
10 Iq => "iq",
11 Message => "message",
12}, Default = Iq);
13
14generate_element_with_only_attributes!(Open, "open", IBB, [
15 block_size: u16 = "block-size" => required,
16 sid: String = "sid" => required,
17 stanza: Stanza = "stanza" => default,
18]);
19
20generate_element_with_text!(Data, "data", IBB,
21 [
22 seq: u16 = "seq" => required,
23 sid: String = "sid" => required
24 ],
25 data: Base64<Vec<u8>>
26);
27
28generate_element_with_only_attributes!(Close, "close", IBB, [
29 sid: String = "sid" => required,
30]);
31
32#[cfg(test)]
33mod tests {
34 use super::*;
35 use try_from::TryFrom;
36 use minidom::Element;
37 use error::Error;
38 use std::error::Error as StdError;
39
40 #[test]
41 fn test_simple() {
42 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='3' sid='coucou'/>".parse().unwrap();
43 let open = Open::try_from(elem).unwrap();
44 assert_eq!(open.block_size, 3);
45 assert_eq!(open.sid, "coucou");
46 assert_eq!(open.stanza, Stanza::Iq);
47
48 let elem: Element = "<data xmlns='http://jabber.org/protocol/ibb' seq='0' sid='coucou'>AAAA</data>".parse().unwrap();
49 let data = Data::try_from(elem).unwrap();
50 assert_eq!(data.seq, 0);
51 assert_eq!(data.sid, "coucou");
52 assert_eq!(data.data, vec!(0, 0, 0));
53
54 let elem: Element = "<close xmlns='http://jabber.org/protocol/ibb' sid='coucou'/>".parse().unwrap();
55 let close = Close::try_from(elem).unwrap();
56 assert_eq!(close.sid, "coucou");
57 }
58
59 #[test]
60 fn test_invalid() {
61 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb'/>".parse().unwrap();
62 let error = Open::try_from(elem).unwrap_err();
63 let message = match error {
64 Error::ParseError(string) => string,
65 _ => panic!(),
66 };
67 assert_eq!(message, "Required attribute 'block-size' missing.");
68
69 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='-5'/>".parse().unwrap();
70 let error = Open::try_from(elem).unwrap_err();
71 let message = match error {
72 Error::ParseIntError(error) => error,
73 _ => panic!(),
74 };
75 assert_eq!(message.description(), "invalid digit found in string");
76
77 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='128'/>".parse().unwrap();
78 let error = Open::try_from(elem).unwrap_err();
79 let message = match error {
80 Error::ParseError(error) => error,
81 _ => panic!(),
82 };
83 assert_eq!(message, "Required attribute 'sid' missing.");
84 }
85
86 #[test]
87 fn test_invalid_stanza() {
88 let elem: Element = "<open xmlns='http://jabber.org/protocol/ibb' block-size='128' sid='coucou' stanza='fdsq'/>".parse().unwrap();
89 let error = Open::try_from(elem).unwrap_err();
90 let message = match error {
91 Error::ParseError(string) => string,
92 _ => panic!(),
93 };
94 assert_eq!(message, "Unknown value for 'stanza' attribute.");
95 }
96}