pep/xmpp-rs-mirror
1
0
Fork 0
mirror of https://gitlab.com/xmpp-rs/xmpp-rs.git synced 2024-07-12 22:21:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

minidom: Make ElementBuilder::prefix fail on adding duplicate prefix

Browse source 
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2020-03-29 21:35:53 +02:00
parent 770dff7cb0
commit 937e2380b9
3 changed files with 21 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
minidom-rs/src/element.rs
View file

@ -194,7 +194,7 @@ impl Element {
name.into(),
namespace.into(),
None,
BTreeMap::new(),
None,
BTreeMap::new(),
Vec::new(),
)
@ -987,10 +987,12 @@ pub struct ElementBuilder {
impl ElementBuilder {
/// Sets a custom prefix. It is not possible to set the same prefix twice.
pub fn prefix<S: Into<Namespace>>(mut self, prefix: Prefix, namespace: S) -> ElementBuilder {
// TODO: Make this actually send back an error when a duplicate prefix is added.
pub fn prefix<S: Into<Namespace>>(mut self, prefix: Prefix, namespace: S) -> Result<ElementBuilder> {
if let Some(_) = self.root.prefixes.get(&prefix) {
return Err(Error::DuplicatePrefix);
}
self.root.prefixes.insert(prefix, namespace.into());
self
Ok(self)
}
/// Sets an attribute.

5
minidom-rs/src/error.rs
View file

@ -44,6 +44,9 @@ pub enum Error {
/// An error which is returned when a comment is to be parsed by minidom
NoComments,
/// An error which is returned when a prefixed is defined twice
DuplicatePrefix,
}
impl StdError for Error {
@ -58,6 +61,7 @@ impl StdError for Error {
Error::InvalidPrefix => None,
Error::MissingNamespace => None,
Error::NoComments => None,
Error::DuplicatePrefix => None,
}
}
}
@ -84,6 +88,7 @@ impl std::fmt::Display for Error {
fmt,
"a comment has been found even though comments are forbidden"
),
Error::DuplicatePrefix => write!(fmt, "the prefix is already defined"),
}
}
}

20
minidom-rs/src/tests.rs
View file

@ -77,8 +77,8 @@ fn test_real_data() {
.append(correction)
.build();
let stream = Element::builder("stream", "http://etherx.jabber.org/streams")
.prefix(Some(String::from("stream")), "http://etherx.jabber.org/streams")
.prefix(None, "jabber:client")
.prefix(Some(String::from("stream")), "http://etherx.jabber.org/streams").unwrap()
.prefix(None, "jabber:client").unwrap()
.append(message)
.build();
println!("{}", String::from(&stream));
@ -109,8 +109,8 @@ fn test_real_data() {
.append(pubsub)
.build();
let stream = Element::builder("stream", "http://etherx.jabber.org/streams")
.prefix(Some(String::from("stream")), "http://etherx.jabber.org/streams")
.prefix(None, "jabber:client")
.prefix(Some(String::from("stream")), "http://etherx.jabber.org/streams").unwrap()
.prefix(None, "jabber:client").unwrap()
.append(iq)
.build();
@ -141,8 +141,8 @@ fn writer_with_decl_works() {
#[test]
fn writer_with_prefix() {
let root = Element::builder("root", "ns1")
.prefix(Some(String::from("p1")), "ns1")
.prefix(None, "ns2")
.prefix(Some(String::from("p1")), "ns1").unwrap()
.prefix(None, "ns2").unwrap()
.build();
assert_eq!(String::from(&root),
r#"<p1:root xmlns="ns2" xmlns:p1="ns1"/>"#,
@ -168,7 +168,7 @@ fn writer_no_prefix_namespace_child() {
assert_eq!(String::from(&root), r#"<root xmlns="ns1"><child/></root>"#);
let child = Element::builder("child", "ns2")
.prefix(None, "ns3")
.prefix(None, "ns3").unwrap()
.build();
let root = Element::builder("root", "ns1")
.append(child)
@ -181,7 +181,7 @@ fn writer_no_prefix_namespace_child() {
fn writer_prefix_namespace_child() {
let child = Element::builder("child", "ns1").build();
let root = Element::builder("root", "ns1")
.prefix(Some(String::from("p1")), "ns1")
.prefix(Some(String::from("p1")), "ns1").unwrap()
.append(child)
.build();
assert_eq!(String::from(&root), r#"<p1:root xmlns:p1="ns1"><p1:child/></p1:root>"#);
@ -193,8 +193,8 @@ fn writer_with_prefix_deduplicate() {
// .prefix(Some(String::from("p1")), "ns1")
.build();
let root = Element::builder("root", "ns1")
.prefix(Some(String::from("p1")), "ns1")
.prefix(None, "ns2")
.prefix(Some(String::from("p1")), "ns1").unwrap()
.prefix(None, "ns2").unwrap()
.append(child)
.build();
assert_eq!(String::from(&root),