97 lines
3.1 KiB
Rust
97 lines
3.1 KiB
Rust
|
use std::io::Cursor;
|
||
|
|
||
|
use std::iter::Iterator;
|
||
|
|
||
|
use xml::reader::EventReader;
|
||
|
use xml::writer::EventWriter;
|
||
|
|
||
|
use element::Element;
|
||
|
|
||
|
const TEST_STRING: &'static str = r#"<?xml version="1.0" encoding="utf-8"?><root xmlns="root_ns" a="b">meow<child c="d" /><child xmlns="child_ns" d="e" />nya</root>"#;
|
||
|
|
||
|
fn build_test_tree() -> Element {
|
||
|
let mut root = Element::builder("root")
|
||
|
.ns("root_ns")
|
||
|
.attr("a", "b")
|
||
|
.build();
|
||
|
root.append_text_node("meow");
|
||
|
let child = Element::builder("child")
|
||
|
.attr("c", "d")
|
||
|
.build();
|
||
|
root.append_child(child);
|
||
|
let other_child = Element::builder("child")
|
||
|
.ns("child_ns")
|
||
|
.attr("d", "e")
|
||
|
.build();
|
||
|
root.append_child(other_child);
|
||
|
root.append_text_node("nya");
|
||
|
root
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn reader_works() {
|
||
|
let mut reader = EventReader::new(Cursor::new(TEST_STRING));
|
||
|
assert_eq!(Element::from_reader(&mut reader).unwrap(), build_test_tree());
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn writer_works() {
|
||
|
let root = build_test_tree();
|
||
|
let mut out = Vec::new();
|
||
|
{
|
||
|
let mut writer = EventWriter::new(&mut out);
|
||
|
root.write_to(&mut writer).unwrap();
|
||
|
}
|
||
|
assert_eq!(String::from_utf8(out).unwrap(), TEST_STRING);
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn builder_works() {
|
||
|
let elem = Element::builder("a")
|
||
|
.ns("b")
|
||
|
.attr("c", "d")
|
||
|
.append(Element::builder("child"))
|
||
|
.append("e")
|
||
|
.build();
|
||
|
assert_eq!(elem.name(), "a");
|
||
|
assert_eq!(elem.ns(), Some("b"));
|
||
|
assert_eq!(elem.attr("c"), Some("d"));
|
||
|
assert_eq!(elem.attr("x"), None);
|
||
|
assert_eq!(elem.text(), "e");
|
||
|
assert!(elem.has_child("child", "b"));
|
||
|
assert!(elem.is("a", "b"));
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn children_iter_works() {
|
||
|
let root = build_test_tree();
|
||
|
let mut iter = root.children();
|
||
|
assert!(iter.next().unwrap().is("child", "root_ns"));
|
||
|
assert!(iter.next().unwrap().is("child", "child_ns"));
|
||
|
assert_eq!(iter.next(), None);
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn get_child_works() {
|
||
|
let root = build_test_tree();
|
||
|
assert_eq!(root.get_child("child", "inexistent_ns"), None);
|
||
|
assert_eq!(root.get_child("not_a_child", "root_ns"), None);
|
||
|
assert!(root.get_child("child", "root_ns").unwrap().is("child", "root_ns"));
|
||
|
assert!(root.get_child("child", "child_ns").unwrap().is("child", "child_ns"));
|
||
|
assert_eq!(root.get_child("child", "root_ns").unwrap().attr("c"), Some("d"));
|
||
|
assert_eq!(root.get_child("child", "child_ns").unwrap().attr("d"), Some("e"));
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn namespace_propagation_works() {
|
||
|
let mut root = Element::builder("root").ns("root_ns").build();
|
||
|
let mut child = Element::bare("child");
|
||
|
let grandchild = Element::bare("grandchild");
|
||
|
child.append_child(grandchild);
|
||
|
root.append_child(child);
|
||
|
assert_eq!(root.get_child("child", "root_ns").unwrap().ns(), root.ns());
|
||
|
assert_eq!(root.get_child("child", "root_ns").unwrap()
|
||
|
.get_child("grandchild", "root_ns").unwrap()
|
||
|
.ns(), root.ns());
|
||
|
}
|