o
    i	                     @   sH   d dl mZ d dlZddlmZ dZdd Zdd	 Zej	j
d
d ZdS )    )mockN   )client)pytest_asyncioc               	   C   s   t jjtddd4 tjdddid} tjtdd	 | j}W d    n1 s(w   Y  W d    d S W d    d S 1 s@w   Y  d S )
N!_interactions_experimental_warnedF)newplaceholderapi_versionv1alphaapi_keyhttp_optionszInteractions.*experimental)match)	r   patchobject
client_libClientpytestwarnsUserWarninginteractions)r   _ r   n/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/google/genai/tests/interactions/test_integration.pytest_client_future_warning   s    	"r   c                  C   sv   t jjtddd(} tjddddd}|j}| jt jddt jt jd	t jt jd
 W d    d S 1 s4w   Y  d S )NGeminiNextGenAPIClientTspec_setr   r
     r	   timeoutr         @base_urlr   r	   default_headershttp_clientr    max_retriesclient_adapter)r   r   r   r   r   r   assert_called_once_withANYmock_nextgen_clientr   r   r   r   r   test_client_timeout&   s(   "r,   c                     sz   t jjtddd)} tjddddd}|jj}| jt jddt jt jd	t jt jd
 W d    d S 1 s6w   Y  d S )NAsyncGeminiNextGenAPIClientTr   r   r
   r   r   r   r!   r"   )	r   r   r   r   r   aior   r(   r)   r*   r   r   r   test_async_client_timeout>   s*   "r/   )unittestr   r    r   r   pytest_pluginsr   r,   markasyncior/   r   r   r   r   <module>   s   