관리-도구
편집 파일: test_utils.cpython-37.opt-1.pyc
B �k�a�R � @ sP d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z ddl mZ ddlmZmZmZmZmZmZmZmZmZ ddlmZ ddlm Z ddl m Z m Z dd l m!Z! ddl"Z"dd l#m$Z$m%Z% d d l&m'Z'm(Z( d d l m)Z) d dl*m+Z+ d dl,m-Z- d dl.m/Z/m0Z0 d dl1m2Z2m3Z3 d dl4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z< d dl=m>Z> e�r|ddl?m@Z@ ndZ@e/�r�ddlmAZB n ddlCmBZB ejDdk�o�ejEdkZFejGfeHejIejd�dd�ZJeHeKejIejd �d d �ZLeKd �d d!�ZMG d"d#� d#e �ZNG d$d%� d%eN�ZOG d&d'� d'eN�ZPG d(d)� d)�ZQG d*d+� d+eB�ZReeeed,�d-d.�ZSeg ejTf ZUejVejWd/feUeXeejT d0�d1d2��ZYejWfeUejTd3�d4d5�ZZdGejTeXdd6�d7d8�Z[ej\d �d9d:�Z]dHee@ ej^d;�d<d=�Z_dIe0e2d d �d/de0e0e0e0dd>d?d@� eHeHeee2eXeeeeeee@ eKee8dA�dBdC�Z`e0e0feeedD�dEdF�ZadS )JzUtilities shared by tests.� N)�ABC�abstractmethod)� TracebackType) � TYPE_CHECKING�Any�Callable�Iterator�List�Optional�Type�Union�cast)�mock)�Signal)� CIMultiDict�CIMultiDictProxy)�URL)�_RequestContextManager�_WSRequestContextManager� )� ClientSession�hdrs)�AbstractCookieJar)�ClientResponse)�ClientWebSocketResponse)�PY_38�sentinel)� HttpVersion�RawRequestMessage)� Application� AppRunner� BaseRunner�Request�Server� ServerRunner�SockSite�UrlMappingMatchInfo)�_RequestHandler)� SSLContext)�IsolatedAsyncioTestCase)�TestCase�posix�cygwin)�host�family�returnc C s t | d|�S )Nr )�get_port_socket)r- r. � r1 �E/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/test_utils.py�get_unused_port_socketA s r3 )r- �portr. r/ c C s6 t � |t j�}tr$|�t jt jd� |�| |f� |S )Nr )�socket� SOCK_STREAM� REUSE_ADDRESSZ setsockoptZ SOL_SOCKETZ SO_REUSEADDR�bind)r- r4 r. �sr1 r1 r2 r0 G s r0 )r/ c C s8 t � t jt j�� } | �d� tt| �� d �S Q R X dS )z1Return a port that is unused on the current host.)z 127.0.0.1r r N)r5 �AF_INETr6 r8 r �int� getsockname)r9 r1 r1 r2 � unused_portT s r= c @ sF e Zd ZdZedddded�eeef e e j ee e e eee ejgejf edd�dd�Zd#e e j edd�d d �Zeeed �d d ��Zeed�dd�Zee d�dd��Zee d�dd��Zee d�dd��Z dd�dd�Z dd�dd�Z e e e! e e! e e" dd �d d �Z#d d�d d �Z$e e e! e e! e e" dd �d!d"�Z%dS )$�BaseTestServerFNz 127.0.0.1)�scheme�loopr- r4 �skip_url_asserts�socket_factory)r? r@ r- r4 rA rB �kwargsr/ c K s: || _ d | _d | _|| _|| _d| _|| _|| _|| _d S )NF) �_loop�runner�_rootr- r4 �_closedr? rA rB )�selfr? r@ r- r4 rA rB rC r1 r1 r2 �__init__^ s zBaseTestServer.__init__)r@ rC r/ c � sV | j r d S || _|�dd �| _| jf |�I d H | _ | j �� I d H | jsNd| _yt�| j �j }W n t k rx d}Y nX |dkr�t j nt j}| �| j | j|�}|�� d d� \| _ | _t| j || jd�}|�� I d H |j}|d k s�t�|j}|d k s�t�|d �� d | _| jtk�r2| j�r(d} nd } | | _t| j� d | j � d | j� ��| _d S ) N�sslr � � � )ZsockZ ssl_contextr Zhttps�httpz://�:)rE rD �popZ_ssl� _make_runnerZsetupr4 � ipaddressZ ip_addressr- �version� ValueErrorr5 ZAF_INET6r: rB r< r% �start�_server�AssertionError�socketsr? r r rF ) rH r@ rC rS r. Z_sockZsite�serverrX r? r1 r1 r2 � start_serveru s8 zBaseTestServer.start_server)rC r/ c � s d S )Nr1 )rH rC r1 r1 r2 rQ � s zBaseTestServer._make_runner)�pathr/ c C sJ | j d k st�t|�}| js4|�� r(t�| j �|�S tt| j �| �S d S )N)rF rW r rA Z is_absolute�join�str)rH r[ Zurlr1 r1 r2 �make_url� s zBaseTestServer.make_url)r/ c C s | j d k S )N)rE )rH r1 r1 r2 �started� s zBaseTestServer.startedc C s | j S )N)rG )rH r1 r1 r2 �closed� s zBaseTestServer.closedc C s& | j }|d k st�|jd k s t�|jS )N)rE rW rY )rH rE r1 r1 r2 �handler� s zBaseTestServer.handlerc � s@ | j r<| js<| jdk st�| j�� I dH d| _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)r_ r` rE rW ZcleanuprF r4 rG )rH r1 r1 r2 �close� s zBaseTestServer.closec C s t d��d S )NzUse async with instead)� TypeError)rH r1 r1 r2 � __enter__� s zBaseTestServer.__enter__)�exc_type� exc_value� tracebackr/ c C s d S )Nr1 )rH re rf rg r1 r1 r2 �__exit__� s zBaseTestServer.__exit__c � s | j | jd�I d H | S )N)r@ )rZ rD )rH r1 r1 r2 � __aenter__� s zBaseTestServer.__aenter__c � s | � � I d H d S )N)rb )rH re rf rg r1 r1 r2 � __aexit__� s zBaseTestServer.__aexit__)N)&�__name__� __module__� __qualname__�__test__r r0 r r] �objectr �asyncio�AbstractEventLoopr; �boolr r5 � AddressFamilyr rI rZ r r! rQ r r^ �propertyr_ r` r# ra rb rd r � BaseExceptionr rh ri rj r1 r1 r1 r2 r> [ s> 6 r> c sP e Zd Zeddd�eeeef eee e d�� fdd�Z e e d�dd �Z � ZS ) � TestServerz 127.0.0.1N)r? r- r4 )�appr? r- r4 rC c s$ || _ t� jf |||d�|�� d S )N)r? r- r4 )rw �superrI )rH rw r? r- r4 rC )� __class__r1 r2 rI � s zTestServer.__init__)rC r/ c � s t | jf|�S )N)r rw )rH rC r1 r1 r2 rQ � s zTestServer._make_runner)rk rl rm r r r r] ro r r; r rI r! rQ � __classcell__r1 r1 )ry r2 rv � s rv c sV e Zd Zeddd�eeeef eee e dd�� fdd�Z d e e e d�d d �Z� ZS ) � RawTestServerz 127.0.0.1N)r? r- r4 )ra r? r- r4 rC r/ c s$ || _ t� jf |||d�|�� d S )N)r? r- r4 )�_handlerrx rI )rH ra r? r- r4 rC )ry r1 r2 rI � s zRawTestServer.__init__T)�debugrC r/ c � s. t | jf| j|d�|��}t|fd|i|��S )N)r@ r} r} )r# r| rD r$ )rH r} rC Zsrvr1 r1 r2 rQ � s zRawTestServer._make_runner)T)rk rl rm r r' r r] ro r r; r rI rr r$ rQ rz r1 r1 )ry r2 r{ � s r{ c @ s� e Zd ZdZdZddd�eee eej e dd�dd�Z dd�d d �Z e ed�d d ��Ze ee d�d d��Ze ed�dd��Ze ee d�dd��Ze ed�dd��Zeed�dd�Zeee ed�dd�Zeee ed�dd �Z ee ed �d d �Z ee ed �d d!�Z ee ed �d"d#�Z ee 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-d d�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@ )rY r r@ rC r/ c K sh t |t�stdt|� ��|| _|| _|d kr<tjd|d�}tf ||d�|��| _ d| _ g | _ g | _ d S )Nz2server must be TestServer instance, found type: %rT)Zunsafer@ )r@ r F) � isinstancer> rc �typerV rD �aiohttpZ CookieJarr �_sessionrG � _responses� _websockets)rH rY r r@ rC r1 r1 r2 rI s zTestClient.__init__)r/ c � s | j j| jd�I d H d S )N)r@ )rV rZ rD )rH r1 r1 r2 rZ " s zTestClient.start_serverc C s | j jS )N)rV r- )rH r1 r1 r2 r- % s zTestClient.hostc C s | j jS )N)rV r4 )rH r1 r1 r2 r4 ) s zTestClient.portc C s | j S )N)rV )rH r1 r1 r2 rY - s zTestClient.serverc C s t tt t| jdd ��S )Nrw )r r r �getattrrV )rH r1 r1 r2 rw 1 s zTestClient.appc C s | j S )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� )rH r1 r1 r2 �session5 s zTestClient.session)r[ r/ c C s | j �|�S )N)rV r^ )rH r[ r1 r1 r2 r^ @ s zTestClient.make_url)�methodr[ rC r/ c � s. | j j|| �|�f|�I d H }| j�|� |S )N)r� �requestr^ r� �append)rH r� r[ rC �respr1 r1 r2 �_requestC s zTestClient._requestc K s t | 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. )r r� )rH r� r[ rC r1 r1 r2 r� I s zTestClient.request)r[ rC r/ c K s t | jtj|f|��S )z Perform an HTTP GET request.)r r� r ZMETH_GET)rH r[ rC r1 r1 r2 �getS s zTestClient.getc K s t | jtj|f|��S )z Perform an HTTP POST request.)r r� r Z METH_POST)rH r[ rC r1 r1 r2 �postW s zTestClient.postc K s t | jtj|f|��S )z Perform an HTTP OPTIONS request.)r r� r Z METH_OPTIONS)rH r[ rC r1 r1 r2 �options[ s zTestClient.optionsc K s t | jtj|f|��S )z Perform an HTTP HEAD request.)r r� r Z METH_HEAD)rH r[ rC r1 r1 r2 �head_ s zTestClient.headc K s t | jtj|f|��S )z Perform an HTTP PUT request.)r r� r ZMETH_PUT)rH r[ rC r1 r1 r2 �putc s zTestClient.putc K s t | jtj|f|��S )z Perform an HTTP PATCH request.)r r� r Z METH_PATCH)rH r[ rC r1 r1 r2 �patchg s zTestClient.patchc K s t | jtj|f|��S )z Perform an HTTP PATCH request.)r r� r Z METH_DELETE)rH r[ rC r1 r1 r2 �deletek s zTestClient.deletec K s t | j|f|��S )zjInitiate websocket connection. The api corresponds to aiohttp.ClientSession.ws_connect. )r � _ws_connect)rH r[ rC r1 r1 r2 � ws_connecto s zTestClient.ws_connectc � s, | j j| �|�f|�I d H }| j�|� |S )N)r� r� r^ r� r� )rH r[ rC �wsr1 r1 r2 r� w s zTestClient._ws_connectc � sf | j sbx| jD ] }|�� qW x | jD ]}|�� I dH q&W | j�� I dH | j�� I dH d| _ 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)rG r� rb r� r� rV )rH r� r� r1 r1 r2 rb | s zTestClient.closec C s t d��d S )NzUse async with instead)rc )rH r1 r1 r2 rd � s zTestClient.__enter__)re �exc�tbr/ c C s d S )Nr1 )rH re r� r� r1 r1 r2 rh � s zTestClient.__exit__c � s | � � I d H | S )N)rZ )rH r1 r1 r2 ri � s zTestClient.__aenter__c � s | � � I d H d S )N)rb )rH re r� r� r1 r1 r2 rj � s zTestClient.__aexit__)0rk rl rm �__doc__rn r> r r rp rq r rI rZ rt r] r- r; r4 rY r rw r r� r r^ r r� r r� r� r� r� r� r� r� r� r r� r r� rb rd r ru r rh ri rj r1 r1 r1 r2 r~ sP r~ c @ s� e Zd ZdZed�dd�Zed�dd�Zdd�dd �Zdd�d d �Zdd�d d �Z dd�dd�Z ee d�dd�Z e e d�dd�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_app)rH r1 r1 r2 �get_application� s z AioHTTPTestCase.get_applicationc C s t d��dS )zqObsolete method used to constructing web application. Use .get_application() coroutine instead. z+Did you forget to define get_application()?N)� RuntimeError)rH r1 r1 r2 r� � s zAioHTTPTestCase.get_appNc C sJ yt �� | _W n& ttfk r4 t �� �� | _Y nX | j�| �� � d S )N) rp Zget_running_loopr@ �AttributeErrorr� �get_event_loop_policyZget_event_loop�run_until_complete� setUpAsync)rH r1 r1 r2 �setUp� s zAioHTTPTestCase.setUpc � sL | � � I d H | _| �| j�I d H | _| �| j�I d H | _| j�� I d H d S )N)r� rw � get_serverrY � get_client�clientrZ )rH r1 r1 r2 r� � s zAioHTTPTestCase.setUpAsyncc C s | j �| �� � d S )N)r@ r� � tearDownAsync)rH r1 r1 r2 �tearDown� s zAioHTTPTestCase.tearDownc � s | j �� I d H d S )N)r� rb )rH r1 r1 r2 r� � s z AioHTTPTestCase.tearDownAsync)rw r/ c � s t || jd�S )z Return a TestServer instance.)r@ )rv r@ )rH rw r1 r1 r2 r� � s zAioHTTPTestCase.get_server)rY r/ c � s t || jd�S )z Return a TestClient instance.)r@ )r~ r@ )rH rY r1 r1 r2 r� � s zAioHTTPTestCase.get_client)rk rl rm r� r r� r� r� r� r� r� rv r� r~ r� r1 r1 r1 r2 r� � s r� )�func�argsrC r/ 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+rM )� stacklevel)�warnings�warn�DeprecationWarning)r� r� rC r1 r1 r2 �unittest_run_loop� s r� F)� loop_factory�fastr/ c 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� r@ r1 r1 r2 � loop_context� s r� )r� r/ c C s� | � }y|j j}d|k}W n tk r2 d}Y nX t�|� tjdkr�|s�t�� }y t�� }W n tk r| t� � }Y nX |� |� t � t �� |�|� W dQ R X |S )z�Create and return an asyncio.BaseEventLoop instance. The caller should also call teardown_test_loop, once they are done with the loop. ZuvloopT�win32N)ry rl r� rp �set_event_loop�sys�platformr� ZThreadedChildWatcherZSafeChildWatcherZ attach_loop� contextlib�suppress�NotImplementedErrorZset_child_watcher)r� r@ �moduleZ skip_watcherZpolicyZwatcherr1 r1 r2 r� s" r� )r@ r� r/ c C sB | � � }|s(| �| j� | �� | �� |s4t�� t�d� dS )z>Teardown and cleanup an event_loop created by setup_test_loop.N) Z is_closedZ call_soon�stopZ run_foreverrb �gcZcollectrp r� )r@ r� r` r1 r1 r2 r� ' s r� c C s^ t tt d�dd�} t tt d d�dd�}t�� }i |_| |_||_d|_t|�|_ |j � � |S )N)rw �keyr/ c S s | j | S )N)� __app_dict)rw r� r1 r1 r2 �get_dict6 s z"_create_app_mock.<locals>.get_dict)rw r� �valuer/ c S s || j |< d S )N)r� )rw r� r� r1 r1 r2 �set_dict9 s z"_create_app_mock.<locals>.set_dictF) r r] r � MagicMockr� � __getitem__� __setitem__Z_debugr Zon_response_prepareZfreeze)r� r� rw r1 r1 r2 �_create_app_mock5 s r� )� sslcontextr/ c s, t �� }ttt d�� fdd� }||j_|S )N)r� r/ c s | dkr � S d S d S )Nr� r1 )r� )r� r1 r2 �get_extra_infoJ s z)_create_transport.<locals>.get_extra_info)r �Mockr] r r( r� Z side_effect)r� � transportr� r1 )r� r2 �_create_transportG s r� i .) � match_inforS �closingrw �writer�protocolr� �payloadr� �client_max_sizer@ )r� r[ �headersr� rS r� rw r� r� r� r� r� r� r@ r/ c C st t �� }| dkr t �� } d| j_|tdd�k r2d}|rZtt|��}tdd� |�� D ��}ntt� �}d}d|� t j d�� � 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�rTi 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. .r1 r Tc s s& | ] \}}|� d �|� d �fV qdS )zutf-8N)�encode)�.0�k�vr1 r1 r2 � <genexpr>u s z&make_mocked_request.<locals>.<genexpr>�chunked� NF)r� ) r r� Z create_future� return_valuer r r �tuple�itemsr� r ZTRANSFER_ENCODING�lowerr r r� r r� r� �make_mocked_coroZ write_headers�writeZ write_eofZdrainr� r" r& Zadd_appZ _match_info)r� r[ r� r� rS r� rw r� r� r� r� r� r� r@ ZtaskZraw_hdrsr� �messageZreqr1 r1 r2 �make_mocked_requestT sb r� )r� �raise_exceptionr/ c s$ t t t d�� �fdd� }tj|d�S )zCreates a coroutine mock.)r� rC r/ c � s( � t k r � �t���s�S �I d H d S )N)r �inspectZ isawaitable)r� rC )r� r� r1 r2 � mock_coro� s z#make_mocked_coro.<locals>.mock_coro)�wraps)r r r� )r� r� r� r1 )r� r� r2 r� � s r� )F)N)N)br� rp r� r� r� rR �osr5 r� r� �abcr r �typesr �typingr r r r r r r r r Zunittestr Z aiosignalr Z multidictr r Zyarlr r� Zaiohttp.clientr r r� r r r Z client_reqrepr Z client_wsr Zhelpersr r rN r r Zwebr r r! r" r# r$ r% r& Z web_protocolr' rJ r( r) r* Z asynctest�namer� r7 r: r] rs r3 r; r0 r= r> rv r{ r~ r� r� rq Z _LOOP_FACTORY�contextmanagerZnew_event_looprr r� r� r� r� r� r� r� r� r� r1 r1 r1 r2 �<module> s� , ( (= &K