o
    pkj                     @   s   d dl mZ ddlmZ d dlZddlmZ dZ	dd Z
ejjd	d
 Zejddd Zejddd Zdd Zdd ZdS )    )mock   )google_genaiN)client)pytest_asyncioc               
   C   s   t jjtddd0} t  }|| _tjddddd}|jj	}| j
t jt jt jt jt jdt jd	 W d    d S 1 s<w   Y  d S )
NGenAITspec_setplaceholderv1alpha  api_versiontimeoutapi_keyhttp_options)securityr   user_project
server_urlr   
timeout_msretry_config)r   patchobjectgaos_google_genaiMockreturn_value
client_libClientinteractionswith_raw_responseassert_called_once_withANY)
mock_genaimock_clientr   _ r&   n/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/tests/interactions/test_integration.pytest_client_timeout   s*   "r(   c               
      s   t jjtddd1} t  }|| _tjddddd}|jj	j
}| jt jt jt jt jt jdt jd	 W d    d S 1 s>w   Y  d S )
N
AsyncGenAITr   r
   r   r   r   r   )r   r   r   r   async_clientr   r   )r   r   r   r   r   r   r   r   aior   r    r!   r"   )mock_async_genair$   r   r%   r&   r&   r'   test_async_client_timeout4   s,   
"r-   errorc                  C   s4   ddl m}  | ddd}| }|d dksJ d S )Nr   CreateModelInteractiongemini-3.5-flashhellomodelinputr4   )/_gaos.types.interactions.createmodelinteractionr0   
model_dump)r0   objdumpedr&   r&   r'   %test_unrecognized_model_serializationO   s   r:   c                  C   sN   ddl m}  ddlm} |ddd}| |d}| }|d d	 dks%J d S )
Nr   )CreateInteractionRequestr/   r1   r2   r3   )bodyr<   r4   )_gaos.models.createinteractionr;   r6   r0   r7   )r;   r0   r<   reqr9   r&   r&   r'   -test_unrecognized_model_request_serializationX   s   
r?   c                  C   s\   ddl m}  | dddid}|jdksJ |jddiksJ | }|d ddiks,J d S Nr   )AllowlistEntryz
github.comAuthorizationzBearer TOKEN)domain	transformrD   '_gaos.types.interactions.allowlistentryrA   rC   rD   r7   rA   entryr9   r&   r&   r'   (test_allowlist_entry_with_dict_transformb   s   rI   c                  C   sb   ddl m}  | dddigd}|jdksJ |jddigks J | }|d ddigks/J d S r@   rE   rG   r&   r&   r'   (test_allowlist_entry_with_list_transformq   s   
rJ   )unittestr   _gaosr   r   pytest r   r   pytest_pluginsr(   markasyncior-   filterwarningsr:   r?   rI   rJ   r&   r&   r&   r'   <module>   s   




	