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

split up constants & make them public, add test for emojis in the title

Browse Source
This commit is contained in:
OMGeeky
2023-06-18 00:09:22 +02:00
parent fbc4ec1ecd
commit 507c98b850
2 changed files with 42 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/lib.rs
View File

@@ -760,13 +760,15 @@ pub fn get_video_title_from_twitch_video(
}
/// The maximum length a youtube video can have
const MAX_VIDEO_TITLE_LENGTH: usize = 100;
pub const MAX_VIDEO_TITLE_LENGTH: usize = 100;
/// the length my prefix might be (not necessarily all are used)
const PREFIX_LENGTH: usize = 24;
pub static PREFIX_LENGTH: usize = DATE_PREFIX_LENGTH + PART_PREFIX_LENGTH;
pub const DATE_PREFIX_LENGTH: usize = 12;
pub const PART_PREFIX_LENGTH: usize = 12;
/// The length of the separator between the prefix & the title
const SEPARATOR_LEN: usize = 1;
pub const SEPARATOR_LEN: usize = 1;
/// The length of the ... that gets put at the end if it is shortened
const DOTDOTDOT_LEN: usize = 3;
pub const DOTDOTDOT_LEN: usize = 3;
pub fn get_playlist_title_from_twitch_video(video: &data::VideoData) -> Result<String> {
trace!("get playlist title from twitch video");

37
tests/lib_tests.rs
View File

@@ -9,7 +9,7 @@ use downloader;
use downloader::data::{Streamers, VideoData, VideoMetadata, Videos};
use downloader::{
get_playlist_title_from_twitch_video, get_video_prefix_from_twitch_video,
get_video_title_from_twitch_video,
get_video_title_from_twitch_video, MAX_VIDEO_TITLE_LENGTH, PART_PREFIX_LENGTH,
};
fn init_console_logging(log_level: LevelFilter) {
@@ -75,6 +75,16 @@ const LONG_TITLE: &'static str =
"long title with over a hundred characters that is definitely going to \
be cut of because it does not fit into the maximum length that youtube requires";
const LONG_TITLE_ONLY_EMOJI: &'static str = "🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠\
🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠";
#[tokio::test]
async fn get_video_title() {
init_console_logging(LevelFilter::Debug);
@@ -94,6 +104,31 @@ async fn get_video_long_title() {
info!("part title: {}", title);
assert_eq!(title, "[2021-01-01][Part 05/20] long title with over a hundred characters that is definitely going to be...");
}
#[tokio::test]
async fn get_video_long_title_only_emoji() {
init_console_logging(LevelFilter::Debug);
let client = get_sample_client().await;
let mut video = get_sample_video(&client);
video.video.title = Some(LONG_TITLE_ONLY_EMOJI.to_string());
let title = get_video_title_from_twitch_video(&video, 1, 1).unwrap();
info!("part title: {}", title);
assert_eq!(
MAX_VIDEO_TITLE_LENGTH - PART_PREFIX_LENGTH,
title.chars().count()
); //this is 88 chars long to leave space for the part prefix part, with that it should be exactly 100 chars
assert_eq!(
"[2021-01-01] 🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠...",
title,
);
video.video.title = Some(LONG_TITLE_ONLY_EMOJI.to_string());
let title = get_video_title_from_twitch_video(&video, 5, 20).unwrap();
info!("part title: {}", title);
assert_eq!(MAX_VIDEO_TITLE_LENGTH, title.chars().count());
assert_eq!(
"[2021-01-01][Part 05/20] 🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠🔴🟠...",
title,
);
}
#[tokio::test]
async fn get_video_title_single_part() {