o vZh@svddlZddlZddlmZddlmZmZddlmZddl m Z  ej dZ e r/e dkr9eZee ne dkrAeZne d krIe Zn d d e d DZeZejjejjd eeedddZejjddZddZddZejjddZejjddZejjddZddZdd Z ejjd!d"Z!ejjd#d$Z"e#d%kre$d&e%gdSdS)'N)Mock)gTTS gTTSError) _main_langs) _extra_langsZ TEST_LANGSallmainextracCsi|]}|r||qSr ).0lr r B/var/www/auris/lib/python3.10/site-packages/gtts/tests/test_tts.py $sr,lang)ZidscCsNd} dD]}|d|}t|||dd}|||jdks$JqdS)z*Test all supported languages and file savezThis is a test)FTz test_{}_.mp3F)textrslow lang_checkiN)formatrsavestatst_size)tmp_pathrrrfilenamettsr r r test_TTS(s rcCsJd}d}d}ttt|||dWddS1swYdS)z>Raise ValueError on unsupported language (with language check)xx Lorem ipsumTrrrN)pytestraises ValueErrorr)rrcheckr r r test_unsupported_language_check9s  "r#cCs>d}ttt|dWddS1swYdS)z$Raise AssertionError on empty stringrN)rr AssertionErrorrr%r r r test_empty_stringCs  "r'cCsPd}tt|d}t|d}||WddS1s!wYdS)zBRaises AssertionError on no content to send to API (no text_parts)zn .., zno_content.txtr%N)rr r&rr)rrrrr r r test_no_text_partsJs    "r(cCsDtdd}tt|dWddS1swYdS)z=Raise TypeError if fp is not a file-like object (no .write())testr%N)rrr TypeErrorZ write_to_fprr r r test_bad_fp_typeVs   "r-cCs2|d}tdd}|||jdksJdS)zSave .mp3 file successfullysave.mp3r)r%iN)rrrr)rrrr r r test_save_s  r/cCs6tdddd}|d}d|vsJd|vsJdS)zget request bodies listr)comen)rtldrrz%5C%22en%5C%22N)rZ get_bodies)rbodyr r r test_get_bodiesks  r4cCs.td}d|jks Jt}|jdusJdS)zQTest gTTsError internal exception handling Set exception message successfullyr)N)rmsg)Zerror1Zerror2r r r test_msgusr6cCstdd}t|d}|jdksJt}tddd}t||d}|jd ks'Jtd d d }td dd}t||d}|jdks@Jt}tddd} t|| d} | jdksVJt} tddd} t| | d} | jdkslJdS)z+Infer message successfully based on contextinvalid)r2r,z\Failed to connect. Probable cause: Host 'https://translate.google.invalid/' is not reachableiZaaa) status_codereason)rresponsezI403 (aaa) from TTS API. Probable cause: Bad token or upstream API changesrF)rrbbbz^200 (bbb) from TTS API. Probable cause: No audio stream in response. Unsupported language 'xx'iccczL500 (ccc) from TTS API. Probable cause: Upstream API error. Try again later.dZdddz/100 (ddd) from TTS API. Probable cause: UnknownN)rrr5)ZttsTLDZerrorTLDZtts403Z response403Zerror403Ztts200Z response404Zerror200Ztts500Z response500Zerror500Ztts100Z response100Zerror100r r r test_infer_msgs6           r?cCs^d} d}d}tt|d}t|||d}||WddS1s(wYdS)zTest Web RequestsrrFzxx.txtrN)rr rrr)rrrr"rrr r r test_WebRequests  "r@cCstdd}|jdus Jd}td|d}|j|ksJtddd}|d}tt||WddS1s;wYdS)Nr)r%g333333?)rtimeoutgư>r.)rrArr rr)rrrArr r r test_timeouts     "rB__main__z-x)&osrZ unittest.mockrZgtts.ttsrrZ gtts.langsrZ gtts.langrenvirongetenvZlangsupdatesplitZ env_langsmarknetZ parametrizekeyslistvaluesrr#r'r(r-r/r4r6r?r@rB__name__r__file__r r r r sN             1