
    hj                     `   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                            ej        d
d          d             Zej        j        ej                            ej        d
d          d                         Z ej        e e            g           ZdS )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        r*d|j        j         d|j        j	         d|j        j         }nd	}t           j
                            t          d
          5 }t          dt          dd          ddid          |_        |j                            |           |                                 |j        d         d         }t'          |j                  | d| dk    sJ |                                 t          dt          dd          ddid          |_        |j                            |           |                                 |j        d         d         }t'          |j                  | d| dk    sJ |                                 t          dt          dd                    |_        |j                            |           |                                 |j        d         d         }t'          |j                  | d| k    sJ 	 d d d            d S # 1 swxY w Y   d S 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 zcontent-typezapplication/jsonz{"status": "completed"})requestheaderscontent)idr   z/interactions/z?stream=falsePOSTz/cancelDELETE)r   )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          n/var/www/app.qaans.com/backend/venv/lib/python3.11/site-packages/google/genai/tests/interactions/test_paths.pytest_interactions_pathsr6      s+   *NJ#JJ"6J&0.%A"" O uv'9'B  u  upv  qC  qK  u  u  X^  Xj  Xs  u  uN			:v	.	. X)!)#wub7I7ITbdvSw  B[  "\  "\  "\	>222$$&&&%a(+7;&7#d#d~#d#d#ddddd!)#wvr7J7JUcewTx  C\  "]  "]  "]	""n"555$$&&&%a(+7;&7#^#^~#^#^#^^^^^!)#wx7L7L!M!M!M	""n"555$$&&&%a(+7;&7#W#W~#W#WWWWWW'X X X X X X X X X X X X X X X X X Xs   GI!!I%(I%c                 "  K   d}t          j                    }d|_        d|_        d|_        |df| _        |j        j        r*d|j        j         d|j        j	         d|j        j         }nd	}t           j
                            t          d
          5 }t          dt          dd          ddid          |_        |j        j                            |           d {V  |                                 |j        d         d         }t)          |j                  | d| dk    sJ |                                 t          dt          dd          ddid          |_        |j        j                            |           d {V  |                                 |j        d         d         }t)          |j                  | d| dk    sJ |                                 t          dt          dd                    |_        |j        j                            |           d {V  |                                 |j        d         d         }t)          |j                  | d| k    sJ 	 d d d            d S # 1 swxY w Y   d S 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          r5   test_async_interactions_pathsr9   @   sn      +NJ#JJ"6J&0.%A"" O uv'9'B  u  upv  qC  qK  u  u  X^  Xj  Xs  u  uN			+V	4	4 X	!)#wub7I7ITbdvSw  B[  "\  "\  "\	j%))^)<<<<<<<<<$$&&&%a(+7;&7#d#d~#d#d#ddddd!)#wvr7J7JUcewTx  C\  "]  "]  "]	j%,,,?????????$$&&&%a(+7;&7#^#^~#^#^#^^^^^!)#wx7L7L!M!M!M	j%,,,?????????$$&&&%a(+7;&7#W#W~#W#WWWWWW'X X X X X X X X X X X X X X X X X Xs   G#JJJ)fileglobals_for_file
test_table)__doc__unittestr   pytesthttpxr   r   r   r   r   r$   r   r
   google.authgoogler"   r#   authr6   markasyncior9   setup__file__globals
pytestmark     r5   <module>rL      s0  $ , +        # # # # # # # # + + + + + + & & & & & &          6;	D99!X !X :9!XF 6;	D99!X !X :9 !XF !] 	WYY  


rK   