File: //opt/alt/python37/lib64/python3.7/site-packages/aiohttp/__pycache__/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-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/ ZfunctionrM �getrF rG Z _fixtureinfor0 r rJ )rO r4 Z
existing_loopr# Ztestargsr )rO r �pytest_pyfunc_call� s
&rR 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. ZDefaultEventLoopPolicyrT ZEventLoopPolicyrU �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)rS 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 WindowsProactorEventLoopPolicyr` r Znew_event_loopra )rb 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 rg s rg 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)ri r )r � start_server�append)Zappri r) �server)r �serversr r �go s
zaiohttp_server.<locals>.go)�returnc � s x� r� � � �� I d H qW d S )N)�pop�closer )rm r r �finalize s z aiohttp_server.<locals>.finalize)r )r rn rr r )r rm r �aiohttp_server s
rs c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_server fixture insteadrc )rd )r@ re rf )rs r r r � test_server" s
rt 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)ri c � s4 t | |d�}|jf d� i|��I d H ��|� |S )N)ri r )r rj rk )Zhandlerri r) rl )r rm r r rn 4 s
z aiohttp_raw_server.<locals>.go)ro c � s x� r� � � �� I d H qW d S )N)rp rq r )rm r r rr < s z$aiohttp_raw_server.<locals>.finalize)r )r rn rr r )r rm r �aiohttp_raw_server, s
ru c C s t jdtdd� | S )Nz2Deprecated, use aiohttp_raw_server fixture insteadrc )rd )r@ re rf )ru r r r �raw_test_serverC s
rv )r ro 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( rw r) ro 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\ �typerj rk )rx rw r( r) rl Zclient)�clientsr r r rn [ s
zaiohttp_client.<locals>.go)ro c � s x� r� � � �� I d H qW d S )N)rp rq r )r| r r rr y s z aiohttp_client.<locals>.finalize) r r r
r r r �strr r )r rn rr r )r| r r �aiohttp_clientO s
r~ c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_client fixture insteadrc )rd )r@ re rf )r~ r r r � test_client� s
r )F)5r. �
contextlibr@ Zcollections.abcr �typingr r r r r r ZpytestZaiohttp.helpersr r
Z aiohttp.webr Z
test_utilsr
r r r r r r r rh rT � ImportErrorrU Z
AiohttpClientr r1 Zfixturer4 r5 �contextmanagerrF rG rK rR r_ r rL rg rs rt ru rv ZAbstractEventLoopr~ r r r r r �<module> sN $ (
9 "
.