yt-transcript-rs 0.1.8

A Rust library for fetching and working with YouTube video transcripts
Documentation 
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

#[allow(unused_imports)]
use crate::proxies::*;

#[test]
fn test_generic_proxy_config() {
    // Test with both HTTP and HTTPS
    let proxy_config = GenericProxyConfig::new(
        Some("http://proxy.example.com:8080".to_string()),
        Some("https://secure-proxy.example.com:8443".to_string()),
    )
    .unwrap();

    let requests_dict = proxy_config.to_requests_dict();

    assert_eq!(
        requests_dict.get("http").unwrap(),
        "http://proxy.example.com:8080"
    );
    assert_eq!(
        requests_dict.get("https").unwrap(),
        "https://secure-proxy.example.com:8443"
    );

    // By default, keep-alive should be enabled
    assert!(!proxy_config.prevent_keeping_connections_alive());

    // By default, retries should be 0
    assert_eq!(proxy_config.retries_when_blocked(), 0);
}

#[test]
fn test_generic_proxy_config_http_only() {
    // Test with HTTP only
    let proxy_config =
        GenericProxyConfig::new(Some("http://proxy.example.com:8080".to_string()), None).unwrap();

    let requests_dict = proxy_config.to_requests_dict();

    assert_eq!(
        requests_dict.get("http").unwrap(),
        "http://proxy.example.com:8080"
    );
    assert_eq!(
        requests_dict.get("https").unwrap(),
        "http://proxy.example.com:8080"
    );
}

#[test]
fn test_generic_proxy_config_https_only() {
    // Test with HTTPS only
    let proxy_config = GenericProxyConfig::new(
        None,
        Some("https://secure-proxy.example.com:8443".to_string()),
    )
    .unwrap();

    let requests_dict = proxy_config.to_requests_dict();

    assert_eq!(
        requests_dict.get("http").unwrap(),
        "https://secure-proxy.example.com:8443"
    );
    assert_eq!(
        requests_dict.get("https").unwrap(),
        "https://secure-proxy.example.com:8443"
    );
}

#[test]
fn test_generic_proxy_config_invalid() {
    // Test with neither HTTP nor HTTPS
    let proxy_config = GenericProxyConfig::new(None, None);
    assert!(proxy_config.is_err());

    let error = proxy_config.unwrap_err();
    assert!(format!("{}", error).contains("requires you to define at least one"));
}

#[test]
fn test_webshare_proxy_config() {
    // Test with default values
    let proxy_config =
        WebshareProxyConfig::new("user123".to_string(), "pass456".to_string(), 5, None, None);

    let requests_dict = proxy_config.to_requests_dict();

    // Check that both HTTP and HTTPS are set to the same URL
    let expected_url = "http://user123-rotate:pass456@p.webshare.io:80/";
    assert_eq!(requests_dict.get("http").unwrap(), expected_url);
    assert_eq!(requests_dict.get("https").unwrap(), expected_url);

    // Check URL generation
    assert_eq!(proxy_config.url(), expected_url);

    // Keep-alive should be disabled
    assert!(proxy_config.prevent_keeping_connections_alive());

    // Retries should be set to 5
    assert_eq!(proxy_config.retries_when_blocked(), 5);
}

#[test]
fn test_webshare_proxy_config_custom() {
    // Test with custom domain and port
    let proxy_config = WebshareProxyConfig::new(
        "user123".to_string(),
        "pass456".to_string(),
        10,
        Some("custom.webshare.io".to_string()),
        Some(8080),
    );

    let requests_dict = proxy_config.to_requests_dict();

    // Check that both HTTP and HTTPS are set to the same URL
    let expected_url = "http://user123-rotate:pass456@custom.webshare.io:8080/";
    assert_eq!(requests_dict.get("http").unwrap(), expected_url);
    assert_eq!(requests_dict.get("https").unwrap(), expected_url);

    // Check URL generation
    assert_eq!(proxy_config.url(), expected_url);

    // Retries should be set to 10
    assert_eq!(proxy_config.retries_when_blocked(), 10);
}