File: //opt/alt/python38/lib64/python3.8/site-packages/aiohttp/__pycache__/pytest_plugin.cpython-38.pyc
U
'�7`+ � @ s� d dl Z d dlZd dlZd dlmZ d dlZd dlmZmZ d dl m
Z
ddl m Z m
Z
mZmZmZmZmZmZ z d dlZW n ek
r� dZY nX z d dlZW n ek
r� dZY nX dd� Zd d
� Zejd d � �Zejd
d� �Z ej dd� �Z ej d.dd��Z dd� Z dd� Z!dd� Z"ejdd� �Z#ejd d � �Z$ejd d � �Zejd d!� �Z%ejd"d#� �Z&ejd$d%� �Z'ejd&d'� �Z(ejd(d)� �Z)ejd*d+� �Z*ejd,d-� �Z+dS )/� N)�Callable)�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 ) N�--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 all�--aiohttp-enable-loop-debugz enable event loop debug mode)Z addoption)�parser� r �H/opt/alt/python38/lib64/python3.8/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 )zS
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* z� � ��� �W S tk
r$ Y nX d S �N)�run_until_complete� __anext__�StopAsyncIterationr ��_loop�genr r � finalizer` s z8pytest_fixture_setup.<locals>.wrapper.<locals>.finalizer)� fixturenames� ExceptionZgetfixturevalueZ addfinalizerr r )�args�kwargsr r% ��funcZ is_async_genZ
strip_requestr" r �wrapperK s
�
z%pytest_fixture_setup.<locals>.wrapper)r+ r �asyncio�iscoroutinefunction�argnames)Z
fixturedefr, r r* r �pytest_fixture_setup6 s
r0 c C s | j �d�S )z--fast config optionr ��config� getoptionr r r r �fastn s r4 c C s | j �d�S )z!--enable-loop-debug config optionr r1 r r r r �
loop_debugt s r5 c c s` t jdd��J} dV dd� | D �}|rRtd�t|�t|�dkrBdnd d
�|����W 5 Q R X dS ) z�
Context manager which checks for RuntimeWarnings, 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)� _warnings�rwr r r �_runtime_warning_contextz s � ��rG Fc c s( | r | V nt � } | V t| |d� dS )z{
setups 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
rI 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 rM c
s� � j �d�}t�� j�r�� j�d�p0� j�dd�}t� �F t||d��.}� fdd�� j j
D �}|� � j f |�� W 5 Q R X W 5 Q R X dS dS ) zL
Run coroutines in an event loop instead of a normal function call.
r �
proactor_loopr NrH c s i | ]}|� j | �qS r )�funcargs)r; �arg��
pyfuncitemr r �
<dictcomp>� s � z&pytest_pyfunc_call.<locals>.<dictcomp>T)
r2 r3 r- r. �functionrO �getrG rI Z _fixtureinfor/ r rL )rR r4 Z
existing_loopr# Ztestargsr rQ r �pytest_pyfunc_call� s � �
�&rV c C s� d| j krd S | jjj}dtji}td k r4tj|d< td k rFtj|d< |dkrRd}i }|� d�D ]N}|�
d� }|� d �}||kr�|r`t d
|t
|�� �f ��nq`|| ||< q`| jdt
|�� �t
|�� �d � d S ) N� loop_factoryr �uvloop�tokio�allzpyloop,uvloop?,tokio?�,�?z ?z&Unknown loop '%s', available loops: %s)Zids)r& r2 �optionZ aiohttp_loopr- ZDefaultEventLoopPolicyrX ZEventLoopPolicyrY �split�endswith�strip�
ValueErrorrJ �keysZ parametrize�values)ZmetafuncZloopsZavail_factoriesZ factoriesrK �requiredr r r �pytest_generate_tests� s8
��
�re c c sH | � }t �|� t|d��$}|r*|�d� t �|� |V W 5 Q R X dS )z%Return an instance of the event loop.rH TN)r- �set_event_loop_policyr Z set_debug�set_event_loop)rW r4 r5 �policyr# r r r r � s
r c c sR t st�� } tj| _nt�� } t�| � t| j��}t� |� |V W 5 Q R X d S r )
r r- Zget_event_loop_policyZProactorEventLoopZ
_loop_factoryZ WindowsProactorEventLoopPolicyrf r Znew_event_looprg )rh r# r r r rN � s
rN 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 ro s ro c # s: g �dd�� �fdd�
}|V �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 �Nrq r )r
� start_server�append)Zapprr r) �server�r �serversr r �go s
zaiohttp_server.<locals>.goc � s � r� � � �� I d H q d S r ��pop�closer �rx r r �finalize s z aiohttp_server.<locals>.finalize�r �r ry r~ r rw r �aiohttp_server s
r� c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_server fixture insteadri rj rl )r� r r r � test_server s �r� c # s: g �dd�� �fdd�
}|V �fdd�}� � |� � dS )zpFactory to create a RawTestServer instance, given a web handler.
aiohttp_raw_server(handler, **kwargs)
Nrq c � s4 t | |d�}|jf d� i|��I d H ��|� |S rs )r rt ru )�handlerrr r) rv rw r r ry 2 s
z aiohttp_raw_server.<locals>.goc � s � r� � � �� I d H q d S r rz r r} r r r~ : s z$aiohttp_raw_server.<locals>.finalizer r� r rw r �aiohttp_raw_server* s
r� c C s t jdtdd� | S )Nz2Deprecated, use aiohttp_raw_server fixture insteadri rj rl )r� r r r �raw_test_serverA s �r� c # s: g � dd�� �fdd�
}|V � fdd�}�� |� � dS )z�Factory to create a TestClient instance.
aiohttp_client(app, **kwargs)
aiohttp_client(server, **kwargs)
aiohttp_client(raw_server, **kwargs)
N)�
server_kwargsc � 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 ra �typert ru )Z__paramr� r( r) rv Zclient��clientsr r r ry U s" �
zaiohttp_client.<locals>.goc � s � r� � � �� I d H q d S r rz r )r� r r r~ n s z aiohttp_client.<locals>.finalizer r� r r� r �aiohttp_clientK s
r� c C s t jdtdd� | S )Nz.Deprecated, use aiohttp_client fixture insteadri rj rl )r� r r r � test_clientu s �r� )F),r- �
contextlibr@ �collections.abcr �pytestZaiohttp.helpersr r Z aiohttp.webr Z
test_utilsr r r r
r r r
r rp rX � ImportErrorrY r r0 Zfixturer4 r5 �contextmanagerrG rI rM rV re r rN ro r� r� r� r� r� r� r r r r �<module> s` (
8
"
)