File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/test_utils.cpython-310.pyc
o
;j<` � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z ddl
mZmZ ddlmZ ddlmZmZmZmZmZmZmZ ddlmZm Z ddl m Z dd l m Z m!Z! dd
l"m#Z# ddl$Z$dd l%m&Z&m'Z'm(Z(m)Z) d d
l*m+Z+m,Z, d dl
m-Z- d dl.m/Z/ d dl0m1Z1 d dl2m3Z3 d dl4m5Z5m6Z6 d dl7m8Z8m9Z9 d dl:m;Z; d dl<m=Z=m>Z>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZE d dlFmGZG er�ddlHmIZI neZIejJdkr�er�ddlmKZK ejJdkr�ddlmLZL neZLede=d�ZMed e?d �ZNejOd k�oejPd kZQejRfd eSd!ejTd"ejfd#d$�ZUd eSd%eVd!ejTd"ejfd&d'�ZWd"eVfd(d)�ZXG d*d+� d+e �ZYG d,d-� d-eY�ZZG d.d/� d/eY�Z[G d0d1� d1eeNeMf �Z\G d2d3� d3e�Z]d4ed5ed6ed"efd7d8�Z^eg ej_f Z`ejaejbd9fd:e`d;ecd"eej_ fd<d=��Zdejbfd:e`d"ej_fd>d?�Zed]d@ej_d;ecd"dfdAdB�Zfd"e jgfdCdD�Zhd^dEeIdB d"e jifdFdG�Zj d^e3e5d d �d9de3e3e3e8ddHdIdJ� dKeSdLeSdMedNedOe5dPecdQedRedSedTedUe9dEeIdB dVeVd@ed"eAf dWdX�Zke3e3fdYedZed"efd[d\�ZldS )_zUtilities shared by tests.� N)�ABC�abstractmethod)�Callable�Iterator)�
TracebackType)�
TYPE_CHECKING�Any�Generic�Literal�TypeVar�cast�overload)�IsolatedAsyncioTestCase�mock)�Signal)� CIMultiDict�CIMultiDictProxy)�URL)�_BaseRequestContextManager�_RequestContextManager�_RequestOptions�_WSRequestContextManager� )�
ClientSession�hdrs)�AbstractCookieJar)�ClientResponse)�ClientWebSocketResponse)�sentinel)� HttpVersion�RawRequestMessage)�
EMPTY_PAYLOAD� StreamReader)�StrOrURL) � Application� AppRunner� BaseRequest�
BaseRunner�Request�Server� ServerRunner�SockSite�UrlMappingMatchInfo)�_RequestHandler)�
SSLContext�� � )�Unpack)�Self�_ApplicationNone�_Request)�bound�posix�cygwin�host�family�returnc C s t | d|�S )Nr )�get_port_socket)r9 r: � r= �F/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/test_utils.py�get_unused_port_socketF s r? �portc C s6 t � |t j�}tr|�t jt jd� |�| |f� |S )Nr )�socket� SOCK_STREAM�
REUSE_ADDRESS�
setsockopt�
SOL_SOCKET� SO_REUSEADDR�bind)r9 r@ r: �sr= r= r> r< L s
r< c C sP t � t jt j��} | �d� tt| �� d �W d � S 1 s!w Y dS )z1Return a port that is unused on the current host.)� 127.0.0.1r r N)rA �AF_INETrB rG r �int� getsockname)rH r= r= r> � unused_portY s
$�rM c @ sV e Zd ZdZddddded�dedejdB ded edB d
e d e
eee j ge j f d e
d
dfdd�Z d*dejdB d e
d
dfdd�Zed e
d
efdd��Zded
efdd�Zed
e fdd��Zed
e fdd��Zed
efdd ��Zd+d d �Zd+d d �Z d!e e dB d"e dB d#e dB d
dfd$d%�Z d,d&d'�Z!d!e e dB d"e dB d#e dB d
dfd(d)�Z"dS )-�BaseTestServerF� NrI )�scheme�loopr9 r@ �skip_url_asserts�socket_factoryrP rQ r9 r@ rR rS �kwargsr; c K s: || _ d | _d | _|| _|| _d| _|| _|| _|| _d S )NF) �_loop�runner�_rootr9 r@ �_closedrP rR rS )�selfrP rQ r9 r@ rR rS rT r= r= r> �__init__c s
zBaseTestServer.__init__c
� sj �| j rd S || _|�dd �| _| jdddi|��I d H | _ | j �� I d H | js,d| _| j}z t� | j�j
}W n t yC d}Y nw |dkrOd| j� d�}|dkrVt j
nt j}| �| j| j|�}|�� d d � \| _| _t| j || jd
�}|�� I d H |j}|d us�J �|j} | d us�J �| d �� d | _| js�| jr�d nd
| _t| j� d|� d| j� ��| _d S )N�sslZhandler_cancellationTr � � �[�]� )�sock� ssl_contextr �https�httpz://�:r= )rV rU �pop�_ssl� _make_runner�setupr@ r9 � ipaddress�
ip_address�version�
ValueErrorrA �AF_INET6rJ rS rL r+ �start�_server�socketsrP r rW )
rY rQ rT Z
absolute_hostrl r: �_sock�site�serverrq r= r= r> � start_serverz s< � � "zBaseTestServer.start_serverc � � �d S �Nr= �rY rT r= r= r> rh � s �zBaseTestServer._make_runner�pathc C sH | j d usJ �t|�}| js|jrJ �| j �|�S tt| j �t|� �S rw )rW r rR �absolute�join�str)rY ry �urlr= r= r> �make_url� s
zBaseTestServer.make_urlc C s
| j d uS rw )rV �rY r= r= r> �started� s
zBaseTestServer.startedc C � | j S rw )rX r r= r= r> �closed� � zBaseTestServer.closedc C s& | j }|d us J �|jd usJ �|jS rw )rV rt )rY rV r= r= r> �handler� s zBaseTestServer.handlerc � sJ �| j r!| js#| jdusJ �| j�� I dH d| _d| _d| _dS dS dS )a` Close all fixtures created by the test client.
After that point, the TestClient is no longer usable.
This is an idempotent function: running close multiple times
will not have any additional effects.
close is also run when the object is garbage collected, and on
exit when used as a context manager.
NT)r� r� rV ZcleanuprW r@ rX r r= r= r> �close� s �
�zBaseTestServer.closec C � t d���NzUse async with instead�� TypeErrorr r= r= r> � __enter__� � zBaseTestServer.__enter__�exc_type� exc_value� tracebackc C � d S rw r= �rY r� r� r� r= r= r> �__exit__� � zBaseTestServer.__exit__c � s �| j | jd�I d H | S �N�rQ )ru rU r r= r= r> �
__aenter__� s �zBaseTestServer.__aenter__c � � �| � � I d H d S rw �r� r� r= r= r> � __aexit__� � �zBaseTestServer.__aexit__rw �r; N)r; rN )#�__name__�
__module__� __qualname__�__test__r< r| �asyncio�AbstractEventLooprK �boolr rA �
AddressFamilyr rZ ru r r' rh r# r r~ �propertyr� r� r) r� r� r� �type�
BaseExceptionr r� r� r� r= r= r= r> rN ` s� �������� �
����
�
���
�
����rN c sR e Zd Zdddd�dededededB d ef
� fd
d �Zd ed efd
d�Z � Z
S )�
TestServerrO rI N�rP r9 r@ �apprP r9 r@ rT c �$ || _ t� jd|||d�|�� d S �Nr� r= )r� �superrZ )rY r� rP r9 r@ rT �� __class__r= r> rZ � � zTestServer.__init__r; c � s �t | jfi |��S rw )r% r� rx r= r= r> rh � s �zTestServer._make_runner) r� r� r� r$ r| rK r rZ r' rh �
__classcell__r= r= r� r> r� � s ������ r� c
s\ e Zd Zdddd�dededededB d ed
df � fd d �Zdded ed
e fdd�Z
� Z S )�
RawTestServerrO rI Nr� r� rP r9 r@ rT r; c r� r� )�_handlerr� rZ )rY r� rP r9 r@ rT r� r= r> rZ � r� zRawTestServer.__init__T�debugc � s0 �t | jf| j|d�|��}t|fd|i|��S )N)rQ r� r� )r) r� rU r* )rY r� rT Zsrvr= r= r> rh s �zRawTestServer._make_runner)T) r� r� r� r- r| rK r rZ r� r* rh r� r= r= r� r> r� � s$ ������� r� c @ s^ e Zd ZdZdZedd�dddededB d ed
df
d d ��Z edd�dd
de
dedB d ed
df
dd ��Z ddd�de
dedB de j dB d ed
df
dd �Z d^dd�Z
ed
efdd��Zed
edB fdd��Zed
e
fdd��Zed
efdd��Zed
efd d ��Zd ed
efd d �Zd!ed ed ed
efd"d#�Z e j d$k�r
e �r
d!ed ed e e! d
e"fd%d&�Z#d ed e e! d
e"fd'd(�Z$d ed e e! d
e"fd)d*�Z%d ed e e! d
e"fd+d,�Z&d ed e e! d
e"fd-d.�Z'd ed e e! d
e"fd/d0�Z(d ed e e! d
e"fd1d2�Z)d ed e e! d
e"fd3d4�Z*nZd!ed ed ed
e"fd5d&�Z#d ed ed
e"fd6d(�Z$d ed ed
e"fd7d.�Z'd ed ed
e"fd8d*�Z%d ed ed
e"fd9d,�Z&d ed ed
e"fd:d0�Z(d ed ed
e"fd;d2�Z)d ed ed
e"fd<d4�Z*ed=d>�d ed?e+d@ d ed
dAfdBdC��Z,ed ed?e+d d ed
dDfdEdC��Z,ed=d>�d ed?e-d ed
dFfdGdC��Z,d@d>�d ed?e-d ed
dFfdHdC�Z,ed=d>�d ed?e+d@ d ed
dIfdJdK��Z.ed ed?e+d d ed
dLfdMdK��Z.ed=d>�d ed?e-d ed
dNfdOdK��Z.d@d>�d ed?e-d ed
dNfdPdK�Z.d^dQdR�Z/d^dSdT�Z0dUe1e2 dB dVe2dB dWe3dB d
dfdXdY�Z4d
e5fdZd[�Z6dUe1e2 dB dVe2dB dWe3dB d
dfd\d]�Z7dS )_�
TestClientzb
A test client implementation.
To write functional tests for aiohttp based servers.
FN)�
cookie_jarrY z TestClient[Request, Application]rt r� rT r; c K r� rw r= �rY rt r� rT r= r= r> rZ r� zTestClient.__init__zTestClient[_Request, None]c K r� rw r= r� r= r= r> rZ r� )r� rQ rQ c K sp t |t�s
tdt|� ��|| _|| _|d u r tjd|d�}td||d�|��| _ d| j _
d| _ g | _ g | _
d S )Nz2server must be TestServer instance, found type: %rT)ZunsaferQ )rQ r� Fr= )�
isinstancerN r� r� rp rU �aiohttpZ CookieJarr �_sessionZ_retry_connectionrX �
_responses� _websockets)rY rt r� rQ rT r= r= r> rZ ! s
�
c � s �| j j| jd�I d H d S r� )rp ru rU r r= r= r> ru 7 s �zTestClient.start_serverc C � | j jS rw )rp r9 r r= r= r> r9 : � zTestClient.hostc C r� rw )rp r@ r r= r= r> r@ > r� zTestClient.portc C r� rw )rp r r= r= r> rt B r� zTestClient.serverc C s t | jdd �S )Nr� )�getattrrp r r= r= r> r� F s zTestClient.appc C r� )z�An internal aiohttp.ClientSession.
Unlike the methods on the TestClient, client session requests
do not automatically include the host in the url queried, and
will require an absolute path to the resource.
)r� r r= r= r> �sessionJ s zTestClient.sessionry c C s | j �|�S rw )rp r~ )rY ry r= r= r> r~ U s zTestClient.make_url�methodc � s4 �| j j|| �|�fi |��I d H }| j�|� |S rw )r� �requestr~ r� �append)rY r� ry rT �respr= r= r> �_requestX s �" zTestClient._requestr/ c K r� rw r= �rY r� ry rT r= r= r> r� b s zTestClient.requestc K r� rw r= �rY ry rT r= r= r> �getf � zTestClient.getc K r� rw r= r� r= r= r> �optionsl r� zTestClient.optionsc K r� rw r= r� r= r= r> �headr r� zTestClient.headc K r� rw r= r� r= r= r> �postx r� zTestClient.postc K r� rw r= r� r= r= r> �put~ r� zTestClient.putc K r� rw r= r� r= r= r> �patch� r� zTestClient.patchc K r� rw r= r� r= r= r> �delete� r� zTestClient.deletec K s t | j||fi |���S )z�Routes a request to tested http server.
The interface is identical to aiohttp.ClientSession.request,
except the loop kwarg is overridden by the instance used by the
test server.
)r r� r� r= r= r> r� � s
c K � t | jtj|fi |���S )z Perform an HTTP GET request.)r r� r ZMETH_GETr� r= r= r> r� � � c K r� )z Perform an HTTP POST request.)r r� r Z METH_POSTr� r= r= r> r� � r� c K r� )z Perform an HTTP OPTIONS request.)r r� r Z METH_OPTIONSr� r= r= r> r� � � �c K r� )z Perform an HTTP HEAD request.)r r� r Z METH_HEADr� r= r= r> r� � r� c K r� )z Perform an HTTP PUT request.)r r� r ZMETH_PUTr� r= r= r> r� � r� c K r� �z Perform an HTTP PATCH request.)r r� r Z
METH_PATCHr� r= r= r> r� � r� c K r� r� )r r� r Z METH_DELETEr� r= r= r> r� � r� .)� decode_textr� TzB_BaseRequestContextManager[ClientWebSocketResponse[Literal[True]]]c K r� rw r= �rY ry r� rT r= r= r> �
ws_connect� � zTestClient.ws_connectzC_BaseRequestContextManager[ClientWebSocketResponse[Literal[False]]]c K r� rw r= r� r= r= r> r� � r� z9_BaseRequestContextManager[ClientWebSocketResponse[bool]]c K r� rw r= r� r= r= r> r� � r� c K s t | j|fd|i|���S )zjInitiate websocket connection.
The api corresponds to aiohttp.ClientSession.ws_connect.
r� )r � _ws_connectr� r= r= r> r� � s �z&ClientWebSocketResponse[Literal[True]]c � rv rw r= r� r= r= r> r� � � �zTestClient._ws_connectz'ClientWebSocketResponse[Literal[False]]c � rv rw r= r� r= r= r> r� � r� z ClientWebSocketResponse[bool]c � rv rw r= r� r= r= r> r� � r� c � s6 �| j j| �|�fd|i|��I d H }| j�|� |S )Nr� )r� r� r~ r� r� )rY ry r� rT �wsr= r= r> r� � s ��� � c � sd �| j s0| jD ]}|�� q| jD ] }|�� I dH q| j�� I dH | j�� I dH d| _ dS dS )aH Close all fixtures created by the test client.
After that point, the TestClient is no longer usable.
This is an idempotent function: running close multiple times
will not have any additional effects.
close is also run on exit when used as a(n) (asynchronous)
context manager.
NT)rX r� r� r� r� rp )rY r� r� r= r= r> r� � s �
�zTestClient.closec C r� r� r� r r= r= r> r� r� zTestClient.__enter__r� �exc�tbc C r� rw r= �rY r� r� r� r= r= r> r� r� zTestClient.__exit__c � s �| � � I d H | S rw )ru r r= r= r> r� s �zTestClient.__aenter__c � r� rw r� r� r= r= r> r� r� zTestClient.__aexit__r� )8r� r� r� �__doc__r� r
r� r r rZ rN r� r� ru r� r| r9 rK r@ rt r4 r� r r� r# r r~ r r� �sys� version_infor r2 r r r� r� r� r� r� r� r� r� r
r� r� r� r� r� r� r� r r� r3 r� r� r= r= r= r> r� s� ����� ������ � �����
�
���
����
���
���
���
���
���
���
��� ����
� ���� ���� ����� �����
� ���� ���� ����� �����
�
���
�
����r� c @ s� e Zd ZdZdefdd�Zdefdd�Zddd �Zdd
d �Zdd d
�Z ddd�Z
dede fdd�Z de de
eef fdd�ZdS )�AioHTTPTestCasea A base class to allow for unittest web applications using aiohttp.
Provides the following:
* self.client (aiohttp.test_utils.TestClient): an aiohttp test client.
* self.loop (asyncio.BaseEventLoop): the event loop in which the
application and server are running.
* self.app (aiohttp.web.Application): the application returned by
self.get_application()
Note that the TestClient's methods are asynchronous: you have to
execute function on the test client using asynchronous methods.
r; c � s
�| � � S )z�Get application.
This method should be overridden
to return the aiohttp.web.Application
object to test.
)�get_appr r= r= r> �get_application0 s �z AioHTTPTestCase.get_applicationc C r� )zqObsolete method used to constructing web application.
Use .get_application() coroutine instead.
z+Did you forget to define get_application()?)� RuntimeErrorr r= r= r> r� 9 s zAioHTTPTestCase.get_appNc � s �t �� | _| �� I d H S rw )r� �get_running_looprQ �
setUpAsyncr r= r= r> �
asyncSetUp@ s �
zAioHTTPTestCase.asyncSetUpc � sN �| � � I d H | _| �| j�I d H | _| �| j�I d H | _| j�� I d H d S rw )r� r� �
get_serverrt �
get_client�clientru r r= r= r> r� D s
�zAioHTTPTestCase.setUpAsyncc � s �| � � I d H S rw )�
tearDownAsyncr r= r= r> �
asyncTearDownK s �z AioHTTPTestCase.asyncTearDownc � s �| j �� I d H d S rw )r� r� r r= r= r> r� N s �z AioHTTPTestCase.tearDownAsyncr� c � � �t || jd�S )z Return a TestServer instance.r� )r� rQ )rY r� r= r= r> r� Q � �zAioHTTPTestCase.get_serverrt c � r� )z Return a TestClient instance.r� )r� rQ )rY rt r= r= r> r� U r� zAioHTTPTestCase.get_clientr� )r� r� r� r� r$ r� r� r� r� r� r� r� r� r� r( r� r= r= r= r> r� ! s
r� �func�argsrT c O s t jdtdd� | S )zw
A decorator dedicated to use with asynchronous AioHTTPTestCase test methods.
In 3.8+, this does nothing.
zBDecorator `@unittest_run_loop` is no longer needed in aiohttp 3.8+r` )�
stacklevel)�warnings�warn�DeprecationWarning)r� r� rT r= r= r> �unittest_run_loopZ s �r� F� loop_factory�fastc c s � t | �}|V t||d� dS )zzA contextmanager that creates an event_loop, for test purposes.
Handles the creation and cleanup of a test loop.
)r� N)�setup_test_loop�teardown_test_loop)r� r� rQ r= r= r> � loop_contextk s �r� c C s | � }t �|� |S )z�Create and return an asyncio.BaseEventLoop instance.
The caller should also call teardown_test_loop,
once they are done with the loop.
)r� �set_event_loop)r� rQ r= r= r> r� x s
r� rQ c C sB | � � }|s| �| j� | �� | �� |st�� t�d� dS )z>Teardown and cleanup an event_loop created by setup_test_loop.N) Z is_closedZ call_soon�stopZ run_foreverr� �gcZcollectr� r� )rQ r� r� r= r= r> r� � s r� c C sl dt dtdt fdd�} dt dtdt dd fdd�}tjtd �}i |_| |_||_d
|_t |�|_
|j
� � |S ) Nr� �keyr; c S s
| j | S rw ��
__app_dict)r� r� r= r= r> �get_dict� s
z"_create_app_mock.<locals>.get_dict�valuec S s || j |< d S rw r� )r� r� r r= r= r> �set_dict� s z"_create_app_mock.<locals>.set_dict)�specF) r r| r � MagicMockr$ r� � __getitem__� __setitem__Z_debugr Zon_response_prepareZfreeze)r� r r� r= r= r> �_create_app_mock� s
r �
sslcontextc s: t �� }dtdttttf B d B f� fdd� }||j_|S )Nr� r; c s | dkr� S | dkr dS d S )Nr �sockname)rI �P r= )r� �r r= r> �get_extra_info� s z)_create_transport.<locals>.get_extra_info)r �Mockr| r. �tuplerK r Z side_effect)r � transportr r= r
r> �_create_transport� s &r i .) �
match_inforl �closingr� �writer�protocolr �payloadr �client_max_sizerQ r� ry �headersr rl r r� r r r r r c C s� t �� }|
du r zt�� }
W n ty t �� }
d|
j_Y nw |tdd�k r)d}|r=tt |��}t
dd� |� � D ��}ntt � �}d}d|� t
jd��� v }|� t
jd��� d koat|� t
j��}t| |||||d
||t|��
}|d
u rxt� }| tu r�t| �} |tu r�t �� }d |_d |_d |_| |_t j | � d
�d�t |�_ t j | � d�d�t |�_ t j | d�t |�_!|tu r�t �� }t"d
�|_#t"d
�|_$t"d
�|_%t"d
�|_&| |_| |_||_'t(||
||||
| d�}t)|tu r�i n|t �� �}|�*|� ||_+|S )z�Creates mocked web.Request testing purposes.
Useful in unit tests, when spinning full web server is overkill or
specific conditions and errors are hard to trigger.
.r= r Tc s s( � | ]\}}|� d �|� d �fV qdS )zutf-8N)�encode)�.0�k�vr= r= r> � <genexpr>� s �
�z&make_mocked_request.<locals>.<genexpr>�chunkedrO �upgradeNi� � �peername)� return_valuer )r ),r r r� r� r� Z
create_futurer r r r r
�itemsr� r ZTRANSFER_ENCODING�lowerZ
CONNECTIONr� ZUPGRADEr r r r r Zmax_field_size�max_line_lengthZ max_headersr Z PropertyMockr r� r r rb �make_mocked_coroZ
write_headers�write� write_eofZdrainr r( r, Zadd_appZ _match_info)r� ry r r rl r r� r r r r r r rQ ZtaskZraw_hdrsr r �message�reqr= r= r> �make_mocked_request� s� �
�
�� � �
��
r) r �raise_exceptionc s( dt dt dt f� �fdd� }tj|d�S )zCreates a coroutine mock.r� rT r; c � s* �� t ur� �t���s�S �I d H d S rw )r �inspectZ isawaitable)r� rT �r* r r= r> � mock_coro! s �
z#make_mocked_coro.<locals>.mock_coro)�wraps)r r r )r r* r- r= r, r> r$ s r$ )Frw )mr� r� �
contextlibr� r+ rj �osrA r� r� �abcr r �collections.abcr r �typesr �typingr r r r
r r r
Zunittestr r Z aiosignalr Z multidictr r Zyarlr r� Zaiohttp.clientr r r r rO r r r Z
client_reqrepr Z client_wsr Zhelpersr rd r r Zstreamsr! r" Ztypedefsr# Zwebr$ r% r& r' r( r) r* r+ r, Z web_protocolr- r[ r. r� r2 r3 r4 r5 �name�platformrC rJ r| r� r? rK r< rM rN r� r� r� r� r� r� Z
_LOOP_FACTORY�contextmanagerZnew_event_loopr� r� r� r� r r r r r) r$ r= r= r= r> �<module> s $ ,
���
����
�
9��� �
��
�
��������� �
� � �
���
�l����