File: //opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/__pycache__/pytest_plugin.cpython-310.pyc
o
;j�2 �
@ sd d dl Z d dlZd dlZd dlZd dlmZmZmZ d dlm Z m
Z
m Z d dl Z ddl
mZmZmZmZmZmZmZmZ ddlmZmZmZ ddlm Z zd dl Z W n e y] dZ Y nw G dd � d e
�Z G d
d � d e
�Z G d d
� d
e
�Z!dd� Z"dd� Z#e j$dd� �Z%e j$dd� �Z&ej'dd� �Z(ej'd7dd��Z)dd � Z*d d � Z+d d � Z,e j$d!eg e j-f de.de.d"ee j- fd#d$��Z/e j$d"ee j- fd%d&��Z0e j$d'eg e1f d"eg e1f fd(d)��Ze j$d"eg e1f fd*d'��Z2e j$d$e j-d"ee fd+d,��Z3e j$d-d.� �Z4e j$d$e j-d"ee! fd/d0��Z5e j$d1d2� �Z6e j$d$e j-d"ee fd3d4��Z7e j$d5d6� �Z8dS )8� N)� Awaitable�Callable�Iterator)�Any�Protocol�overload� )�BaseTestServer�
RawTestServer�
TestClient�
TestServer� loop_context�setup_test_loop�teardown_test_loop� unused_port)� Application� BaseRequest�Request)�_RequestHandlerc @ s| e Zd Zedd�dedeeef dB dedee ef fdd��Z
edd�de deeef dB dedee df fd d��Z
dS )
�
AiohttpClientN��
server_kwargs�_AiohttpClient__paramr �kwargs�returnc � � �d S �N� ��selfr r r r r �I/opt/saltstack/salt/lib/python3.10/site-packages/aiohttp/pytest_plugin.py�__call__ � �zAiohttpClient.__call__c � r r r r r r r r! & r" )
�__name__�
__module__� __qualname__r r �dict�strr r r r! r r r r r r r s. ����
�����
�r c
@ �4 e Zd Zdd�dededB dedee fdd�ZdS ) �
AiohttpServerN��port�appr+ r r c K � d S r r )r r, r+ r r r r r! 1 � zAiohttpServer.__call__) r# r$ r% r �intr r r r! r r r r r) 0 � �����r) c
@ r( ) �AiohttpRawServerNr* �handlerr+ r r c K r- r r )r r2 r+ r r r r r! 7 r. zAiohttpRawServer.__call__) r# r$ r% r r/ r r r
r! r r r r r1 6 r0 r1 c 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�pyloopz3run tests with specific loop: pyloop, uvloop or all�--aiohttp-enable-loop-debugz enable event loop debug mode)Z addoption)�parserr r r �pytest_addoption< s$ ��
�r< c sd | j � t�� �r d�n
t�� �rd�ndS d�d| jvr%| jd7 _d�� ��fdd�}|| _ dS )zjSet up pytest fixture.
Allow fixtures to be coroutines. Run coroutine fixtures in an event loop.
TFN�request�r= c sz |d }�r |d= d|j vrtd��|�d�� �r3�| i |���� �fdd�}|�|� � ���� �S � ��| i |���S )Nr= �loopz^Asynchronous fixtures must depend on the 'loop' fixture or be used in tests depending from it.c s& z� � ��� �W S ty Y d S w r )�run_until_complete� __anext__�StopAsyncIterationr ��_loop�genr r � finalizer| s
�z8pytest_fixture_setup.<locals>.wrapper.<locals>.finalizer)� fixturenames� ExceptionZgetfixturevalueZ addfinalizerr@ rA )�argsr r= rF ��funcZ is_async_genZ
strip_requestrC r �wrapperg s
�
z%pytest_fixture_setup.<locals>.wrapper)rK �inspectZisasyncgenfunction�iscoroutinefunction�argnames)Z
fixturedefrL r rJ r �pytest_fixture_setupQ s
rP c C � | j �d�S )z--fast config optionr3 ��config� getoptionr> r r r �fast� � rU c C rQ )z!--enable-loop-debug config optionr: rR r> r r r �
loop_debug� rV rW c c sx � t jdd��+} dV dd� | D �}|r*td�t|�t|�dkr"dnd d
�|����W d � dS 1 s5w Y 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 s0 g | ]}|j tkr|j� d |j� d |j� ��qS )�:)�category�RuntimeWarning�filename�lineno�message)�.0�wr r r �
<listcomp>� s
�z,_runtime_warning_context.<locals>.<listcomp>z{} Runtime Warning{},
{}r � �s�
)�warnings�catch_warnings� RuntimeError�format�len�join)� _warnings�rwr r r �_runtime_warning_context� s � � ���"�rm Fc c s, � | r| V dS t � } | 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.
�rU N)r r )r? rU r r r �_passthrough_loop_context� s �
ro c C s, | � |�rt�|�rt| �||��S dS dS )z%Fix pytest collecting for coroutines.N)ZfuncnamefilterrM rN �listZ
_genfunctions)Z collector�name�objr r r �pytest_pycollect_makeitem� s �rs c s� � j �d�}t�� j�rut�dt� � j� d�p*� j� d�p*� j� d�p*� j� dd�}t
� �= t ||d�� }� fd d
�� j j
D �}|�� jd i |��� W d � n1 sVw Y W d � d S W d � d S 1 snw Y d S dS )
zBRun coroutines in an event loop instead of a normal function call.r3 zKaiohttp.pytest_plugin will be removed in v4. Please install pytest-aiohttp.�
proactor_loopZ
selector_loopZ uvloop_loopr? Nrn c s i | ]}|� j | �qS r )�funcargs)r_ �arg��
pyfuncitemr r �
<dictcomp>� s
��z&pytest_pyfunc_call.<locals>.<dictcomp>Tr )rS rT rM rN �functionre �warn�DeprecationWarningru �getrm ro Z _fixtureinforO r@ rr )rx rU Z
existing_looprD Ztestargsr rw r �pytest_pyfunc_call� s8 �
�
� �
� �
��
���r~ c C s� d| j vrd S | jjj}dtji}td urtj|d< |dkr d}i }|�d�D ]&}|�d� }|� d�}||vrG|rFt
d |t |� � �f ��q'|| ||< q'| j
dt |�� �t |� � �d
� d S ) N� loop_factoryr9 �uvloop�allzpyloop,uvloop?�,�?z ?z&Unknown loop '%s', available loops: %s)Zids)rG rS �optionZ aiohttp_loop�asyncioZnew_event_loopr� �split�endswith�strip�
ValueErrorrp �keysZ parametrize�values)ZmetafuncZloopsZavail_factoriesZ factoriesrq �requiredr r r �pytest_generate_tests� s0
��
�r� r r c c sR � t | |d��}|r|�d� t�|� |V W d � dS 1 s"w Y dS )z%Return an instance of the event loop.rn TN)r
Z set_debugr� �set_event_loop)r rU rW rD r r r r? � s �
"�r? c c sF � t j} t| ��}t �|� |V W d � d S 1 s w Y d S r )r� ZProactorEventLoopr
r� )�factoryrD r r r rt
s �
"�rt �aiohttp_unused_portc C � t jdtdd� | S )Nz3Deprecated, use aiohttp_unused_port fixture instead� ��
stacklevel�re r{ r| )r� r r r r � �r c C s t S )z1Return a port that is unused on the current host.)� _unused_portr r r r r� s c # sZ � g �ddd�dt dtdtdB dtdtf
� �fd d
�}|V d
�fd d � }� �|� � dS )z^Factory to create a TestServer instance, given an app.
aiohttp_server(app, **kwargs)
z 127.0.0.1N��hostr+ r, r� r+ r r c � s8 �t | ||d�}|jdd� i|��I d H ��|� |S )Nr� r? r )r � start_server�append)r, r� r+ r �server�r? �serversr r �go. s
�
zaiohttp_server.<locals>.goc � �$ �� r� � � �� I d H � sd S d S r ��pop�closer �r� r r �finalize< � � �z aiohttp_server.<locals>.finalize�r N)r r' r/ r r r@ �r? r� r� r r� r �aiohttp_server& s$ ������� r� c C r� )Nz.Deprecated, use aiohttp_server fixture insteadr� r� r� )r� r r r � test_serverC r� r� c # sT � g �dd�dt dtdB dtdtf� �fdd�}|V d �fd d
� }� �|� � dS ) zpFactory to create a RawTestServer instance, given a web handler.
aiohttp_raw_server(handler, **kwargs)
Nr* r2 r+ r r c � s6 �t | |d�}|jdd� i|��I d H ��|� |S )Nr* r? r )r
r� r� )r2 r+ r r� r� r r r� U s
�
z aiohttp_raw_server.<locals>.goc � r� r r� r r� r r r� _ r� z$aiohttp_raw_server.<locals>.finalizer� )r r/ r r
r@ r� r r� r �aiohttp_raw_serverM s ������r� c C r� )Nz2Deprecated, use aiohttp_raw_server fixture insteadr� r� r� )r� r r r �raw_test_serverf s �r� c
# s� � g � t dd�dtdtttf dB dtdtttf fdd��}t dd�dtdtttf dB dtdttdf fd d��}dd�dttB d
tdtttf dB dtdtttf f
� �fd d�}|V d� fd d
� }�� |� � dS )z�Factory to create a TestClient instance.
aiohttp_client(app, **kwargs)
aiohttp_client(server, **kwargs)
aiohttp_client(raw_server, **kwargs)
Nr �__paramr r r c � r r r �r� r r r r r r� | � �zaiohttp_client.<locals>.goc � r r r r� r r r r� � r� rI c � s� �t | t�rt | ttf�s| �g|�R i |��} i }n|r!J d��t | t�r?|p)i }t| fd�i|��}t|fd�i|��}nt | t�rOt| fd�i|��}ntdt| � ��|�� I d H � � |� |S )Nzargs should be emptyr? zUnknown argument type: %r)
�
isinstancer r r r r r� �typer� r� )r� r rI r r� �client��clientsr? r r r� � s" � �
c � r� r r� r )r� r r r� � r� z aiohttp_client.<locals>.finalizer� )
r r r& r' r r r r r r@ r� r r� r �aiohttp_clientr sN �����
�����
�
�����
�r� c C r� )Nz.Deprecated, use aiohttp_client fixture insteadr� r� r� )r� r r r � test_client� r� r� )F)9r� �
contextlibrM re �collections.abcr r r �typingr r r ZpytestZ
test_utilsr r
r r r
r r r r� Zwebr r r Z web_protocolr r� � ImportErrorr r) r1 r< rP ZfixturerU rW �contextmanagerrm ro rs r~ r� ZAbstractEventLoop�boolr? rt r/ r� r� r� r� r� r� r� r r r r �<module> sv (
�9
��� �
$
=