5bf14b0b22
This bumps the minimum supported stable Rust version to 1.34. The TryFrom and TryInto traits are still reexported to not break the API, but these reexports are deprecated and will be removed in a future release.
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: Required<u16> = "bytes",
|
||
|
||
/// The width of the image in pixels.
|
||
width: Option<u16> = "width",
|
||
|
||
/// The height of the image in pixels.
|
||
height: Option<u16> = "height",
|
||
|
||
/// The SHA-1 hash of the image data for the specified content-type.
|
||
id: Required<Sha1HexAttribute> = "id",
|
||
|
||
/// The IANA-registered content type of the image data.
|
||
type_: Required<String> = "type",
|
||
|
||
/// The http: or https: URL at which the image data file is hosted.
|
||
url: Option<String> = "url",
|
||
]
|
||
);
|
||
|
||
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 std::convert::TryFrom;
|
||
|
||
#[cfg(target_pointer_width = "32")]
|
||
#[test]
|
||
fn test_size() {
|
||
assert_size!(Metadata, 12);
|
||
assert_size!(Info, 64);
|
||
assert_size!(Data, 12);
|
||
}
|
||
|
||
#[cfg(target_pointer_width = "64")]
|
||
#[test]
|
||
fn test_size() {
|
||
assert_size!(Metadata, 24);
|
||
assert_size!(Info, 120);
|
||
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.")
|
||
}
|
||
}
|