OMGeeky/downloader
1
0
Fork 0
mirror of https://github.com/OMGeeky/downloader.git synced 2026-01-20 02:11:17 +01:00
Code Issues 1 Packages Projects Releases Wiki Activity

fix youtube title prefix bug & split tests into multiple

Browse Source
This commit is contained in:
OMGeeky
2023-04-16 00:39:56 +02:00
parent 3aeaf51240
commit 713ec66ef1
2 changed files with 25 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/lib.rs
View File

@@ -664,7 +664,6 @@ pub fn get_video_description_from_twitch_video(
.clone()
.unwrap_or("<NO DESCRIPTION FOUND>".to_string());
let description = description
.replace("$$video_title$$", &video_title)
.replace("$$video_url$$", &video_url)
@@ -689,7 +688,7 @@ pub fn get_video_title_from_twitch_video(
) -> Result<String> {
trace!("get video title from twitch video");
let prefix = match total_parts {
1 => "".to_string(),
1 => format!("{} ", get_date_string_from_video(&video)?),
_ => format!(
"{} ",
get_video_prefix_from_twitch_video(video, part, total_parts)?

30
tests/lib_tests.rs
View File

@@ -79,11 +79,16 @@ const LONG_TITLE: &'static str =
async fn get_video_title() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
let video = get_sample_video(&client);
let title = get_video_title_from_twitch_video(&video, 5, 20).unwrap();
assert_eq!(title, "[2021-01-01][Part 05/20] Test Video");
}
#[tokio::test]
async fn get_video_long_title() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
video.video.title = Some(LONG_TITLE.to_string());
let title = get_video_title_from_twitch_video(&video, 5, 20).unwrap();
info!("part title: {}", title);
@@ -94,17 +99,23 @@ async fn get_video_title() {
async fn get_video_title_single_part() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
let video = get_sample_video(&client);
let title = get_video_title_from_twitch_video(&video, 1, 1).unwrap();
assert_eq!(title, "Test Video");
assert_eq!(title, "[2021-01-01] Test Video");
}
#[tokio::test]
async fn get_video_long_title_single_part() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
video.video.title = Some(LONG_TITLE.to_string());
let title = get_video_title_from_twitch_video(&video, 1, 1).unwrap();
info!("single part title: {}", title);
assert_eq!(
title,
"long title with over a hundred characters that is definitely going to be..."
"[2021-01-01] long title with over a hundred characters that is definitely going to be..."
);
}
@@ -112,10 +123,17 @@ async fn get_video_title_single_part() {
async fn get_playlist_title() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
let video = get_sample_video(&client);
let title = get_playlist_title_from_twitch_video(&video).unwrap();
assert_eq!("[2021-01-01] Test Video", title);
}
#[tokio::test]
async fn get_playlist_long_title() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
video.video.title = Some(LONG_TITLE.to_string());
let title = get_playlist_title_from_twitch_video(&video).unwrap();