HOME

Mini Shell 1.0
DIR:/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/__pycache__/
Current File : //opt/alt/python38/lib64/python3.8/site-packages/aiohttp/__pycache__/test_utils.cpython-38.pyc
U

'�7`O�@s dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
mZmZddl
mZddlmZmZmZmZmZmZmZmZddl	mZddlmZmZddlmZddlZdd	lm Z m!Z!m"Z"d
dl#m$Z$m%Z%d
dl
m&Z&d
d
l'm Z d
dl(m)Z)d
dl*m+Z+d
dl,m-Z-m.Z.d
dl/m0Z0d
dl1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9d
dl:m;Z;e�rxddl<m=Z=ndZ=ej>dk�o�ej?dkZ@eAejd�dd�ZBeAeCejd�dd�ZDeCd�dd�ZEGd d!�d!e�ZFGd"d#�d#eF�ZGGd$d%�d%eF�ZHGd&d'�d'�ZIGd(d)�d)e	jJ�ZKeeeed*�d+d,�ZLegejMfZNejOejPd-feNeQeejMd.�d/d0��ZRejPfeNejMd1�d2d3�ZSdEejMeQdd4�d5d6�ZTejUd�d7d8�ZVdFee=ejWd9�d:d;�ZXdGe+e-d
d
�d-de+e+e+e+dd<d=d>�eAeAeee-eQeeeeeee=eCee5d?�d@dA�ZYe+e+feeedB�dCdD�ZZdS)HzUtilities shared by tests.�N)�ABC�abstractmethod)�
TracebackType)�
TYPE_CHECKING�Any�Callable�Iterator�List�Optional�Type�Union)�mock)�CIMultiDict�CIMultiDictProxy)�URL)�ClientResponse�_RequestContextManager�_WSRequestContextManager�)�
ClientSession�hdrs)�AbstractCookieJar)r)�ClientWebSocketResponse)�sentinel)�HttpVersion�RawRequestMessage)�Signal)�Application�	AppRunner�
BaseRunner�Request�Server�ServerRunner�SockSite�UrlMappingMatchInfo)�_RequestHandler)�
SSLContext�posix�cygwin)�host�returncCs
t|d�S)Nr)�get_port_socket)r)�r,�E/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/test_utils.py�get_unused_port_socket7sr.)r)�portr*cCs8t�tjtj�}tr&|�tjtjd�|�||f�|S)Nr)�socket�AF_INET�SOCK_STREAM�
REUSE_ADDRESSZ
setsockoptZ
SOL_SOCKETZSO_REUSEADDR�bind)r)r/�sr,r,r-r+;s
r+�r*c
Cs>t�tjtj��$}|�d�|��dW5QR�SQRXdS)z1Return a port that is unused on the current host.)�	127.0.0.1rrN)r0r1r2r4�getsockname)r5r,r,r-�unused_portFs
r9c	@s0eZdZdZeddddd�eeefee	j
eeeee
dd�dd�Zd#ee	j
e
dd�d	d
�Zee
ed�dd
��Zeed�dd�Zeed�dd��Zeed�dd��Zeed�dd��Zdd�dd�Zdd�dd�Zeeeeeeedd�dd�Zdd�dd �Z eeeeeeedd�d!d"�Z!dS)$�BaseTestServerFNr7)�scheme�loopr)r/�skip_url_asserts)r;r<r)r/r=�kwargsr*cKs4||_d|_d|_||_||_d|_||_||_dS)NF)�_loop�runner�_rootr)r/�_closedr;r=)�selfr;r<r)r/r=r>r,r,r-�__init__Ps
zBaseTestServer.__init__)r<r>r*c�s|jr
dS||_|�dd�|_|jf|�IdH|_|j��IdH|jsNd|_t|j|j�}|�	�dd�\|_|_t
|j||jd�}|��IdH|j}|dk	s�t
�|j}|dk	s�t
�|d�	�d|_|jtkr�|jr�d}nd}||_t|j�d|j�d	|j���|_dS)
N�sslr�)ZsockZssl_contextrZhttps�httpz://�:)r@r?�popZ_ssl�_make_runnerZsetupr/r+r)r8r#�start�_server�AssertionError�socketsr;rrrA)rCr<r>Z_sockZsite�serverrNr;r,r,r-�start_servercs.
zBaseTestServer.start_server�r>r*c�sdS�Nr,�rCr>r,r,r-rJszBaseTestServer._make_runner��pathr*cCsJ|jdk	st�t|�}|js4|��r(t�|j�|�Stt|j�|�SdSrR)rArMrr=Zis_absolute�join�str)rCrUZurlr,r,r-�make_url�szBaseTestServer.make_urlr6cCs
|jdk	SrR)r@�rCr,r,r-�started�szBaseTestServer.startedcCs|jSrR)rBrYr,r,r-�closed�szBaseTestServer.closedcCs&|j}|dk	st�|jdk	s t�|jSrR)r@rMrO)rCr@r,r,r-�handler�szBaseTestServer.handlerc�s@|jr<|js<|jdk	st�|j��IdHd|_d|_d|_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)rZr[r@rMZcleanuprAr/rBrYr,r,r-�close�szBaseTestServer.closecCstd��dS�NzUse async with instead��	TypeErrorrYr,r,r-�	__enter__�szBaseTestServer.__enter__)�exc_type�	exc_value�	tracebackr*cCsdSrRr,�rCrbrcrdr,r,r-�__exit__�szBaseTestServer.__exit__c�s|j|jd�IdH|S�N�r<)rPr?rYr,r,r-�
__aenter__�szBaseTestServer.__aenter__c�s|��IdHdSrR�r]rer,r,r-�	__aexit__�szBaseTestServer.__aexit__)N)"�__name__�
__module__�__qualname__�__test__rrrW�objectr
�asyncio�AbstractEventLoop�int�boolrrDrPrrrJrrX�propertyrZr[r!r\r]rar�
BaseExceptionrrfrirkr,r,r,r-r:MsV�
���	
�	
�r:csPeZdZeddd�eeeefeee	e
d��fdd�Ze
ed�dd	�Z
�ZS)
�
TestServerr7N�r;r)r/)�appr;r)r/r>cs$||_t�jf|||d�|��dS�Nrx)ry�superrD)rCryr;r)r/r>��	__class__r,r-rD�s	zTestServer.__init__rQc�st|jf|�SrR)rryrSr,r,r-rJ�szTestServer._make_runner)rlrmrnrrrrWrpr
rsrrDrrJ�
__classcell__r,r,r|r-rw�s�
�rwcsVeZdZeddd�eeeefeee	e
dd��fdd�Zdee
e
d�d	d
�Z�ZS)�
RawTestServerr7Nrx)r\r;r)r/r>r*cs$||_t�jf|||d�|��dSrz)�_handlerr{rD)rCr\r;r)r/r>r|r,r-rD�s	zRawTestServer.__init__T)�debugr>r*c�s.t|jf|j|d�|��}t|fd|i|��S)N)r<r�r�)r!r�r?r")rCr�r>Zsrvr,r,r-rJ�szRawTestServer._make_runner)T)rlrmrnrr%rrWrpr
rsrrDrtr"rJr~r,r,r|r-r�s�
�rc@s�eZdZdZdZddd�eeeeej	e
dd�dd�Zdd�d	d
�Ze
ed�dd��Ze
eed�d
d��Ze
ed�dd��Ze
ed�dd��Ze
ed�dd��Zeed�dd�Zeee
ed�dd�Zeee
ed�dd�Zee
ed�dd�Zee
ed�d d!�Zee
ed�d"d#�Zee
ed�d$d%�Z ee
ed�d&d'�Z!ee
ed�d(d)�Z"ee
ed�d*d+�Z#ee
e$d�d,d-�Z%ee
e&d�d.d/�Z'dd�d0d1�Z(dd�d2d3�Z)ee*e+ee+ee,dd4�d5d6�Z-dd�d7d8�Z.ee*e+ee+ee,dd4�d9d:�Z/dS);�
TestClientzb
    A test client implementation.

    To write functional tests for aiohttp based servers.

    FN)�
