ifKdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZd Ze ejd d ZGd d ejZiddddddddddddddddddddddddd dd!d"d#dd$dd%d&d'dd(dd)Zgd*Zddd+d,d-d.d.d/d0d1d2d2d2d3 Zd4Zd5ZdTd6Zd7Zd8Zd9Zd:Z d;Z!d<Z"hd=Z#hd>Z$dUd?Z%dVd@Z&GdAdBe'Z(dCZ)ej*Z+dDZ,dEZ-GdFdGe.Z/dHZ0dWdJZ1dKZ2dLZ3dMZ4GdNdOe'Z5GdPdQe5ej6Z7GdRdSej8Z9dS)Xz .. caution:: This module imports `subprocess` and `threading`; this can break monkey-patched unittests. Specifically, ``test_threading.ThreadTests.test_import_from_another_thread``. N)sleep)six) validate_bool) perf_counter) get_originalFGEVENTTEST_QUIET0c"eZdZdZdZdZdZdS)Popenz Depending on when we're imported and if the process has been monkey-patched, this could use cooperative or native Popen. Nc|SNselfs >/usr/local/lib/python3.11/dist-packages/gevent/testing/util.py __enter__zPopen.__enter__&s c$t|dSr)kill)rargss r__exit__zPopen.__exit__)s T r)__name__ __module__ __qualname____doc__timerrrrrrr r sC Err normaldefaultz actual-outputredzcharacter-diffsmagentadebugcyanz diff-chunkerror brightredz error-number exceptionzexpected-outputgreenzfailed-examplefilename lightblueinfolinenolightrednumberz ok-numberskipped brightyellow brightmagenta lightcyan) slow-testsuboptimal-behaviourtestnamewarning))darkz0;)light1;)brightr8)boldr8 !"#$%) rrblackr r'brownyellowbluer!r#greygraywhitecd}tD]5\}}||r|t|d}|}n6t|}d||dS)Nz[m) _prefixes startswithlen _colorcodes)color prefix_codeprefixcode color_codes r _color_coderV^srK!    F # # #f++,,'EK E U#JJ&; 33rc6tt|Sr)rV _colorscheme)whats r_colorrZhs |D) * **rcFt||zt|zSr)rZ)rYmessagers r _colorizer]ks $<<' !F6NN 22rc|dd}|r||z}n|}t||}t5tj|dzddddS#1swxYwYdS)zV Log a *message* :keyword str color: One of the values from _colorscheme rQr N)popr] output_locksysstderrwrite)r\rkwargsrQstrings rlogrgns JJw ) )E 4 uf % %F (( $'''((((((((((((((((((s#A((A,/A,cdts(|ddt|g|Ri|dSdS)z< Log the *message* only if we're not in quiet mode. rQr"N)QUIET setdefaultrg)r\rres rr"r"sP &'7+++ G%d%%%f%%%%%&&rcdttdsdS tj|dS#t$r@}|jdkr*t d|t |j|Yd}~dSYd}~dSd}~wt$r/}t d|t |j|Yd}~dSd}~wwxYw)Nkillpg zkillpg(%r, 9) failed: %s: %s) hasattrosrlOSErrorerrnorgtyper Exception)pidexs rrlrls 2x Hya   LLL 8q== .T"XX5F K K K K K K K K K ===== HHH *Cb1BBGGGGGGGGGHs . B//A33 B/$B**B/cd|z}td|ztjd}|r*||dfvrt d|dSdSdS)Nz"ERROR: The process "%s" not found.ztaskkill /F /PID %s /T)rcrrKz%r)r subprocessPIPE communicatestriprg)ru ignore_msgerrs rkill_processtreer~sx5;J (3.z G G G S S U UVW XC syy{{:r"222 D#22rct|drL |dS#t$r(}|jdkrYd}~dS|jdkrYd}~dSd}~wwxYw t j|jddS#t $rYdSwxYw)Nrrn rm)rorrqrrrpruEnvironmentError)popenrvs r_killrsuf   JJLLLLL   x1}}x2~~     GEIq ! ! ! ! !    DD s2( A A AAAA:: BBcd|j |jd|_|dSd|_ t |ddrt |jn3tj drt|jn##t$rtj YnwxYw t|n##t$rtj YnwxYw |dS#t$rtj YdSwxYw)NTsetpgrp_enabledwin)rcancelpoll was_killedgetattrrlrurbplatformrNr~rt traceback print_excrwait)rs rrrsQ {   zz||E 5+T 2 2 ( 59     \ $ $U + + ( UY ' ' '  e  s7AB C?CCC43C48DD/.D/> GEVENT_DEBUGr GEVENTSETUP_EMBEDGEVENTSETUP_EV_VERIFY>GEVENT_RESOLVERthreadGEVENTTEST_USE_RESOURCESallGEVENT_RESOLVER_NAMESERVERSz8.8.8.8cg}|p tj}||pit |D]K\}}|ds|tvr%||ftvr1| |d|Lt|tj r| |n| |d|S)NGEVENT= )rpenvironcopyupdatesorteditemsrNIGNORED_GEVENT_ENV_KEYSIGNORED_GEVENT_ENV_ITEMSappend isinstancer string_typesextendjoin)commandenvsetenvresultkeyvalues rgetnamers F  "* " " $ $CJJv|SYY[[)).. U~~h''   ) ) )  <3 3 3  ee,----'3+,, g g 88F  rc |dd}d}tjdst tdd}|dd}|ddpi}t |||}|d|d<|rI|r|ntj}|||std|t|f||d |}||_ |du|_ d |_ |Ktd d |t|f } | |_d| _| | |_|S)NtimeoutDO_NOT_SETPGRPsetpgrprr)rr1z+ %s) preexec_fnrF threadingTimer)rT)r`rprgetrrrrrgr namerrrrrdaemonstart) rquietrerrrrrrts rrrsmjjD))GJ :>>* + +2RD11 **UD ! !C ZZ$ ' ' -2F 7F 3 3 3D#&  6chhjjjRZ__%6%6 6  FD ' Djc D DV D DEEJ&d2EE .Lg . .wE9 M M M    LrcPeZdZdZ d dZedZdZeZdZ dZ dS) RunResultz The results of running an external command. If the command was successful, this has a boolean value of True; otherwise, a boolean value of false. The integer value of this object is the command's exit code. Nrc ||_||_||_||_||_||_||_||_| |_dSr) r run_kwargsrToutputr$r run_count skipped_count run_duration) rrrrTrr$rrrrs r__init__zRunResult.__init__sK $    "*(rc4|jSr)r splitlinesrs r output_lineszRunResult.output_lines$s{%%'''rc,t|j Sr)boolrTrs r__bool__zRunResult.__bool__(s ??""rc|jSr)rTrs r__int__zRunResult.__int__-s yrc Xd|jd|jd|jd|jd|jd S)NzRunResult of: z Code: z kwargs: z Output: ---- z---- Error: ---- z---- )rrTrrr$rs r__repr__zRunResult.__repr__0s= LLL III OOO KKK JJJ#  r)NNNrrr) rrrrrpropertyrr __nonzero__rrrrrrrs,-))))(((X(###K     rrcFd|vr|dd}|dd}|dd}|dd}|dd}|dd}|d d}d|vS) NWarningzWarning -- reap_childrenNADAzWarning -- threading_cleanupz/DeprecationWarning: Using or importing the ABCsz UserWarning: libuv only supportsz&ImportWarning: Not importing directoryz*DeprecationWarning: 'U' mode is deprecatedz#DeprecationWarning: dns.hash module)replace)outs r_should_show_warning_outputrFsCkk4f==kk8&AAkkKVTTkk>M JJw & &EiG ZZ% ( (F!::&>J.   "uzz||3FFZZ\\F U {"""" U {""""""" &\\ @f,@@F MiikkS))Lccszz'8/L/L ##'#%@%E%E#S[[v...jjll t  Jc"""'83'G'G$FI} % vv#?'')? A A A A A % ItV$$$  + + +# #   s /A(D00E ceZdZdZdS)NoSetupPyFoundzRaised by find_setup_py_aboveN)rrrrrrrrrs####rrctjtj|}tjtj|dsv|tj|}}||krt d|tjtj|dv|S)zAReturn the directory containing setup.py somewhere above *a_file*zsetup.pyz!could not find my setup.py above )rppathdirnameabspathexistsrr)a_filerootprevs rfind_setup_py_abovers 7??27??622 3 3DgnnRW\\$ ;;<<T27??400d 4<< .&&!RSS S gnnRW\\$ ;;<<T KrTc| t|S#t$rYnwxYw|: ttj|jjS#t$rYnwxYw|5 ttj|jS#t$rYnwxYw|rtdStdt )Nz./dnezAfter checking )rrrbmodulesr__file__locals)r a_module_namea_class climb_cwds rsearch_for_setup_pyrs  &v.. .    D  &s{73E'F'OPP P    D   &s{='A'JKK K    D ,"7+++ .: ; ;;s/  (A AA"#B BBcdtjddz}dtjddz}ttdr |dz }|dz }||fS)N%s.%s%s.%s.%srnpypy_version_infopypy)rb version_inforo) directoryfull_directorys r_version_dir_componentsrs`#*2A2..I#"22A2"66Ns'((!V &  n $$rctt}tj|dd}t \}}dt jddz}dt jddz}tt d r |d z }|d z }tj||}tj||}||fS) z Return a sequence of directories that could contain stdlib tests for the running version of Python. The most specific tests are at the end of the sequence. No checks are performed on existence of the directories. )rsrc greentestrNr r rnr r ) rrrprrrrbr ro)setup_pyrrrs rfind_stdlib_testsrs#(333H Xuk::I!8 9 9I~#*2A2..I#"22A2"66Ns'((!V &  Y 22IW\\)^<z'absolute_pythonpath..s& D D D1BGOOA   D D Dr)rprsplitrpathsepr)rs rabsolute_pythonpathrs] 2:%%t :l #D D D 27?(C(C D D DD 7?   % %%rcjeZdZdZgZdZdZdZdZdZ e dZ e dZ dZ d Zd ZdS) ExampleMixinz Something that uses the ``examples/`` directory from the root of the gevent distribution. The `cwd` property is set to the root of the gevent distribution. rng?NcHttt|S)z(Return the directory containing setup.py)rr)rrrsrs r find_setup_pyzExampleMixin.find_setup_pys%"JJ    rc |}n)#t$r}tjd|d}~wwxYwtj|dS)N"Unable to locate file/dir to run: examples)r"runittestSkipTestrprr)rres rcwdzExampleMixin.cwdsm S%%''DD S S S##aa$QRR R Sw||D*---s =8=c.t}|rd|indS)a" Returns a dictionary of environment variables to set for the child in addition to (or replacing) the ones already in the environment. Since the child is run in `cwd`, relative paths in ``PYTHONPATH`` need to be converted to absolute paths. rN)r)rabs_pythonpaths rrzExampleMixin.setenv&s$-..1?I n--TIrct|ddrtd||jt|ddrtd||j |j}|j}n)#t $r}tjd|d}~wwxYw|j pi}|j }|r%d|vr|d |n||d<|tj ddd|g|jzfd |i|S) Nrz Invalid testserverr$rz-Wrz-ur))rAssertionErrorrr-exampler)rr&r' start_kwargsrrrb executable example_args)rmethr- server_dirr(rers r_startzExampleMixin._start3s8 4 & & B tyAA A 44 ( ( D t{CC C S\FJJ S S S##aa$QRR R S"(b  *6!!x ''////#)x t ^T8T6 :T=N N      sA B)BBc8|tSN)r3)r5rrs r start_examplezExampleMixin.start_exampleMs{{{&&&rc8|tSr7)r5rrs r run_examplezExampleMixin.run_examplePs{{{$$$r)rrrrr2 before_delay after_delayr/r0r"rr)rr5r8r:rrrr r sLLKGL   ..X. J JX J   4'''%%%%%rr c0eZdZdZdZdZdZdZdZdS) TestServerNc>ddlm}|fd}|S)Nr)contextmanagerc3K5}|_dVddddS#1swxYwYdSr)r8rbeforeafter)rrs rrunning_serverz1TestServer.running_server..running_server[s##%% "                     s4AA A) contextlibr@)rr@rDs` rrDzTestServer.running_serverXsH------        ~rc|5|ddddS#1swxYwYdSr)rD_run_all_testsrs rtestzTestServer.testds  " " " "    ! ! ! " " " " " " " " " " " " " " " " " "s 7;;c|jt|j||j|jd|jdSNz died with code )r;r assertIsNonerrr/rs rrBzTestServer.beforehsk   ( $# $ $ $ $*//++"lllDJOO,=,=,=     rc|jt|j||j|jd|jdSrJ)r<rrKrrr/rs rrCzTestServer.afterpsk   ' $" # # # $*//++"lllDJOO,=,=,=     rcd}tt|D]B}|dr+t||}t |r |d}C|sJdS)NF_testT)rdirrNrcallable)rranmethodfunctions rrGzTestServer._run_all_testsxswSYY''  F  )) "400H%%HJJJC r) rrrrrDrHrBrCrGrrrr>r>Tsd E    """rr>ceZdZdZdZdS)alarmctj|d|_||_|dS)NT)rThreadrrrr)rrs rrzalarm.__init__s8!!$'''   rct|jtjdt jddS)Nz Timeout. )rrrbrcrdrp_exitrs rrz alarm.runs9 dl &&&  rN)rrrrrrrrrUrUs2 rrU)r)NN)F)NNNT):rrrbrprr&rrxtimerrKrgevent._configrgevent._compatr gevent.monkeyrrrrrir rXrMrPrVrZr]rgr"rlr~rrrrrrobjectrrLockrarrrtrrrrrrr TestCaser>rWrUrrrrbs ((((((''''''&&&&&&   bjnn%7==>>     J    $ h yU  y   V ) [Kwf  H j  g!"#$~%&!%- 2       2   R   444+++3333((("&&& H H H"0 .:= = = = = = = = @.in ,,,,;;;|$$$$$Y$$$<<<<0%%%%%%4 & & &J%J%J%J%J%6J%J%J%Z,,,,,",,,^     I      r