mirror of
https://gitlab.com/xmpp-rs/xmpp-rs.git
synced 2024-07-12 22:21:53 +00:00
127 lines
3.8 KiB
Rust
127 lines
3.8 KiB
Rust
// Copyright (c) 2019 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
|
||
//
|
||
// This Source Code Form is subject to the terms of the Mozilla Public
|
||
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
||
use crate::hashes::Sha1HexAttribute;
|
||
use crate::pubsub::PubSubPayload;
|
||
use crate::util::helpers::WhitespaceAwareBase64;
|
||
|
||
generate_element!(
|
||
/// Communicates information about an avatar.
|
||
Metadata, "metadata", AVATAR_METADATA,
|
||
children: [
|
||
/// List of information elements describing this avatar.
|
||
infos: Vec<Info> = ("info", AVATAR_METADATA) => Info
|
||
]
|
||
);
|
||
|
||
impl PubSubPayload for Metadata {}
|
||
|
||
generate_element!(
|
||
/// Communicates avatar metadata.
|
||
Info, "info", AVATAR_METADATA,
|
||
attributes: [
|
||
/// The size of the image data in bytes.
|
||
bytes: u16 = "bytes" => required,
|
||
|
||
/// The width of the image in pixels.
|
||
width: Option<u8> = "width" => optional,
|
||
|
||
/// The height of the image in pixels.
|
||
height: Option<u8> = "height" => optional,
|
||
|
||
/// The SHA-1 hash of the image data for the specified content-type.
|
||
id: Sha1HexAttribute = "id" => required,
|
||
|
||
/// The IANA-registered content type of the image data.
|
||
type_: String = "type" => required,
|
||
|
||
/// The http: or https: URL at which the image data file is hosted.
|
||
url: Option<String> = "url" => optional,
|
||
]
|
||
);
|
||
|
||
generate_element!(
|
||
/// The actual avatar data.
|
||
Data, "data", AVATAR_DATA,
|
||
text: (
|
||
/// Vector of bytes representing the avatar’s image.
|
||
data: WhitespaceAwareBase64<Vec<u8>>
|
||
)
|
||
);
|
||
|
||
impl PubSubPayload for Data {}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
use crate::hashes::Algo;
|
||
use crate::util::error::Error;
|
||
use minidom::Element;
|
||
use try_from::TryFrom;
|
||
|
||
#[cfg(target_pointer_width = "32")]
|
||
#[test]
|
||
fn test_size() {
|
||
assert_size!(Metadata, 12);
|
||
assert_size!(Info, 60);
|
||
assert_size!(Data, 12);
|
||
}
|
||
|
||
#[cfg(target_pointer_width = "64")]
|
||
#[test]
|
||
fn test_size() {
|
||
assert_size!(Metadata, 24);
|
||
assert_size!(Info, 112);
|
||
assert_size!(Data, 24);
|
||
}
|
||
|
||
#[test]
|
||
fn test_simple() {
|
||
let elem: Element = "<metadata xmlns='urn:xmpp:avatar:metadata'>
|
||
<info bytes='12345' width='64' height='64'
|
||
id='111f4b3c50d7b0df729d299bc6f8e9ef9066971f'
|
||
type='image/png'/>
|
||
</metadata>"
|
||
.parse()
|
||
.unwrap();
|
||
let metadata = Metadata::try_from(elem).unwrap();
|
||
assert_eq!(metadata.infos.len(), 1);
|
||
let info = &metadata.infos[0];
|
||
assert_eq!(info.bytes, 12345);
|
||
assert_eq!(info.width, Some(64));
|
||
assert_eq!(info.height, Some(64));
|
||
assert_eq!(info.id.algo, Algo::Sha_1);
|
||
assert_eq!(info.type_, "image/png");
|
||
assert_eq!(info.url, None);
|
||
assert_eq!(
|
||
info.id.hash,
|
||
[
|
||
17, 31, 75, 60, 80, 215, 176, 223, 114, 157, 41, 155, 198, 248, 233, 239, 144, 102,
|
||
151, 31
|
||
]
|
||
);
|
||
|
||
let elem: Element = "<data xmlns='urn:xmpp:avatar:data'>AAAA</data>"
|
||
.parse()
|
||
.unwrap();
|
||
let data = Data::try_from(elem).unwrap();
|
||
assert_eq!(data.data, b"\0\0\0");
|
||
}
|
||
|
||
#[cfg(not(feature = "disable-validation"))]
|
||
#[test]
|
||
fn test_invalid() {
|
||
let elem: Element = "<data xmlns='urn:xmpp:avatar:data' id='coucou'/>"
|
||
.parse()
|
||
.unwrap();
|
||
let error = Data::try_from(elem).unwrap_err();
|
||
let message = match error {
|
||
Error::ParseError(string) => string,
|
||
_ => panic!(),
|
||
};
|
||
assert_eq!(message, "Unknown attribute in data element.")
|
||
}
|
||
}
|