File: //usr/local/lib/python3.10/dist-packages/anyio/__pycache__/pytest_plugin.cpython-310.pyc
o
���g8 � @ s~ U d dl mZ d dlZd dlmZmZ d dlmZmZ d dl m
Z
m Z m Z d dl
mZmZ d dlZd dlZd dlmZ d dlmZ d d
lmZmZ d d lmZ d d lm Z ej d
k r`d dl m Z da de!d< da"de!d< d a#d>dd�Z$ed?d d ��Z%d@d#d$�Z&ej'd%d&�dAd*d+��Z(ej'd%d,�dBd0d1��Z)ej'd%d,�dCd4d5��Z*ej+d6e� d7�dDd8d9��Z,ej+dEd:d;��Z-ej+dFd<d=��Z.dS )G� )� annotationsN)� Generator�Iterator)� ExitStack�contextmanager)�isasyncgenfunction�iscoroutinefunction�ismethod)�Any�cast)�
SubRequest)�Exit� )�get_all_backends�get_async_backend)�iterate_exceptions)�
TestRunner)� � )�ExceptionGroupzTestRunner | None�_current_runnerzExitStack | None�
_runner_stack�backend�object�return�tuple[str, dict[str, Any]]c C sf t | t�r | i fS t | t�r/t| �dkr/t | d t�r/t | d t�r/ttttttf f | �S td��)N� r r z@anyio_backend must be either a string or tuple of (string, dict))�
isinstance�str�tuple�len�dictr r
� TypeError)r � r# �>/usr/local/lib/python3.10/dist-packages/anyio/pytest_plugin.py�extract_backend_and_options s
r% � backend_namer �backend_options�dict[str, Any]�Iterator[TestRunner]c c s� � t d u r.t| �}t� atj�d �d u r"tj�| �}t�tjj |� |p%i }t�
|� |��a t d7 a z t V W t d8 a t sMtd usCJ �t�
� d aa d S d S t d8 a t sctd us[J �t�
� d aa w �Nr )r r r r �sniffio�current_async_library_cvar�get�set�callback�reset�
enter_context�create_test_runner�_runner_leases�close)r& r' �asynclib�tokenr# r# r$ �
get_runner$ s2 � � ��
r7 �configr
�Nonec C s | � dd� d S )N�markerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.)�addinivalue_line)r8 r# r# r$ �pytest_configureB s �r< T)� hookwrapper�
fixturedef�request�Generator[Any]c # s� � d � ��fdd � }| j � t� �st� �rNd|jv rN|| _ | j}d| jv �s.| jd
7 _d| jv �s<| jd 7 _z
d V W � | _ || _S � | _ || _w d V S )
N�argsr
�
anyio_backendr? r �kwargsr c ? s� � |j rt� �rt� j�t|j �u r� j�|j �}n� }t| �\}}�r(| |d< �r.||d< t||��&}t|�rB|� ||�E d H n|�
||�V W d � d S W d � d S 1 s\w Y d S )NrB r? ) �instancer �type�__self__�__func__�__get__r% r7 r �run_asyncgen_fixture� run_fixture)rB r? rA rC �
local_funcr& r' �runner��func�has_backend_arg�has_request_argr# r$ �wrapperL s( ��� �"�z%pytest_fixture_setup.<locals>.wrapper�rB �r? )
rA r
rB r
r? r rC r
r r
)rN r r � fixturenames�argnames)r> r? rQ �original_argnamer# rM r$ �pytest_fixture_setupJ s$ �
�rW )�tryfirst� collector�name�objc C sv | � ||�r5t|d�r|jjn|}t|�r7| �d�}t|dd�}|s+tdd� |D ��r9tj �
d�|� d S d S d S d S )N�
hypothesis�anyio�
pytestmarkr# c s s � | ]}|j d kV qdS )r] N)rZ )�.0�markerr# r# r$ � <genexpr>� s � z,pytest_pycollect_makeitem.<locals>.<genexpr>rB ) �istestfunction�hasattrr\ �
inner_testr �get_closest_marker�getattr�any�pytest�mark� usefixtures)rY rZ r[ �
inner_funcr` � own_markersr# r# r$ �pytest_pycollect_makeitem} s
�rm �
pyfuncitem� bool | Nonec s d � ��fdd� }| j �d�}|r�t|�\� �t| jd�r3| jjj��j|jkr1t��r1|| jj_d S t| j�r�| j ��fd d
�| j j
D �}t � ���2}z |� | j|� W n t
yt } zt|�D ]
}t|tttf�rn||�qa� d }~ww W d � d S 1 s�w Y d S d S )
NrC r
r r9 c s< t � ���}|��| � W d � d S 1 sw Y d S �N)r7 �run_test)rC rL )r& r' �
original_funcr# r$ �run_with_hypothesis� s "�z/pytest_pyfunc_call.<locals>.run_with_hypothesisrB r\ c s i | ]}|� | �qS r# r# )r_ �arg)�funcargsr# r$ �
<dictcomp>� s z&pytest_pyfunc_call.<locals>.<dictcomp>T)rC r
r r9 )ru r- r% rc r[ r\ rd � __qualname__r � _fixtureinforU r7 rq r r r r
�KeyboardInterrupt�
SystemExit)rn rs r �testargsrL �excgrp�excr# )r& r' ru rr r$ �pytest_pyfunc_call� s<
����
�
�
r~ �module)�scope�paramsc C s | j S rp )�paramrS r# r# r$ rB � s rB c C s t | t�r| S | d S )Nr �r r rR r# r# r$ �anyio_backend_name� �
r� c C s t | t�ri S | d S r* r� rR r# r# r$ �anyio_backend_options� r� r� )r r r r )r& r r' r( r r) )r8 r
r r9 )r> r
r? r
r r@ )rY r
rZ r
r[ r
r r9 )rn r
r ro )r? r
r r
)rB r
r r )rB r
r r( )/�
__future__r �sys�collections.abcr r �
contextlibr r �inspectr r r �typingr
r rh r+ �_pytest.fixturesr �_pytest.outcomesr
�_core._eventloopr r �_core._exceptionsr �abcr � version_info�exceptiongroupr r �__annotations__r r3 r% r7 r< �hookimplrW rm r~ �fixturerB r� r� r# r# r# r$ �<module> sD
2
$