use skyscraper::{
html::{self, trim_internal_whitespace},
xpath,
};
#[test]
fn leading_slash_should_select_html_node() {
// arrange
let text = r###"
"###;
let document = html::parse(&text).unwrap();
let xpath_item_tree = xpath::XpathItemTree::from(&document);
let xpath = xpath::parse("/html").unwrap();
// act
let nodes = xpath.apply(&xpath_item_tree).unwrap();
// assert
assert_eq!(nodes.len(), 1);
let mut nodes = nodes.into_iter();
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "html");
}
}
#[test]
fn leading_double_slash_should_select_all() {
// arrange
let text = r###"
"###;
let document = html::parse(&text).unwrap();
let xpath_item_tree = xpath::XpathItemTree::from(&document);
let xpath = xpath::parse("//p").unwrap();
// act
let nodes = xpath.apply(&xpath_item_tree).unwrap();
// assert
assert_eq!(nodes.len(), 6);
let mut nodes = nodes.into_iter();
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("1".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("2".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("3".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("4".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("5".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("6".to_string()));
}
}
#[test]
fn double_slash_should_select_all() {
// arrange
let text = r###"
"###;
let document = html::parse(&text).unwrap();
let xpath_item_tree = xpath::XpathItemTree::from(&document);
let xpath = xpath::parse("/html/footer/div//p").unwrap();
// act
let nodes = xpath.apply(&xpath_item_tree).unwrap();
// assert
assert_eq!(nodes.len(), 3);
let mut nodes = nodes.into_iter();
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("4".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("5".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element = tree_node.extract_as_element_node();
assert_eq!(element.name, "p");
assert_eq!(tree_node.text(&xpath_item_tree), Some("6".to_string()));
}
}
#[test]
fn document_order_preserved_in_results() {
// arrange
let text = r###"
1
2
3
"###;
let document = html::parse(&text).unwrap();
let xpath_item_tree = xpath::XpathItemTree::from(&document);
let xpath = xpath::parse("/html/body//span").unwrap();
// act
let nodes = xpath.apply(&xpath_item_tree).unwrap();
// assert
assert_eq!(nodes.len(), 3);
let mut nodes = nodes.into_iter();
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element_node = tree_node.extract_as_element_node();
assert_eq!(element_node.name, "span");
assert_eq!(tree_node.text(&xpath_item_tree), Some("1".to_string()));
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element_node = tree_node.extract_as_element_node();
assert_eq!(element_node.name, "span");
assert_eq!(
trim_internal_whitespace(&tree_node.text(&xpath_item_tree).unwrap()),
"2"
);
}
// assert node
{
let tree_node = nodes.next().unwrap().extract_into_node();
let element_node = tree_node.extract_as_element_node();
assert_eq!(element_node.name, "span");
assert_eq!(
trim_internal_whitespace(&tree_node.text(&xpath_item_tree).unwrap()),
"3"
);
}
}