<|hGBddlmZddlZddlZddlmZddlmZddlmZ ddl Zddl Z ddlm Z ddl Z Gdde jZdZGd d eZGd d e jZGd deZd3dZdZdZdZd\ZZZdZdZe j<r e j>se j@s e jBrdZne jDs e jFrdZGdde jZ$e jJGdde$Z&e jJGdde$Z'e jJGdde$Z(Gd d!e$Z)Gd"d#e jZ*Gd$d%ejjVjXZ-Gd&d'ejjVjXZ.Gd(d)e jZ/Gd*d+e jZ0d,Z1Gd-d.e jZ2d/Z3Gd0d1e jZ4e5d2k(re jlyy)4)timeN)Event)QueueExpectedExceptionceZdZejj ZdZdZdZ de _ dZ dZ dZ ejdd Zy ) TestCoroutinePoolctfd}|jd}|j|djy)Nc&jyN)set)_dones V/var/www/html/test/engine/venv/lib/python3.12/site-packages/gevent/tests/test__pool.py some_workz5TestCoroutinePool.test_apply_async..some_works HHJx)rklass apply_asyncwait)selfrpoolrs @rtest_apply_asyncz"TestCoroutinePool.test_apply_asyncs7w zz!} G, rc|dfd}|jd}|j|}|j|y)Nz return valuecSr )valuesrrz/TestCoroutinePool.test_apply..some_worksLrrrapply assertEqual)rrrresultrs @r test_applyzTestCoroutinePool.test_applys: zz!}I& 'rc|jd}d} |j||jdy#t$rYywxYw)Nctr rrrrraiserz3TestCoroutinePool.test_apply_raises..raiser)s #% %rz$Should have raised ExpectedException)rr!failr)rrr(s rtest_apply_raisesz#TestCoroutinePool.test_apply_raises&sHzz!} & > JJv  II< =!   s9 AAFctgfd}fd}|jd}|j|}|j||j |j gdy)Ncrtjdjdjy)NMbP?prod)geventsleepappendr evtresultssrproducerz7TestCoroutinePool.test_multiple_coros..producer:s$ LL  NN6 " GGIrcjjdjjdy)Ncons1cons2)r1rr2srconsumerz7TestCoroutinePool.test_multiple_coros..consumer?s$ NN7 # HHJ NN7 #rr)r7r.r8)rrspawnrgetr")rr5r9rrr3r4s @@rtest_multiple_corosz%TestCoroutinePool.test_multiple_coros6s^g   $ zz!}zz(# "   3W=rcgfdfd}|jd}|j|tjd|j gy)Nc(jdy)NT)r1) timer_firedsr fire_timerz.fire_timerMs   t $rc2tjdy)Nr)r/timer)r@srrz;TestCoroutinePool.dont_test_timer_cancel..some_workPs LLJ 'rrr)rr!r/r0r")rrrr@r?s @@rdont_test_timer_cancelz(TestCoroutinePool.dont_test_timer_cancelJsH  % (zz!} 9 Q b)rc|jd}|j|jddf}|j|dt}|j |j |j y)Nr&c |dzS)Nr&ras rz2TestCoroutinePool.test_reentrant..Zs 1q5r)rr!r"rrr r)rrr#r3s rtest_reentrantz TestCoroutinePool.test_reentrantXsZzz!}DJJ%(@A #g !  rzDoes not work on PyPycddl}|jd}|j} t|_|j t }t jd5|jt|jdddt jd|j|jdt jjd5|jt jdddd||_|j!y#1swYxYw#1swY-xYw#||_|j!wxYw)Nrr&sizer皙?)r)sysrstderrFakeFiler:crashr/Timeout assertRaises RuntimeErrorr;r0r" free_count start_newr!join)rrQr normal_errwaiters rtest_stderr_raisingz%TestCoroutinePool.test_stderr_raising`s zzqz!ZZ  !CJZZ&F" < !!, ;  < LLO   T__. 2))#. 0 6<</ 0$CJ IIK) < <  0 0$CJ IIKs<9D3!D>AD3!D';D3D$ D3'D0,D33E N)__name__ __module__ __qualname__r/rPoolrrr$r* error_fatalr<rCrL greentest skipOnPyPyr]rrrr r s\ KK  E( >%*!>( *Y12!3!rr ctd)Nz*Raising an error from the crash() functionrW)_args_kws rrTrTs C DDrceZdZdZy)rSctd)NzWriting to the file failedrf)rrgs rwritezFakeFile.writes788rN)r^r_r`rkrrrrSrSs9rrSceZdZejj ZdZdZdZ e jdZ e jdZ e jdZdZdZy ) PoolBasicTestsc|jd}|j|jdg}|j|jd}|j|jd|j |j|dgt jd|j|jd|j|jd|jd|j|j|dg|j|jd|jd|j|j|dg|j|jd|j|dgt jd|jt|gd y) NrrNr&r)r)){Gz?)r&rrorp) rr"rXr:r1r;r/r0rsorted)rprfirsts rtest_execute_asyncz!PoolBasicTests.test_execute_asyncsM JJAJ  + !$ +  QC  Q + ahh& ALLN+ QC  ahh& ALLN+ QC  ahh& QC  T L1rcN|jd}|jtjd}|j ||j |j ||jt|d|j|jjdy)Nr&rNr) rr:r/r0discardkill assertFalser"len _semaphorecounter)rrsrus r test_discardzPoolBasicTests.test_discardsw JJAJ  d+ %   Q# --q1rc|jd}tjtjd} tjtjd} |j |j d|j t |d|j||j |j d|j t |d|jtj5tjd5|j|dddddd|j |j d|j t |d|j |jy#1swYnxYw#1swYrxYw#|jwxYw#|jwxYw)Nr&rNrxrrP) rr/r:r0r"rXr|addrVrUrzrrsruseconds rtest_add_methodzPoolBasicTests.test_add_methodsJ JJAJ  V\\40 \\&,,5F   3  Q+e   3  Q+&&v~~6&,&f &&  3  Q+ JJL&&&& JJLsU$GB&F0F$F+F$3AF06GF! F$$F-)F00GGGc|jd}tjtjd} tjtjd} |j ||j tj j5|j |dddd|j |jy#1swY+xYw#|jwxYw#|jwxYw)Nr&rNrxF)blocking rr/r:r0rrVrPoolFullrzrs rtest_add_method_non_blockingz+PoolBasicTests.test_add_method_non_blockings JJAJ  V\\40 \\&,,5F e &&v{{';';<2EE&5E12 JJL 22 JJL;$C6:C!C+C!3C6CC!!C33C66Dc|jd}tjtjd} tjtjd} |j ||j tj j5|j |dddd|j |jy#1swY+xYw#|jwxYw#|jwxYwNr&rNrxrPtimeoutrrs rtest_add_method_timeoutz&PoolBasicTests.test_add_method_timeouts JJAJ  V\\40 \\&,,5F e &&v{{';';<1EE&%E01 JJL 11 JJLrc|jd}tjtjd} tjtjd} |j ||j tjj5|j|dddd|j |jy#1swY+xYw#|jwxYw#|jwxYwr) rr/r:r0GreenletrrVrrstartrzrs rtest_start_method_timeoutz(PoolBasicTests.test_start_method_timeouts JJAJ  V\\40 __V\\48F e &&v{{';';<3GGFEG23 JJL 33 JJLrcn|j}|jdd}|j|dy)Nc d|fS)NfoorrFs rrHz+PoolBasicTests.test_apply..s E1:r)r&)rr&r )rrsr#s rr$zPoolBasicTests.test_applys. JJL-u5 ,rcTd|_|jt|jdy)NF)switch_expectedrV ValueErrorrrs rtest_init_errorzPoolBasicTests.test_init_errors $ *djj"5rN)r^r_r`r/rrarrvrrrcignores_leakcheckrrrr$rrrrrmrmsx KK  E262.   !    !    ! - 6rrmceZdZdZdZy) TimingWrapperc ||_d|_yr )funcelapsed)rrs r__init__zTimingWrapper.__init__ s  rct} |j|i|t|z |_S#t|z |_wxYwr )rrr)rargskwdsts r__call__zTimingWrapper.__call__ s< F &499d+d+6A:DL46A:DLs 0AN)r^r_r`rrrrrrrs &rrc6tj|||zSr )r/r0)rrs rsqrrs LL q5Lrc ||zSr rrs rsquaredrs q5Lrc`tjtjdz||zS)NrP)r/r0randomrs rsqr_random_sleeprs" LL3&' q5Lrc#`KtdEd{tjdy7w)Nro皙?)ranger/r0rrr final_sleepr#s$Qx LLs .,.)gˡE?gQ?gQ? rxdceZdZejZdZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZdZdZdZdZy)TestPoolr&ctjj|tjj |j |_yr )rcTestCasesetUpr/rrarOrs rrzTestPool.setUp8s0  &KK$$TYY/ rc8|jjyr )rrZrs rcleanupzTestPool.cleanup<s rc|jj}|j|tdd|j|tdddidy)NrIrrrro )rr!r"r)rpapplys rr$zTestPool.test_apply?sC T*B/ R#q2A6rc X|jj}|j|tt t t ttt t |j|tt dt ttt dy)Nr)rmapr"rr SMALL_RANGElistr)rpmaps rtest_mapzTestPool.test_mapDshyy}} c5#56S%P[J\=]8^_ c5:.S%*5M0NOrc|jjtdtf}t |j }|j |d|j|jtdy)N1r&) rrrTIMEOUT1rr;r"assertTimeoutAlmostEqualrrresr;s r test_asynczTestPool.test_asyncIsRii##C!X8CGG$ # %%ckk8Q?rcVg}|jjtdtf|j}t |j }|j|d|j|jtdtjd|j|dgy)Nr)callbackrr&r) rrrrr1rr;r"rrr/r0)rr#rr;s rtest_async_callbackzTestPool.test_async_callbackOs}ii##C!X&--#PCGG$ # %%ckk8Q? Q "&rcB|jjtdtdzf}t |j }|j tj|t|j|jtd|jjy)NrKrrr&) rrrTIMEOUT2rr;rVr/rUrrrZrs rtest_async_timeoutzTestPool.test_async_timeoutXslii##C!X^)<=CGG$ &..#x@ %%ckk8Q? rc |jjttt}|j t |t ttttyr )rimaprrrr"rrrits rtest_imap_list_smallzTestPool.test_imap_list_small_s@ YY^^C{!3 4 b4C{1C(D#EFrc|jjttt}ttD] }|j t |||z"|jtt |yr ) rrrrrr"nextrV StopIterationrris rtest_imap_it_smallzTestPool.test_imap_it_smallcY YY^^C{!3 4{# .A   T"Xq1u - . -r2rc|jjttt}ttD] }|j t |||z"|jtt |yr ) rrrr LARGE_RANGEr"rrVrrs rtest_imap_it_largezTestPool.test_imap_it_largeirrc |jjttt}|j t |t ttttyr ) rrrrrr"rrrrs rtest_imap_randomzTestPool.test_imap_randomosA YY^^,eK.@ A b4GU;5G(H#IJrc |jjttt}|j t |ttttt|jjttt}|j t |tttttyr ) rimap_unorderedrrrr"rrrrrrs rtest_imap_unorderedzTestPool.test_imap_unorderedss YY % %c5+= > T#gu[7I*J%KL YY % %c5+= > T#gu[7I*J%KLrc |jjttt}|j t |tttttyr ) rrrrrr"rrrrrrs rtest_imap_unordered_randomz#TestPool.test_imap_unordered_randomzsC YY % %&6k8J K T#gu[7I*J%KLrcz|jjtg}|jt |gyr )rrrr"rrs rtest_empty_imap_unorderedz"TestPool.test_empty_imap_unordered~s, YY % %c2 . b2&rcz|jjtg}|jt |gyr )rrrr"rrs rtest_empty_imapzTestPool.test_empty_imaps* YY^^C $ b2&rcd|j|jjtggyr )r"rrrrs rtest_empty_mapzTestPool.test_empty_maps! sB/4rcR|jjtjdg|jxsddzz}tjdt |jj }||j|jdd|jy)NrPrr?) r map_asyncr/r0rOrrzassertTimeWithinRangerrZ)rr#rzs rtest_terminatezTestPool.test_terminatess$$V\\C5TYY_"PQ>Q 'H'H *rct}|jdtjd|jtt |j jd|}|j|dgy)N{rPc|Sr rrs rrHz,TestPool.test_imap_no_stop..sqr) rputr/ spawn_laterrrrrr"rqr#s rtest_imap_no_stopzTestPool.test_imap_no_stopsU G c 3}5diinn[!45 #'rct}|jdtjd|jtt |j jd|}|j|dgy)NirPc|Sr rrs rrHz6TestPool.test_imap_unordered_no_stop..sr) rrr/rrrrrr"rs rtest_imap_unordered_no_stopz$TestPool.test_imap_unordered_no_stopsW G d 3}5dii..{A>? $(rct|jjtt }|j |gdyN)rr&rp)rrrrrr"rr#s rtest_imap_final_sleepzTestPool.test_imap_final_sleeps-diinnS+-89 +rct|jjtt }|j |gdyr )rrrrrr"rs rtest_imap_unordered_final_sleepz(TestPool.test_imap_unordered_final_sleeps/dii..sKMBC +rc$ttd}dgfd}fd}jjjjfD]=}dd<||||d}||}|}j t ||?y)Nrrc"dxxdz cc<|S)Nrr&r)r_jrunnings rshort_running_funczFTestPool.test_imap_unordered_bounded_queue..short_running_funcs AJ!OJHrcfd}|S)Nc0g}tD]\}}jd|dzkd|j|tjdjt j dkt j |S)Nrrrq) enumerate assertTruer1r/r0r|queue)r#rrmappingrrs rreaderzOTestPool.test_imap_unordered_bounded_queue..make_reader..readers%g.QDAqOOGAJ!a%$7DMM!$LL&OOC $6!$;S=OP Q  rr)rrrrs` r make_readerz?TestPool.test_imap_unordered_bounded_queue..make_readers Mrr&)maxsize)rrrrrr"rr) riterablerrmethrrlrs ` @r!test_imap_unordered_bounded_queuez*TestPool.test_imap_unordered_bounded_queuesc ##  II,,diinn< 2DGAJ-x#$&G!)FA   VAY 1 2rN)r^r_r`rc LARGE_TIMEOUT __timeout__rOrrr$rrrrrrrrrrrrrrr0rrr rrr#rrrrr4s))K D07 P @ 'G3 3 KMM''5+(),, 2rrceZdZdZy) TestPool2rNr^r_r`rOrrrr'r' Drr'ceZdZdZy) TestPool3roNr(rrrr+r+r)rr+ceZdZdZy) TestPool10rNr(rrrr-r-s Drr-ceZdZdZy)TestPoolUnlimitNr(rrrr/r/s Drr/ceZdZdZdZy) TestPool0rctjjd}|jd|j |j |j |jd|jdy)NrrNrqr)r/rrar"rXrfullwait_availablerrss rtest_wait_fullzTestPool0.test_wait_fulls_ KK  !  $ ALLN+ ! A,,T,:;rN)r^r_r`rOr6rrrr1r1s Drr@ceZdZdZdZy) TestJoinEmptyFctjj}|j}|j |dy)Nzempty should return true)r/rrarZrrrsrs rtestzTestJoinEmpty.test s. KK   ffh 78rN)r^r_r`rrGrrrrDrD s O9rrDceZdZdZdZdZy) TestSpawnTctjjd}|jt |d|j tj d|jt |d|j tj d|jt |dtj tjsdnd|jt |dy)Nr&rrPgRQ?r) r/rrar"r|r:r0rcEXPECT_POOR_TIMER_RESOLUTIONr5s rrGzTestSpawn.tests KK  Q  Q#  c" Q#  c" Q# !G!GTSQ Q#rctjjd}|jt |d|j tj d|jt |d|jd}|j|d|j}|j|d|jt |dy)Nr&rrPrqz*waiting on a full pool should return falsez waiting to finish should be true) r/rrar"r|r:r0rZr{rrFs rtestSpawnAndWaitzTestSpawn.testSpawnAndWaits KK  Q  Q#  c" Q#ffTl JKffh ?@ Q#rN)r^r_r`rrGrMrrrrIrIsO$ $rrIc#&Kddtw)Nr&rrrrr error_iterrO*s G G sceZdZdZdZdZy)TestErrorInIteratorFctjjd}|jt|j dt tjdy)Nrocyr rrs rrHz*TestErrorInIterator.test..5rr-)r/rrarVrrrOr0r5s rrGzTestErrorInIterator.test3s= KK  Q  +QUUNJLQ Urctjjdfd}|jt|tj dy)NrocLtjdtS)Ncyr rrs rrHzGTestErrorInIterator.test_unordered..unordered..<rTr)rrrO)rssr unorderedz5TestErrorInIterator.test_unordered..unordered;s((FG Grr-)r/rrarVrr0)rrXrss @rtest_unorderedz"TestErrorInIterator.test_unordered8s< KK  Q  H +Y7 UrN)r^r_r`rbrGrYrrrrQrQ0sK rrQc d|z S)N?rrs r divide_byr\Bs 7Nrc"eZdZdZdZdZdZy)TestErrorInHandlerFctjjd}|jt|j t gdy)Nror&rr)r/rrarVZeroDivisionErrorrr\r5s rrzTestErrorInHandler.test_mapIs/ KK  Q  +QUUIyIrcJtjjd}|jtgd}|j t |d|jtt ||j t |d|jtt |yNr&r`r[r) r/rrarr\r"rrVrarrrsrs r test_imapzTestErrorInHandler.test_imapMss KK  Q  VVIy ) b3' +T26 b3' -r2rcJtjjd}|jtgd}|j t |d|jtt ||j t |d|jtt |yrc) r/rrarr\r"rrVrarrds rrz&TestErrorInHandler.test_imap_unorderedUsu KK  Q    i 3 b3' +T26 b3' -r2rN)r^r_r`rbrrerrrrr^r^FsKJ33rr^__main__)r)7rr/ gevent.pool gevent.eventr gevent.queuergevent.testingtestingrcgevent.testing.timingrrunittestrr rTobjectrSrmrrrrrrrTIMEOUT3rrPYPYWINRUN_LEAKCHECKS RUN_COVERAGE RUNNING_ON_CIrKrrr'r+r-r/r1timingAbstractGenericWaitTestCaser8r@rDrIrOrQr\r^r^mainrrrrys  " ,s))slE9v9 r6Y''r6p &F &   2(H   NNy}})A)AYE[E[K F FKb2y!!b2H         h< ""<FNN))EE 5 5 Q Q9I&&9$ ""$0 ),,$3++30 zINNr