cookie_jarr<)rOr�r<r>r*cKsht|t�stdt|���||_||_|dkr<tjd|d�}tf||d�|��|_	d|_
g|_g|_dS)Nz2server must be TestServer instance, found type: %rT)Zunsafer<)r<r�F)
�
isinstancer:r`�typerLr?�aiohttpZ	CookieJarr�_sessionrB�
_responses�_websockets)rCrOr�r<r>r,r,r-rD�s

�zTestClient.__init__r6c�s|jj|jd�IdHdSrg)rLrPr?rYr,r,r-rPszTestClient.start_servercCs|jjSrR)rLr)rYr,r,r-r)szTestClient.hostcCs|jjSrR)rLr/rYr,r,r-r/szTestClient.portcCs|jSrR)rLrYr,r,r-rOszTestClient.servercCst|jdd�S)Nry)�getattrrLrYr,r,r-ryszTestClient.appcCs|jS)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�rYr,r,r-�sessions	zTestClient.sessionrTcCs|j�|�SrR)rLrX)rCrUr,r,r-rX)szTestClient.make_url)�methodrUr>r*c�s.|jj||�|�f|�IdH}|j�|�|SrR)r��requestrXr��append)rCr�rUr>�respr,r,r-�_request,szTestClient._requestcKst|j||f|��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.

        )rr�)rCr�rUr>r,r,r-r�2szTestClient.request)rUr>r*cKst|jtj|f|��S)zPerform an HTTP GET request.)rr�rZMETH_GET�rCrUr>r,r,r-�get<szTestClient.getcKst|jtj|f|��S)zPerform an HTTP POST request.)rr�rZ	METH_POSTr�r,r,r-�post@szTestClient.postcKst|jtj|f|��S)z Perform an HTTP OPTIONS request.)rr�rZMETH_OPTIONSr�r,r,r-�optionsDszTestClient.optionscKst|jtj|f|��S)zPerform an HTTP HEAD request.)rr�rZ	METH_HEADr�r,r,r-�headHszTestClient.headcKst|jtj|f|��S)zPerform an HTTP PUT request.)rr�rZMETH_PUTr�r,r,r-�putLszTestClient.putcKst|jtj|f|��S�zPerform an HTTP PATCH request.)rr�rZ
METH_PATCHr�r,r,r-�patchPszTestClient.patchcKst|jtj|f|��Sr�)rr�rZMETH_DELETEr�r,r,r-�deleteTszTestClient.deletecKst|j|f|��S)zjInitiate websocket connection.

        The api corresponds to aiohttp.ClientSession.ws_connect.

        )r�_ws_connectr�r,r,r-�
ws_connectXszTestClient.ws_connectc�s,|jj|�|�f|�IdH}|j�|�|SrR)r�r�rXr�r�)rCrUr>�wsr,r,r-r�`szTestClient._ws_connectc�s^|jsZ|jD]}|��q|jD]}|��IdHq |j��IdH|j��IdHd|_dS)aHClose 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)rBr�r]r�r�rL)rCr�r�r,r,r-r]es


