o
    Y€Æi%	  ã                   @   sˆ   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d
ZdZdd„ Zdd„ Zejjdd„ ƒZdS )zTest files upload method.é    Né   )Ú_transformers)Útypesé   )Úpytest_helperÚ
test_tablezt.t_file_name)ÚfileZglobals_for_fileZtest_methodr   )Zpytest_asyncioc                 C   s¶   t  | t¡K | j ¡ D ]	}|jd ur nqtdƒ‚|j d¡d }tj	|jd}tj
|d}|||j|j|j||fD ]}t |¡}||ksHJ ‚q;W d   ƒ d S 1 sTw   Y  d S )Nú%No files found with a `download_uri`.ú/éÿÿÿÿ)Úuri)Úvideo)r   Úexception_if_vertexÚ
ValueErrorÚfilesÚlistÚdownload_uriÚnameÚsplitr   ZVideoZGeneratedVideor   ÚtZt_file_name)Úclientr   Zfile_idr   Zgenerated_videoÚfr   © r   ú`/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/files/test_download.pyÚtest_name_transform_name'   s*   
ÿù
	ö"ör   c                 C   sz   t  | t¡- | j ¡ D ]	}|jd ur nqtdƒ‚| jj|d}|dd… dks+J ‚W d   ƒ d S 1 s6w   Y  d S ©Nr	   )r   é   é   s   ftyp)r   r   r   r   r   r   Údownload©r   r   Úcontentr   r   r   Útest_basic_download?   s   
ÿ"ør!   c                 Ã   s–   t  | t¡: | jj ¡ I d H 2 z3 d H W }|jd ur nq6 tdƒ‚| jjj|dI d H }|dd… dks9J ‚W d   ƒ d S 1 sDw   Y  d S r   )r   r   r   Zaior   r   r   r   r   r   r   r   Útest_basic_download_asyncK   s   €
ÿÿ"ør"   )Ú__doc__ÚpathlibZpytestÚ r   r   r   r   r   r   ZTestTableItemÚ__annotations__ÚsetupÚ__file__ÚglobalsZ
pytestmarkZpytest_pluginsr   r!   ÚmarkZasyncior"   r   r   r   r   Ú<module>   s$   ü