
    עiY                     v   d Z ddlmZ ddlZddlmZmZ ddlmZ ddlm	Z
 dd	lmZ ddlZej                  j!                  ej"                  d
d      d        Zej&                  j(                  ej                  j!                  ej"                  d
d      d               Z ej,                  e e       g       Zy)z%Tests for Interactions API URL paths.    )mockN)RequestResponse   )AsyncHttpxClient)Client   )pytest_helperdefaultT)autospecc                    d}t        j                         }d|_        d|_        d|_        |df| _        |j                  j                  rHd|j                  j                   d|j                  j                   d|j                  j                   }nd	}t         j                  j                  t        d
      5 }t        dt        dd            |_        |j                  j!                  |       |j#                          |j$                  d   d   }t'        |j(                        | d| k(  sJ |j+                          t        dt        dd            |_        |j                  j-                  |       |j#                          |j$                  d   d   }t'        |j(                        | d| dk(  sJ |j+                          t        dt        dd            |_        |j                  j/                  |       |j#                          |j$                  d   d   }t'        |j(                        | d| k(  sJ 	 d d d        y # 1 sw Y   y xY wNztest-interaction-idz
test-tokenFztest-quota-projectztest-projectzhttps://z,-aiplatform.googleapis.com/v1beta1/projects/z/locations/z0https://generativelanguage.googleapis.com/v1betasend   GET )request)idr   z/interactions/POSTz/cancelDELETE)r   Mocktokenexpiredquota_project_idreturn_value_api_clientvertexailocationprojectpatchobject
HTTPClientr   r   interactionsgetassert_called_once	call_argsstrurl
reset_mockcanceldeletemock_auth_defaultclientinteraction_id
mock_credsexpected_base_url	mock_sendr   s          k/sites/domain-asset-master/.venv/lib/python3.12/site-packages/google/genai/tests/interactions/test_paths.pytest_interactions_pathsr4      sg   *NJ#JJ"6J&0.%A"""&v'9'9'B'B&CCopv  qC  qC  qK  qK  pL  LW  X^  Xj  Xj  Xs  Xs  Wt  uN			:v	. X)!)#wub7I!J	>2$$&%%a(+7;;&7%8~FV#WWWW!)#wvr7J!K	""n"5$$&%%a(+7;;&7%8~FVV]#^^^^!)#wx7L!M	""n"5$$&%%a(+7;;&7%8~FV#WWWW'X X Xs   9FIIc                 :  K   d}t        j                         }d|_        d|_        d|_        |df| _        |j                  j                  rHd|j                  j                   d|j                  j                   d|j                  j                   }nd	}t         j                  j                  t        d
      5 }t        dt        dd            |_        |j                  j                   j#                  |       d {    |j%                          |j&                  d   d   }t)        |j*                        | d| k(  sJ |j-                          t        dt        dd            |_        |j                  j                   j/                  |       d {    |j%                          |j&                  d   d   }t)        |j*                        | d| dk(  sJ |j-                          t        dt        dd            |_        |j                  j                   j1                  |       d {    |j%                          |j&                  d   d   }t)        |j*                        | d| k(  sJ 	 d d d        y 7 7 7 T# 1 sw Y   y xY wwr   )r   r   r   r   r   r   r   r   r   r   r    r!   r   r   r   aior#   r$   r%   r&   r'   r(   r)   r*   r+   r,   s          r3   test_async_interactions_pathsr7   @   s     +NJ#JJ"6J&0.%A"""&v'9'9'B'B&CCopv  qC  qC  qK  qK  pL  LW  X^  Xj  Xj  Xs  Xs  Wt  uN			+V	4 X	!)#wub7I!J	jj%%))^)<<<$$&%%a(+7;;&7%8~FV#WWWW!)#wvr7J!K	jj%%,,,???$$&%%a(+7;;&7%8~FVV]#^^^^!)#wx7L!M	jj%%,,,???$$&%%a(+7;;&7%8~FV#WWWW'X X< 	@ 	@!X XsX   B9J;AJJBJJBJ8J9AJ?	JJJJJJ)fileglobals_for_file
test_table)__doc__unittestr   pytesthttpxr   r   r   r   r   r"   r   r
   google.authgoogler    r!   authr4   markasyncior7   setup__file__globals
pytestmark     r3   <module>rJ      s   $ ,   # + &  6;;	D9!X :!XF 6;;	D9!X : !XF !]  	Y
rI   