Add prefix support to decoder to accept xml:lang in presence statuses.

This commit is contained in:
O01eg 2019-01-08 13:41:39 +03:00
parent 7bb4bd1094
commit 983078f120
No known key found for this signature in database
GPG key ID: 8FBA74B3E78B4677

View file

@ -104,10 +104,14 @@ impl ParserSink {
"xmlns" => (), "xmlns" => (),
_ if is_prefix_xmlns(attr) => (), _ if is_prefix_xmlns(attr) => (),
_ => { _ => {
if let Some(ref prefix) = attr.name.prefix {
el_builder = el_builder.attr(format!("{}:{}", prefix, attr.name.local), attr.value.as_ref());
} else {
el_builder = el_builder.attr(attr.name.local.as_ref(), attr.value.as_ref()); el_builder = el_builder.attr(attr.name.local.as_ref(), attr.value.as_ref());
} }
} }
} }
}
el_builder.build() el_builder.build()
}; };
@ -428,6 +432,28 @@ mod tests {
}); });
} }
/// test case for https://gitlab.com/xmpp-rs/tokio-xmpp/issues/3
#[test]
fn test_atrribute_prefix() {
let mut c = XMPPCodec::new();
let mut b = BytesMut::with_capacity(1024);
b.put(r"<?xml version='1.0'?><stream:stream xmlns:stream='http://etherx.jabber.org/streams' version='1.0' xmlns='jabber:client'>");
let r = c.decode(&mut b);
assert!(match r {
Ok(Some(Packet::StreamStart(_))) => true,
_ => false,
});
b.clear();
b.put(r"<status xml:lang='en'>Test status</status>");
let r = c.decode(&mut b);
assert!(match r {
Ok(Some(Packet::Stanza(ref el))) if el.name() == "status" && el.text() == "Test status" && el.attr("xml:lang").map_or(false, |a| a == "en") => true,
_ => false,
});
}
/// By default, encode() only get's a BytesMut that has 8kb space reserved. /// By default, encode() only get's a BytesMut that has 8kb space reserved.
#[test] #[test]
fn test_large_stanza() { fn test_large_stanza() {