관리-도구
편집 파일: pytest_plugin.cpython-37.pyc
B �k�a�- � @ s d dl Z d dlZd dlZd dlmZ d dlmZmZmZmZm Z m Z m Z d dl Z d dl mZmZ d dlmZ ddlmZmZmZmZmZmZmZmZ y d dl Z W n e k r� dZ Y nX y d dl Z W n e k r� dZ Y nX ee eef gee f Z dd � Z d d � Z!e j"d d � �Z#e j"dd� �Z$ej%dd� �Z&ej%d0dd��Z'dd� Z(dd� Z)dd� Z*e j"dd � �Z+e j"d d � �Z,e j"d d � �Ze j"d!d"� �Z-e j"d#d$� �Z.e j"d%d&� �Z/e j"d'd(� �Z0e j"d)d*� �Z1e j"e j2e e ddf d+�d,d-��Z3e j"d.d/� �Z4dS )1� N)�Callable)�Any� Awaitabler �Dict� Generator�Optional�Union)�PY_37�isasyncgenfunction)� Application� )�BaseTestServer� RawTestServer� TestClient� TestServer� loop_context�setup_test_loop�teardown_test_loop� unused_portc C s: | j ddddd� | j dddd d� | j d ddd d� d S ) Nz--aiohttp-fast� store_trueFz*run tests faster by disabling extra checks)�action�default�helpz--aiohttp-loop�store�pyloopz:run tests with specific loop: pyloop, uvloop, tokio or allz--aiohttp-enable-loop-debugz enable event loop debug mode)Z addoption)�parser� r �H/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/pytest_plugin.py�pytest_addoption$ s r c sb | j � t� �rd�nt�� �r$d�ndS d�d| jkrH| jd7 _d�� ��fdd�}|| _ dS )zjSet up pytest fixture. Allow fixtures to be coroutines. Run coroutine fixtures in an event loop. TFN�request)r c sv |d }�r|d= d|j kr$td��|�d�� �rb�| |��� �fdd�}|�|� � ���� �S � ��| |��S d S )Nr �loopz^Asynchronous fixtures must depend on the 'loop' fixture or be used in tests depending from it.c s( y� � ��� �S tk r" Y nX d S )N)�run_until_complete� __anext__�StopAsyncIterationr )�_loop�genr r � finalizerd s z8pytest_fixture_setup.<locals>.wrapper.<locals>.finalizer)� fixturenames� ExceptionZgetfixturevalueZ addfinalizerr! r" )�args�kwargsr r& )�func� is_async_gen� strip_request)r$ r% r �wrapperO s z%pytest_fixture_setup.<locals>.wrapper)r+ r �asyncio�iscoroutinefunction�argnames)Z fixturedefr. r )r+ r, r- r �pytest_fixture_setup9 s r2 c C s | j �d�S )z--fast config optionz--aiohttp-fast)�config� getoption)r r r r �fastr s r5 c C s | j �d�S )z!--enable-loop-debug config optionz--aiohttp-enable-loop-debug)r3 r4 )r r r r � loop_debugx s r6 c c s` t jdd��J} dV dd� | D �}|rRtd�t|�t|�dkrBdnd d �|����W dQ R X dS ) z�Context manager which checks for RuntimeWarnings. This exists specifically to avoid "coroutine 'X' was never awaited" warnings being missed. If RuntimeWarnings occur in the context a RuntimeError is raised. T)�recordNc S s" g | ]}|j tkrd j|d��qS )z#{w.filename}:{w.lineno}:{w.message})�w)�category�RuntimeWarning�format)�.0r8 r r r � <listcomp>� s z,_runtime_warning_context.<locals>.<listcomp>z{} Runtime Warning{}, {}r � �s� )�warnings�catch_warnings� RuntimeErrorr; �len�join)� _warnings�rwr r r �_runtime_warning_context~ s rH Fc c s( | r | V nt � } | V t| |d� dS )z�Passthrough loop context. Sets up and tears down a loop unless one is passed in via the loop argument when it's passed straight through. )r5 N)r r )r r5 r r r �_passthrough_loop_context� s rI c C s( | � |�r$t�|�r$t| �||��S dS )z%Fix pytest collecting for coroutines.N)Zfuncnamefilterr/ r0 �listZ _genfunctions)Z collector�name�objr r r �pytest_pycollect_makeitem� s rM c s� � j �d�}t�� j�r�� j�d�p0� j�dd�}t� �F t||d��.}� fdd�� j j D �}|� � j f |�� W dQ R X W dQ R X dS dS ) zBRun coroutines in an event loop instead of a normal function call.z--aiohttp-fast� proactor_loopr N)r5 c s i | ]}� j | |�qS r )�funcargs)r< �arg)� pyfuncitemr r � <dictcomp>� s z&pytest_pyfunc_call.<locals>.<dictcomp>T) r3 r4 r/ r0 �functionrO �getrH rI Z _fixtureinfor1 r! rL )rQ r5 Z existing_loopr$ Ztestargsr )rQ r �pytest_pyfunc_call� s &rU c C s� d| j krd S | jjj}dtji}td k r4tj|d< td k rFtj|d< |dkrRd}i }x\|� d�D ]N}|� d� }|� d �}||kr�|rbt d |t |�� �f ��nqb|| ||< qbW | jdt |�� �t |�� �d � d S ) N� loop_factoryr �uvloop�tokio�allzpyloop,uvloop?,tokio?�,�?z ?z&Unknown loop '%s', available loops: %s)Zids)r' r3 �optionZ aiohttp_loopr/ ZDefaultEventLoopPolicyrW ZEventLoopPolicyrX �split�endswith�strip� ValueErrorrJ �keysZ parametrize�values)ZmetafuncZloopsZavail_factoriesZ factoriesrK �requiredr r r �pytest_generate_tests� s. rd c c sH | � }t �|� t|d��$}|r*|�d� t �|� |V W dQ R X dS )z%Return an instance of the event loop.)r5 TN)r/ �set_event_loop_policyr Z set_debug�set_event_loop)rV r5 r6 �policyr$ r r r r � s r c c sR t st�� } tj| _nt�� } t�| � t| j��}t� |� |V W d Q R X d S )N) r r/ Zget_event_loop_policyZProactorEventLoopZ _loop_factoryZ WindowsProactorEventLoopPolicyre r Znew_event_looprf )rg r$ r r r rN � s rN c C s t jdtdd� | S )Nz3Deprecated, use aiohttp_unused_port fixture instead� )� stacklevel)rA �warn�DeprecationWarning)�aiohttp_unused_portr r r r � s r c C s t S )z1Return a port that is unused on the current host.)� _unused_portr r r r rl s rl c # s@ g �dd�� �fdd� }|V dd��fdd� }� � |� � dS )z^Factory to create a TestServer instance, given an app. aiohttp_server(app, **kwargs) N)�portc � s4 t | |d�}|jf d� i|��I d H ��|� |S )N)rn r )r � start_server�append)Zapprn r* �server)r �serversr r �go s zaiohttp_server.<locals>.go)�returnc � s x� r� � � �� I d H qW d S )N)�pop�closer )rr r r �finalize s z aiohttp_server.<locals>.finalize)r! )r rs rw r )r rr r �aiohttp_server s rx c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_server fixture insteadrh )ri )rA rj rk )rx r r r � test_server" s ry c # s@ g �dd�� �fdd� }|V dd��fdd� }� � |� � dS )zpFactory to create a RawTestServer instance, given a web handler. aiohttp_raw_server(handler, **kwargs) N)rn c � s4 t | |d�}|jf d� i|��I d H ��|� |S )N)rn r )r ro rp )�handlerrn r* rq )r rr r r rs 4 s z aiohttp_raw_server.<locals>.go)rt c � s x� r� � � �� I d H qW d S )N)ru rv r )rr r r rw < s z$aiohttp_raw_server.<locals>.finalize)r! )r rs rw r )r rr r �aiohttp_raw_server, s r{ c C s t jdtdd� | S )Nz2Deprecated, use aiohttp_raw_server fixture insteadrh )ri )rA rj rk )r{ r r r �raw_test_serverC s r| )r rt c # sb g � dd�t ttf tttttf ttd�� �fdd�}|V dd�� fdd� }��|� � dS ) z�Factory to create a TestClient instance. aiohttp_client(app, **kwargs) aiohttp_client(server, **kwargs) aiohttp_client(raw_server, **kwargs) N)� server_kwargs)�__paramr) r} r* rt c � s� t | t�r.t | ttf�s.| �f|�|�} i }n |r:td��t | t�rv|pJi }t| fd�i|��}t|fd�i|��}n0t | t�r�t| fd�i|��}ntdt| � ��|� � I d H � � |� |S )Nzargs should be emptyr zUnknown argument type: %r) � isinstancer r r �AssertionErrorr r r` �typero rp )r~ r} r) r* rq Zclient)�clientsr r r rs [ s zaiohttp_client.<locals>.go)rt c � s x� r� � � �� I d H qW d S )N)ru rv r )r� r r rw y s z aiohttp_client.<locals>.finalize) r r r r r r �strr r! )r rs rw r )r� r r �aiohttp_clientO s r� c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_client fixture insteadrh )ri )rA rj rk )r� r r r � test_client� s r� )F)5r/ � contextlibrA �collections.abcr �typingr r r r r r ZpytestZaiohttp.helpersr r Z aiohttp.webr Z test_utilsr r r r r r r r rm rW � ImportErrorrX Z AiohttpClientr r2 Zfixturer5 r6 �contextmanagerrH rI rM rU rd r rN rl rx ry r{ r| ZAbstractEventLoopr� r� r r r r �<module> sN $ ( 9 " .