관리-도구
편집 파일: pytest_plugin.cpython-37.opt-1.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-loopZstore�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 r1 c C s | j �d�S )z--fast config optionz--aiohttp-fast)�config� getoption)r r r r �fastr s r4 c C s | j �d�S )z!--enable-loop-debug config optionz--aiohttp-enable-loop-debug)r2 r3 )r r r r � loop_debugx s r5 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)�.0r7 r r r � <listcomp>� s z,_runtime_warning_context.<locals>.<listcomp>z{} Runtime Warning{}, {}r � �s� )�warnings�catch_warnings� RuntimeErrorr: �len�join)� _warningsZrwr r r �_runtime_warning_context~ s rF 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. )r4 N)r r )r r4 r r r �_passthrough_loop_context� s rG c C s( | � |�r$t�|�r$t| �||��S dS )z%Fix pytest collecting for coroutines.N)Zfuncnamefilterr. r/ �listZ _genfunctions)Z collector�name�objr r r �pytest_pycollect_makeitem� s rK 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)r4 c s i | ]}� j | |�qS r )�funcargs)r; �arg)� pyfuncitemr r � <dictcomp>� s z&pytest_pyfunc_call.<locals>.<dictcomp>T) r2 r3 r. r/ �functionrM �getrF rG Z _fixtureinfor0 r rJ )rO r4 Z existing_loopr# Ztestargsr )rO r �pytest_pyfunc_call� s &rS 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& r2 ZoptionZ aiohttp_loopr. ZDefaultEventLoopPolicyrU ZEventLoopPolicyrV �split�endswith�strip� ValueErrorrH �keysZ parametrize�values)ZmetafuncZloopsZavail_factoriesZ factoriesrI Zrequiredr r r �pytest_generate_tests� s. r` c c sH | � }t �|� t|d��$}|r*|�d� t �|� |V W dQ R X dS )z%Return an instance of the event loop.)r4 TN)r. �set_event_loop_policyr Z set_debug�set_event_loop)rT r4 r5 �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 WindowsProactorEventLoopPolicyra r Znew_event_looprb )rc r# r r r rL � s rL c C s t jdtdd� | S )Nz3Deprecated, use aiohttp_unused_port fixture instead� )� stacklevel)r@ �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 rh s rh 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)rj r )r � start_server�append)Zapprj r) �server)r �serversr r �go s zaiohttp_server.<locals>.go)�returnc � s x� r� � � �� I d H qW d S )N)�pop�closer )rn r r �finalize s z aiohttp_server.<locals>.finalize)r )r ro rs r )r rn r �aiohttp_server s rt c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_server fixture insteadrd )re )r@ rf rg )rt r r r � test_server" s ru 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)rj c � s4 t | |d�}|jf d� i|��I d H ��|� |S )N)rj r )r rk rl )�handlerrj r) rm )r rn r r ro 4 s z aiohttp_raw_server.<locals>.go)rp c � s x� r� � � �� I d H qW d S )N)rq rr r )rn r r rs < s z$aiohttp_raw_server.<locals>.finalize)r )r ro rs r )r rn r �aiohttp_raw_server, s rw c C s t jdtdd� | S )Nz2Deprecated, use aiohttp_raw_server fixture insteadrd )re )r@ rf rg )rw r r r �raw_test_serverC s rx )r rp 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( ry r) rp 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] �typerk rl )rz ry r( r) rm Zclient)�clientsr r r ro [ s zaiohttp_client.<locals>.go)rp c � s x� r� � � �� I d H qW d S )N)rq rr r )r~ r r rs y s z aiohttp_client.<locals>.finalize) r r r r r r �strr r )r ro rs r )r~ r r �aiohttp_clientO s r� c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_client fixture insteadrd )re )r@ rf rg )r� r r r � test_client� s r� )F)5r. � contextlibr@ �collections.abcr �typingr r r r r r ZpytestZaiohttp.helpersr r Z aiohttp.webr Z test_utilsr r r r r r r r ri rU � ImportErrorrV Z AiohttpClientr r1 Zfixturer4 r5 �contextmanagerrF rG rK rS r` r rL rh rt ru rw rx ZAbstractEventLoopr� r� r r r r �<module> sN $ ( 9 " .