o
óT·j; ã @ s| 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 dd„ deƒZdS ) é N)ÚPackageNotFoundErrorÚversion)ÚTestCase)Ú MagicMockÚpatch)ÚYouTubeTranscriptApiÚVideoUnavailableÚFetchedTranscriptÚFetchedTranscriptSnippet)ÚYouTubeTranscriptClic @ sÚ e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd d „ Zd
d„ Z dd„ Z
dd„ Z dd„ Z dd„ Z
dd„ Zdd„ Zdd „ Zd d „ Zd d „ Zd!d"„ Zd#d$„ Zd%d&„ Zd'd(„ Zd)d*„ Zejjd+d,d-d.„ ƒZd/d0„ Z d1d2„ Z d3S )4ÚTestYouTubeTranscriptClic C s® t ƒ | _t ttddddtddddtddd dgd
d d d
dd| j_t | jd| j_t ƒ | _t | jd| j_t | jd| j_t | jd| j_ t d dt
_ t | jdt
_ d S )NzHey, this is just a testg g¤p=
×£ø?)ÚtextÚstartÚdurationz*this is not the original transcriptg¤p=
×£@z-just something shorter, I made up for testinggÍÌÌÌÌÌ@g¶óýÔxé @ÚEnglishÚenTÚ GJLlxj_dtq8)ÚsnippetsÚlanguageÚ
language_codeÚ is_generatedÚvideo_id)Ú return_value)
r Útranscript_mockr r
ÚfetchÚ translateÚtranscript_list_mockÚfind_generated_transcriptÚ find_manually_created_transcriptÚfind_transcriptr Ú__init__Úlist©Úself© r$ úg/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/youtube_transcript_api/test/test_cli.pyÚsetUp sL ýýýõë
ÿ
ÿ
ÿ
ÿ z TestYouTubeTranscriptCli.setUpc C s– t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd¡ | |jd¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd¡ | |jd¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd¡ | |jd¡ t d
¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd ¡ | |jd ¡ t d
¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |j d¡ | |j
d¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd ¡ | |jd¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd ¡ | |jd¡ d S )Nz%v1 v2 --format json --languages de enÚv1Úv2ÚjsonÚder Ú ú%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 passwordÚusernameÚpasswordzOv1 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_idsÚformatÚ languagesÚ
http_proxyÚ https_proxyÚwebshare_proxy_usernameÚwebshare_proxy_password©r# Ú parsed_argsr$ r$ r% Útest_argument_parsing> s ÿþÿþÿþýüýüÿþÿþz.TestYouTubeTranscriptCli.test_argument_parsingc C sD t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jdg¡ d S )Núv1 v2r' r( Úprettyr ©r r1 r2 r3 r4 r5 r6 r; r$ r$ r% Ú%test_argument_parsing__only_video_ids‚ ó z>TestYouTubeTranscriptCli.test_argument_parsing__only_video_idsc C sD t d ¡ ƒ ¡ }| |jg d¢¡ | |jd¡ | |jdg¡ d S )Nz\-v1 \-\-v2 \--v3)z-v1z--v2z--v3r? r r@ r; r$ r$ r% Ú3test_argument_parsing__video_ids_starting_with_dashˆ rB zLTestYouTubeTranscriptCli.test_argument_parsing__video_ids_starting_with_dashc C s@ | t¡ td ¡ ƒ ¡ W d ƒ d S 1 sw Y d S )Nz
--format json)Ú assertRaisesÚ
SystemExitr r1 r2 r" r$ r$ r% Ú-test_argument_parsing__fail_without_video_idsŽ s "ÿzFTestYouTubeTranscriptCli.test_argument_parsing__fail_without_video_idsc C s„ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jdg¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jdg¡ d S )Nzv1 v2 --format jsonr' r( r) r z--format json v1 v2r@ r; r$ r$ r% Útest_argument_parsing__json’ s z4TestYouTubeTranscriptCli.test_argument_parsing__jsonc C sF t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ d S )Núv1 v2 --languages de enr' r( r? r* r r@ r; r$ r$ r% Ú test_argument_parsing__languages s ÿþz9TestYouTubeTranscriptCli.test_argument_parsing__languagesc C s˜ t d ¡ ƒ ¡ }| |jd¡ t d ¡ ƒ ¡ }| |jd¡ t d ¡ ƒ ¡ }| |jd¡ | |jd¡ t d ¡ ƒ ¡ }| |jd¡ | |jd¡ d S )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 r1 r2 r3 r7 r8 r; r$ r$ r% Ú test_argument_parsing__proxies¥ s, ÿþÿþÿþz7TestYouTubeTranscriptCli.test_argument_parsing__proxiesc C s` t d ¡ ƒ ¡ }| |jddg¡ | |j¡ t d ¡ ƒ ¡ }| |jddg¡ | |j¡ d S )Nú--list-transcripts v1 v2r' r( zv1 v2 --list-transcripts)r r1 r2 r3 r4 Ú
assertTrueÚlist_transcriptsr; r$ r$ r% Ú'test_argument_parsing__list_transcriptsº s ÿþ ÿþz@TestYouTubeTranscriptCli.test_argument_parsing__list_transcriptsc C s¤ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd¡ t d ¡ ƒ ¡ }| |jddg¡ | |jd¡ | |jddg¡ | |jd¡ d S ) Nú&v1 v2 --languages de en --translate czr' r( r? r* r Úczz&v1 v2 --translate cz --languages de en)r r1 r2 r3 r4 r5 r6 r r; r$ r$ r% Ú test_argument_parsing__translateÇ s$ ÿþÿþz9TestYouTubeTranscriptCli.test_argument_parsing__translatec C s² t d ¡ ƒ ¡ }| |jddg¡ | |j¡ | |j¡ t d ¡ ƒ ¡ }| |jddg¡ | |j¡ | |j¡ t d ¡ ƒ ¡ }| |jddg¡ | |j¡ | |j¡ d S )Nz v1 v2 --exclude-manually-createdr' r( zv1 v2 --exclude-generatedz4v1 v2 --exclude-manually-created --exclude-generated) r r1 r2 r3 r4 rL Úexclude_manually_createdÚ assertFalseÚexclude_generatedr; r$ r$ r% Ú,test_argument_parsing__manually_or_generatedØ s0 ÿþ ÿþ ÿþ zETestYouTubeTranscriptCli.test_argument_parsing__manually_or_generatedc C s> t d ¡ ƒ ¡ tj d¡ tj d¡ | jj ddg¡ d S )NrH r' r( r* r )r r1 Úrunr r! Úassert_any_callr r r" r$ r$ r% Útest_runî s z!TestYouTubeTranscriptCli.test_runc C s8 t tdƒdt_td ¡ ƒ ¡ }| |ttdƒƒ¡ d S )Nr ©Ú side_effectzv1 --languages de en) r r r r! r r1 rV r3 Ústr©r# Úoutputr$ r$ r% Ú test_run__failing_transcriptsö s z6TestYouTubeTranscriptCli.test_run__failing_transcriptsc C ó& t d ¡ ƒ ¡ | jj ddg¡ d S )Nz+v1 v2 --languages de en --exclude-generatedr* r )r r1 rV r r rW r" r$ r$ r% Útest_run__exclude_generatedý ó ÿÿz4TestYouTubeTranscriptCli.test_run__exclude_generatedc C r_ )Nz2v1 v2 --languages de en --exclude-manually-createdr* r )r r1 rV r r rW r" r$ r$ r% Ú"test_run__exclude_manually_created ra z;TestYouTubeTranscriptCli.test_run__exclude_manually_createdc C s | td ¡ ƒ ¡ d¡ d S )NzFv1 v2 --languages de en --exclude-manually-created --exclude-generatedr+ )r3 r r1 rV r" r$ r$ r% Ú0test_run__exclude_manually_created_and_generated s ÿüzITestYouTubeTranscriptCli.test_run__exclude_manually_created_and_generatedc C s$ t d ¡ ƒ ¡ f | jj d¡ d S )NrO rP )r r1 rV r r rW r" r$ r$ r% Útest_run__translate s z,TestYouTubeTranscriptCli.test_run__translatec C s, t d ¡ ƒ ¡ tj d¡ tj d¡ d S )NrK r' r( )r r1 rV r r! rW r" r$ r$ r% Útest_run__list_transcripts s z3TestYouTubeTranscriptCli.test_run__list_transcriptsc C s t d ¡ ƒ ¡ }t |¡ d S )Nr, )r r1 rV r) Úloadsr\ r$ r$ r% Útest_run__json_output" s ÿþz.TestYouTubeTranscriptCli.test_run__json_outputc C óJ t d ¡ ƒ ¡ tjjj d¡}| |¡ | |j
d¡ | |j d¡ d S )Nz]v1 v2 --languages de en --webshare-proxy-username username --webshare-proxy-password passwordÚ proxy_configr/ r0 ) r r1 rV r r Ú call_argsÚkwargsÚgetÚassertIsNotNoner3 Úproxy_usernameÚproxy_password©r# ri r$ r$ r% Ú test_run__webshare_proxy_config* ó û
ÿ
z8TestYouTubeTranscriptCli.test_run__webshare_proxy_configc C rh )Nzmv1 v2 --languages de en --http-proxy http://user:pass@domain:port --https-proxy https://user:pass@domain:portri r- r. ) r r1 rV r r rj rk rl rm r3 Úhttp_urlÚ https_urlrp r$ r$ r% Ú test_run__generic_proxy_config; rr z7TestYouTubeTranscriptCli.test_run__generic_proxy_configzfThis test is temporarily disabled because cookie auth is currently not working due to YouTube changes.)Úreasonc C s$ t d ¡ ƒ ¡ tjjd dd d S )Nz.v1 v2 --languages de en --cookies blahblah.txtz blahblah.txt)ri Ú cookie_path)r r1 rV r r rW r" r$ r$ r% Útest_run__cookiesL s ÿ
þz*TestYouTubeTranscriptCli.test_run__cookiesc C sL dt dƒ› }tjddgddddj ¡ }||ks$J d|› d|› d ƒ‚d
S ) zo
`youtube_transcript_api --version` should return the same version as in the package metadata.
z youtube_transcript_api, version zyoutube-transcript-apiÚyoutube_transcript_apiz --versionT)Úcapture_outputr
ÚcheckzExpected version 'z ', but got 'ú'N)r Ú
subprocessrV ÚstdoutÚstrip)r# Úexpected_version_msgÚcli_version_msgr$ r$ r% Ú test_version_matches_metadataZ s ÿüû
ÿz6TestYouTubeTranscriptCli.test_version_matches_metadatac C sJ t dtd tg ƒ}| ¡ dksJ ‚W d ƒ d S 1 s w Y d S )Nz#youtube_transcript_api._cli.versionrY Úunknown)r r r Ú _get_version)r# Úclir$ r$ r% Ú"test_get_version_package_not_foundm s ÿ"üz;TestYouTubeTranscriptCli.test_get_version_package_not_foundN) Ú__name__Ú
__module__Ú __qualname__r& r= rA rC rF rG rI rJ rN rQ rU rX r^ r` rb rc rd re rg rq ru ÚpytestÚmarkÚskiprx r‚ r† r$ r$ r$ r% r s8 +D
ÿ
r )rŠ Úimportlib.metadatar r Úunittestr Ú
unittest.mockr r r) r} ry r r r r
Úyoutube_transcript_api._clir r r$ r$ r$ r% Ú s