From 562b62a0b638d2ca0731f83476a2e5f74757962a Mon Sep 17 00:00:00 2001 From: Sven-Hendrik Haase Date: Sat, 13 Jan 2024 06:11:59 +0100 Subject: Clean up default order function I removed the stringly typing as we already have enums for this that we can make use of. --- tests/utils/mod.rs | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'tests/utils/mod.rs') diff --git a/tests/utils/mod.rs b/tests/utils/mod.rs index 64433fc..baffc29 100644 --- a/tests/utils/mod.rs +++ b/tests/utils/mod.rs @@ -13,16 +13,16 @@ pub fn get_link_from_text(document: &Document, text: &str) -> Option { 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 { +/// Return the href attributes of all links that start with the specified `prefix`. +pub fn get_link_hrefs_with_prefix(document: &Document, prefix: &str) -> Vec { let mut vec: Vec = Vec::new(); - let a_elem = document.find(Name("a")); + let a_elements = 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()); + for element in a_elements { + let s = element.attr("href").unwrap_or(""); + if s.to_string().starts_with(prefix) { + vec.push(s.to_string()); } } -- cgit v1.2.3