zTestClient.closecCstd��dSr^r_rYr,r,r-razszTestClient.__enter__)rb�exc�tbr*cCsdSrRr,�rCrbr�r�r,r,r-rf}szTestClient.__exit__c�s|��IdH|SrR)rPrYr,r,r-ri�szTestClient.__aenter__c�s|��IdHdSrRrjr�r,r,r-rk�szTestClient.__aexit__)0rlrmrn�__doc__ror:r
rrqrrrrDrPrurWr)rsr/rOrryrr�rrXrr�rr�r�r�r�r�r�r�r�rr�rr�r]rarrvrrfrirkr,r,r,r-r��s^��


�	
�r�c@s�eZdZdZed�dd�Zed�dd�Zdd�dd	�Zdd�d
d�Zdd�dd
�Z	dd�dd�Z
eed�dd�Zee
d�dd�ZdS)�AioHTTPTestCaseaA 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.
    r6c�s|��S)zy
        This method should be overridden
        to return the aiohttp.web.Application
        object to test.

        )�get_apprYr,r,r-�get_application�szAioHTTPTestCase.get_applicationcCstd��dS)zqObsolete method used to constructing web application.

        Use .get_application() coroutine instead

        z+Did you forget to define get_application()?N)�RuntimeErrorrYr,r,r-r��szAioHTTPTestCase.get_appNcCslt�|_|j�|���|_|j�|�|j��|_|j�|�|j��|_|j�|j�	��|j�|�
��dSrR)�setup_test_loopr<�run_until_completer�ry�
get_serverrO�
get_client�clientrP�
setUpAsyncrYr,r,r-�setUp�szAioHTTPTestCase.setUpc�sdSrRr,rYr,r,r-r��szAioHTTPTestCase.setUpAsynccCs0|j�|���|j�|j���t|j�dSrR)r<r��
tearDownAsyncr�r]�teardown_test_looprYr,r,r-�tearDown�szAioHTTPTestCase.tearDownc�sdSrRr,rYr,r,r-r��szAioHTTPTestCase.tearDownAsync)ryr*c�st||jd�S)zReturn a TestServer instance.rh)rwr<)rCryr,r,r-r��szAioHTTPTestCase.get_server)rOr*c�st||jd�S)zReturn a TestClient instance.rh)r�r<)rCrOr,r,r-r��szAioHTTPTestCase.get_client)rlrmrnr�rr�r�r�r�r�r�rwr�r�r�r,r,r,r-r��s	r�)�func�argsr>r*cs.tj�f|�|�ttttd��fdd��}|S)z�A decorator dedicated to use with asynchronous methods of an
    AioHTTPTestCase.

    Handles executing an asynchronous function, using
    the self.loop of the AioHTTPTestCase.
    )rC�
