U lu f@sdZddlZddlZddlZddlZddlZddlmZddlm Z zddl Z ddl m Z Wne k rtdZ YnXe dkrdZ ddl mZddlZddlZddlZejdZd Zd d Zejd d ZddZejddZejddZejddZejddZejddZejddZejddZ ddZ!d dd!d"d#d$Z"ee"d%d&Z#ejd'd(Z$ejd)d*Z%ejd+d,Z&ejd-d.Z'dS)/z!pytest configuration and fixturesN)partial)Thread) version_info))IOLoopZZMQ_TEST_TIMEOUT cCs4|D]*}t|jr|dt|jrtqdS)a+This function is automatically run by pytest passing all collected test functions. We use it to add asyncio marker to all async tests and assert we don't use test functions that are async generators which wouldn't make sense. It is no longer required with pytest-asyncio >= 0.17 asyncioN)inspectiscoroutinefunctionobjZ add_markerisasyncgenfunctionAssertionError)itemsitemr6/tmp/pip-unpacked-wheel-h6ekxre8/zmq/tests/conftest.pypytest_collection_modifyitemss  rcsPtdkrttt|ks(tj|ks6tfdd}| |S)z4Create tornado io_loop on current asyncio event loopNcsjdddS)NT)Zall_fds)closerio_looprr_close4szio_loop.._close) tornadopytestskiprcurrentrZget_event_loopr Z asyncio_loopZ addfinalizer) event_looprequestrrrrr+s  rcCsNt|jd}d|_||j|d|rJdtjjj _ t d|ddS)z"Terminate a context with a timeout)targetTtimeoutNzcontext z8 could not terminate, open sockets likely remain in test) rZtermdaemonstartjoinis_alivezmqsugarcontextContext _instance RuntimeError)ctxrtrrr term_context;s    r,ccs.t}|V|ttjjiks*tdSN)rZnew_event_looprdictr$Z _selectorsr )ZlooprrrrIsrcCs6ttdrtsdSdd}ttj|ttdS)z~Set timeout using SIGALRM Avoids infinite hang in context.term for an unclean context, raising an error instead. SIGALRMNcWstdtddS)NzTest did not complete in z seconds) TimeoutErrortest_timeout_seconds)argsrrr_alarm_timeout\sz'sigalrm_timeout.._alarm_timeout)hasattrsignalr1r/alarm)r3rrrsigalrm_timeoutRs r7cCstjS)zaContext class fixture Override in modules to specify a different class (e.g. zmq.green) )r$r'rrrrr'csr'c csLt}|V|D]6}zt|tWqtk rDdtjjj_YqXqdS)zFFixture to track contexts used in tests For cleanup purposes N) setr,teardown_timeout Exceptionr$r%r&r'r()r7contextsr*rrrr;ls r;cCs|}|||S)zFixture for shared context)add)r'r;r*rrrr&}s r&ccs:g}|V|D]}||jq|D]}|jddq$dS)Nr)linger)r<r&r)r;socketssocketrrrr>s r>csfdd}|S)z:Fixture to create sockets, while tracking them for cleanupcsj||}||Sr-)r?append)r2kwargssr&r>rr new_sockets  zsocket..new_socketr)r&r>rDrrCrr?sr?c csxz dVWnPtjk rZ}z0|j|ksJtdt|dt|jW5d}~XYnXtdt|ddS)Nzwrong error raised, expected z got z Expected z, no error raised)r$ZZMQErrorerrnor rZfail)rEerrrassert_raises_errnos rGrF)rflags multipartc Ksttdkrtdtj|ggg|d\}}}|s.create_bound_pair)r$ZPAIR)r?r[rrZrr[s r[cCs|Sr-rr[rrr bound_pairsr]cCs|tjtjSr-)r$ZPUSHZPULLr\rrr push_pullsr^cCs|tjtjSr-)r$ZDEALERZROUTERr\rrr dealer_routersr_)(__doc__rr osr5rL functoolsr threadingrrr ImportErrorZtornado.iolooprrr$Z zmq.asyncioenvironrOr1r9rZfixturerr,rr7r'r;r&r>r?rGrQrPr[r]r^r_rrrrsb