From e5e3ad2fed80bea3f2c9c90f87a405bca077691e Mon Sep 17 00:00:00 2001 From: elandsborough Date: Wed, 10 Jan 2024 19:53:36 +0000 Subject: Set default sorting order and method with arguments --- tests/utils/mod.rs | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'tests/utils/mod.rs') diff --git a/tests/utils/mod.rs b/tests/utils/mod.rs index 734d33b..64433fc 100644 --- a/tests/utils/mod.rs +++ b/tests/utils/mod.rs @@ -12,3 +12,19 @@ pub fn get_link_from_text(document: &Document, text: &str) -> Option { .next()?; Some(a_elem.attr("href")?.to_string()) } + +/// Return the href attributes of all links that start with the specified prefix `text`. +pub fn get_link_hrefs_from_text_with_prefix(document: &Document, text: &str) -> Vec { + let mut vec: Vec = Vec::new(); + + let a_elem = document.find(Name("a")); + + for element in a_elem { + let str = element.attr("href").unwrap_or(""); + if str.to_string().starts_with(text) { + vec.push(str.to_string()); + } + } + + return vec; +} -- cgit v1.2.3