o >h&k@sdZddlZddlZddlmZddlmZddlmZddl 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%d%Z`ee(jaGd&d'd'e7ZbGd(d)d)eZZcGd*d+d+eZdGd,d-d-eZeGd.d/d/eZfGd0d1d1egZhGd2d3d3ZieGd4d5d5ZjGd6d7d7ZkGd8d9d9egZlGd:d;d;ZmeGdd?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@s<eZdZUdZeeedZee e d<ddZ ddZ dS) FakeTransportz* A simple fake process transport. default_closedcCsdS)z% Ignore write calls. N)selffddatarFrF/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_dist/test/test_disttrial.py writeToChildIszFakeTransport.writeToChildcCs|j|dS)z> Mark one of the child descriptors as closed. N)rEadd)rGrHrFrFrJ closeChildFDNszFakeTransport.closeChildFDN) __name__ __module__ __qualname____doc__rr setrErint__annotations__rKrMrFrFrFrJrBAs  rBc@sLeZdZdZdZdZdZddZdidddddfddZd d Z d d Z dS) CountingReactorz A fake reactor that counts the calls to L{IReactorCore.run}, L{IReactorCore.stop}, and L{IReactorProcess.spawnProcess}. rcCst|||_dSN)r$__init___workers)rGworkersrFrFrJrW`s  zCountingReactor.__init__rFNc Cs*|j||t|jd7_dS)z See L{IReactorProcess.spawnProcess}. @param workerProto: See L{IReactorProcess.spawnProcess}. @param args: See L{IReactorProcess.spawnProcess}. @param kwargs: See L{IReactorProcess.spawnProcess}. r@N)rXappendmakeConnectionrB spawnCount) rG workerProto executableargsenvpathuidgidusePTYchildFDsrFrFrJ spawnProcessds  zCountingReactor.spawnProcesscCs>t|d|jvr|jdddd|jd7_dS)z+ See L{IReactorCore.stop}. beforeshutdownrr@N)r$stoptriggers stopCountrGrFrFrJri{s  zCountingReactor.stopcCsT|jd7_d|_d|_|jD] \}}}||i|q||jd8_dS)z* See L{IReactorCore.run}. r@TN)runCountrunninghasRunwhenRunningHooksrirk)rGfr_kwargsrFrFrJrunszCountingReactor.run) rNrOrPrQr\rkrmrWrfrirsrFrFrFrJrUUs   rUc@s8eZdZdZddZddZddZdd Zd d Zd S) CountingReactorTestsz' Tests for L{CountingReactor}. cCsg|_t|j|_dSrV)rYrUreactorrlrFrFrJsetUpszCountingReactorTests.setUpcCsttj|jdS)zJ L{CountingReactor} instances provide L{IReactorProcess}. N)r verifyObjectrIReactorProcessrurlrFrFrJtest_providesIReactorProcesssz1CountingReactorTests.test_providesIReactorProcesscCsj||jjt}dD]&}|jj|tjtjgd||j| |j |g|| |jj|q 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. r@)r_N) assertFalserur\r"rfsysr^ assertTrue transport assertEqualrY)rGprotocountrFrFrJtest_spawnProcesss z&CountingReactorTests.test_spawnProcesscCs6||jjdD]}|j||jj|q dS)zB Stopping the reactor increments its C{stopCount} rzN)r|rurkrir)rGrrFrFrJ test_stops  zCountingReactorTests.test_stopcCsj||jjg}|j|jddD]}|j||jj|||jjd|t||qdS)z Running the reactor increments its C{runCount}, does not imply C{stop}, and calls L{IReactorCore.callWhenRunning} hooks. Nrzr) r|rurmcallWhenRunningrZrsrrklen)rGwhenRunningCallsrrFrFrJtest_runs zCountingReactorTests.test_runN) rNrOrPrQrvryrrrrFrFrFrJrts rtc@sreZdZdZddZddZddZdd Zd d Zd d Z e e e dde dgdededdfddZdS)WorkerPoolTestsz" Tests for L{WorkerPool}. cCs@t||_|jd|_td|jgdd|_t|j|_dS)N _trial_tempout.log) numWorkersworkingDirectoryworkerArgumentslogFile) r&mktempparentchildrr,configr+poolrlrFrFrJrvszWorkerPoolTests.setUpcCsRddtdD}|j|tdt}|D]}||tq|dt|dS)zw C{_createLocalWorkers} iterates the list of protocols and create one L{LocalWorker} for each. cSg|]}tqSrF)object).0xrFrFrJ z;WorkerPoolTests.test_createLocalWorkers..rraN) ranger_createLocalWorkersr&rassertIsInstancer2rr)rG protocolsrYsrFrFrJtest_createLocalWorkerss z'WorkerPoolTests.test_createLocalWorkerscsddtdD}gididddddffdd }|j||d g|dd |tjd |d d |tj t jd dS)z Given a C{spawnProcess} function, C{_launchWorkerProcess} launches a python process with an existing path as its argument. cSrrF)r!rirFrFrJrrz>WorkerPoolTests.test_launchWorkerProcesses..rrFNrc s"|||dSrV)rZextendupdate) processProtocolr^r_r`rarbrcrdre arguments environmentrFrJfakeSpawnProcesss zDWorkerPoolTests.test_launchWorkerProcesses..fakeSpawnProcessfoor@r{r= PYTHONPATH) rr_launchWorkerProcessesrr~osraexistspathsepjoinr})rGrrrFrrJtest_launchWorkerProcessess z*WorkerPoolTests.test_launchWorkerProcessescsZ|jg|jtg}||}|fdd}||tt|j j dS)zY C{run} dispatches the given action to each of its workers exactly once. cst|SrV)rrZ)wrYrFrJsz*WorkerPoolTests.test_run..N) rmakedirsrstartrUsuccessResultOfrsrrrr)rGstartingstartedrnrFrrJr s   zWorkerPoolTests.test_runcCsh|jt|jjd}||||jt g}| |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'rrar~lock addCleanupunlockrUrrrrsibling)rGr fakeReactorrrFrFrJtest_runUsedDirectorys   z%WorkerPoolTests.test_runUsedDirectorycCs|jtg}||j|}t|}| ||j D](}t |j j tt|j j D]}||q0|j|jfD] }|ttdq>q!t ||tt |jjtdt |jjtddS)z L{StartedWorkerPool.join} causes all of the workers to exit, closes the log file, and unlocks the test directory. rTFN)rrrUrrrr fromCoroutinerassertNoResultrXrrrErr(childConnectionLost processExited processEndedr%r rtestLogclosedr testDirLocklocked)rGrurjoiningrrHrqrFrFrJ test_join3s     zWorkerPoolTests.test_joinrsubdirabsoluterreturnNcCsx|r |jjt|}t|j|d}|rt|}n tt|jjt t|}t |}| | t g}t|jj|dS)z L{WorkerPool.start} creates a L{StartedWorkerPool} configured with a log file based on the L{WorkerPoolConfig.logFile}. )rN)rrarr rrrrrrr+rrrUrrname)rGrrrmatchesrrrFrFrJ test_logFileHs   zWorkerPoolTests.test_logFile)rNrOrPrQrvrrrrrrrrrboolstrrrFrFrFrJrs"  !  rc@seZdZdZeedgZddZddZ ddZ d%d d Z d dZ ddZ d%ddZd%ddZd%ddZd%ddZddZddZd%ddZd%dd Zd%d!d"Zd#d$Zd S)&DistTrialRunnerTestsz' Tests for L{DistTrialRunner}. test_fooc Ks@tt|tdgttddtgd}||tdi|S)z. Create a runner for testing. rT)autostop)reporterFactoryrstream maxWorkersrworkerPoolFactoryruNrF) dictr8rrrLocalWorkerPoolrUrr*)rG overridesr_rFrFrJ getRunnerrs  zDistTrialRunnerTests.getRunnercCs:t}tt|}|}||||dkdS)ze L{DistTrialRunner.writeResults} writes to the stream specified in the init. rN)rr)r6r writeResultsr~tell)rGstringIOresultrunnerrFrFrJtest_writeResultss   z&DistTrialRunnerTests.test_writeResultscsfdfdd}d}d}|j||d}tddt|D}|||tjd jt|dS) zq L{DistTrialRunner.runAsync} doesn't try to start more workers than the number of tests. Nct|ddi|S)NrTrakwrrFrJrecordingFactoryzADistTrialRunnerTests.test_minimalWorker..recordingFactoryr=)rrcSrrFr)rnrFrFrJrrz;DistTrialRunnerTests.test_minimalWorker..r) rr;rrrunAsyncr_startedrYr)rGrrnumTestsrsuiterFrrJtest_minimalWorkers z'DistTrialRunnerTests.test_minimalWorkerrNcCs@|jdd}||j}||}||t||jtdS)z Running with the C{unclean-warnings} option makes L{DistTrialRunner} uses the L{UncleanWarningsReporterWrapper}. T)uncleanWarningsN)rrrrrr)originalr9)rGrdrrFrFrJtest_runUncleanWarningss    z,DistTrialRunnerTests.test_runUncleanWarningscCsTt}|j|d}||t}||t|}|d||d|dS)zA L{DistTrialRunner} can run an empty test suite. rRunning 0 testPASSEDN) rrrrr;rr)getvalueassertIn)rGrrroutputrFrFrJtest_runWithoutTests   z(DistTrialRunnerTests.test_runWithoutTestcCstdttd}t}|j|d}|||}||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% RuntimeErrorrrrrrr)rr)rGerrrrrrrFrFrJ!test_runWithoutTestButWithAnErrors      z6DistTrialRunnerTests.test_runWithoutTestButWithAnErrorcCsP|jtd}|||j}|jj}t|tdt|ddj t t dS)z If for some reasons we can't connect to the worker process, the error is recorded in the result object. rr@rN) rBrokenWorkerPoolrrrrerrorsrrtyperWorkerPoolBrokenrGrrrrFrFrJtest_runUnexpectedErrors z,DistTrialRunnerTests.test_runUnexpectedErrorcCsH|jtd}|t||jWddS1swYdS)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. rN)rr assertRaisesrrsr)rGrrFrFrJtest_runUnexpectedErrorCtrlCs  "z1DistTrialRunnerTests.test_runUnexpectedErrorCtrlCcCsZ|jtttddd}|||j}|jj}t |t dt |ddj t t dS)z~ If for some reason the worker process cannot run a test, the error is recorded in the result object. T) workerFactoryrrr@rN)rrr_BrokenLocalWorkerrrrrrrrrr WorkerBrokenrrFrFrJtest_runUnexpectedWorkerErrorsz2DistTrialRunnerTests.test_runUnexpectedWorkerErrorcsdfdd}|j|d}t||j}dur |ddus&Jjdj}||| d| |}| |t dS)zG L{DistTrialRunner} waits for the worker pool to stop. Ncr)NrFrrrrFrJrrzPDistTrialRunnerTests.test_runWaitForProcessesDeferreds..recordingFactoryrzworker pool was never createdr) rrrrrfailr_stoppedrcallbackrrr))rGrrrstoppedrrFrrJ!test_runWaitForProcessesDeferredss       z6DistTrialRunnerTests.test_runWaitForProcessesDeferredscCsft}ttdtdtdg}|j|ddd}||}||}t |j t dt ddd 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 exitFirstrr@) successesfailuresN) rr;r?FooTestr>TestRegularFailrrrrrrAr)rGrrrrrrFrFrJtest_exitFirst s"  z#DistTrialRunnerTests.test_exitFirstcCst}td}|j|d}|j|dd}||}|d|j|| | }|| d|jdd| d |t dd D] }| d ||qD|| d |jd dS)z L{DistTrialRunner} can run in C{untilFailure} mode where it will run the given tests until they fail. test_itrT) untilFailurerr@z expected to see PASSED in outputFAILz Test Pass zRan 1 tests inz2expected to see per-iteration test count in outputN)rr>EventuallyFailingTestCaserrrrrr| wasSuccessfulrrrr)rGrcaserrrrrrFrFrJtest_runUntilFailure"s*    z)DistTrialRunnerTests.test_runUntilFailurecCs:|}||j}t|tdt|jtddS)zw L{DistTrialRunner.run} returns a L{DistReporter} containing the result of the test suite run. Tr@N)rrsrrr#rr)rGrrrFrFrJrEs zDistTrialRunnerTests.test_runcCstg}t||jdd}Wdn1swY||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. NruTr@) rUr#rrsrrrrrr#rrmrk)rGrurrrFrFrJtest_installedReactorOs  z*DistTrialRunnerTests.test_installedReactorc CslGdddt}ttjGddd}Gddd}t|#|t|jddWdn1s7wYWdn1sFwYt|#|t|jddWdn1sgwYWdn1svwYt|,|t|jddWdn1swYWddSWddS1swYdS) z L{DistTrialRunner} raises L{TypeError} if the installed reactor provides neither L{IReactorCore} nor L{IReactorProcess} and no other reactor is given. c@seZdZddZdS)z=DistTrialRunnerTests.test_wrongInstalledReactor..CorecSdSrVrFrlrFrFrJ installWakergszJDistTrialRunnerTests.test_wrongInstalledReactor..Core.installWakerN)rNrOrPr)rFrFrFrJCorefs r*c@s"eZdZ      dddZdS)z@DistTrialRunnerTests.test_wrongInstalledReactor..ProcessNFc Sr(rVrF) rGrr^r_r`rarbrcrdrerFrFrJrfls zMDistTrialRunnerTests.test_wrongInstalledReactor..Process.spawnProcess)NNNNFN)rNrOrPrfrFrFrFrJProcessjsr+c@ eZdZdS)z@DistTrialRunnerTests.test_wrongInstalledReactor..NeitherNrNrOrPrFrFrFrJNeitherzr.Nr&)rr rrxr#r  TypeErrorr)rGr*r+r.rFrFrJtest_wrongInstalledReactor_s.      "z/DistTrialRunnerTests.test_wrongInstalledReactorcsdGdddtfdd}|j|d}|||jWddS1s+wYdS)z If there is an unexpected exception running the test suite then it is re-raised by L{DistTrialRunner.run}. c@r,)z;DistTrialRunnerTests.test_runFailure..BrokenFactoryNr-rFrFrFrJ BrokenFactoryr/r2csrVrF)r_rrr2rFrJ brokenFactorysz;DistTrialRunnerTests.test_runFailure..brokenFactoryrN) Exceptionrr rsr)rGr4rrFr3rJtest_runFailures    "z$DistTrialRunnerTests.test_runFailurerN)rNrOrPrQr;r?rrrrrrrrr r rrrr%rr'r1r6rFrFrFrJrks&     #  -rc@sBeZdZdZdddZdddZddd Zdd d Zdd d ZdS)FunctionalTestsz8 Tests for the functional helpers that need it. rNcCs,ttddtdttddtddS)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. r@Nr{)rr/rrlrFrFrJtest_fromOptionalsz!FunctionalTests.test_fromOptionalcCs8t}t|}|||dt||tdS)z ``discardResult`` accepts an awaitable and returns a ``Deferred`` that fires with ``None`` after the awaitable completes. rN)rr.rrrrr)rGrrrFrFrJtest_discardResults   z"FunctionalTests.test_discardResultcCsRt}t}tt||}|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)rrr1rrrrr)rGrbcrFrFrJ test_sequences   zFunctionalTests.test_sequencecCsltttg}dd}tt|t|j}|d|||dt||t ddS)z ``iterateWhile`` executes the actions from its factory until the predicate does not match an action result. cSs|dkS)Nr;rF)valuerFrFrJ predicatesz4FunctionalTests.test_iterateWhile..predicaterr;N) rrr0listpoprrrrr)rGactionsrArrFrFrJtest_iterateWhiles z!FunctionalTests.test_iterateWhilecCs:tdtdtfdd}tddD] }t|t|qdS)z ``countingCalls`` decorates a function so that it is called with an increasing counter and passes the return value through. rrcSs|dS)Nr@rF)rrFrFrJtargetsz2FunctionalTests.test_countingCalls..targetr@ N)r-rSrrr)rGrFexpectedrFrFrJtest_countingCallss z"FunctionalTests.test_countingCallsr7) rNrOrPrQr9r:r?rErIrFrFrFrJr8s  r8c@eZdZdZdS)rzx An exception for ``StartedWorkerPoolBroken`` to fail with to allow tests to exercise exception code paths. NrNrOrPrQrFrFrFrJrrc@s0eZdZdZdedddfddZd ddZdS) StartedWorkerPoolBrokenzn A broken, started worker pool. Its workers cannot run actions. They always raise an exception. workerActionNrctrV)r)rGrNrFrFrJrsszStartedWorkerPoolBroken.runcsdSrVrFrlrFrFrJrszStartedWorkerPoolBroken.joinr7)rNrOrPrQr5rsrrFrFrFrJrMsrMc@s.eZdZUdZeed<dejdefddZ dS)rzF A worker pool that has workers with a broken ``run`` method. _configrurcstSrV)rM)rGrurFrFrJr szBrokenWorkerPool.startN) rNrOrPrQr,rTrrxrMrrFrFrFrJrs rc@s&eZdZdZdededefddZdS) _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.. r$rrcst|g|ddiS)z8 Directly run C{case} in the usual way. successT)r;rsrGr$rrFrFrJrssz_LocalWorker.runN)rNrOrPrQPyUnitTestCaser7r3rsrFrFrFrJrQsrQc@rJ)rzE A worker tried to run a test case but the worker is broken. NrKrFrFrFrJr rLrc@s&eZdZdZdededdfddZdS)r z: A L{Worker} that always fails to run test cases. r$rrNcrO)z@ Raise an exception instead of running C{case}. )rrSrFrFrJrs+sz_BrokenLocalWorker.run)rNrOrPrQrTr7rsrFrFrFrJr &sr c@sTeZdZUdZeeed<eeed<e ded<de dddfdd Z d d Z dS) StartedLocalWorkerPoolz' A started L{LocalWorkerPool}. rrYNrrNrcs |jD] }||IdHqdS)z8 Run the action with each local worker. Nr)rGrNworkerrFrFrJrs<s zStartedLocalWorkerPool.runcs|jIdHdSrV)rrlrFrFrJrCszStartedLocalWorkerPool.join) rNrOrPrQr&rrTrr4rr5rsrrFrFrFrJrU2s     rUc@sheZdZUdZeed<eeedZ e e ed<dZ e ed<eZegefed<dejd e fd d Zd S) rz_ Implement a worker pool that runs tests in-process instead of in child processes. rPrCrF _autostop_workerFactoryrurcsLfddtjjD}tjj|jrtdnt}j ||S)Ncsg|]}qSrF)rXrrlrFrJrVsz)LocalWorkerPool.start..) rrPrrUrrWrrrrZ)rGrurYrrFrlrJrSs zLocalWorkerPool.startN)rNrOrPrQr,rTrr rBrrrUrWrrQrXrr4rrxrrFrFrFrJrGs  r)prQrr} functoolsrioros.pathrtypingrrrunittestr rTzope.interfacer r attrsr r rrhamcrestrrrrrrrhamcrest.core.core.allofr hypothesisrhypothesis.strategiesrrtwisted.internetrtwisted.internet.basertwisted.internet.deferrrrtwisted.internet.errorr twisted.internet.protocolr!r"#twisted.internet.test.modulehelpersr#twisted.internet.testingr$twisted.python.failurer%twisted.python.filepathr&twisted.python.lockfiler'twisted.trial._distr( twisted.trial._dist.distreporterr)twisted.trial._dist.disttrialr*r+r,twisted.trial._dist.functionalr-r.r/r0r1twisted.trial._dist.workerr2r3r4r5twisted.trial.reporterr6r7r8r9twisted.trial.runnerr:r;twisted.trial.unittestr<testr>r?matchersrArBrxrUrtrrr8r5rrMrrQrr rUrrFrFrFrJsp    $            B86N