1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
extern crate reqwest;
extern crate scraper;
extern crate select;
mod search;
use search::*;
pub fn search<T: Into<Option<u32>>, U: Into<Option<String>>>(
query: &str,
limit: T,
agent: U,
) -> Result<Vec<SearchResult>, Error> {
let limit: Option<u32> = limit.into();
let agent: Option<String> = agent.into();
google(
query,
limit.unwrap_or(10),
agent.unwrap_or(
"Mozilla/5.0 (X11; Linux x86_64; rv:77.0) Gecko/20100101 Firefox/77.0".to_string(),
),
)
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn search_test() {
let results = search("minecraft", None, None);
if let Ok(result_list) = results {
assert!(result_list[0].title.starts_with("Minecraft"));
assert!(result_list[0].link.len() > 0);
assert!(result_list[0].description.len() > 0);
}
}
#[test]
fn limit_test() {
let results = search("minecraft", 4, None);
if let Ok(result_list) = results {
assert!(result_list.len() <= 4);
}
}
}