jl6 ^ d dl mZ d dlmZ d dlZd dlmZmZmZm Z d dl
m Z G d de Z y) )TestCase) MagicMockN)YouTubeTranscriptApiVideoUnavailableFetchedTranscriptFetchedTranscriptSnippet)YouTubeTranscriptClic e Zd Zd Zd Zd Zd Zd Zd Zd Z d Z
d Z d
Z d Z
d Zd
Zd Zd Zd Zd Zd Zd Zd Zd Zd Zy)TestYouTubeTranscriptClic p t | _ t t t ddd t ddd t ddd gd
d d d
| j _ t | j | j _ t | _ t | j | j _ t | j | j _ t | j | j _ t d t _ t | j t _ y )NzHey, this is just a testg gp=
ף?)textstartdurationz*this is not the original transcriptgp=
ף@z-just something shorter, I made up for testingg@gx @EnglishenT GJLlxj_dtq8)snippetslanguage
language_code is_generatedvideo_id) return_value)
r transcript_mockr r fetch translatetranscript_list_mockfind_generated_transcript find_manually_created_transcriptfind_transcriptr __init__listselfs e/root/aria/tools/markitdown-venv/lib/python3.12/site-packages/youtube_transcript_api/test/test_cli.pysetUpz TestYouTubeTranscriptCli.setUp s ({ %.*, 7 !!%
- I "!%
- L !!&" # " !&+&
"2 *3@T@T)U &$-K!>G --?
!!; FO --F
!!B 5> --5
!!1 )2t(D %$-4;T;T$U ! c
t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t d j j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t d
j j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t d
j j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d | j |j d y )Nz%v1 v2 --format json --languages de env1v2jsonder %v1 v2 --languages de en --format jsonz& --format json v1 v2 --languages de enz{v1 v2 --languages de en --format json --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:port http://user:pass@domain:port https://user:pass@domain:portzkv1 v2 --languages de en --format json --webshare-proxy-username username --webshare-proxy-password passwordusernamepasswordzOv1 v2 --languages de en --format json --http-proxy http://user:pass@domain:portzQv1 v2 --languages de en --format json --https-proxy https://user:pass@domain:port) r split _parse_args assertEqual video_idsformat languages
http_proxy https_proxywebshare_proxy_usernamewebshare_proxy_passwordr# parsed_argss r$ test_argument_parsingz.TestYouTubeTranscriptCli.test_argument_parsing; s2 * 3 9 9 ;
+-
..t = ++V4 ..t = //4 00"5* 3 9 9 ;
+-
..t = ++V4 ..t = //4 00"5* 4 : : <
+-
..t = ++V4 ..t = //4 00"5*
::?%'
+-
..t = ++V4 ..t = //1OP 002QR*
116
+-
..t = ++V4 ..t = <TestYouTubeTranscriptCli.test_argument_parsing__only_video_ids sc *7==?;GGI ..t = ++X6 ..7r& c t dj j }| j |j g d | j |j
d | j |j dg y )Nz\-v1 \-\-v2 \--v3)z-v1z--v2z--v3rA r rB r< s r$ 3test_argument_parsing__video_ids_starting_with_dashzLTestYouTubeTranscriptCli.test_argument_parsing__video_ids_starting_with_dash sd *+?+E+E+GHTTV ..0GH ++X6 ..7r& c | j t 5 t dj j d d d y # 1 sw Y y xY w)Nz
--format json) assertRaises
SystemExitr r2 r3 r" s r$ -test_argument_parsing__fail_without_video_idszFTestYouTubeTranscriptCli.test_argument_parsing__fail_without_video_ids sC
z
* H !6!6!8 9 E E G H H Hs (AAc t dj j }| j |j ddg | j |j
d | j |j dg t dj j }| j |j ddg | j |j
d | j |j dg y )Nzv1 v2 --format jsonr( r) r* r z--format json v1 v2rB r< s r$ test_argument_parsing__jsonz4TestYouTubeTranscriptCli.test_argument_parsing__json s *+@+F+F+HIUUW ..t = ++V4 ..7*+@+F+F+HIUUW ..t = ++V4 ..7r& c t dj j }| j |j ddg | j |j
d | j |j ddg y )Nv1 v2 --languages de enr( r) rA r+ r rB r< s r$ test_argument_parsing__languagesz9TestYouTubeTranscriptCli.test_argument_parsing__languages sn * % + + -
+-
..t = ++X6 ..t =r& c t dj j }| j |j d t dj j }| j |j
d t dj j }| j |j d | j |j
d t dj j }| j |j d | j |j
d y )Nz/v1 v2 --http-proxy http://user:pass@domain:portr. z1v1 v2 --https-proxy https://user:pass@domain:portr/ z[v1 v2 --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:portr@ r, )r r2 r3 r4 r8 r9 r< s r$ test_argument_parsing__proxiesz7TestYouTubeTranscriptCli.test_argument_parsing__proxies s * = C C E
+-
//1OP* ? E E G
+-
002QR* i o o q
+-
//1OP 002QR*7==?;GGI //4 00"5r& c t dj j }| j |j ddg | j |j t dj j }| j |j ddg | j |j y )N--list-transcripts v1 v2r( r) zv1 v2 --list-transcripts)r r2 r3 r4 r5
assertTruelist_transcriptsr< s r$ 'test_argument_parsing__list_transcriptsz@TestYouTubeTranscriptCli.test_argument_parsing__list_transcripts s * & , , .
+-
..t = 445* & , , .
+-
..t = 445r& c p t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d t dj j }| j |j ddg | j |j
d | j |j ddg | j |j d y ) N&v1 v2 --languages de en --translate czr( r) rA r+ r czz&v1 v2 --translate cz --languages de en)r r2 r3 r4 r5 r6 r7 r r< s r$ test_argument_parsing__translatez9TestYouTubeTranscriptCli.test_argument_parsing__translate s * 4 : : <
+-
..t = ++X6 ..t = ..5* 4 : : <
+-
..t = ++X6 ..t = ..5r& c t dj j }| j |j ddg | j |j | j |j t dj j }| j |j ddg | j |j | j |j t dj j }| j |j ddg | j |j | j |j y )Nz v1 v2 --exclude-manually-createdr( r) zv1 v2 --exclude-generatedz4v1 v2 --exclude-manually-created --exclude-generated) r r2 r3 r4 r5 rS exclude_manually_created assertFalseexclude_generatedr< s r$ ,test_argument_parsing__manually_or_generatedzETestYouTubeTranscriptCli.test_argument_parsing__manually_or_generated s& * . 4 4 6
+-
..t = <<= 667* ' - - /
+-
..t = ==> 556* B H H J
+-
..t = <<= 556r& c t dj j t j j d t j j d | j j j ddg y )NrM r( r) r+ r )r r2 runr r! assert_any_callr r r" s r$ test_runz!TestYouTubeTranscriptCli.test_run sc 6 < < >?CCE !!11$7 !!11$7 !!11AA4,Or& c t t d t _ t dj j
}| j |t t d y )Nr ) side_effectzv1 --languages de en) r r r r! r r2 r` r4 strr# outputs r$ test_run__failing_transcriptsz6TestYouTubeTranscriptCli.test_run__failing_transcripts sN $-:J::V$W !%&<&B&B&DEIIK %5j%A!BCr& c t dj j | j j j ddg y )Nz+v1 v2 --languages de en --exclude-generatedr+ r )r r2 r` r r ra r" s r$ test_run__exclude_generatedz4TestYouTubeTranscriptCli.test_run__exclude_generated s> 9 ? ? A
#% !!BBRR
4L
r& c t dj j | j j j ddg y )Nz2v1 v2 --languages de en --exclude-manually-createdr+ r )r r2 r` r r ra r" s r$ "test_run__exclude_manually_createdz;TestYouTubeTranscriptCli.test_run__exclude_manually_created s> @ F F H
#% !!;;KK
4L
r& c r | j t dj j d y )NzFv1 v2 --languages de en --exclude-manually-created --exclude-generatedr, )r4 r r2 r` r" s r$ 0test_run__exclude_manually_created_and_generatedzITestYouTubeTranscriptCli.test_run__exclude_manually_created_and_generated s. X^^`
ce
r& c t dj j f | j j j d y )NrW rX )r r2 r` r r ra r" s r$ test_run__translatez,TestYouTubeTranscriptCli.test_run__translate s: F L L N O S S UW &&66t>EEII
\* 44jA 44jAr& c J t dj j t j j
j j d }| j | | j |j d | j |j d y )Nzmv1 v2 --languages de en --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:portrw r. r/ ) r r2 r` r r rx ry rz r{ r4 http_url https_urlr~ s r$ test_run__generic_proxy_configz7TestYouTubeTranscriptCli.test_run__generic_proxy_config8 s > eg
#%+44>>EEII
\* ..0NO //1PQr& c t dj j t j j d d y )Nz.v1 v2 --languages de en --cookies blahblah.txtz blahblah.txt)rw cookie_path)r r2 r` r r ra r" s r$ test_run__cookiesz*TestYouTubeTranscriptCli.test_run__cookiesI s<
@ G G I
#% %%55 & 6
r& N)__name__
__module__ __qualname__r% r> rC rE rI rK rN rP rU rY r^ rb rh rj rl rn rp rr ru r r r r& r$ r r sy )VVB5H8 8 H 8>6* 66"7,PD
=
8 B"R"
r& r )
unittestr
unittest.mockr r* youtube_transcript_apir r r r youtube_transcript_api._clir r r r& r$ r s+ # =B
x B
r&