עi fUdZddlZddlZddlZddlmZddlmZddlmZgZe eje d<eje ed e Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"ejFjHdZ%ejFjHdZ&ejFjHd Z'ejFjHd!Z(ejFjHd"Z)ejFjHd#Z*y)$zTest files upload method.N)types)errors) pytest_helper test_tablez files.upload)fileglobals_for_file test_methodrctj|t5|jj d}|j j dsJ dddy#1swYyxYwNtests/data/google.pngr files/rexception_if_vertex ValueErrorfilesuploadname startswithclientr s e/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/files/test_upload.pytest_image_png_uploadr%W((<* <<  $;  :CC C  CcBtj|t5tdd5}|jj |t jd}dddjjdsJ dddy#1swY0xYw#1swYyxYwr*) rrrr.rrrr&rr)rr3r s rtest_image_png_upload_with_fdr7>s((<* %t , \\ ''+>!d 99   )) )****s" B2B %B B BBctj|t5|jj dt j d}|jjdsJ dddy#1swYyxYw)Nrtest_image_pngr r"r rrrrrrr&rrrs r!test_image_png_upload_with_configr;Hl((<* <<   $%%3CD  D 99   )) ) *** AA44A=ctj|t5|jj dddi}|j j dsJ dddy#1swYyxYw)Nrr!r9r"rrrs r&test_image_png_upload_with_config_dictr?Qe((<* <<   $n>N-O  D 99   )) ) *** =A""A+ctj|t5|jj d}|j j dsJ dddy#1swYyxYw)Ntests/data/google.jpgrrrrs rtest_image_jpg_uploadrDYrrctj|t5|jj dt j d}|jjdsJ dddy#1swYyxYw)NrCtest_image_jpgr r"rr:rs r!test_image_jpg_upload_with_configrG_r<r=ctj|t5|jj dddi}|j j dsJ dddy#1swYyxYw)NrCr!rFr"rrrs r&test_image_jpg_upload_with_config_dictrIhr@rActj|t5|jj d}|j j dsJ dddy#1swYyxYw)Ntests/data/story.pdfrrrrs r test_application_pdf_file_uploadrLpsW((<* <<  $:  ;D 99   )) )***rctj|t5|jj dt j d}|jjdsJ dddy#1swYyxYw)NrKtest_application_pdfr r"rr:rs r'test_application_pdf_upload_with_configrOvsl((<* <<   #%%3IJ  D 99   )) ) ***r=ctj|t5|jj dddi}|j j dsJ dddy#1swYyxYw)NrKr!rNr"rrrs r,test_application_pdf_upload_with_config_dictrQsf((<* <<   # 67  D 99   )) ) ***rActj|t5|jj d}|j j dsJ dddy#1swYyxYw)Ntests/data/animal.mp4rrrrs rtest_video_mp4_file_uploadrTrrctj|t5|jj dt j d}|jjdsJ dddy#1swYyxYw)NrStest_video_mp4r r"rr:rs r!test_video_mp4_upload_with_configrWr<r=ctj|t5|jj dddi}|j j dsJ dddy#1swYyxYw)NrSr!rVr"rrrs r&test_video_mp4_upload_with_config_dictrYr@rActj|t5|jj dt j d}|jjdsJ dddy#1swYyxYw)Ntests/data/pixel.m4a audio/mp4r,r"rr:rs rtest_audio_m4a_file_uploadr]sk((<* <<   #%% <  D 99   )) ) ***r=ctj|t5|jj dt j dd}|jjdsJ dddy#1swYyxYwNr[test_audio_m4ar\)r!r-r"rr:rs r!test_audio_m4a_upload_with_configrasp((<* <<   #%%)[   D 99   )) )***s AA55A>ctj|t5|jj dddd}|j j dsJ dddy#1swYyxYwr_rrs r&test_audio_m4a_upload_with_config_dictrcse((<* <<   # 0{K  D 99   )) ) ***s >A##A,c:tj|t5tjt j d5|jjtjdddi}ddddddy#1swYxYw#1swYyxYwNzUnsupported MIME)matchstestr-z bad/mime_typer") rrrpytestraisesrAPIErrorrrr/r0rs rtest_bad_mime_typerjs((< v.@ A \\ zz'"/!ds#&B3B4BB BBcKtj|t5|jjj dd{}|j jdsJ dddy7+#1swYyxYwwr rrraiorrrrrs rtest_image_upload_asyncrnsl((<*!!((.E(F FD 99   )) )** F**s.B*A5A3!A5* B3A55A>:Bc0Ktj|t5|jjj dt jdd{}|jjdsJ dddy7+#1swYyxYww)Nr test_imager r"r) rrrrmrrrr&rrrs r#test_image_upload_with_config_asyncrqs((<*!!(( $%%<@) D 99   )) ) ** **s.B?B B!B ? BB  BBcKtj|t5|jjj ddddidd{}|j jdsJ dddy7+#1swYyxYww)Nrrptimeout8000)r! http_optionsr"rrlrs r(test_image_upload_with_config_dict_asyncrvs((<*!!(( $(&/ ) D 99   )) )** **s.B0A; A9!A;0 B9A;;BBcKtj|t5tdd5}t j |j }ddd|jjjtjdd{}|jjdsJ dddy#1swYsxYw77#1swYyxYwwr*)rrrr.r/r0r1rmrrrr&rrr2s r$test_image_upload_with_bytesio_asyncrxs((<* %t ,$zz!&&(#f$!!(( %%!#) D 99   )) )**$$ **s@C C$CACC !C7 CC CCCcKtj|t5 |jjj dd{dddy7 #t $r}dt|vsJYd}~(d}~wwxYw#1swYyxYww)N unknown_pathrzis not a valid file path)rrrrmrrFileNotFoundErrorstr)res rtest_unknown_path_upload_asyncr~sz((<22 JJ   # # # 888228 2 '3q6 11 1222sQB A=)AA A B A A:!A50A=5A::A==BB cjKtj|t5tjt j d5|jjjtjdddid{}ddddddy7#1swYxYw#1swYyxYwwre) rrrrgrhrrirmrrr/r0rs rtest_bad_mime_type_asyncrs((< v.@ A::##**zz'"/+dsGB3&B'ABBBB' B3BB$ B''B0,B3)+__doc__r/r$rgrrrrlist TestTableItem__annotations__setup__file__globals pytestmarkrr(r5r7r;r?rDrGrIrLrOrQrTrWrYr]rarcrjmarkasynciornrqrvrxr~rrrsh"   13 D,, -2 ] Y  * * .**** *** *** ******* ** * * * *22r