Yf&kdZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z ddl mZddlmZmZdd lmZmZmZmZdd lmZmZmZmZmZmZmZdd l m!Z!dd l"m#Z#dd l$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?ddl@mAZAddlBmCZCmDZDmEZEddlFmGZGmHZHmIZImJZJmKZKddlLmMZMmNZNmOZOmPZPddlQmRZRmSZSmTZTmUZUddlVmWZWmXZXddlYmZZZmZd d!l[m\Z\m]Z]d"d#l^m_Z_eGd$d%Z`ee(jaGd&d'e7ZbGd(d)eZZcGd*d+eZdGd,d-eZeGd.d/eZfGd0d1egZhGd2d3ZieGd4d5ZjGd6d7ZkGd8d9egZlGd:d;ZmeGd<d=ZneGd>d?ZodS)@z- Tests for L{twisted.trial._dist.disttrial}. N)partial)StringIO)sep)CallableListSetTestCase) implementerverify)Factoryassocdefinefield) assert_thatcontains ends_withequal_to has_lengthnone starts_with)AllOf)given)booleans sampled_from) interfaces) ReactorBase)CancelledErrorDeferredsucceed) ProcessDone)ProcessProtocolProtocol)AlternateReactor)MemoryReactorClock)Failure)FilePath)FilesystemLock)_WORKER_AMP_STDIN) DistReporter)DistTrialRunner WorkerPoolWorkerPoolConfig) countingCalls discardResult fromOptional iterateWhilesequence) LocalWorker RunResultWorker WorkerAction)Reporter TestResult TreeReporterUncleanWarningsReporterWrapper) ErrorHolder TrialSuite)SynchronousTestCaser ) erroneoussample)matches_resultc`eZdZUdZeeeZee e d<dZ dZ dS) FakeTransportz* A simple fake process transport. default_closedcdS)z% Ignore write calls. N)selffddatas Y/var/www/html/env/lib/python3.11/site-packages/twisted/trial/_dist/test/test_disttrial.py writeToChildzFakeTransport.writeToChildIsc:|j|dS)z> Mark one of the child descriptors as closed. N)rGadd)rJrKs rM closeChildFDzFakeTransport.closeChildFDNs  rON) __name__ __module__ __qualname____doc__rr setrGrint__annotations__rNrRrIrOrMrDrDAsiggcll333GSX333   rOrDcFeZdZdZdZdZdZdZdidddddfdZdZ dZ dS) CountingReactorz A fake reactor that counts the calls to L{IReactorCore.run}, L{IReactorCore.stop}, and L{IReactorProcess.spawnProcess}. rc<tj|||_dSN)r%__init___workers)rJworkerss rMr^zCountingReactor.__init__`s#D))) rOrINc |j||t|xjdz c_dS)z See L{IReactorProcess.spawnProcess}. @param workerProto: See L{IReactorProcess.spawnProcess}. @param args: See L{IReactorProcess.spawnProcess}. @param kwargs: See L{IReactorProcess.spawnProcess}. rAN)r_appendmakeConnectionrD spawnCount) rJ workerProto executableargsenvpathuidgidusePTYchildFDss rM spawnProcesszCountingReactor.spawnProcessdsF& [)))""=??333 1rOctj|d|jvr'|jdddd|xjdz c_dS)z+ See L{IReactorCore.stop}. beforeshutdownrrAN)r%stoptriggers stopCountrJs rMrrzCountingReactor.stop{s\ %%% t} $ $ 5DM( #J / 21 5 7 7 7 !rOc|xjdz c_d|_d|_|jD]\}}}||i|||xjdzc_dS)z* See L{IReactorCore.run}. rATN)runCountrunninghasRunwhenRunningHooksrrrt)rJfrgkwargss rMrunzCountingReactor.runsu    #4  OAtV At v      !rO) rSrTrUrVrdrtrwr^rnrrr}rIrOrMr[r[Us JIH       .   rOr[c0eZdZdZdZdZdZdZdZdS)CountingReactorTestsz' Tests for L{CountingReactor}. cFg|_t|j|_dSr])r`r[reactorrus rMsetUpzCountingReactorTests.setUps &t|44 rOcNtjtj|jdS)zJ L{CountingReactor} instances provide L{IReactorProcess}. N)r verifyObjectrIReactorProcessrrus rMtest_providesIReactorProcessz1CountingReactorTests.test_providesIReactorProcesss" J6 EEEEErOc||jjt}dD]}|j|t jt jg||j| |j |g|z| |jj|dS)z The process protocol for a spawned process is connected to a transport and appended onto the provided C{workers} list, and the reactor's C{spawnCount} increased. rA)rgN) assertFalserrdr#rnsysrf assertTrue transport assertEqualr`)rJprotocounts rMtest_spawnProcessz&CountingReactorTests.test_spawnProcesss 0111  = =E L % %eS^3>BR % S S S OOEO , , ,   T\E7U? ; ; ;   T\4e < < < <  = =rOc||jjdD];}|j||jj|z;WorkerPoolTests.test_createLocalWorkers..s000!VXX000rOrriN) ranger_createLocalWorkersr'rassertIsInstancer3rr)rJ protocolsr`ss rMtest_createLocalWorkersz'WorkerPoolTests.test_createLocalWorkerss 10uQxx000 )// 8F;K;KXZZXX 2 2A  ! !![ 1 1 1 1 CLL)))))rOc dtdD}gididddddffd }|j||dg|dd|t jd |dd |t j tjd dS) z Given a C{spawnProcess} function, C{_launchWorkerProcess} launches a python process with an existing path as its argument. c*g|]}tSrI)r")ris rMrz>WorkerPoolTests.test_launchWorkerProcesses..s9991_&&999rOrrINrc  | | |dSr])rbextendupdate) processProtocolrfrgrhrirjrkrlrm arguments environments rMfakeSpawnProcesszDWorkerPoolTests.test_launchWorkerProcesses..fakeSpawnProcesssG   Z ( ( (   T " " "   s # # # # #rOfoorArr> PYTHONPATH) rr_launchWorkerProcessesrrosriexistspathsepjoinr)rJrrrrs @@rMtest_launchWorkerProcessesz*WorkerPoolTests.test_launchWorkerProcessess :9a999    $ $ $ $ $ $ $ (()99ugNNN 1y|444 y|44555  ! --- 22K 4MNNNNNrOc^|jg|jt g}||}|fd}||tt|j j dS)zY C{run} dispatches the given action to each of its workers exactly once. cHt|Sr])r rb)wr`s rMz*WorkerPoolTests.test_run..sq0A0A(B(BrON) rmakedirsrstartr[successResultOfr}rrrr)rJstartingstartedrxr`s @rMrzWorkerPoolTests.test_run s 9???2#6#677&&x00++BBBBCC W%%%GZ (>??@@@@@rOc|jt|jjdz}||||jtg}| |j |}| |j|jddS)z L{WorkerPool.start} checks if the test directory is already locked, and if it is generates a name based on it. z.lockz _trial_temp-1N)rrr(rrirlock addCleanupunlockr[rrrrsibling)rJr fakeReactorrs rMtest_runUsedDirectoryz%WorkerPoolTests.test_runUsedDirectorys d387BCC  $$$  $$$&b)) &&ty{'C'CDD   $  ! ) )/ : :     rOc @|jtg}||j|}t j|}| ||j D]}t|j j tt|j j D]}|||j|jfD]'}|t't)d(t||t+t|jjt1dt|jjt1ddS)z L{StartedWorkerPool.join} causes all of the workers to exit, closes the log file, and unlocks the test directory. rTFN)rrr[rrrr fromCoroutinerassertNoResultr_rrrGrr)childConnectionLost processExited processEndedr&r!rtestLogclosedr testDirLocklocked)rJrrjoiningrrKr{s rM test_joinzWorkerPoolTests.test_join3sj !"%%&&tyw'?'?@@(88 G$$$! + +A  +X6G-H-H I I Ik) * *%%b))))oq~6 + +'+a..))**** +D((11466:::GO*HTNN;;;G'.@@@@@rOrsubdirabsoluterreturnNc|r|jjtz|z}t|j|}|rt |}n.recordingFactorys"A;;;;DKrOr>)rrc*g|]}tSrIr )rns rMrz;DistTrialRunnerTests.test_minimalWorker..s@@@1HJJ@@@rOr) rr<rrrunAsyncr_startedr`r)rJrrnumTestsrsuiters @rMtest_minimalWorkerz'DistTrialRunnerTests.test_minimalWorkers        !5E   @@h@@@AA V__U33444DM!$,j.B.BCCCCCrOrNc|d}||j}||}||t ||jtdS)z Running with the C{unclean-warnings} option makes L{DistTrialRunner} uses the L{UncleanWarningsReporterWrapper}. T)uncleanWarningsN)rr r rrr*originalr:)rJrdrs rMtest_runUncleanWarningsz,DistTrialRunnerTests.test_runUncleanWarningssr 55 OODJ ' '%%a(( fl333 fo/MNNNNNrOclt}||}||t }||t |}|d||d|dS)zA L{DistTrialRunner} can run an empty test suite. rRunning 0 testPASSEDN) rrrr r<rr*getvalueassertIn)rJrrroutputs rMtest_runWithoutTestz(DistTrialRunnerTests.test_runWithoutTestsv..%%foojll&C&CDD fl333"" &/// h'''''rOc,tdttd}t}||}|||}||t| }| d|| d|| d|| d|| d|dS)z Even if there is no test, the suite can contain an error (most likely, an import error): this should make the run fail, and the error should be printed. zan errorzfoo barrrzerrors=1FAILEDN) r;r& RuntimeErrorrrrr rr*rr)rJerrrrrrs rM!test_runWithoutTestButWithAnErrorz6DistTrialRunnerTests.test_runWithoutTestButWithAnErrors *gl9.E.E&F&FGGv..%%fooc&:&:;; fl333"" &/// i((( j&))) j&))) h'''''rOcN|t}|||j}|jj}t|tdt|ddj ttdS)z If for some reasons we can't connect to the worker process, the error is recorded in the result object. rrArN) rBrokenWorkerPoolrr r rerrorsrrtyperWorkerPoolBrokenrJrrr#s rMtest_runUnexpectedErrorz,DistTrialRunnerTests.test_runUnexpectedErrors 2BCC%%foodj&A&ABB'FJqMM***F1IaL%x0@'A'ABBBBBrOc|t}|t5||jddddS#1swxYwYdS)z If the reactor is stopped by C-c (i.e. `run` returns before the test case's Deferred has been fired) we should cancel the pending test run. r!N)rr assertRaisesrr}r )rJrs rMtest_runUnexpectedErrorCtrlCz1DistTrialRunnerTests.test_runUnexpectedErrorCtrlCs /BB   ~ . . # # JJtz " " " # # # # # # # # # # # # # # # # # #sAA"%A"cx|tttd}|||j}|jj}t|tdt|ddj ttdS)z~ If for some reason the worker process cannot run a test, the error is recorded in the result object. T) workerFactoryrr!rArN)rrr_BrokenLocalWorkerrr r rr#rrr$r WorkerBrokenr&s rMtest_runUnexpectedWorkerErrorz2DistTrialRunnerTests.test_runUnexpectedWorkerErrors %/AD    %%foodj&A&ABB'FJqMM***F1IaL%x '='=>>>>>rOcdfd}||}tj||j}|dJjdj}||| d| |}| |tdS)zG L{DistTrialRunner} waits for the worker pool to stop. Nc$t|ddi|S)NrFrrs rMrzPDistTrialRunnerTests.test_runWaitForProcessesDeferreds..recordingFactorys"A<<<#> ? ? < II5 6 6 6-"+ A%%a(( fl33333rOc t}ttjdt jdtjdg}||dd}||}||}t|j tdtdd S) z L{DistTrialRunner} can run in C{exitFirst} mode where it will run until a test fails and then abandon the rest of the suite. r test_failtest_barTr)r exitFirstrrA) successesfailuresN) rr<r@FooTestr?TestRegularFailrr rrrrBr)rJrr rrrs rMtest_exitFirstz#DistTrialRunnerTests.test_exitFirst s z**)+66z**   v!LL OOE " "%%a(( O #A        rOct}tjd}||}||d}||}|d|j|| | }|| d|jdz d| d |tdd D]}| d |||| d |jd dS)z L{DistTrialRunner} can run in C{untilFailure} mode where it will run the given tests until they fail. test_itrT) untilFailurerrAz expected to see PASSED in outputFAILz Test Pass zRan 1 tests inz2expected to see per-iteration test count in outputN)rr?EventuallyFailingTestCaserr rrrr wasSuccessfulrrrr)rJrcaserrrrrs rMtest_runUntilFailurez)DistTrialRunnerTests.test_runUntilFailure"s_ 29==v.. OODtO 4 4%%a(( DF### --//000""  LL " " FQJ .   ff%%%q! 4 4A MM*q**F 3 3 3 3  LL) * * F @     rOc|}||j}t|t dt|jt ddS)zw L{DistTrialRunner.run} returns a L{DistReporter} containing the result of the test suite run. TrAN)rr}r rrGrr;)rJrrs rMrzDistTrialRunnerTests.test_runEsg !!DJ''F((**HTNN;;;F$hqkk22222rOctg}t|5|d}dddn #1swxYwY||j}t |jtgt |jtgt | tdt |j tdt |j tdt |j tddS)z L{DistTrialRunner.run} uses the installed reactor L{DistTrialRunner} was constructed without a reactor. NrTrA) r[r$rr}r rr#rr<rGr;rwrt)rJrrrs rMtest_installedReactorz*DistTrialRunnerTests.test_installedReactorOs= ""%% g & & 2 2^^D^11F 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2DJ''FM8B<<000FOXb\\222F((**HTNN;;;F$hqkk222G$hqkk222G%x{{33333sAA AcjGddt}ttjGdd}Gdd}t |5|t 5|ddddn #1swxYwYdddn #1swxYwYt |5|t 5|ddddn #1swxYwYdddn #1swxYwYt |5|t 5|ddddn #1swxYwYddddS#1swxYwYdS) z L{DistTrialRunner} raises L{TypeError} if the installed reactor provides neither L{IReactorCore} nor L{IReactorProcess} and no other reactor is given. ceZdZdZdS)=DistTrialRunnerTests.test_wrongInstalledReactor..CorecdSr]rIrus rM installWakerzJDistTrialRunnerTests.test_wrongInstalledReactor..Core.installWakergsrON)rSrTrUrRrIrOrMCorerPfs#     rOrSc"eZdZ ddZdS)@DistTrialRunnerTests.test_wrongInstalledReactor..ProcessNFc dSr]rI) rJrrfrgrhrirjrkrlrms rMrnzMDistTrialRunnerTests.test_wrongInstalledReactor..Process.spawnProcessls rO)NNNNFN)rSrTrUrnrIrOrMProcessrUjs<      rOrWceZdZdS)@DistTrialRunnerTests.test_wrongInstalledReactor..NeitherNrSrTrUrIrOrMNeitherrYz DrOr[NrL)rr rrr$r) TypeErrorr)rJrSrWr[s rMtest_wrongInstalledReactorz/DistTrialRunnerTests.test_wrongInstalledReactor_s[     ;    Z/ 0 0        1 0         ggii ( ( - -""9-- - -t,,, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - ddff % % - -""9-- - -t,,, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - ggii ( ( - -""9-- - -t,,, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -s$B9?B" B9"B& &B9)B& *B99B=B=D06D D0D D0 D !D00D47D4F(-F F(F F(F F((F,/F,cGddtfd}||}|5||jddddS#1swxYwYdS)z If there is an unexpected exception running the test suite then it is re-raised by L{DistTrialRunner.run}. ceZdZdS);DistTrialRunnerTests.test_runFailure..BrokenFactoryNrZrIrOrM BrokenFactoryrar\rOrbcr]rI)rgr|rbs rM brokenFactoryz;DistTrialRunnerTests.test_runFailure..brokenFactorys-// !rOr!N) Exceptionrr)r}r )rJrdrrbs @rMtest_runFailurez$DistTrialRunnerTests.test_runFailures     I    " " " " "-@@   } - - # # JJtz " " " # # # # # # # # # # # # # # # # # #sA--A14A1rN)rSrTrUrVr<r@r=r rrr rrrr'r*r/r6r?rIrrMr^rfrIrOrMrrksc Jz223 4 4E''' - - -DDD, O O O O ( ( ((((& C C C C#### ? ? ? ?44442   2! ! ! F33334444 +-+-+-+-Z#####rOrc:eZdZdZd dZd dZd dZd dZd dZdS) FunctionalTestsz8 Tests for the functional helpers that need it. rNcttddtdttddtddS)z ``fromOptional`` accepts a default value and an ``Optional`` value of the same type and returns the default value if the optional value is ``None`` or the optional value otherwise. rANr)rr0rrus rMtest_fromOptionalz!FunctionalTests.test_fromOptionalsJ LD))8A;;777LA&& 44444rOct}t|}|||dt ||t dS)z ``discardResult`` accepts an awaitable and returns a ``Deferred`` that fires with ``None`` after the awaitable completes. rN)rr/rr4rrr)rJrrs rMtest_discardResultz"FunctionalTests.test_discardResultsh $:: !   A 8D((++TVV44444rOc`t}t}tjt||}|d|||dt ||tddS)z ``sequence`` accepts two awaitables and returns an awaitable that waits for the first one to complete and then completes with the result of the second one. *helloN)rrr2r4rrrr)rJrbcs rM test_sequencezFunctionalTests.test_sequences $::#::  "8Aq>> 2 2 2 A 7D((++Xb\\:::::rOctttg}d}tjt|t|j}|d|||dt||tddS)z ``iterateWhile`` executes the actions from its factory until the predicate does not match an action result. c|dkS)NrorI)values rM predicatez4FunctionalTests.test_iterateWhile..predicates B; rOrroN) rrr1listpopr4rrrr)rJactionsrwrs rMtest_iterateWhilez!FunctionalTests.test_iterateWhiles )1 HJJ 'K   $1 DMM$5 6 6    q!!! A  r"""D((++Xb\\:::::rOctdtdtfd}tddD]'}t|t |(dS)z ``countingCalls`` decorates a function so that it is called with an increasing counter and passes the return value through. rrc |dzS)NrArI)rs rMtargetz2FunctionalTests.test_countingCalls..targets q5LrOrA N)r.rXrrr)rJr~expecteds rMtest_countingCallsz"FunctionalTests.test_countingCallsst  c c     a  6 6H (8"4"4 5 5 5 5 6 6rOrg) rSrTrUrVrkrmrsr{rrIrOrMriris5555 5 5 5 5 ; ; ; ;;;;;2 6 6 6 6 6 6rOriceZdZdZdS)r%zx An exception for ``StartedWorkerPoolBroken`` to fail with to allow tests to exercise exception code paths. NrSrTrUrVrIrOrMr%r%srOr%c6eZdZdZdedddfdZddZdS)StartedWorkerPoolBrokenzn A broken, started worker pool. Its workers cannot run actions. They always raise an exception. workerActionNrc"Ktr])r%)rJrs rMr}zStartedWorkerPoolBroken.runs   rOc KdSr]rIrus rMrzStartedWorkerPoolBroken.joins trOrg)rSrTrUrVr6r}rrIrOrMrrsY !l4&8!T!!!!rOrc8eZdZUdZeed<dejdefdZ dS)r"zF A worker pool that has workers with a broken ``run`` method. _configrrc"KtSr])r)rJrs rMrzBrokenWorkerPool.start s'(((rON) rSrTrUrVr-rYrrrrrIrOrMr"r"sU)!1) ))))))rOr"c&eZdZdZdededefdZdS) _LocalWorkerz A L{Worker} that runs tests in this process in the usual way. This is a test double for L{LocalWorkerAMP} which allows testing worker pool logic without sending tests over an AMP connection to be run somewhere else.. rHrrcTKt|g|ddiS)z8 Directly run C{case} in the usual way. successT)r<r}rJrHrs rMr}z_LocalWorker.runs1 D6v&&&4  rON)rSrTrUrVPyUnitTestCaser8r4r}rIrOrMrrsG!n!j!Y!!!!!!rOrceZdZdZdS)r.zE A worker tried to run a test case but the worker is broken. NrrIrOrMr.r. srOr.c&eZdZdZdededdfdZdS)r-z: A L{Worker} that always fails to run test cases. rHrrNc"Kt)z@ Raise an exception instead of running C{case}. )r.rs rMr}z_BrokenLocalWorker.run+snnrO)rSrTrUrVrr8r}rIrOrMr-r-&sGnjTrOr-cxeZdZUdZeeed<eeed<e ded<de dddfdZ d Z dS) StartedLocalWorkerPoolz' A started L{LocalWorkerPool}. rr`Nr3rrc@K|jD]}||d{VdS)z8 Run the action with each local worker. N)r`)rJrworkers rMr}zStartedLocalWorkerPool.run<sHl ' 'F,v&& & & & & & & & & ' 'rOc$K|jd{VdSr])r3rus rMrzStartedLocalWorkerPool.joinCs&mrO) rSrTrUrVr'rrYrr5rr6r}rrIrOrMrr2ssm### &\tn'l4&8'T''''rOrceZdZUdZeed<eeeZ e e ed<dZ e ed<eZegefed<dejd e fd Zd S) rz_ Implement a worker pool that runs tests in-process instead of in child processes. rrEr F _autostop_workerFactoryrrcKfdtjjD}tjj|jrt dn t}j ||S)Nc8g|]}SrI)r)rrrJs rMrz)LocalWorkerPool.start..Vs%QQQQ4&&((QQQrO) rrrrrrr rr rb)rJrr`rs` rMrzLocalWorkerPool.startSsRQQQ% 8O2P2PQQQ( L ) "n rs &&&&&&&&&&//////........////////////+*****88888888''''''------DDDDDDDDDD......????????@@@@@@777777******,,,,,,222222111111999999WWWWWWWWWWTSSSSSSSSSSS 98888888@@@@@@@@%%%%%%%%$$$$$$& Z '((?????(??)(?D5;5;5;5;5;.5;5;5;pX3X3X3X3X3hX3X3X3vq#q#q#q#q#8q#q#q#h K6K6K6K6K6hK6K6K6\y         ) ) ) ) ) ) ) )!!!!!!!!"9        (rO