inner_args�inner_kwargsr*cs|j��|f|�|��SrR)r<r�)rCr�r��r�r,r-�new_func�sz#unittest_run_loop.<locals>.new_func)�	functools�wrapsr)r�r�r>r�r,r�r-�unittest_run_loop�sr�F)�loop_factory�fastr*ccst|�}|Vt||d�dS)zzA contextmanager that creates an event_loop, for test purposes.

    Handles the creation and cleanup of a test loop.
    )r�N)r�r�)r�r�r<r,r,r-�loop_context�sr�)r�r*c	Cs�|�}z|jj}d|k}Wntk
r2d}YnXt�|�tjdkr�|s�t��}t��}|�	|�t
�t��|�
|�W5QRX|S)z�Create and return an asyncio.BaseEventLoop
    instance.

    The caller should also call teardown_test_loop,
    once they are done with the loop.
    ZuvloopTZwin32)r}rm�AttributeErrorrq�set_event_loop�sys�platformZget_event_loop_policyZSafeChildWatcherZattach_loop�
contextlib�suppress�NotImplementedErrorZset_child_watcher)r�r<�moduleZskip_watcherZpolicyZwatcherr,r,r-r��s	


r�)r<r�r*cCsB|��}|s(|�|j�|��|��|s4t��t�d�dS)zHTeardown and cleanup an event_loop created
    by setup_test_loop.

    N)	Z	is_closedZ	call_soon�stopZrun_foreverr]�gcZcollectrqr�)r<r�r[r,r,r-r�sr�cCs^tttd�dd�}tttdd�dd�}t��}i|_||_||_d|_t|�|_	|j	�
�|S)N)ry�keyr*cSs
|j|SrR��
__app_dict)ryr�r,r,r-�get_dictsz"_create_app_mock.<locals>.get_dict)ryr��valuer*cSs||j|<dSrRr�)ryr�r�r,r,r-�set_dict!sz"_create_app_mock.<locals>.set_dictF)rrWr
�	MagicMockr��__getitem__�__setitem__Z_debugrZon_response_prepareZfreeze)r�r�ryr,r,r-�_create_app_mocks

r�)�
sslcontextr*cs,t��}tttd��fdd�}||j_|S)N)r�r*cs|dkr�SdSdS)Nr�r,)r��r�r,r-�get_extra_info2sz)_create_transport.<locals>.get_extra_info)r
�MockrWr
r&r�Zside_effect)r��	transportr�r,r�r-�_create_transport/sr�i.)�
match_info�version�closingry�writer�protocolr��payloadr��client_max_sizer<)r�rU�headersr�r�r�ryr�r�r�r�r�r�r<r*cCstt��}|
dkr t��}
d|
j_|tdd�kr2d}|rZtt|��}tdd�|��D��}ntt��}d}d|�	t
jd���k}t
||||||d	d	|t|��
}|d
kr�t�}|	tkr�t|�}	|tkr�t��}|	|_|tk�rt��}td
�|_td
�|_td
�|_td
�|_|	|_|	|_||_|
tk�r.t��}
t||
||||
|d�}t|tk�rTin|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,rTcss&|]\}}|�d�|�d�fVqdS)zutf-8N)�encode)�.0�k�vr,r,r-�	<genexpr>^sz&make_mocked_request.<locals>.<genexpr>�chunked�FN)r�)r
r�Z
create_future�return_valuerrr�tuple�itemsr�rZTRANSFER_ENCODING�lowerrrr�rr�r��make_mocked_coroZ
write_headers�writeZ	write_eofZdrainr�r r$Zadd_appZ_match_info)r�rUr�r�r�r�ryr�r�r�r�r�r�r<ZtaskZraw_hdrsr��messageZreqr,r,r-�make_mocked_request<sx�

�





��
r�)r��raise_exceptionr*cs$tttd���fdd�}tj|d�S)zCreates a coroutine mock.)r�r>r*c�s(�tk	r��t���s�S�IdHdSrR)r�inspectZisawaitable)r�r>�r�r�r,r-�	mock_coro�s

z#make_mocked_coro.<locals>.mock_coro)r�)rr
r�)r�r�r�r,r�r-r��sr�)F)N)N)[r�rqr�r�r�r��osr0r�Zunittest�abcrr�typesr�typingrrrrr	r
rrr
Z	multidictrrZyarlrr�Zaiohttp.clientrrrr�rrrZ
client_reqrepZ	client_wsrZhelpersrrGrrZsignalsrZwebrrrr r!r"r#r$Zweb_protocolr%rEr&�namer�r3rWr.rsr+r9r:rwrr�ZTestCaser�r�rrZ
_LOOP_FACTORY�contextmanagerZnew_event_looprtr�r�r�r�r�r�r�r�r�r,r,r,r-�<module>s�((
|(@��
�����]��