
    i1                       U d dl mZ d dlZd dlmc mZ d dlZd dl	Z	d dl
Z
d dlmZmZmZ d dlmZmZ d dlmZmZmZ d dlmZmZ d dlZd dlmZmZ d dlmZ d d	lm Z  d d
l!m"Z" ddl#m$Z$ ddl%m&Z&m'Z'm(Z(m)Z) ddl*m+Z+ ddl,m-Z- e
j\                  dk  rd dl/m0Z0 da1de2d<   da3de2d<   d a4d.dZ5e	 	 	 	 	 	 d/d       Z6d0dZ7d1dZ8 ejr                  d      d2d       Z: ejr                  d      	 	 	 	 	 	 	 	 d3d       Z;d4d Z< ejr                  d      d5d!       Z= ej|                  d" e$       #      d6d$       Z?ej|                  d7d%       Z@ej|                  d8d&       ZA G d' d(      ZB ej|                  d)      d9d*       ZC ej|                  d)      d9d+       ZDej|                  d:d,       ZEej|                  d;d-       ZFy)<    )annotationsN)Callable	GeneratorIterator)	ExitStackcontextmanager)isasyncgenfunctioniscoroutinefunctionismethod)Anycast)FuncFixtureInfo
SubRequest)Exit)	CallSpec2)Scope   )get_available_backends)current_async_libraryget_async_backendreset_current_async_libraryset_current_async_library)iterate_exceptions)
TestRunner)      )ExceptionGroupzTestRunner | None_current_runnerzExitStack | None_runner_stackc                   t        | t              r| i fS t        | t              r^t        |       dk(  rPt        | d   t              r=t        | d   t              r*t        t        t        t        t        t        f   f   |       S t        d      )N   r   r   z@anyio_backend must be either a string or tuple of (string, dict))
isinstancestrtuplelendictr   r   	TypeError)backends    L/home/work/apex_v16/venv/lib/python3.12/site-packages/anyio/pytest_plugin.pyextract_backend_and_optionsr*   #   sq    '3{	GU	#G(9gaj#&:gaj$+Gc4S>12G<<
V
WW    c           	   #    K   t         nt        |       }t               at	               %t        |       }t        j                  t        |       |xs i }t        j                  |j                  |            a t        dz  a
	 t          t        dz  a
t        sd }t        |u}|st        j                  d|fdt        |f      dt        j                         v st        j                  t              rt        j                   t              ndt        j                   |      dz  }dd|iz  }t#        t        j$                  |            d x}}t        j'                          d xaa y y # t        dz  a
t        sd }t        |u}|st        j                  d|fdt        |f      dt        j                         v st        j                  t              rt        j                   t              ndt        j                   |      dz  }dd|iz  }t#        t        j$                  |            d x}}t        j'                          d xaa w w xY ww)Nr   )is not)z%(py0)s is not %(py3)sr   )py0py3zassert %(py5)spy5)r   r   r   r   r   r   callbackr   enter_contextcreate_test_runner_runner_leases
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationclose)backend_namebackend_optionsasynclibtoken@py_assert2@py_assert1@py_format4@py_format6s           r)   
get_runnerrF   -   s    
 $\2! "* .l;E""#>F)/R'55''8
 aN3!(,,=,,,,=,,,,,,=,,,=,,,,,,,,,,!.22MO  	!(,,=,,,,=,,,,,,=,,,=,,,,,,,,,,!.22MO s    A>I	E' 	CI	'CII	c                ,    | j                  ddd       y )N
anyio_modestrictz-AnyIO plugin mode (either "strict" or "auto"))defaulthelp)addini)parsers    r)   pytest_addoptionrN   K   s    
MM<  r+   c                    | j                  dd       | j                  d      dk(  rW| j                  j                  d      r;| j                  d      dk(  r&| j	                  t        j                  d      d       y y y y )	NmarkerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.rH   autoasyncioasyncio_modezlAnyIO auto mode has been enabled together with pytest-asyncio auto mode. This may cause unexpected behavior.r   )addinivalue_linegetinipluginmanager
has_pluginissue_config_time_warningpytestPytestConfigWarning)configs    r)   pytest_configurer\   S   s    
W
 	l#v-  ++I6MM.)V3((&&< 	
 4 7 	.r+   T)hookwrapperc              #    K   dfd}| j                   t              st              r~d|j                  v rp|| _         | j                  }d| j                  v xs| xj                  dz  c_        d| j                  v xs| xj                  dz  c_        	 d | _         || _        S d S # | _         || _        w xY ww)Nanyio_backendrequestc              ;    K   |j                   r[t              rPt        j                        t        |j                         u r&j                  j                  |j                         }n}t        |       \  }}r| |d<   	r||d<   t        ||      5 }t        |      r|j                  ||      E d {    n|j                  ||       d d d        y 7 "# 1 sw Y   y xY ww)Nr_   r`   )instancer   type__self____func____get__r*   rF   r	   run_asyncgen_fixturerun_fixture)
