ruạṛ
a �u:j( � @ sR U d dl mZ d dlZd dlZd dlmZmZmZ d dlm Z m Z d dl m Z m Z mZ d dlmZmZ d dlZd dlmZ d dlmZ d d lmZ d d lmZmZm Z m Z d d l m Z d d l m!Z! ej"dk r�d dl#m$Z$ da%de&d< da'de&d< d a(ddd�dd�Z)e dddd �d d ��Z*d d d!�d"d#�Z+d$d d%�d&d'�Z,ej-d(d)�d*d*d+d,�d-d.��Z.ej-d(d/�d0ddd d1�d2d3��Z/ej-d(d/�d*d4d5�d6d7��Z0ej1d8e� d9�d*d*d:�d;d<��Z2ej1d*dd=�d>d?��Z3ej1d*dd=�d@dA��Z4G dBdC� dC�Z5ej1dDdE�dCdF�dGdH��Z6ej1dDdE�dCdF�dIdJ��Z7ej1dKdLdM�dNdO��Z8ej1dKdLdP�dQdR��Z9dS )S� )� annotationsN)�Callable� Generator�Iterator)� ExitStack�contextmanager)�isasyncgenfunction�iscoroutinefunction�ismethod)�Any�cast)� SubRequest)�Exit� )�get_available_backends)�current_async_library�get_async_backend�reset_current_async_library�set_current_async_library)�iterate_exceptions)� TestRunner)� � )�ExceptionGroupzTestRunner | None�_current_runnerzExitStack | None� _runner_stack�objectztuple[str, dict[str, Any]])�backend�returnc C sj t | t�r| i fS t | t�r^t| �dkr^t | d t�r^t | d t�r^ttttttf f | �S td��d S )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& �F/opt/bart/bart_venv/lib/python3.9/site-packages/anyio/pytest_plugin.py�extract_backend_and_options s r( r! zdict[str, Any]zIterator[TestRunner])� backend_name�backend_optionsr c c s� t d u rLt| �}t� at� d u r4t| �}t�t|� |p:i }t�|� |��a t d7 a z2t V W t d8 a t s�td usvJ �t� � d aa n*t d8 a t s�td us�J �t� � d aa 0 d S �Nr ) r r r r r r �callbackr � enter_contextZcreate_test_runner�_runner_leases�close)r) r* Zasynclib�tokenr& r& r' � get_runner* s. � � r1 z pytest.Parser�None)�parserr c C s | j dddd� d S )N� anyio_mode�strictz-AnyIO plugin mode (either "strict" or "auto"))�default�help)Zaddini)r3 r& r& r' �pytest_addoptionH s �r8 z pytest.Config)�configr c C sJ | � dd� | �d�dkrF| j�d�rF| �d�dkrF| �t�d�d� d S ) N�markerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.r4 �autoZasyncioZ asyncio_modezlAnyIO auto mode has been enabled together with pytest-asyncio auto mode. This may cause unexpected behavior.r )Zaddinivalue_line�getiniZ pluginmanagerZ has_pluginZissue_config_time_warning�pytestZPytestConfigWarning)r9 r& r& r' �pytest_configureP s � � � ���r> T)Z hookwrapperr zGenerator[Any])� fixturedef�requestr c # s� ddddd�� ��fdd� }| j � t� �s2t� �r�d|jv r�|| _ | j}d| jv �sd| jd7 _d| jv �s�| jd 7 _zd V W � | _ || _S � | _ || _0 d V S ) Nr r )� anyio_backendr@ �kwargsr c ; s� |j r2t� �r2t� j�t|j �u r2� j�|j �}n� }t| �\}}�rN| |d< �rZ||d< t||��:}t|�r�|� ||�E d H n|� ||�V W d � n1 s�0 Y d S )NrA r@ ) �instancer �type�__self__�__func__�__get__r( r1 r Zrun_asyncgen_fixtureZ run_fixture)rA r@ rB Z local_funcr) r* �runner��funcZhas_backend_argZhas_request_argr& r' �wrappere s" ��� z%pytest_fixture_setup.<locals>.wrapperrA �rA r@ �r@ )rJ r r Z fixturenames�argnames)r? r@ rK Zoriginal_argnamer& rI r' �pytest_fixture_setupc s( � �rO )Ztryfirstz pytest.Module | pytest.Class)� collector�name�objr c C s~ | � ||�rzt|d�r |jjn|}t|�rz| j�d�dk}| �d�}t|dd�}|sj|sjt dd� |D ��rzt j � d �|� d S ) N� hypothesisr4 r; �anyioZ pytestmarkr& c s s | ]}|j d kV qdS )rT N)rQ )�.0�markerr& r& r' � <genexpr>� � z,pytest_pycollect_makeitem.<locals>.<genexpr>rA ) Zistestfunction�hasattrrS � inner_testr r9 r< Zget_closest_marker�getattr�anyr= �markZ usefixtures)rP rQ rR Z inner_funcZanyio_auto_moderV Z own_markersr& r& r' �pytest_pycollect_makeitem� s ���r^ z bool | None)� pyfuncitemr c s( ddd�� ��fdd� }| j �d�}|�r$t|�\� �t| jd�rn| jjj��j|jkrjt��rj|| jj_d S t| j��r$| j ��fdd �| j j D �}t � ���n}z|� | j|� W nJ t y� } z2t|�D ]}t|tttf�r�||�q̂ W Y d }~n d }~0 0 W d � n1 �s0 Y d S d S ) Nr r2 )rB r c s: t � ��� }|��| � W d � n1 s,0 Y d S �N)r1 �run_test)rB rH )r) r* � original_funcr& r' �run_with_hypothesis� s z/pytest_pyfunc_call.<locals>.run_with_hypothesisrA rS c s i | ]}|� | �qS r& r& )rU �arg)�funcargsr& r' � <dictcomp>� rX z&pytest_pyfunc_call.<locals>.<dictcomp>T)re �getr( rY rR rS rZ � __qualname__r Z _fixtureinforN r1 ra r r r r �KeyboardInterrupt� SystemExit)r_ rc r ZtestargsrH Zexcgrp�excr& )r) r* re rb r' �pytest_pyfunc_call� s. 8rl �module)�scope�params)r@ r c C s | j S r` )�paramrM r& r& r' rA � s rA )rA r c C s t | t�r| S | d S d S )Nr �r r! rL r& r& r' �anyio_backend_name� s rr c C s t | t�ri S | d S d S r+ rq rL r& r& r' �anyio_backend_options� s rs c @ sD e Zd ZdZddd�dd�Zedd�dd ��Zdd d d �dd�Zd S )�FreePortFactoryaO Manages port generation based on specified socket kind, ensuring no duplicate ports are generated. This class provides functionality for generating available free ports on the system. It is initialized with a specific socket kind and can generate ports for given address families while avoiding reuse of previously generated ports. Users should not instantiate this class directly, but use the ``free_tcp_port_factory`` and ``free_udp_port_factory`` fixtures instead. For simple uses cases, ``free_tcp_port`` and ``free_udp_port`` can be used instead. zsocket.SocketKindr2 )�kindr c C s || _ tt � | _d S r` )�_kind�set�int� _generated)�selfru r& r& r' �__init__� s zFreePortFactory.__init__�r c C s | j S )z� The type of socket connection (e.g., :data:`~socket.SOCK_STREAM` or :data:`~socket.SOCK_DGRAM`) used to bind for checking port availability )rv )rz r& r& r' ru � s zFreePortFactory.kindNzsocket.AddressFamily | Nonerx )�familyr c C s� |dur|g}nt jg}t jr*|�t j� d}t� ��}|D ]f}|�t � || j��}|t jkr`dnd}z|�||f� W n t y� Y q�Y n0 |s:|� � d }q:|| j vr�| j � |� |W d � S W d � q*1 s�0 Y q*dS )z� Return an unbound port for the given address family. :param family: if omitted, both IPv4 and IPv6 addresses will be tried :return: a port number Nr z::1z 127.0.0.1r ) �socket�AF_INET�has_ipv6�append�AF_INET6r r- rv �bind�OSError� getsocknamery �add)rz r} Zfamilies�port�stack�sock�addrr& r& r' �__call__� s& zFreePortFactory.__call__)N)�__name__� __module__rh �__doc__r{ �propertyru r� r& r& r& r' rt � s rt �session)rn r| c C s t tj�S r` )rt r~ � SOCK_STREAMr& r& r& r' �free_tcp_port_factory s r� c C s t tj�S r` )rt r~ � SOCK_DGRAMr& r& r& r' �free_udp_port_factory" s r� zCallable[[], int]rx )r� r c C s | � S r` r& )r� r& r& r' � free_tcp_port' s r� )r� r c C s | � S r` r& )r� r& r& r' � free_udp_port, s r� ):� __future__r r~ �sys�collections.abcr r r � contextlibr r �inspectr r r �typingr r r= Z_pytest.fixturesr Z_pytest.outcomesr � r Z_core._eventloopr r r r Z_core._exceptionsr �abcr � version_infoZexceptiongroupr r �__annotations__r r. r( r1 r8 r> ZhookimplrO r^ rl ZfixturerA rr rs rt r� r� r� r� r& r& r& r' �<module> sX 0 $=
cải xoăn