Format: application/x-python-code
1 o
2
3 ;'h�$ � @ s� U d dl mZ d dlZd dlZd dlmZmZmZ d dlm Z m
4 Z
5 d dl m Z m
6 Z
7 mZ d dlmZmZ d dlZd dlZd dlmZ d dlmZ d d
8 lmZmZ d d lm Z d d l m Z ej d
9 k rfd dl m!Z! da"de#d< da$de#d< d a%dLdd�Z&e
10 dMd d ��Z'dNd#d$�Z(ej)d%d&�dOd*d+��Z*ej)d%d,�dPd0d1��Z+ej)d%d,�dQd4d5��Z,ej-d6e� d7�dRd8d9��Z.ej-dSd:d;��Z/ej-dTd<d=��Z0G d>d?� d?�Z1ej-d@dA�dUdBdC��Z2ej-d@dA�dUdDdE��Z3ej-dVdHdI��Z4ej-dWdJdK��Z5dS )X� )� annotationsN)�Callable� Generator�Iterator)� ExitStack�contextmanager)�isasyncgenfunction�iscoroutinefunction�ismethod)�Any�cast)�
11 SubRequest)�Exit� )�get_all_backends�get_async_backend)�iterate_exceptions)�
12 TestRunner)� � )�ExceptionGroupzTestRunner | None�_current_runnerzExitStack | None�
13 _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))�
14 isinstance�str�tuple�len�dictr r � TypeError)r � r$ �IC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/pytest_plugin.py�extract_backend_and_options s
15 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�
16 |� |��a t d7 a z t V W t d8 a t sMtd usCJ �t�
17 � d aa d S d S t d8 a t sctd us[J �t�
18 � d aa w �Nr )r r r r �sniffioZcurrent_async_library_cvar�get�set�callback�reset�
19 enter_contextZcreate_test_runner�_runner_leases�close)r' r( Zasynclib�tokenr$ r$ r% �
20 get_runner% s2 � � ��
21 r5 �configr �Nonec C s | � dd� d S )N�markerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.)Zaddinivalue_line)r6 r$ r$ r% �pytest_configureC s �r9 T)Z hookwrapper�
22 fixturedef�request�Generator[Any]c # s� � d � ��fdd � }| j � t� �st� �rNd|jv rN|| _ | j}d| jv �s.| jd
23 7 _d| jv �s<| jd 7 _z
24 d V W � | _ || _S � | _ || _w d V S )
25 N�argsr �
26 anyio_backendr; r
27 �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|�
28 ||�V W d � d S W d � d S 1 s\w Y d S )Nr> r; ) �instancer
29 �type�__self__�__func__�__get__r& r5 r Zrun_asyncgen_fixtureZ run_fixture)r> r; r= r? Z
30 local_funcr' r( �runner��funcZhas_backend_argZhas_request_argr$ r% �wrapperL s( ��� �"�z%pytest_fixture_setup.<locals>.wrapper�r> �r; )
31 r= r r> r r; r
32 r? r r r )rG r r Z fixturenames�argnames)r: r; rH Zoriginal_argnamer$ rF r% �pytest_fixture_setupJ s$ �
33 �rL )Ztryfirst� collector�name�objc C sv | � ||�r5t|d�r|jjn|}t|�r7| �d�}t|dd�}|s+tdd� |D ��r9tj �
34 d�|� d S d S d S d S )N�
35 hypothesis�anyioZ
36 pytestmarkr$ c s s � | ]}|j d kV qdS )rQ N)rN )�.0�markerr$ r$ r% � <genexpr>� s � z,pytest_pycollect_makeitem.<locals>.<genexpr>r> ) Zistestfunction�hasattrrP �
37 inner_testr Zget_closest_marker�getattr�any�pytest�markZ usefixtures)rM rN rO Z
38 inner_funcrS Z own_markersr$ r$ r% �pytest_pycollect_makeitem} s
39 �r[ �
40 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
41 �| j j
42 D �}t � ���2}z |� | j|� W n t
43 yt } zt|�D ]
44 }t|tttf�rn||�qa� d }~ww W d � d S 1 s�w Y d S d S )
45 Nr? r r r7 c s< t � ���}|��| � W d � d S 1 sw Y d S �N)r5 �run_test)r? rE )r' r( �
46 original_funcr$ r% �run_with_hypothesis� s "�z/pytest_pyfunc_call.<locals>.run_with_hypothesisr> rP c s i | ]}|� | �qS r$ r$ )rR �arg)�funcargsr$ r% �
47 <dictcomp>� s z&pytest_pyfunc_call.<locals>.<dictcomp>T)r? r r r7 )rc r- r& rU rO rP rV � __qualname__r Z _fixtureinforK r5 r_ r r r r �KeyboardInterrupt�
48 SystemExit)r\ ra r ZtestargsrE Zexcgrp�excr$ )r' r( rc r` r% �pytest_pyfunc_call� s<
49
50
51 ����
52 �
53 �
54 ri �module)�scope�paramsc C � | j S r^ )�paramrJ r$ r$ r% r> � � r> c C s t | t�r| S | d S )Nr �r r rI r$ r$ r% �anyio_backend_name� �
55 rq c C s t | t�ri S | d S r+ rp rI r$ r$ r% �anyio_backend_options� rr rs c @ s4 e Zd ZdZddd�Zeddd ��Zdddd�Zd
56 S )�FreePortFactoryaO
57 Manages port generation based on specified socket kind, ensuring no duplicate
58 ports are generated.
59
60 This class provides functionality for generating available free ports on the
61 system. It is initialized with a specific socket kind and can generate ports
62 for given address families while avoiding reuse of previously generated ports.
63
64 Users should not instantiate this class directly, but use the
65 ``free_tcp_port_factory`` and ``free_udp_port_factory`` fixtures instead. For simple
66 uses cases, ``free_tcp_port`` and ``free_udp_port`` can be used instead.
67 �kind�socket.SocketKindr r7 c C s || _ tt � | _d S r^ )�_kindr. �int�
68 _generated)�selfru r$ r$ r% �__init__� s zFreePortFactory.__init__c C rm )z�
69 The type of socket connection (e.g., :data:`~socket.SOCK_STREAM` or
70 :data:`~socket.SOCK_DGRAM`) used to bind for checking port availability
71
72 )rw )rz r$ r$ r% ru � s zFreePortFactory.kindN�family�socket.AddressFamily | Nonerx c
73 C s� |dur|g}n
74 t jg}t jr|�t j� d}t� �P}|D ]1}|�t � || j��}|t jkr1dnd}z |�||f� W n
75 t yF Y n w |sO|�
76 � d }q || j vrd| j � |� |W d � S W d � n1 snw Y q)z�
77 Return an unbound port for the given address family.
78
79 :param family: if omitted, both IPv4 and IPv6 addresses will be tried
80 :return: a port number
81
82 NTr z::1z 127.0.0.1r )
83 �socket�AF_INET�has_ipv6�append�AF_INET6r r1 rw �bind�OSError� getsocknamery �add)rz r| Zfamilies�port�stack�sock�addrr$ r$ r% �__call__� s4 � �
84 �� �zFreePortFactory.__call__)ru rv r r7 )r rv r^ )r| r} r rx )�__name__�
85 __module__re �__doc__r{ �propertyru r� r$ r$ r$ r% rt � s
86
87 rt �session)rk c C �
88 t tj�S r^ )rt r~ � SOCK_STREAMr$ r$ r$ r% �free_tcp_port_factory� �
89 r� c C r� r^ )rt r~ �
90 SOCK_DGRAMr$ r$ r$ r% �free_udp_port_factory r� r� �Callable[[], int]rx c C � | � S r^ r$ )r� r$ r$ r% �
91 free_tcp_port ro r� c C r� r^ r$ )r� r$ r$ r% �
92 free_udp_port ro r� )r r r r )r' r r( r) r r* )r6 r r r7 )r: r r; r r r< )rM r rN r rO r r r7 )r\ r r r] )r; r r r )r> r r r )r> r r r) )r rt )r� r� r rx )r� r� r rx )6�
93 __future__r r~ �sys�collections.abcr r r �
94 contextlibr r �inspectr r r
95 �typingr r rY r, Z_pytest.fixturesr
96 Z_pytest.outcomesr Z_core._eventloopr r Z_core._exceptionsr �abcr � version_infoZexceptiongroupr r �__annotations__r r2 r& r5 r9 ZhookimplrL r[ ri Zfixturer> rq rs rt r� r� r� r� r$ r$ r$ r% �<module> sX
97
98
99
100
101
102 2
103
104 $
105 =
106