i\>ddlmZddlmZdZeeddedDZ ddedDZ ddlm Z e j Z ddl Z ddlZddlZddlZddlZddlZddlZddlZddlmZd Z dd lmZd ZGd d eZGddejZeGddej Z!eGddej Z"eGddej Z#eGddej$Z$eGddej%Z%eGddej%Z&eGddej'Z'eGddej%Z(eGd d!ej)Z)eGd"d#ej*Z*eGd$d%ej+Z+e,d&krej-dSdS)')xrangeNafrom gevent import monkey; monkey.patch_all() from gevent.event import Event from gevent.lock import RLock, Semaphore, BoundedSemaphore from gevent.thread import allocate_lock as Lock import threading threading.Event = Event threading.Lock = Lock # NOTE: We're completely patching around the allocate_lock # patch we try to do with RLock; our monkey patch doesn't # behave this way, but we do it in tests to make sure that # our RLock implementation behaves correctly by itself. # However, we must test the patched version too, so make it # available. threading.NativeRLock = threading.RLock threading.RLock = RLock threading.Semaphore = Semaphore threading.BoundedSemaphore = BoundedSemaphore  c# K|] }d|zV dS)z %sN.0lines I/usr/local/lib/python3.11/dist-packages/gevent/tests/test__threading_2.py r s(KK$t+KKKKKKc# K|] }d|zV dS)z %sNrrs r r r s(OOD(4/OOOOOOr )support) lock_testsF)patchcjtjtjo tjd|S)Nz#SKIPPED: Timeout on PyPy3 on Travis)unittestskipIf greentestPYPY3 RUNNING_ON_CI)clss r skipDueToHangr3s9 8?3I3-      r c&eZdZdZdZdZdZdS)Countercd|_dS)Nrvalueselfs r __init__zCounter.__init__;s  r c&|xjdz c_dSNrrs r incz Counter.inc> a r c&|xjdzc_dSr"rrs r decz Counter.decAr%r c|jSNrrs r getz Counter.getDs zr N)__name__ __module__ __qualname__r r$r'r*rr r rr9sPr rceZdZdZdZdS) TestThreadctj||||_||_||_||_dS)N)name) threadingThreadr testcasesemamutexnrunning)rr1r4r5r6r7s r r zTestThread.__init__Is>!!$T!222      r ctjdz }trtd|j|dzfz|j5|j5|jtr't|jd|j |jddddn #1swxYwYtj |trtd|jd|j5|j |j |jdtr0td |j|jfzdddn #1swxYwYddddS#1swxYwYdS) Ng@ztask %s will run for %.1f usecg.Aztasks are runningtaskdonerz$%s is finished. %d tasks are running)randomverboseprintr1r5r6r7r$r*r4assertLessEqualtimesleepr'assertGreaterEqual)rdelays r runzTestThread.runQs ')  ) 2 53;6(( ) ) )Y 9 9 F F !!###D$-++--/BCCC --dm.?.?.A.A1EEE  F F F F F F F F F F F F F F F Ju    1fdi000 9 9 !!### 001B1B1D1DaHHH9@ 4=#4#4#6#6D88999  9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s\F9 A:C F9C F9C ;F9BF! F9!F% %F9(F% )F99F=F=N)r+r,r-r rDrr r r/r/Hs2!!!99999r r/ceZdZdZdZdZdZdZdZdZ dZ d Z e j d d Ze jeed  d dZeedsdZdSdS) ThreadTestsNc ~d}tjd}tj}t}g}d}tjdddkrd}t |D]}td|z||||}||d |_ | |j | |j | tj|t!|t!||t$rt'd |D]}||| ||jt/|d rl||j d | |j du| tjd t!|t$rt'd||d dS)N r9rz)r9 zz Fz!waiting for all tasks to completeidentrzzall tasks done)r2BoundedSemaphoreRLockrsys version_inforanger/appenddaemon assertIsNonerK assertFalse assertTruerematchreprstartr=r>joinis_alive__dict__hasattrassertNotEqual assertEqualr*) rNUMTASKSr5r6 numrunningthreads initial_regexits r test_various_opszThreadTests.test_various_opsos)222!!YY 6  BQB ' ) );Mx  A=1,dD%LLA NN1   AH   ag & & &   QX & & & OOBH]DGG<< GG % % % GGIIII  7 5 6 6 6 U UA FF8      QZZ\\1: 6 6 6q'"" U##AGQ///  D111)I4PQ77 S STTT  $ " # # # ))1-----r ctj}||jdut |t |fd}tjgtj|d |ddutj d=dS)Nctj}|jt |t |dSr))r2current_threadrQrKstrrXset)rer;rKs r fz9ThreadTests.test_ident_of_no_threading_threads..fsL(**A LL ! ! ! FFF GGG HHJJJJJr rr) r2rirTrKrjrXEventthreadstart_new_threadwait_active)rrerlr;rKs @@r "test_ident_of_no_threading_threadsz.ThreadTests.test_ident_of_no_threading_threadss  $ & & D))) A Q        2&&&  qT)***  eAh ' ' 'r ctrtd tjdn,#tj$rtrtdYdSwxYw|tjddS)Nzwith 256kB thread stack size...i4platform does not support changing thread stack sizerr=r>r2 stack_sizernerrorrfrs r test_various_ops_small_stackz(ThreadTests.test_various_ops_small_stacks  5 3 4 4 4    ( ( ( (|    NLMMM FF  Q-%AActrtd tjdn,#tj$rtrtdYdSwxYw|tjddS)Nzwith 1MB thread stack size...irtrrurs r test_various_ops_large_stackz(ThreadTests.test_various_ops_large_stacks  3 1 2 2 2    * * * *|    NLMMM FF  Qrychd}tj}|tj||f}|||tj|tj|tjtj|=dS)NcTtj|dSr))r2rirelease)r6s r rlz*ThreadTests.test_foreign_thread..fs$  $ & & & MMOOOOOr ) r2LockacquirernroassertInrqassertIsInstance _DummyThread)rrlr6tids r test_foreign_threadzThreadTests.test_foreign_threads       %a%22  c9,--- i/4'4 6 6 6  c " " "r c ddl}n'#t$rtrtdYdSwxYw|jj}Gddt |}tj tj G fddtj }|}d|_ | trtdtrtd || d |}||dtrtd  trtd ||jtrtd || |j|}||dtrtdd||jtrtd|jr|dSdS)Nrz2test_PyThreadState_SetAsyncExc can't import ctypesceZdZdS)AThreadTests.SKIP_test_PyThreadState_SetAsyncExc..AsyncExcN)r+r,r-rr r AsyncExcrs Dr rc&eZdZdZdZfdZdS)?ThreadTests.SKIP_test_PyThreadState_SetAsyncExc..WorkerNFctj|_d|_ t jd)#$rd|_YdSwxYw)NFTg?)rn get_identidfinishedrkr@rA)rrworker_saw_exceptionworker_starteds r rDzCThreadTests.SKIP_test_PyThreadState_SetAsyncExc..Worker.runs *,, % /(&**,,, 3( ///$(DM(,,....../s*A A0/A0)r+r,r-rrrD)rrrsr WorkerrsBBH / / / / / / / / /r rTz started worker threadz trying nonsensical thread idz, waiting for worker thread to get startedz" verifying worker hasn't exitedz2 attempting to raise asynch exception in workerr#z5 waiting for worker to say it caught the exceptionrH)timeoutz all OK -- joining worker)ctypes ImportErrorr=r> pythonapiPyThreadState_SetAsyncExc Exception py_objectr2rmr3rRrYc_longr_rprTrrrUrZ) rr set_async_exc exceptionrreresultrrrs @@@r #SKIP_test_PyThreadState_SetAsyncExcz/ThreadTests.SKIP_test_PyThreadState_SetAsyncExcs  MMMM    LJKKK FF  (B      y   $$X.. #**(00 / / / / / / / / /Y% / / / FHH   / - . . .  6 4 5 5 5v}}R00)<< ###  B @ A A A  8 6 7 7 7 $$$  H F G G Gv}}QT22I>> ###  K I J J J!!"!---  ###  2 0 1 1 1 :  FFHHHHH  s  ..cd}ttdrtjtd|}ntjtd|}||j|tjd}|tj |j| |tj vddS)Nc(tjr))rnrw)_args_kws r fail_new_threadz7ThreadTests.test_limbo_cleanup..fail_new_thread*s,.. r _start_new_thread)new_start_joinable_threadcdSr)rrr r z0ThreadTests.test_limbo_cleanup..5sDr targetz:Failed to cleanup _limbo map on failure of Thread.start().) r]r2Patchobject addCleanupstoprYr3 assertRaisesrnrwrT_limbo)rrpatcherres r test_limbo_cleanupzThreadTests.test_limbo_cleanup(s ! ! ! 91 2 2 ]l9.AWWWGGl9.FO\\\G  %%%   LL 1 1 1 &,000  ! ! H J J J J Jr cF ddl}t|dt|jdn.#ttf$rt rt dYdSwxYw~ddl}|tj ddddtzg}| |d dS) NrrPyGILState_Ensurez6test_finalize_with_runnning_thread can't import ctypes-Wignore-caif 1: %s import ctypes, sys, time try: import thread except ImportError: import _thread as thread # Py3 # This lock is used as a simple event variable. ready = thread.allocate_lock() ready.acquire() # Module globals are cleared before __del__ is run # So we save the functions in class dict class C: ensure = ctypes.pythonapi.PyGILState_Ensure release = ctypes.pythonapi.PyGILState_Release def __del__(self): state = self.ensure() self.release(state) def waitingThread(): x = C() ready.release() time.sleep(100) thread.start_new_thread(waitingThread, ()) ready.acquire() # Be sure the other thread is waiting. sys.exit(42) *) rgetattrrrAttributeErrorr=r> subprocesscallrN executablesetup_3r_)rrrrcs r test_finalize_runnning_threadz)ThreadTests.test_finalize_runnning_thread;s  MMM FK ( ( ( F$&9 : : : :^,    PNOOO FF   __cndHdE:;E< R     s),'AAhangscZddl}dtz}|tjddd|g|j|j}|\}}|}|d}|d}| d|dS) Nraif 1: %s import threading from time import sleep def child(): sleep(0.3) # As a non-daemon thread we SHOULD wake up and nothing # should be torn down yet print("Woke up, sleep function is: %%s.%%s" %% (sleep.__module__, sleep.__name__)) threading.Thread(target=child).start() raise SystemExit rrr)stdoutstderrutf-8z,Woke up, sleep function is: gevent.hub.sleep) rsetup_4PopenrNrPIPE communicatestripdecoder_)rrscriptprrs r test_join_nondaemon_on_shutdownz+ThreadTests.test_join_nondaemon_on_shutdownks       cndHdFK$.O$.O  5 5w''w''  :      r getcheckintervalzNeeds sys.getcheckintervalc\tj}ddl}|5|dt t j} tddD]~}t j |dztj d}| | |}| ||vd||fz t j |n#t j |wxYw ddddS#1swxYwYdS) Nrrr#dcdSr)rrr r rz7ThreadTests.test_enumerate_after_join..sr rz� triggered after %d trials: %s)r2 enumeratewarningscatch_warnings simplefilterDeprecationWarningrNrrsetcheckintervalr3rYrZrT)renumr old_intervalrdrels r test_enumerate_after_joinz%ThreadTests.test_enumerate_after_joins"  $ $ & & 3 3  ! !(,> ? ? ?/11L 33 X XA(a000!( ===AGGIIIFFHHHA$$Q!V%MQRTUPV%VXXXX X$\2222$\22222# 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3 3s*/D!BC:%D!:DD!!D%(D%pypy_version_infocGddt}|d}tj|}|j~||dt j|z|d}tj|}|j~||dt j|zdS)NceZdZdZdZdS)DThreadTests.test_no_refcycle_through_target..RunSelfFunctionc||_tj|j|fd|i|_|jdS)N _yet_another)rargskwargs) should_raiser2r3_runrnrY)rrs r r zMThreadTests.test_no_refcycle_through_target..RunSelfFunction.__init__sU)5D%"+"2$)9=;I4:P#R#R#RDKK%%'''''r c"|jrtdSr))r SystemExit)r _other_refrs r rzIThreadTests.test_no_refcycle_through_target..RunSelfFunction._runs()(())r N)r+r,r-r rrr r RunSelfFunctionrs2((()))))r rF)rz%d references still around)msgT)rweakrefrefrnrZrSrN getrefcount)rr cyclic_objectweak_cyclic_objectraising_cyclic_objectweak_raising_cyclic_objects r test_no_refcycle_through_targetz+ThreadTests.test_no_refcycle_through_targetsJ ) ) ) ) )& ) ) ),O???M!(]!;!;   % % ' ' '   0022#?#&?3E3E3G3G#H#H$I  K K K%4O$F$F$F !)05J)K)K & ! ( - - / / /%   88::#?#&?3M3M3O3O#P#P$Q  S S S S Sr )r+r,r-maxDiffrfrrrxr{rrrrrskipOnLibuvOnPyPyOnWinrrr]rNrrrr r rFrFis?G(.(.(.T(((2        ###*EEEPJJJ&.!.!.!`&Y%g../.HY GC+ , ,-$33 30 73+ , ,S S S S S SSSr rFceZdZdZdZejddZejddZ dZ dS)ThreadJoinOnShutdownNcdtz|z}ddl}|tjddd|g|j}|d}|j dd }|j | d }| |d | |d d | |dddS)Nzif 1: %s import sys, os, time, threading # a thread, which waits for the main program to terminate def joiningfunc(mainthread): mainthread.join() print('end of thread') rrrr)r< r rzend of main end of thread rIzinterpreter was blockedzUnexpected error)rrrrNrrrprreadreplacecloserr_r^)rrrrrdatas r _run_and_joinz"ThreadJoinOnShutdown._run_and_joins!!    cndHdFKT^Tc  d d VVBZZx}}&&uc22 {{7## =>>> B#<=== Q 233333r rc4d}||dS)Nzif 1: import os t = threading.Thread(target=joiningfunc, args=(threading.current_thread(),)) t.start() time.sleep(0.2) print('end of main') )r)rrs r test_1_join_on_shutdownz,ThreadJoinOnShutdown.test_1_join_on_shutdowns% 6"""""r zSometimes randomly times outc`ddl}t|dsdSd}||dS)NrforkaGif 1: childpid = os.fork() if childpid != 0: os.waitpid(childpid, 0) sys.exit(0) t = threading.Thread(target=joiningfunc, args=(threading.current_thread(),)) t.start() print('end of main') osr]rrrrs r test_2_join_in_forked_processz2ThreadJoinOnShutdown.test_2_join_in_forked_processsG  r6""  F  6"""""r c`ddl}t|dsdSd}||dS)Nrraif 1: main_thread = threading.current_thread() def worker(): childpid = os.fork() if childpid != 0: os.waitpid(childpid, 0) sys.exit(0) t = threading.Thread(target=joiningfunc, args=(main_thread,)) print('end of main') t.start() # Should not block: main_thread object is already stopped # because we're running in a non-main thread of # the child process t.join(30) assert not t.is_alive() w = threading.Thread(target=worker) w.start() import sys if sys.version_info[:2] >= (3, 7) or (sys.version_info[:2] >= (3, 5) and hasattr(sys, 'pypy_version_info') and sys.platform != 'darwin'): w.join(45) if w.is_alive(): print('OH NO, failed to join worker') rrs r !test_3_join_in_forked_from_threadz6ThreadJoinOnShutdown.test_3_join_in_forked_from_threadsH  r6""  F*F 6"""""r ) r+r,r-rrrrrskipOnPyPy3OnCIr r rr r rrsG444(&Y%g.. # #/. #Y=>>##?>#$=#=#=#=#=#r rc&eZdZdZdZdZdZdS)ThreadingExceptionTestsctj}||t|jdSr))r2r3rYr RuntimeErrorrthread_s r test_start_thread_againz/ThreadingExceptionTests.test_start_thread_againJs;"$$  , 66666r cltj}|t|jdSr))r2rirrrZ)rris r test_joining_current_threadz3ThreadingExceptionTests.test_joining_current_threadOs/"133 ,(;<<<<r?sT&%%%%%"""""" &V ))KK T8J8JKKK K K ))OOFLL