
    עi%	                         U d Z ddlZddlZddlmZ ddlmZ ddlmZ g Ze	ej                     ed<    ej                  e e       d	e
      ZdZd Zd Zej&                  j(                  d        Zy)zTest files upload method.    N   )_transformers)types   )pytest_helper
test_tablezt.t_file_name)fileglobals_for_filetest_methodr   )pytest_asyncioc                    t        j                  | t              5  | j                  j	                         D ]  }|j
                   n t        d      |j                  j                  d      d   }t        j                  |j
                        }t        j                  |      }|||j                  |j                  |j
                  ||fD ]  }t        j                  |      }||k(  rJ  	 d d d        y # 1 sw Y   y xY w)N%No files found with a `download_uri`./)uri)video)r   exception_if_vertex
ValueErrorfileslistdownload_urinamesplitr   VideoGeneratedVideor   tt_file_name)clientr	   file_idr   generated_videofr   s          g/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/files/test_download.pytest_name_transform_namer#   '   s    ((< !!# @				&@ >??iiooc"2&GKKD--.E**7O		 
 ]]1dW__
  s   *C:B&C:-C::Dc                     t        j                  | t              5  | j                  j	                         D ]  }|j
                   n t        d      | j                  j                  |      }|dd dk(  sJ 	 d d d        y # 1 sw Y   y xY wNr   )r	         s   ftyp)r   r   r   r   r   r   downloadr   r	   contents      r"   test_basic_downloadr+   ?   s    ((< #!!# @				&@ >??ll###.G1Q<7"""# # #s   *B4BBc                   K   t        j                  | t              5  | j                  j                  j                          d {   2 3 d {   }|j                   | j                  j                  j                  |       d {   }|dd dk(  sJ 	 d d d        y 7 ]7 V6 t        d      7 (# 1 sw Y   y xY wwr%   )r   r   r   aior   r   r   r(   r)   s      r"   test_basic_download_asyncr.   K   s     ((< #"JJ,,1133 @ @d				& JJ$$--4-88G1Q<7"""# #3 @3 >??8# #sd   C(B5B#B5
B'B%B'B5 *B5
B3B5	C#B5%B''B55B>:C)__doc__pathlibpytest r   r   r   r   r   r   TestTableItem__annotations__setup__file__globals
pytestmarkpytest_pluginsr#   r+   markasyncior.        r"   <module>r>      s   "     "   13
D,,- 2 ]  	Y	
 %0	# 	# 	#r=   