o
    Yi	                     @   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)Z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 )
NZ!_interactions_experimental_warnedF)newplaceholderapi_versionv1alphaapi_keyZhttp_optionszInteractions.*experimental)match)	r   patchobject
client_libClientpytestZwarnsUserWarninginteractions)r   _ r   j/home/kim/smarthome/.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 )NZGeminiNextGenAPIClientTZspec_setr   r     r   timeoutr	         @base_urlr
   r   default_headershttp_clientr   max_retriesZclient_adapter)r   r   r   r   r   r   assert_called_once_withANYZ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 )NZAsyncGeminiNextGenAPIClientTr   r   r   r   r   r	   r   r   )	r   r   r   r   r   Zaior   r!   r"   r#   r   r   r   test_async_client_timeout>   s*   "r%   )Zunittestr   r    r   r   Zpytest_pluginsr   r$   markZasyncior%   r   r   r   r   <module>   s   