o d @sUddlmZddlmZmZddlmZmZmZm Z m Z m Z m Z ddl Z ddlZddlmZddlmZmZddlmZerEdd lmZdae eed <d ed e eeeefffd dZededeeefd e eedffddZd,ddZdeded dfddZe j dddededed dfd d!Z!e j ddd"ed e e"fd#d$Z#e j$ed%ded efd&d'Z%e j$d'ed efd(d)Z&e j$d'ed eeeffd*d+Z'dS)-)contextmanager)isasyncgenfunctioniscoroutinefunction) TYPE_CHECKINGAnyDict GeneratorOptionalTuplecastN)FixtureRequest)get_all_backends get_asynclib) TestRunner)Config_current_runnerbackendreturncCsft|tr |ifSt|tr/t|dkr/t|dtr/t|dtr/ttttttff|St d)Nrr z@anyio_backend must be either a string or tuple of (string, dict)) isinstancestrtuplelendictr r rr TypeError)rr>/usr/local/lib/python3.10/dist-packages/anyio/pytest_plugin.pyextract_backend_and_optionss r backend_namebackend_optionsc cstrtVdSt|}d}tjddurtj|}z0|p i}|jdi| }|a|VWdn1s9wYWda|rKtj|dSdSda|rXtj|ww)Nr)rrsniffiocurrent_async_library_cvargetsetrreset)rr asynclibtokenrunnerrrr get_runners. r)configrcCs|dddS)NmarkerszManyio: mark the (coroutine function) test to be run asynchronously via anyio.)addinivalue_line)r*rrrpytest_configure6sr- fixturedefrequestcs`fdd}|jtstr,d|jvr*d|jv||_s.|jd7_dSdSdSdS)Nc?st|\}}r ||d<t||&}tr!||EdHn||VWddSWddS1s;wYdS)N anyio_backend)rr)rrun_asyncgen_fixture run_fixture)r0argskwargsrr r(funchas_backend_argrrwrapper?s  "z%pytest_fixture_setup..wrapperr0r0)r6rr fixturenamesargnames)r.r/r8rr5rpytest_fixture_setup>s   r<T)tryfirst collectornameobjcCsv|||r5t|dr|jjn|}t|r7|d}t|dd}|s+tdd|Dr9tj d|dSdSdSdS)N hypothesisanyio pytestmarkrcss|]}|jdkVqdS)rBN)r?).0markerrrr \sz,pytest_pycollect_makeitem..r0) istestfunctionhasattrrA inner_testrget_closest_markergetattranypytestmark usefixtures)r>r?r@ inner_funcrE own_markersrrrpytest_pycollect_makeitemUs   rR pyfuncitemcsdtddffdd }|jd}|rit|\t|jdr7|jjjj|jkr5t r5||jj_dSt |jri|jfdd|j j D}t }| |j|Wdd S1sbwYd SdS) Nr4rcs<t}||WddS1swYdSN)r)run_test)r4r()rr original_funcrrrun_with_hypothesisbs "z/pytest_pyfunc_call..run_with_hypothesisr0rAcsi|]}||qSrr)rDarg)funcargsrr usz&pytest_pyfunc_call..T)rrYr#rrHr@rArI __qualname__r _fixtureinfor;r)rU)rSrWrtestargsr(r)rr rYrVrpytest_pyfunc_call`s(         r^)paramscCs|jSrT)param)r/rrrr0~sr0cCst|tr|S|dS)Nrrrr9rrranyio_backend_name rbcCst|triS|dS)Nr rar9rrranyio_backend_optionsrcrd)r*rrN)( contextlibrinspectrrtypingrrrrr r r rMr!_pytest.fixturesr _core._eventlooprrabcr_pytest.configrr__annotations__objectrrr)r-r<hookimplrRboolr^fixturer0rbrdrrrrs@$   "