about summary refs log tree commit diff stats
path: root/crates/yt_dlp/src/tests.rs
blob: b48deb4521aac2d45a3f1b16517c40dd4d0603f0 (plain) (blame)
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
// yt - A fully featured command line YouTube client
//
// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
// SPDX-License-Identifier: GPL-3.0-or-later
//
// This file is part of Yt.
//
// You should have received a copy of the License along with this program.
// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.

use std::sync::LazyLock;

use serde_json::{json, Value};
use url::Url;

static YT_OPTS: LazyLock<serde_json::Map<String, Value>> = LazyLock::new(|| {
    match json!({
        "playliststart": 1,
        "playlistend": 10,
        "noplaylist": false,
        "extract_flat": false,
    }) {
        Value::Object(obj) => obj,
        _ => unreachable!("This json is hardcoded"),
    }
});

#[tokio::test]
async fn test_extract_info_video() {
    let info = crate::extract_info(
        &YT_OPTS,
        &Url::parse("https://www.youtube.com/watch?v=dbjPnXaacAU").expect("Is valid."),
        false,
        false,
    )
    .await
    .map_err(|err| format!("Encountered error: '{err}'"))
    .unwrap();

    println!("{info:#?}");
}

#[tokio::test]
async fn test_extract_info_url() {
    let err = crate::extract_info(
        &YT_OPTS,
        &Url::parse("https://google.com").expect("Is valid."),
        false,
        false,
    )
    .await
    .map_err(|err| format!("Encountered error: '{err}'"))
    .unwrap();

    println!("{err:#?}");
}

#[tokio::test]
async fn test_extract_info_playlist() {
    let err = crate::extract_info(
        &YT_OPTS,
        &Url::parse("https://www.youtube.com/@TheGarriFrischer/videos").expect("Is valid."),
        false,
        true,
    )
    .await
    .map_err(|err| format!("Encountered error: '{err}'"))
    .unwrap();

    println!("{err:#?}");
}
#[tokio::test]
async fn test_extract_info_playlist_full() {
    let err = crate::extract_info(
        &YT_OPTS,
        &Url::parse("https://www.youtube.com/@NixOS-Foundation/videos").expect("Is valid."),
        false,
        true,
    )
    .await
    .map_err(|err| format!("Encountered error: '{err}'"))
    .unwrap();

    println!("{err:#?}");
}