use crate::html::{ get_node_name, get_parent_node, html_to_dom, is_icon, stringify_document, walk_and_embed_assets, }; use html5ever::rcdom::{Handle, NodeData}; use html5ever::serialize::{serialize, SerializeOpts}; #[test] fn test_is_icon() { assert_eq!(is_icon("icon"), true); assert_eq!(is_icon("Shortcut Icon"), true); assert_eq!(is_icon("ICON"), true); assert_eq!(is_icon("mask-icon"), true); assert_eq!(is_icon("fluid-icon"), true); assert_eq!(is_icon("stylesheet"), false); assert_eq!(is_icon(""), false); } #[test] fn test_get_parent_node_name() { let html = "

"; let dom = html_to_dom(&html); let mut count = 0; fn test_walk(node: &Handle, i: &mut i8) { *i += 1; match &node.data { NodeData::Document => { for child in node.children.borrow().iter() { test_walk(child, &mut *i); } } NodeData::Element { ref name, .. } => { let node_name = name.local.as_ref().to_string(); let parent_node_name = get_node_name(&get_parent_node(node)); if node_name == "head" || node_name == "body" { assert_eq!(parent_node_name, "html"); } else if node_name == "div" { assert_eq!(parent_node_name, "body"); } else if node_name == "p" { assert_eq!(parent_node_name, "div"); } println!("{}", node_name); for child in node.children.borrow().iter() { test_walk(child, &mut *i); } } _ => (), }; } test_walk(&dom.document, &mut count); assert_eq!(count, 7); } #[test] fn test_walk_and_embed_assets() { let html = "

"; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "

" ); } #[test] fn test_walk_and_embed_assets_ensure_no_recursive_iframe() { let html = "

"; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "

" ); } #[test] fn test_walk_and_embed_assets_no_css() { let html = "\ \
"; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = true; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "\ \ \ \ \ \
\ \ " ); } #[test] fn test_walk_and_embed_assets_no_images() { let html = "\
"; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = true; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "\ \ \ \ \
\ \
\ \ " ); } #[test] fn test_walk_and_embed_assets_no_frames() { let html = ""; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = false; let opt_no_frames: bool = true; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "" ); } #[test] fn test_walk_and_embed_assets_no_js() { let html = "
\ \ \
"; let dom = html_to_dom(&html); let url = "http://localhost"; let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = true; let opt_no_images: bool = false; let opt_silent = true; let opt_insecure = false; walk_and_embed_assets( &url, &dom.document, opt_no_css, opt_no_js, opt_no_images, "", opt_silent, opt_insecure, opt_no_frames, ); let mut buf: Vec = Vec::new(); serialize(&mut buf, &dom.document, SerializeOpts::default()).unwrap(); assert_eq!( buf.iter().map(|&c| c as char).collect::(), "
\
" ); } #[test] fn test_stringify_document() { let html = "
"; let dom = html_to_dom(&html); let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_isolate: bool = false; assert_eq!( stringify_document( &dom.document, opt_no_css, opt_no_frames, opt_no_js, opt_no_images, opt_isolate, ), "
" ); } #[test] fn test_stringify_document_isolate() { let html = "Isolated document\ \ \
"; let dom = html_to_dom(&html); let opt_no_css: bool = false; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_isolate: bool = true; assert_eq!( stringify_document( &dom.document, opt_no_css, opt_no_frames, opt_no_js, opt_no_images, opt_isolate, ), "\ \ \ Isolated document\ \ \ \ \
\ \
\ \ " ); } #[test] fn test_stringify_document_no_css() { let html = "\ Unstyled document\ \
"; let dom = html_to_dom(&html); let opt_no_css: bool = true; let opt_no_frames: bool = false; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_isolate: bool = false; assert_eq!( stringify_document( &dom.document, opt_no_css, opt_no_frames, opt_no_js, opt_no_images, opt_isolate, ), "\ \ \ \ Unstyled document\ \ \
\ " ); } #[test] fn test_stringify_document_no_frames() { let html = "\ Frameless document\ \
"; let dom = html_to_dom(&html); let opt_no_css: bool = false; let opt_no_frames: bool = true; let opt_no_js: bool = false; let opt_no_images: bool = false; let opt_isolate: bool = false; assert_eq!( stringify_document( &dom.document, opt_no_css, opt_no_frames, opt_no_js, opt_no_images, opt_isolate, ), "\ \ \ \ Frameless document\ \ \
\ " ); } #[test] fn test_stringify_document_isolate_no_frames_no_js_no_css_no_images() { let html = "\ no-frame no-css no-js no-image isolated document\ \ \
\ \ \ \
"; let dom = html_to_dom(&html); let opt_isolate: bool = true; let opt_no_css: bool = true; let opt_no_frames: bool = true; let opt_no_js: bool = true; let opt_no_images: bool = true; assert_eq!( stringify_document( &dom.document, opt_no_css, opt_no_frames, opt_no_js, opt_no_images, opt_isolate, ), "\ \ \ \ no-frame no-css no-js no-image isolated document\ \ \ \ \
\ \ \ \
\ \ " ); }