
    Qi1                    r   U d dl 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jP                  dk  rd dl)m*Z* da+de,d<   da-de,d<   d a.d.dZ/e	 	 	 	 	 	 d/d       Z0d0dZ1d1dZ2 ejf                  d      d2d       Z4 ejf                  d      	 	 	 	 	 	 	 	 d3d       Z5d4d Z6 ejf                  d      d5d!       Z7 ejp                  d" e       #      d6d$       Z9ejp                  d7d%       Z:ejp                  d8d&       Z; G d' d(      Z< ejp                  d)      d9d*       Z= ejp                  d)      d9d+       Z>ejp                  d:d,       Z?ejp                  d;d-       Z@y)<    )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    >/app/.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        s!t        J t        j                          d xaa y y # t        dz  a
t        s!t        J t        j                          d xaa w w xY wwNr   )r   r   r   r   r   r   callbackr   enter_contextcreate_test_runner_runner_leasesclose)backend_namebackend_optionsasynclibtokens       r)   
get_runnerr7   -   s     
 $\2! "* .l;E""#>F)/R'55''8
 aN3! ,,,!.22MO  	! ,,,!.22MO s   A>C/B: 	1C/:2C,,C/c                ,    | j                  ddd       y )N
anyio_modestrictz-AnyIO plugin mode (either "strict" or "auto"))defaulthelp)addini)parsers    r)   pytest_addoptionr?   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.r9   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_configurerM   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)NrP   rQ   )instancer   type__self____func____get__r*   r7   r	   run_asyncgen_fixturerun_fixture)
rP   rQ   kwargs
local_funcr3   r4   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<<	 76J 76s0   BC'!C6C7C	C'CC$ C'rP   rQ   )rP   r   rQ   r   rZ   r   returnr   )r]   r	   r
   fixturenamesargnames)
fixturedefrQ   r`   original_argnamer]   r^   r_   s       @@@r)   pytest_fixture_setuprh   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
hypothesisr9   rB   anyio
pytestmark c              3  :   K   | ]  }|j                   d k(    yw)rl   N)name).0markers     r)   	<genexpr>z,pytest_pycollect_makeitem.<locals>.<genexpr>   s     HK&v{{g-Ks   rP   )istestfunctionhasattrrk   
inner_testr
   rL   rF   get_closest_markergetattranyrJ   markusefixtures)	collectorrp   obj
inner_funcanyio_auto_moderr   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)Nrl   rP   
_arg2scope)paramsindicesr   _idlistmarks)funcargsr   r   	arg2scopeidlistr   )re   initialnamesnames_closurename2fixturedefs[])rp   callspeccallobjfixtureinfokeywordsoriginalnamer   ) reversedlist	enumerateitemsr"   rJ   Functionr
   functionrw   rd   dataclassesfieldsr   rp   r'   setr   r   Module_fixtureinfor   appendr   re   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:;4tV__-#DMM2''0<t'8'88I"-8-?-?	-JK-JQVV-J	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 )LR (1GMM!a!e$m < 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)r7   run_test)rZ   r\   r3   r4   original_funcs     r)   run_with_hypothesisz/pytest_pyfunc_call.<locals>.run_with_hypothesis   s)    o6&OOM62 766s   *3rP   rk   T)rZ   r   rc   None)r   getr*   ru   r}   rk   rv   __qualname__r
   r   re   r7   r   r   r   r"   r   KeyboardInterrupt
SystemExit)
pyfuncitemr   r(   r   argtestargsr\   excgrpexcr3   r4   r   s            @@@r)   pytest_pyfunc_callr      sJ   3 !!%%o6G(CG(L%o:>><0&NN55@@M))-@-M-MM&}5;NJNN--8z~~.!**H6@6M6M6V6VW6VsXc]*6VHWL/:fOOJNNH= ;  X & 1&9%cD2CZ+PQ"%61  :  ; s6   'D+E0D00	E-9)E(#E((E--E00E9module)scoper   c                    | j                   S r   )paramrb   s    r)   rP   rP     s    ==r+   c                0    t        | t              r| S | d   S )Nr   r"   r#   ra   s    r)   anyio_backend_namer     s    -%Qr+   c                0    t        | t              ri S | d   S r-   r   ra   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   r/   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.SocketKindrc   r   )rc   r   r   )r   zsocket.AddressFamily | Nonerc   r   )__name__
__module__r   __doc__r   propertyr   r   rn   r+   r)   r   r     s%    %  $r+   r   )r   c                 4    t        t        j                        S r   )r   r   SOCK_STREAMrn   r+   r)   free_tcp_port_factoryr   Z  s    6--..r+   c                 4    t        t        j                        S r   )r   r   
SOCK_DGRAMrn   r+   r)   free_udp_port_factoryr   _  s    6,,--r+   c                     |        S r   rn   )r   s    r)   free_tcp_portr   d       ""r+   c                     |        S r   rn   )r   s    r)   free_udp_portr   i  r   r+   )r(   objectrc   ztuple[str, dict[str, Any]])r3   r#   r4   dict[str, Any]rc   zIterator[TestRunner])r>   zpytest.Parserrc   r   )rL   zpytest.Configrc   r   )rf   r   rQ   r   rc   zGenerator[Any])r|   zpytest.Module | pytest.Classrp   r#   r}   r   rc   r   )r   zpytest.Sessionrc   r   )r   r   rc   zbool | None)rQ   r   rc   r   )rP   r   rc   r#   )rP   r   rc   r   )rc   r   )r   Callable[[], int]rc   r   )r   r   rc   r   )A
__future__r   r   r   syscollections.abcr   r   r   
contextlibr   r   inspectr	   r
   r   typingr   r   rJ   _pytest.fixturesr   r   _pytest.outcomesr   _pytest.pythonr   _pytest.scoper    r   _core._eventloopr   r   r   r   _core._exceptionsr   abcr   version_infoexceptiongroupr   r   __annotations__r   r1   r*   r7   r?   rM   hookimplrh   r   r   r   fixturerP   r   r   r   r   r   r   r   rn   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+   