r_   r`   kwargs
local_funcr>   r?   runnerfunchas_backend_arghas_request_args
          r)   wrapperz%pytest_fixture_setup.<locals>.wrapperh   s      T]]#tG,<,<'==..w/?/?@JJ(CM(R%o&3F?# 'F9o6 	=&!*-!66z6JJJ((V<<		= 	=J	= 	=s0   BC'!C6C7C	C'CC$ C'r_   r`   )r_   r   r`   r   ri   r   returnr   )rl   r	   r
   fixturenamesargnames)
fixturedefr`   ro   original_argnamerl   rm   rn   s       @@@r)   pytest_fixture_setuprw   f   s     =4 ??D$#6t#<g222%JO)22'6*:M:M'MMOM##'99#'0J4G4G'GGOG##|3#7"&
&6
#N #'
&6
#s   BCB2 C2CC)tryfirstc                x   | j                  ||      rt        |d      r|j                  j                  n|}t	        |      rx| j
                  j                  d      dk(  }| j                  d      }t        |dd      }|s|st        d |D              r%t        j                  j                  d      |       y y y y )	N
hypothesisrH   rQ   anyio
pytestmark c              3  :   K   | ]  }|j                   d k(    yw)r{   N)name).0markers     r)   	<genexpr>z,pytest_pycollect_makeitem.<locals>.<genexpr>   s     H&v{{g-Hs   r_   )istestfunctionhasattrrz   
inner_testr
   r[   rU   get_closest_markergetattranyrY   markusefixtures)	collectorr   obj
inner_funcanyio_auto_moder   own_markerss          r)   pytest_pycollect_makeitemr      s     T*29#|2LS^^..RU
z*'..55lCvMO11':F!#|R8KHKHH''8= I + +r+   sessionc                P   t        t        t        | j                                    D ]  \  }}t	        |t
        j                        s"t        |j                        s8|j                  d      Jd|j                  vsYg }	 t        j                  t              D ch c]  }|j                   }}t        t#                     D ](  \  }}d|v r&t        d|id|idt$        j&                  i|gg       }n&t        i d|id|idt$        j&                  i|gg       }|j(                  }	t        |	j*                        }
d|
vr|
j-                  d       t/        |	j0                  |	j2                  |
|	j4                        }t
        j                  j7                  |j8                  |j:                   d| d||j<                  ||j>                  |j:                  	      }|j-                  |       + || j                  ||d
