o ;'h$@sUddlmZddlZddlZddlmZmZmZddlm Z m Z ddl m Z m Z mZddlmZmZddlZddlZddlmZddlmZd d lmZmZd d lmZd d lmZejd krfddl m!Z!da"de#d<da$de#d<da%dLddZ&e dMddZ'dNd#d$Z(ej)d%d&dOd*d+Z*ej)d%d,dPd0d1Z+ej)d%d,dQd4d5Z,ej-d6ed7dRd8d9Z.ej-dSd:d;Z/ej-dTdd?d?Z1ej-d@dAdUdBdCZ2ej-d@dAdUdDdEZ3ej-dVdHdIZ4ej-dWdJdKZ5dS)X) annotationsN)Callable GeneratorIterator) ExitStackcontextmanager)isasyncgenfunctioniscoroutinefunctionismethod)Anycast) SubRequest)Exit)get_all_backendsget_async_backend)iterate_exceptions) TestRunner) )ExceptionGroupzTestRunner | None_current_runnerzExitStack | None _runner_stackbackendobjectreturntuple[str, dict[str, Any]]cCsft|tr |ifSt|tr/t|dkr/t|dtr/t|dtr/ttttttff|Std)Nrrz@anyio_backend must be either a string or tuple of (string, dict)) isinstancestrtuplelendictr r TypeError)rr$IC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/pytest_plugin.pyextract_backend_and_optionss r& backend_namerbackend_optionsdict[str, Any]Iterator[TestRunner]ccstdur.t|}tatjddur"tj|}ttjj ||p%i}t | |at d7a ztVWt d8a t sMtdusCJt daadSdSt d8a t sctdus[Jt daawNr)rrrrsniffioZcurrent_async_library_cvargetsetcallbackreset enter_contextZcreate_test_runner_runner_leasesclose)r'r(Zasynclibtokenr$r$r% get_runner%s2     r5configr NonecCs|dddS)NmarkerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.)Zaddinivalue_line)r6r$r$r%pytest_configureCsr9T)Z hookwrapper fixturedefrequestGenerator[Any]c#sd fdd }|jtstrNd|jvrN||_|j}d|jvs.|jd 7_d|jvs<|jd 7_z dVW|_||_S|_||_wdVS) Nargsr anyio_backendr;r kwargsrc?s|jrtrtjt|jurj|j}n}t|\}}r(||d<r.||d<t||&}t|rB| ||EdHn| ||VWddSWddS1s\wYdS)Nr>r;) instancer type__self____func____get__r&r5rZrun_asyncgen_fixtureZ run_fixture)r>r;r=r?Z local_funcr'r(runnerfuncZhas_backend_argZhas_request_argr$r%wrapperLs(  "z%pytest_fixture_setup..wrapperr>r;) r=r r>r r;r r?r rr )rGrr Z fixturenamesargnames)r:r;rHZoriginal_argnamer$rFr%pytest_fixture_setupJs$ rL)Ztryfirst collectornameobjcCsv|||r5t|dr|jjn|}t|r7|d}t|dd}|s+tdd|Dr9tj d|dSdSdSdS)N hypothesisanyioZ pytestmarkr$css|]}|jdkVqdS)rQN)rN).0markerr$r$r% sz,pytest_pycollect_makeitem..r>) ZistestfunctionhasattrrP inner_testr Zget_closest_markergetattranypytestmarkZ usefixtures)rMrNrOZ inner_funcrSZ own_markersr$r$r%pytest_pycollect_makeitem}s   r[ pyfuncitem bool | Nonec sd fdd }|jd}|rt|\t|jdr3|jjjj|jkr1tr1||jj_dSt|jr|jfd d |j j D}t 2}z | |j|Wnt yt}zt|D] }t|tttfrn||qad}~wwWdd S1swYd SdS) Nr?r rr7cs<t}||WddS1swYdSN)r5run_test)r?rE)r'r( original_funcr$r%run_with_hypothesiss "z/pytest_pyfunc_call..run_with_hypothesisr>rPcsi|]}||qSr$r$)rRarg)funcargsr$r% sz&pytest_pyfunc_call..T)r?r rr7)rcr-r&rUrOrPrV __qualname__r Z _fixtureinforKr5r_rrrrKeyboardInterrupt SystemExit)r\rarZtestargsrEZexcgrpexcr$)r'r(rcr`r%pytest_pyfunc_calls<            rimodule)scopeparamscC|jSr^)paramrJr$r$r%r>r>cCst|tr|S|dS)NrrrrIr$r$r%anyio_backend_name rqcCst|triS|dSr+rprIr$r$r%anyio_backend_optionsrrrsc@s4eZdZdZdddZeddd ZddddZd 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. kindsocket.SocketKindrr7cCs||_tt|_dSr^)_kindr.int _generated)selfrur$r$r%__init__szFreePortFactory.__init__cCrm)z The type of socket connection (e.g., :data:`~socket.SOCK_STREAM` or :data:`~socket.SOCK_DGRAM`) used to bind for checking port availability )rw)rzr$r$r%ruszFreePortFactory.kindNfamilysocket.AddressFamily | Nonerxc Cs|dur|g}n tjg}tjr|tj d}tP}|D]1}|t||j}|tjkr1dnd}z |||fWn t yFYnw|sO| d}q||j vrd|j ||WdSWdn1snwYq)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 NTrz::1z 127.0.0.1r) socketAF_INEThas_ipv6appendAF_INET6rr1rwbindOSError getsocknameryadd)rzr|Zfamiliesportstacksockaddrr$r$r%__call__s4     zFreePortFactory.__call__)rurvrr7)rrvr^)r|r}rrx)__name__ __module__re__doc__r{propertyrurr$r$r$r%rts   rtsession)rkcC ttjSr^)rtr~ SOCK_STREAMr$r$r$r%free_tcp_port_factory rcCrr^)rtr~ SOCK_DGRAMr$r$r$r%free_udp_port_factoryrrCallable[[], int]rxcC|Sr^r$)rr$r$r% free_tcp_port rorcCrr^r$)rr$r$r% free_udp_portror)rrrr)r'rr(r)rr*)r6r rr7)r:r r;r rr<)rMr rNr rOr rr7)r\r rr])r;r rr )r>r rr)r>r rr))rrt)rrrrx)rrrrx)6 __future__rr~syscollections.abcrrr contextlibrrinspectrr r typingr r rYr,Z_pytest.fixturesr Z_pytest.outcomesrZ_core._eventlooprrZ_core._exceptionsrabcr version_infoZexceptiongrouprr__annotations__rr2r&r5r9ZhookimplrLr[riZfixturer>rqrsrtrrrrr$r$r$r%sX              2  $    =