
    UEj                         d dl mZ ddlmZ d dlZddlmZ dZ	d Z
ej                  j                  d        Zej                  j                  d	      d
        Zej                  j                  d	      d        Zy)    )mock   )google_genaiN)client)pytest_asyncioc            
         t         j                  j                  t        dd      5 } t        j                         }|| _        t        j                  dddd      }|j                  j                  }| j                  t         j                  t         j                  t         j                  t         j                  t         j                  dt         j                  	       d d d        y # 1 sw Y   y xY w)
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   _s       f/var/www/html/qr/venv/lib/python3.12/site-packages/google/genai/tests/interactions/test_integration.pytest_client_timeoutr)      s    zz4   ))+K)J%.4@F 	--A&&HHXX88xxXX '   s   B8C((C1c            
        K   t         j                  j                  t        dd      5 } t        j                         }|| _        t        j                  dddd      }|j                  j                  j                  }| j                  t         j                  t         j                  t         j                  t         j                  t         j                  dt         j                  	       d d d        y # 1 sw Y   y xY ww)
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'   s       r(   test_async_client_timeoutr/   5   s     zz   ))+K$/!%.4@F 	

11A,,HHXX88XXXX -   s   'D CC4+	D 4C=9D errorc                  X    ddl m}   | dd      }|j                         }|d   dk(  sJ y )Nr   CreateModelInteractiongemini-3.5-flashhellomodelinputr7   )/_gaos.types.interactions.createmodelinteractionr3   
model_dump)r3   objdumpeds      r(   %test_unrecognized_model_serializationr=   P   s1    W%7wG#>>&	.	..	.    c                  |    ddl m}  ddlm}  |dd      } | |      }|j	                         }|d   d	   dk(  sJ y )
Nr   )CreateInteractionRequestr2   r4   r5   r6   )bodyrA   r7   )_gaos.models.createinteractionr@   r9   r3   r:   )r@   r3   rA   reqr<   s        r(   -test_unrecognized_model_request_serializationrD   Y   sC    HW	&8	H$ d+#>>&		 $6	66	6r>   )unittestr   _gaosr   r   pytest r   r   pytest_pluginsr)   markasyncior/   filterwarningsr=   rD    r>   r(   <module>rN      s     6  $ %4  4 G$/ %/ G$7 %7r>   