z     y c c}w # t        $ r t!               }Y pw xY w)Nr{   r_   
_arg2scope)paramsindicesr   _idlistmarks)funcargsr   r   	arg2scopeidlistr   )rt   initialnamesnames_closurename2fixturedefs[])r   callspeccallobjfixtureinfokeywordsoriginalnamer   ) reversedlist	enumerateitemsr"   rY   Functionr
   functionr   rs   dataclassesfieldsr   r   r'   setr   r   Module_fixtureinfor   appendr   rt   r   r   from_parentparentr   r   r   )r   iitem	new_itemsf	cs_fieldsparam_indexr(   r   finew_names_closurenew_fixtureinfonew_items                r)   pytest_collection_finishr      s	   D7==!9:; 614tV__-#DMM2''0<t'8'88I"-8-?-?	-JKQVVK	K )22H2J(K '+$W9,( /9!0+ >$3U\\#B!(	  H  )!# /9!0+ >#2ELL"A 'y  H &&$()9)9$:!"*;;%,,_="1[[!#"3%'%8%8	# "??66KK --.ay:% HH /!]]!%!2!2 7    *O'+R (1GMM!a!e$m61 L "E	"s$   HH	2H	HH%$H%c           	       	
 d	
fd}| j                   j                  d      }|r t        |      \  	
t        | j                  d      r`| j                  j
                  j                  j                  |j                  k7  r&t              r|| j                  j
                  _        y t        | j                        rg| j                   }| j                  j                  D ci c]  }|||   
 }}t        	
      5 }	 |j                  | j                  |       	 d d d        yy c c}w # t        $ r4}t        |      D ]   }t        |t         t"        t$        f      s||  d }~ww xY w# 1 sw Y   yxY w)Nc                 l    t              5 }|j                  |        d d d        y # 1 sw Y   y xY wN)rF   run_test)ri   rk   r>   r?   original_funcs     r)   run_with_hypothesisz/pytest_pyfunc_call.<locals>.run_with_hypothesis   s2    o6 	3&OOM62	3 	3 	3s   *3r_   rz   T)ri   r   rr   None)r   getr*   r   r   rz   r   __qualname__r
   r   rt   rF   r   r   r   r"   r   KeyboardInterrupt
SystemExit)
pyfuncitemr   r(   r   argtestargsrk   excgrpexcr>   r?   r   s            @@@r)   pytest_pyfunc_callr      sQ   3 !!%%o6G(CG(L%o:>><0&NN55@@M))-@-M-MM&}5;NJNN--8z~~.!**H6@6M6M6V6VWsXc]*WHWL/: fOOJNNH=  X & 1&9 2%cD2CZ+PQ"%612  s6   'D+E0D00	E-9)E(#E((E--E00E9module)scoper   c                    | j                   S r   )paramrq   s    r)   r_   r_     s    ==r+   c                0    t        | t              r| S | d   S )Nr   r"   r#   rp   s    r)   anyio_backend_namer     s    -%Qr+   c                0    t        | t              ri S | d   S )Nr   r   rp   s    r)   anyio_backend_optionsr     s    -%	Qr+   c                  4    e Zd ZdZddZedd       Zdd	dZy)
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.
    c                >    || _         t        t                  | _        y r   )_kindr   int
_generated)selfkinds     r)   __init__zFreePortFactory.__init__+  s    
c(*r+   c                    | 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

        )r   )r   s    r)   r   zFreePortFactory.kind/  s     zzr+   Nc                J   ||g}n@t         j                  g}t         j                  r|j                  t         j                         	 d}t               5 }|D ]r  }|j                  t        j                   || j                              }|t         j                  k(  rdnd}	 |j                  ||f       |r`|j                         d   }t || j                  vr&| j                  j                  |       |cddd       S ddd       # t        $ r Y  w xY w# 1 sw Y   xY w)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_ipv6r   AF_INET6r   r2   r   bindOSErrorgetsocknamer   add)r   familyfamiliesportstacksockaddrs          r)   __call__zFreePortFactory.__call__8  s    xH'H0D $& $F ..v}}VTZZ/PQD$*foo$=5;D		4,/  #//1!4$ 4??2++D1#$ $ $  # $ $s7   AD!D	4D7?D		DDDDD")r   socket.SocketKindrr   r   )rr   r   r   )r   zsocket.AddressFamily | Nonerr   r   )__name__
__module__r   __doc__r   propertyr   r   r}   r+   r)   r   r     s%    %  $r+   r   )r   c                 4    t        t        j                        S r   )r   r   SOCK_STREAMr}   r+   r)   free_tcp_port_factoryr   Z  s    6--..r+   c                 4    t        t        j                        S r   )r   r   
SOCK_DGRAMr}   r+   r)   free_udp_port_factoryr   _  s    6,,--r+   c                     |        S r   r}   )r   s    r)   free_tcp_portr   d       ""r+   c                     |        S r   r}   )r   s    r)   free_udp_portr  i  r   r+   )r(   objectrr   ztuple[str, dict[str, Any]])r>   r#   r?   dict[str, Any]rr   zIterator[TestRunner])rM   zpytest.Parserrr   r   )r[   zpytest.Configrr   r   )ru   r   r`   r   rr   zGenerator[Any])r   zpytest.Module | pytest.Classr   r#   r   r  rr   r   )r   zpytest.Sessionrr   r   )r   r   rr   zbool | None)r`   r   rr   r   )r_   r   rr   r#   )r_   r   rr   r  )rr   r   )r   Callable[[], int]rr   r   )r   r  rr   r   )G
__future__r   builtinsr7   _pytest.assertion.rewrite	assertionrewriter5   r   r   syscollections.abcr   r   r   
contextlibr   r   inspectr	   r
   r   typingr   r   rY   _pytest.fixturesr   r   _pytest.outcomesr   _pytest.pythonr   _pytest.scoper    r   _core._eventloopr   r   r   r   _core._exceptionsr   abcr   version_infoexceptiongroupr   r   __annotations__r   r4   r*   rF   rN   r\   hookimplrw   r   r   r   fixturer_   r   r   r   r   r   r   r  r}   r+   r)   <module>r     s   "      
 9 9 0 E E   8 ! $  $  2 g-%)" )"& &X 33(633 3:
& T"- #-` $>+>36>=C>	>  >"71t $!  !H h'='?@ A        :$ :$z i / !/ i . !. # # # #r+   