<|hbUdZddlZddlZ ddlmZmZddlZddl Z ddl m Z ddl m Z dZGddeZGd d e e j"ZGd d eZGd deZGddeZGddeZGddeZGddeZGddeZGddeZGddeZedk(r edyy#e$r ddlmZmZYwxYw)z/ Various tests for synchronization primitives. N)start_new_thread get_ident)support)TimeAssertMixinc.tjdy)Ng{Gz?)timesleepV/var/www/html/test/engine/venv/lib/python3.12/site-packages/gevent/tests/lock_tests.py_waitr sJJtr c*eZdZdZddZdZdZdZy)Bunchz A bunch of threads. c_|_g_g_| _fd}t |D]}t |dy)z Construct a bunch of `n` threads running the same function `f`. If `wait_before_exit` is True, the threads won't terminate until do_finish() is called. cft}jj| jj|jst jsyy#jj|jst jswwxYwN)rstartedappendfinished _can_exitr )tidfselfs r taskzBunch.__init__..task$sx+C LL   $  $$S)..G.. $$S)..G..sA//>B0.B0r N)rnrrrranger)rrrwait_before_exitr_s`` r __init__zBunch.__init__sO   -- q 'A T2 & 'r ct|j|jkr.tt|j|jkr-yyr)lenrrr rs r wait_for_startedzBunch.wait_for_started0s3$,,$&&( G$,,$&&(r ct|j|jkr.tt|j|jkr-yyr)r!rrr r"s r wait_for_finishedzBunch.wait_for_finished4s3$-- 466) G$-- 466)r cd|_yNT)rr"s r do_finishzBunch.do_finish8s r N)F)__name__ __module__ __qualname____doc__rr#r%r(r r r rrs'.r rceZdZdZdZy) BaseTestCasec6tj|_yr)rthreading_setup_threadsr"s r setUpzBaseTestCase.setUp=s//1 r cdtj|jtjyr)rthreading_cleanupr1 reap_childrenr"s r tearDownzBaseTestCase.tearDown@s !!4==1r N)r)r*r+r2r6r r r r.r.<s 2 r r.cFeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) BaseLockTestsz; Tests for both recursive and non-recursive locks. ctrNotImplementedErrorr"s r locktypezBaseLockTests.locktypeJ !##r c&|j}~yr)r<rlocks r test_constructorzBaseLockTests.test_constructorMs}} r cF|j}|j~yr)r<acquirer?s r test_acquire_destroyz"BaseLockTests.test_acquire_destroyQs}}  r cf|j}|j|j~yrr<rCreleaser?s r test_acquire_releasez"BaseLockTests.test_acquire_releaseVs"}}   r c|j}|j|jd|jyNF)r< assertTruerCrGr?s r test_try_acquirezBaseLockTests.test_try_acquire\s+}}  U+, r c|jjgfd}t|dj|j dj y)NcFjjdyrJrrCr@resultsr rz3BaseLockTests.test_try_acquire_contended..fes MM$,,u- .r r)r<rCrr% assertFalserGrrr@rQs @@r test_try_acquire_contendedz(BaseLockTests.test_try_acquire_contendedasN}}  / a %%' # r ct|jjd}fd}t||}|jt |j t |jdj|j|j t |j|y)NcFjjyrrCrGr@sr rz/BaseLockTests.test_acquire_contended..fo LLN LLNr r) r<rCrr#r assertEqualr!rrGr%)rNrbr@s @r test_acquire_contendedz$BaseLockTests.test_acquire_contendedks}}    !QK   QZZ!,   QZZ!,r c|jfd}dfd }|t|dj|jt|tt|djy)NcFjjyrrYrZsr rz"BaseLockTests.test_with..f}r[r c>5|| dddy#1swYyxYwrr )errr@s r _withz&BaseLockTests.test_with.._withs* ?I#   srRr)r<rr% assertRaises TypeError)rrrdr@s @r test_withzBaseLockTests.test_with{sU}}    a %%' )UI6 a %%'r c|jfd}ttj}t |dj |j |ttjy)NcFjjyrrYrZsr rz)BaseLockTests.test_thread_leak..fr[r )r<r! threading enumeraterr%r\)rrrr@s @r test_thread_leakzBaseLockTests.test_thread_leaks\}}   ##% & a &&( C 3 3 567r N) r)r*r+r,r<rArDrHrLrUr_rgrmr r r r8r8Es4$   - ( 8r r8ceZdZdZdZdZy) LockTestszn Tests for non-recursive, weak locks (which can be acquired and released from different threads). c\|jgfd}t|ds ts t|jt dj t dk(rtt dk(r|jt dy)NcjjdjjdyrrCr)r@phasesr rz#LockTests.test_reacquire..fs- LLN LL  LLN LL r r rR)r<rr r\r!rG)rrr@rss @@r test_reacquirezLockTests.test_reacquires}}  B G  UQ' %jAo G%jAo UQ'r c|jjfd}t|d}|jjj y)Nc&jyr)rGrZsr rz*LockTests.test_different_thread..f LLNr rR)r<rCrr%rGrrr^r@s @r test_different_threadzLockTests.test_different_threadsE}}   !QK   r N)r)r*r+r,rurzr r r roros(& r roc(eZdZdZdZdZdZdZy) RLockTestsz$ Tests for recursive locks. c|j}|j|j|j|j|j|jyrrFr?s r ruzRLockTests.test_reacquiresC}}      r cd|j}|jt|j|j |j |j|j |j|j|jt|jyr)r<re RuntimeErrorrGrCr?s r test_release_unacquiredz"RLockTests.test_release_unacquiredsk}} , 5       , 5r c|jfd}t|dd} |jtj|j y#|j wxYw)Nc&jyr)rCrZsr rz+RLockTests.test_different_thread..frxr rRT)r<rrerrGr(rys @r rzz RLockTests.test_different_threadsM}}  !Q     lDLL 9 KKMAKKMs AA(cJ|j|jjj|j jj|j jgfd}t |dj |jdj|j jj|jjy)NcDjjyr)r _is_ownedrPsr rz$RLockTests.test__is_owned..fs MM$..* +r rRr)r<rSrrCrKrr%rGrTs @@r test__is_ownedzRLockTests.test__is_owneds}} )*  ()  () , a %%' #  ()  )*r N)r)r*r+r,rurrzrr r r r|r|s 6 +r r|c.eZdZdZdZdZdZdZdZy) EventTestsz" Tests for Event objects. ctrr:r"s r eventtypezEventTests.eventtyper=r c|j}|j|j|j|j |j|j|j |j|j |j|j|j |j|jyr)rrSis_setsetrKclearrevts r test_is_setzEventTests.test_is_setsnn &    %    %  &  &r c>d}ggfd}t||}|jt|jt dj |j |jdg|z|jdg|zy)NrWcjjjjjjyr)waitrr)rresults1results2sr rz#EventTests._check_notify..f s9 HHJ OOCJJL ) HHJ OOCJJL )r rT)rr#r r\r!rr%)rrr]rr^rrs ` @@r _check_notifyzEventTests._check_notifys  * !QK   X*    D6A:. D6A:.r c|j}|j||j|j|j|yr)rrrrrs r test_notifyzEventTests.test_notifys<nn 3    3r c|jggd}fd}t||j|jdg|zD])\}}|j ||j |dd+ggj t||j|jdg|zD]\}}|j|y)NrWc$jdjjtj}jdj}tj}j|||z fy)Ng皙?)rrrr)t1rt2rrrs r rz"EventTests.test_timeout..f's` HHSM OOCJJL )B HHSM AB OOQRL )r Fg ףp= ? T)rrr%r\rSassertTimeWithinRangerrK)rr]rrdtrrrs @@@r test_timeoutzEventTests.test_timeout"snn  * a %%' E7Q;/ 5EAr   Q   & &r4 4 5   a %%' D6A:. EAr OOA  r N) r)r*r+r,rrrrrr r r rrs $ '/& r rc:eZdZdZdZdZdZdZdZdZ dZ y ) ConditionTestsz( Tests for condition variables. ctrr:rargss r condtypezConditionTests.condtypeCr=r c||j}|j|j|j|jtj}|j|}|j|j |jd|j|j |jd|j |jd|j|5|j |jddddy#1swYyxYwrJ)rrCrGrkLockrSrK)rcondr@s r test_acquirezConditionTests.test_acquireFs}}    ~~}}T"  e,-   U+, e,-   2   T\\%0 1 2 2 2s !D22D;cd|j}|jt|jyr)rrerrrrs r test_unacquired_waitz#ConditionTests.test_unacquired_waitZs!}} , 2r cd|j}|jt|jyr)rrernotifyrs r test_unacquired_notifyz%ConditionTests.test_unacquired_notify^s!}} , 4r cld}ggdfd}t||}|jt|jgj j dtdj tdkrttdkr|jdgdz|jgj j dtdj ttzdkr%tttzdkr%|jdgdzdgdzz|jdgdzj jtdj tdkrttdkr|jdgdzdgdzz|jdgdzdgdzz|jy)NrWrc jjjjjjjjyr)rCrrGr)r phase_numrrsr rz'ConditionTests._check_notify..fgsQ LLN IIK LLN OOI & LLN IIK LLN OOI &r rRrt) rr#r r\rCrrGr! notify_allr%)rrr]rr^rrrs ` @@@r rzConditionTests._check_notifybs   ' !QK   2&  A   (ma G(ma A37+ 2&  A   (mc(m+a/ G(mc(m+a/ A37aS1W#45 A37+     (ma G(ma A37aS1W#45 A37aS1W#45 r ch|j}|j||j|yr)rrrs r rzConditionTests.test_notifys)}} 4  4 r c|jgd}fd}t||j|jt dD]}|j |ddy)NrWcjtj}jdtj}jj ||z y)Nr)rCrrrGr)rrrresultss r rz&ConditionTests.test_timeout..fsF LLNB IIcNB LLN NN27 #r gRQ?@)rrr%r\r!r)rr]rrrrs @@r rzConditionTests.test_timeoutsf}}  $ a %%' Wq) 6B  & &r4 5 6r N) r)r*r+r,rrrrrrrr r r rr>s+$2(350d! 6r rcFeZdZdZdZdZdZdZdZdZ dZ d Z d Z y ) BaseSemaphoreTestszB Common tests for {bounded, unbounded} semaphore objects. ctrr:rs r semtypezBaseSemaphoreTests.semtyper=r c |jt|jd|jttf|jt t dt t dd y)N)valuemaxintmaxsize)re ValueErrorr OverflowErrorgetattrsysr"s r rAz#BaseSemaphoreTests.test_constructorsU *dll"= :}5t||!(hYPT8U!V V  Xr c|jd}|j|j|jd}|j|j|j|jy)NrRrt)rrCrGrsems r rzBaseSemaphoreTests.test_acquiresRll1o  ll1o    r cF|j}|j~yr)rrCrs r rDz'BaseSemaphoreTests.test_acquire_destroyslln r c|jdjggdfd}t|d}|jt t zdkr%t t t zdkr%|j zdgdzdtdD]}jt t zdkr%t t t zdkr%|j tzdgdzdgdzzdtdD]}jt t zd kr%t t t zd kr%|j tzdgdzdgdzzdgdzz|jjd j|jy) Nrcjjjjyrrr)rrrrsr rz4BaseSemaphoreTests.test_acquire_contended..fs- KKM OOI & KKM OOI &r rrR rtF) rrCrr#r!r r\rrGsortedrSr%)rrr^rrrrrs @@@@r r_z)BaseSemaphoreTests.test_acquire_contendedsll1o   ' !RL (mc(m+a/ G(mc(m+a/ H,qcAg6 q A KKM (mc(m+b0 G(mc(m+b0 8 34qcAga6GH q A KKM (mc(m+b0 G(mc(m+b0 8 34qcAga6G1#PQ'6QR U+,  r cF|jd}|j|jd|j|jd|j|jd|j |j|jdy)NrtF)rrKrCrSrGrs r rLz#BaseSemaphoreTests.test_try_acquiresnll1o  E*+  E*+ U+,   E*+r c|jdjgfd}t|dj|j t dgdzdgdzzy)NcjjdjjdyrJrO)rrsr rz8BaseSemaphoreTests.test_try_acquire_contended..fs- NN3;;u- . NN3;;u- .r rWFrTr)rrCrr%r\r)rrrrs @@r rUz-BaseSemaphoreTests.test_try_acquire_contendeds^ll1o  / a %%' 5'A+! *CDr c|jjfd}t|d}|jt |j |j j|jy)NcFjjyrrY)rsr rz0BaseSemaphoreTests.test_default_value..fs KKM KKMr rR) rrCrr#r rSrrGr%)rrr^rs @r test_default_valuez%BaseSemaphoreTests.test_default_valuesalln   !QK   $  r c<jddfd }|jjdjj t |t jjdjy)Nrtc 5jjdj5jjd|r| ddddddy#1swYxYw#1swYyxYwrJ)rKrCrGrS)rcrrs r rdz+BaseSemaphoreTests.test_with.._withsz " E 23 "$$S[[%78! " " """ " "s"3A:%A.A:.A7 3A::BFr)rrKrCrGrerf)rrdrs` @r rgzBaseSemaphoreTests.test_withsjll1o "   E*+  )UI6  E*+ r N) r)r*r+r,rrArrDr_rLrUrrgr r r rrs7$X  !F, E r rceZdZdZdZy)SemaphoreTestsz) Tests for unbounded semaphores. c|jd}|j|j|j|jyNrR)rrGrCrs r rz&SemaphoreTests.test_release_unacquired%s3ll1o    r Nr)r*r+r,rr r r rr s r rceZdZdZdZy)BoundedSemaphoreTestsz' Tests for bounded semaphores. c|j}|jt|j|j |j|jt|jyr)rrerrGrCrs r rz-BoundedSemaphoreTests.test_release_unacquired3sGlln *ckk2   *ckk2r Nrr r r rr.s 3r rcneZdZdZdZdZdZdZdZdZ ddZ d Z d Z d Z d Zd ZdZdZdZdZy) BarrierTestsz$ Tests for Barrier objects. rWrc\|j|j|j|_y)Ntimeout) barriertyper]defaultTimeoutbarrierr"s r r2zBarrierTests.setUpBs#''8K8K'L r c8|jjyr)rabortr"s r r6zBarrierTests.tearDownDs r cdt||jdz }||jyr)rr]r%)rrr^s r run_threadszBarrierTests.run_threadsGs' !TVVAX   r c`|jj}|j||jt |D]}|dj d|jt |d||z|jj|dj d|jt |d|dz|z|jj|j|jjd|j|jjy)NrTrR) rpartiesr\r]rrr!r n_waitingrSbroken)rrrmis r multipasszBarrierTests.multipassLs LL  DFF#q A AJ  d #   S_a!e 4 LL    AJ  d #   S_q1uk : LL      //3 ,,-r cBgggfd}j|y)z; Test that a barrier is passed in lockstep c*jyr)r)passesrrsr rz$BarrierTests.test_barrier..f^s NN7F +r Nr)rrrrs`` @r test_barrierzBarrierTests.test_barrierYs!r( , r c$|jdS)zC Test that a barrier works for 10 consecutive runs r)rr"s r test_barrier_10zBarrierTests.test_barrier_10bs  $$r cgfd}j|jtttjy)z9 test the return value from barrier.wait c\jj}j|yr)rrr)rrrs r rz(BarrierTests.test_wait_return..fms! !!#A NN1 r N)rr\sumrr])rrrs` @r test_wait_returnzBarrierTests.test_wait_returnhs@   Ws5='9:r cgfd}jj|fd}j|y)z, Test the 'action' callback c(jdyr')r)rsr actionz(BarrierTests.test_action..actionys NN4 r c\jjtdyr)rr\r!)rrrsr rz#BarrierTests.test_action..f|s LLN   S\1 -r Nrr]r)rrrrrs` @@r test_actionzBarrierTests.test_actionts: !""46662 . r cggfd}j|jtdjtjdz j j j y)zK Test that an abort will put the barrier in a broken state c^ jj}|jdzk(rtjjj dy#t j $rj dYyt$rjjYywxYwNrtT)rrr]rrrkBrokenBarrierErrorr)rrrrs r rz"BarrierTests.test_abort..fs %LL%%' >&& !!#%// &% % ""$ %sAA!!$B,"B,+B,rrRN)rr\r!r]rKrr)rrrrs` @@r test_abortzBarrierTests.test_abortsj %  X* Xq1  ++,r cgggfd}j|jtdjtjdz jtjy)zL Test that a 'reset' on a barrier frees the waiting threads cZjj}|jdzk(r}jjjdz kr.fs !!#ADFFAI~ll,,tvvax7JJu%ll,,tvvax7 ""$*LL%%'OOD) LL    OOD ! !33*OOD)*s,+D$D*)D*rrRN)rr\r!r])rrrrrs` @@@r test_resetzBarrierTests.test_resetsq ""  X* Xq1 X/r cTgggjjfd}j|jt djt jdz jt jy)zF Test that a barrier can be reset after being broken. cH jj}|jdzk(rtjjj djjdzk(rjjjjjj dy#t j $rj dYt$rjjYwxYwr) rrr]rrrkrrr)rbarrier2rrrrs r rz,BarrierTests.test_abort_and_reset..fs %LL%%' >&& !!#%}}$&&!)+ ""$ MMO LL    OOD !// &% % ""$ %sAC$D!<"D! D!rrRN)rr]rr\r!)rrrrrrs` @@@@r test_abort_and_resetz!BarrierTests.test_abort_and_resets##DFF+ " "*  X* Xq1 X/r c2fd}j|y)z$ Test wait(timeout) cjj}|jdzk(rtjdj t jjjdy)Nrt?g?)rrr]rr rerkr)rrs r rz$BarrierTests.test_timeout..fsU !!#ADFFaK 3   i::"ll// 6r Nr)rrs` r rzBarrierTests.test_timeouts 6 r cpjjdfd}j|y)z4 Test the barrier's default timeout g333333?rcj}|jdzk(rtjdj t j jy)Nrtr)rr]rr rerkr)rrrs r rz,BarrierTests.test_default_timeout..fsC ADFFaK 3   i::GLL Ir Nr )rrrs` @r test_default_timeoutz!BarrierTests.test_default_timeouts4 ""4663"7 J r cf|jd}|j|jyr)rr)rr^s r test_single_threadzBarrierTests.test_single_threads$   Q   r N)rR)r)r*r+r,r]rr2r6rrrrrr rrrrrr!r r r rr;s[ ANM .% ; -.0< 0D  r r__main__zSThis module contains no tests; it is used by other test cases like test_threading_2)r,rrthreadrr ImportError_threadrkunittestgevent.testingrgevent.testing.testcaserr objectrTestCaser.r8ror|rrrrrrr)printr r r r,s 42"3$F$N ?H$5$5 Q8LQ8h! !H3+3+lHHVk6\k6\rrh '  3. 3}<}@ z _`e4334sB>> C C