­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U '7`+@sddlZddlZddlZddlmZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZz ddlZWnek rdZYnXz ddlZWnek rdZYnXddZd d Zejd d Zejd dZejddZejd.ddZddZ ddZ!ddZ"ejddZ#ejddZ$ejddZejd d!Z%ejd"d#Z&ejd$d%Z'ejd&d'Z(ejd(d)Z)ejd*d+Z*ejd,d-Z+dS)/N)Callable)PY_37isasyncgenfunction) Application)BaseTestServer RawTestServer TestClient TestServer loop_contextsetup_test_loopteardown_test_loop unused_portcCs:|jddddd|jdddd d|jd ddd ddS) N--aiohttp-fast store_trueFz*run tests faster by disabling extra checks)actiondefaulthelpz--aiohttp-loopstorepyloopz:run tests with specific loop: pyloop, uvloop, tokio or all--aiohttp-enable-loop-debugzenable event loop debug mode)Z addoption)parserrH/opt/alt/python38/lib64/python3.8/site-packages/aiohttp/pytest_plugin.pypytest_addoption!s$rcsb|jtrdntr$dndSdd|jkrH|jd7_dfdd}||_dS)zS Allow fixtures to be coroutines. Run coroutine fixtures in an event loop. TFNrequestrcsv|d}r|d=d|jkr$td|drb||fdd}||S||SdS)Nrloopz^Asynchronous fixtures must depend on the 'loop' fixture or be used in tests depending from it.cs*zWStk r$YnXdSN)run_until_complete __anext__StopAsyncIterationr_loopgenrr finalizer`sz8pytest_fixture_setup..wrapper..finalizer) fixturenames ExceptionZgetfixturevalueZ addfinalizerrr )argskwargsrr%funcZ is_async_genZ strip_requestr"rwrapperKs    z%pytest_fixture_setup..wrapper)r+rasyncioiscoroutinefunctionargnames)Z fixturedefr,rr*rpytest_fixture_setup6s   r0cCs |jdS)z--fast config optionrconfig getoptionrrrrfastnsr4cCs |jdS)z!--enable-loop-debug config optionrr1rrrr loop_debugtsr5c cs`tjddJ}dVdd|D}|rRtdt|t|dkrBdnd d |W5QRXdS) z Context manager which checks for RuntimeWarnings, specifically to avoid "coroutine 'X' was never awaited" warnings being missed. If RuntimeWarnings occur in the context a RuntimeError is raised. T)recordNcSs"g|]}|jtkrdj|dqS)z#{w.filename}:{w.lineno}:{w.message})w)categoryRuntimeWarningformat).0r7rrr s z,_runtime_warning_context..z{} Runtime Warning{}, {}rs )warningscatch_warnings RuntimeErrorr:lenjoin) _warningsrwrrr_runtime_warning_contextzsrGFccs(|r |Vnt}|Vt||ddS)z{ setups and tears down a loop unless one is passed in via the loop argument when it's passed straight through. r4N)r r )rr4rrr_passthrough_loop_contexts rIcCs(||r$t|r$t|||SdS)z/ Fix pytest collecting for coroutines. N)Zfuncnamefilterr-r.listZ _genfunctions)Z collectornameobjrrrpytest_pycollect_makeitemsrMc sjd}tjrjdp0jdd}tFt||d.}fddj j D}| j f|W5QRXW5QRXdSdS) zL Run coroutines in an event loop instead of a normal function call. r proactor_looprNrHcsi|]}|j|qSr)funcargs)r;arg pyfuncitemrr sz&pytest_pyfunc_call..T) r2r3r-r.functionrOgetrGrIZ _fixtureinfor/rrL)rRr4Z existing_loopr#ZtestargsrrQrpytest_pyfunc_calls    &rVcCsd|jkrdS|jjj}dtji}tdk r4tj|d<tdk rFtj|d<|dkrRd}i}| dD]N}| d }| d }||kr|r`t d |t |fnq`||||<q`|jdt |t |d dS) N loop_factoryruvlooptokioallzpyloop,uvloop?,tokio?,?z ?z&Unknown loop '%s', available loops: %s)Zids)r&r2optionZ aiohttp_loopr-ZDefaultEventLoopPolicyrXZEventLoopPolicyrYsplitendswithstrip ValueErrorrJkeysZ parametrizevalues)ZmetafuncZloopsZavail_factoriesZ factoriesrKrequiredrrrpytest_generate_testss8        rec csH|}t|t|d$}|r*|dt||VW5QRXdS)z%Return an instance of the event loop.rHTN)r-set_event_loop_policyr Z set_debugset_event_loop)rWr4r5policyr#rrrrs    rc csRtst}tj|_nt}t|t|j}t ||VW5QRXdSr) rr-Zget_event_loop_policyZProactorEventLoopZ _loop_factoryZWindowsProactorEventLoopPolicyrfr Znew_event_looprg)rhr#rrrrNs    rNcCstjdtdd|S)Nz3Deprecated, use aiohttp_unused_port fixture instead stacklevelr@warnDeprecationWarning)aiohttp_unused_portrrrrs rcCstS)z1Return a port that is unused on the current host.) _unused_portrrrrrosroc#s:gddfdd }|Vfdd}|dS)z^Factory to create a TestServer instance, given an app. aiohttp_server(app, **kwargs) Nportcs4t||d}|jfdi|IdH||SNrqr)r start_serverappend)Zapprrr)serverrserversrrgos  zaiohttp_server..gocsrIdHqdSrpopcloserrxrrfinalizesz aiohttp_server..finalizerrryr~rrwraiohttp_server s  rcCstjdtdd|S)Nz.Deprecated, use aiohttp_server fixture insteadrirjrl)rrrr test_server s rc#s:gddfdd }|Vfdd}|dS)zpFactory to create a RawTestServer instance, given a web handler. aiohttp_raw_server(handler, **kwargs) Nrqcs4t||d}|jfdi|IdH||Srs)rrtru)handlerrrr)rvrwrrry2s  zaiohttp_raw_server..gocsrIdHqdSrrzrr}rrr~:sz$aiohttp_raw_server..finalizerrrrwraiohttp_raw_server*s  rcCstjdtdd|S)Nz2Deprecated, use aiohttp_raw_server fixture insteadrirjrl)rrrrraw_test_serverAs rc#s:gddfdd }|Vfdd}|dS)zFactory to create a TestClient instance. aiohttp_client(app, **kwargs) aiohttp_client(server, **kwargs) aiohttp_client(raw_server, **kwargs) N) server_kwargscst|tr.t|ttfs.|f||}i}n |r:tdt|trv|pJi}t|fdi|}t|fdi|}n0t|trt|fdi|}ntdt|| IdH ||S)Nzargs should be emptyrzUnknown argument type: %r) isinstancerrrAssertionErrorr r ratypertru)Z__paramrr(r)rvZclientclientsrrrryUs"     zaiohttp_client..gocsrIdHqdSrrzr)rrrr~nsz aiohttp_client..finalizerrrrraiohttp_clientKs  rcCstjdtdd|S)Nz.Deprecated, use aiohttp_client fixture insteadrirjrl)rrrr test_clientus r)F),r- contextlibr@collections.abcrpytestZaiohttp.helpersrrZ aiohttp.webrZ test_utilsrrr r r r r rrprX ImportErrorrYrr0Zfixturer4r5contextmanagerrGrIrMrVrerrNrorrrrrrrrrrs`  (     8    "         )