
    עi	                     l    d dl mZ d dlZddlmZ dZd Zd Zej                  j                  d        Zy)	    )mockN   )client)pytest_asyncioc                  $   t         j                  j                  t        dd      5  t        j                  dddi      } t        j                  t        d	      5  | j                  }d d d        d d d        y # 1 sw Y   xY w# 1 sw Y   y xY w)
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   _s     q/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/interactions/test_integration.pytest_client_future_warningr      s    zz55    9
F 
7
  

a   s#   5BA:)B:B	?BBc                     t         j                  j                  t        dd      5 } t        j                  dddd      }|j
                  }| j                  t         j                  ddt         j                  t         j                  d	t         j                  t         j                  
       d d d        y # 1 sw Y   y xY w)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   s      r   test_client_timeoutr.   &   s    zz*T   %.4@F
 	A//HHHHxx 0 	  s   BB66B?c                    K   t         j                  j                  t        dd      5 } t        j                  dddd      }|j
                  j                  }| j                  t         j                  ddt         j                  t         j                  d	t         j                  t         j                  
       d d d        y # 1 sw Y   y xY ww)NAsyncGeminiNextGenAPIClientTr   r
   r   r    r!   r   r#   r$   )	r   r   r   r   r   aior   r*   r+   r,   s      r   test_async_client_timeoutr2   >   s     zz/$   %.4@F
 	

A//HHHHxx 0 	  s   'CBC9	CCC)unittestr   r    r   r   pytest_pluginsr   r.   markasyncior2        r   <module>r:      s=      $$ 0  r9   