}i; d dl Z d dlmZmZ d dlmZ d dlmZmZ d dl Z d dl
Z
d dl m Z m
Z
mZmZ d dlmZ G d de Zy) N)PackageNotFoundErrorversion)TestCase) MagicMockpatch)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 Zej2 j5 d d Zd Z d Z y) 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 O/usr/local/lib/python3.12/dist-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--v3rD r rE 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 r5 r6 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 v2rE 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, rD r. r rE 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:portr1 z1v1 v2 --https-proxy https://user:pass@domain:portr2 z[v1 v2 --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:portrC r/ )r r5 r6 r7 r; r< 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 r5 r6 r7 r8
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, rD r. r czz&v1 v2 --translate cz --languages de en)r r5 r6 r7 r8 r9 r: 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 r5 r6 r7 r8 rV 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 )NrP r+ r, r. r )r r5 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 r5 rc r7 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 r5 rc r r! rd 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 r5 rc r r rd 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/ )r7 r r5 rc 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 )NrZ r[ )r r5 rc r r rd 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:portr{ r1 r2 ) r r5 rc r r# r| r} r~ r r7 http_url https_urlr s r' test_run__generic_proxy_configz7TestYouTubeTranscriptCli.test_run__generic_proxy_config; s > eg
#%+44>>EEII
\* ..0NO //1PQr) zfThis test is temporarily disabled because cookie auth is currently not working due to YouTube changes.)reasonc t dj j t j j d d y )Nz.v1 v2 --languages de en --cookies blahblah.txtz blahblah.txt)r{ cookie_path)r r5 rc r r# rd r% s r' test_run__cookiesz*TestYouTubeTranscriptCli.test_run__cookiesL s>
@ G G I
#% %%55 & 6
r) c dt d }t j ddgddd j j }||k( sJ d| d| d y
) zo
`youtube_transcript_api --version` should return the same version as in the package metadata.
z youtube_transcript_api, version zyoutube-transcript-apiyoutube_transcript_apiz --versionT)capture_outputr checkzExpected version 'z ', but got ''N)r
subprocessrc stdoutstrip)r& expected_version_msgcli_version_msgs r' test_version_matches_metadataz6TestYouTubeTranscriptCli.test_version_matches_metadataZ s{
/w7O/P.Q R %..
%{ 3
&
3 3 U
4 5\/ARRS T U 3r) c t dt 5 t g }|j dk( sJ d d d y # 1 sw Y y xY w)Nz#youtube_transcript_api._cli.versionrg unknown)r r r _get_version)r& clis r' "test_get_version_package_not_foundz;TestYouTubeTranscriptCli.test_get_version_package_not_foundm sH
1?S
3 'r*C##%2 22 3 3 3s !=AN) __name__
__module__ __qualname__r( rA rF rH rL rN rQ rS rX r\ ra re rl rn rp rr rt rv ry r r pytestmarkskipr r r r) r' r r s )VVB5H8 8 H 8>6* 66"7,PD
=
8 B"R" [[*
U&3r) r )r importlib.metadatar r unittestr
unittest.mockr r r- r r r r r
r youtube_transcript_api._clir r r r) r' r s4
< * =`3x `3r)