fix: search for sensitive topics

This commit is contained in:
ThetaDev 2023-12-21 21:09:09 +01:00
parent 7dc47b1090
commit 31a8fcf2fb
No known key found for this signature in database
GPG key ID: 649CA4EBDC338394
3 changed files with 47 additions and 28 deletions

View file

@ -754,7 +754,7 @@ fn get_video_details_no_desc(rp: RustyPipe) {
assert_eq!(details.id, "VYJNSQ_ANyA");
assert_eq!(details.name, "Cricket comedy by Modi");
assert_eq!(details.channel.id, "UC8gBy2lByHxIyoPMglerNWg");
assert_eq!(details.channel.name, "TMP Shorts");
assert_eq!(details.channel.name, "TMP Mix Veg");
assert!(!details.channel.avatar.is_empty(), "no channel avatars");
assert!(
@ -1256,6 +1256,24 @@ fn search_empty(rp: RustyPipe) {
assert!(result.items.is_empty());
}
#[rstest]
#[case::no_filter(false)]
#[case::filter(true)]
fn search_sensitive(rp: RustyPipe, #[case] filter: bool) {
let q = "suicide";
let result = if filter {
tokio_test::block_on(
rp.query()
.search_filter::<YouTubeItem, _>(q, &search_filter::SearchFilter::new()),
)
} else {
tokio_test::block_on(rp.query().search::<YouTubeItem, _>(q))
}
.unwrap();
assert_gte(result.items.count.unwrap(), 10_000, "results");
assert_next(result.items, rp.query(), 10, 2);
}
#[rstest]
fn search_suggestion(rp: RustyPipe) {
let result = tokio_test::block_on(rp.query().search_suggestion("hunger ga")).unwrap();