o ¯bíaã@s´dZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZGd d „d ƒZGd d „d eeƒZGdd„dejƒZGdd„dejƒZdS)z& Tests for L{twisted.runner.procmon}. éN)Ú ProcessDoneÚProcessExitedAlreadyÚProcessTerminated)ÚClock)ÚglobalLogPublisher)ÚFailure)ÚLoggingProtocolÚProcessMonitor)Ú MemoryReactor)Úunittestc@s>eZdZdZdZdZdZ    d dd„Zdd„Zd d „Z dS) Ú DummyProcessa  An incomplete and fake L{IProcessTransport} implementation for testing how L{ProcessMonitor} behaves when its monitored processes exit. @ivar _terminationDelay: the delay in seconds after which the DummyProcess will appear to exit when it receives a TERM signal éNrc Cs@||_||_||_||_||_||_||_||_| |_| |_ dS©N) ÚprotoÚ_reactorÚ _executableÚ_argsÚ _environmentÚ_pathÚ_uidÚ_gidÚ_usePTYÚ _childFDs) ÚselfÚreactorÚ executableÚargsÚ environmentÚpathrÚuidÚgidÚusePTYÚchildFDs©r#úB/usr/lib/python3/dist-packages/twisted/runner/test/test_procmon.pyÚ__init__ s zDummyProcess.__init__cCsP|jdfddœ}|jdurtƒ‚||vr&||\}}|j ||j|¡|_dSdS)añ A partial implementation of signalProcess which can only handle TERM and KILL signals. - When a TERM signal is given, the dummy process will appear to exit after L{DummyProcess._terminationDelay} seconds with exit code 0 - When a KILL signal is given, the dummy process will appear to exit immediately with exit code 1. @param signalID: The signal name or number to be issued to the process. @type signalID: C{str} r©rr )ÚTERMÚKILLN)Ú_terminationDelayÚpidrrÚ callLaterÚ processEndedÚ_signalHandler)rÚsignalIDÚparamsÚdelayÚstatusr#r#r$Ú signalProcess:s   ÿþzDummyProcess.signalProcesscCs,d|_ttdœ}|j t|||ƒƒ¡dS)zC Deliver the process ended event to C{self.proto}. Nr&)r*rrrr,r)rr1Ú statusMapr#r#r$r,Qs þzDummyProcess.processEnded)NNrN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r*rr)r%r2r,r#r#r#r$r s  õ r c@s0eZdZdZdd„Zdidddddfdd„ZdS) ÚDummyProcessReactorz @ivar spawnedProcesses: a list that keeps track of the fake process instances built by C{spawnProcess}. @type spawnedProcesses: C{list} cCst |¡t |¡g|_dSr)r r%rÚspawnedProcesses©rr#r#r$r%ds   zDummyProcessReactor.__init__r#Nrc Cs4t|||||||||| ƒ } | | ¡|j | ¡| S)zz Fake L{reactor.spawnProcess}, that logs all the process arguments and returns a L{DummyProcess}. )r ÚmakeConnectionr9Úappend) rÚprocessProtocolrrÚenvrrr r!r"Úprocr#r#r$Ú spawnProcessjsö z DummyProcessReactor.spawnProcess)r4r5r6r7r%r@r#r#r#r$r8]s ör8c@seZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„Zd.d/„Zd0d1„Zd2d3„Zd4d5„Zd6d7„Zd8d9„Zd:d;„Z dd?„Z"d@dA„Z#dBS)CÚ ProcmonTestsz& Tests for L{ProcessMonitor}. cCs2tƒ|_t|jd|_d|j_d|j_d|j_dS)zL Create an L{ProcessMonitor} wrapped around a fake reactor. ©réé N)r8rr ÚpmÚminRestartDelayÚmaxRestartDelayÚ thresholdr:r#r#r$ÚsetUp‘s  zProcmonTests.setUpcCsL|jjdddgddidt|jƒ}| d|¡| d|¡| d|¡d S) z+ Repr includes all details ÚfooÚarg1Úarg2r rC©rr r>Ú1Ú2N)rEÚ addProcessÚreprÚassertIn©rÚrepresentationr#r#r$Útest_reprLooksGood›s    zProcmonTests.test_reprLooksGoodcCs<|jjdddgidt|jƒ}| d|¡| d|¡dS)z· Repr does not include unneeded details. Values of attributes that just mean "inherit from launching process" do not appear in the repr of a process. rJrKrL)r>ú(ú)N)rErPrQÚ assertNotInrSr#r#r$Útest_simpleReprLooksGood¥s  z%ProcmonTests.test_simpleReprLooksGoodcCsD|jjdddgddid| |j ¡ddddgddifi¡dS) zW The list of monitored processes must be included in the pickle state. rJrKrLr rCrMÚ processesN)rErPÚ assertEqualÚ __getstate__r:r#r#r$Útest_getStateIncludesProcesses±sÿz+ProcmonTests.test_getStateIncludesProcessescCs| d|j ¡¡dS)z~ The private L{ProcessMonitor._reactor} instance variable should not be included in the pickle state. rN)rXrEr\r:r#r#r$Útest_getStateExcludesReactorºóz)ProcmonTests.test_getStateExcludesReactorcCs~|jjdddgddid| |jji¡| |jjdddgddifi¡|j ¡|j d¡| t|jj  ¡ƒdg¡dS) z‡ L{ProcessMonitor.addProcess} only starts the named program if L{ProcessMonitor.startService} has been called. rJrKrLr rCrMrN) rErPr[Ú protocolsrZÚ startServicerÚadvanceÚlistÚkeysr:r#r#r$Útest_addProcessÁs    zProcmonTests.test_addProcessc Cs>|jjdddgddid|jt|jjdddgddiddS)z| L{ProcessMonitor.addProcess} raises a C{KeyError} if a process with the given name already exists. rJrKrLr rCrMN)rErPÚ assertRaisesÚKeyErrorr:r#r#r$Ú test_addProcessDuplicateKeyErrorÍs ÿz-ProcmonTests.test_addProcessDuplicateKeyErrorcCsPddi}|j ¡|jjddgdd|d|j d¡| |jjdj|¡dS) zƒ L{ProcessMonitor.addProcess} takes an C{env} parameter that is passed to L{IReactorProcess.spawnProcess}. ÚKEYÚvaluerJr rCrMrN)rErarPrrbr[r9r)rÚfakeEnvr#r#r$Útest_addProcessEnv×s   zProcmonTests.test_addProcessEnvcCsD|j ¡|jjddgdd|j d¡| |jjdjd¡dS)zƒ L{ProcessMonitor.addProcess} takes an C{cwd} parameter that is passed to L{IReactorProcess.spawnProcess}. rJz /mnt/lala)ÚcwdrN)rErarPrrbr[r9rr:r#r#r$Útest_addProcessCwdâs  zProcmonTests.test_addProcessCwdcCsR|j ¡|j ddg¡| t|jjƒd¡|j d¡| t|jjƒd¡dS)zm L{ProcessMonitor.removeProcess} removes the process from the public processes list. rJr rN)rErarPr[ÚlenrZÚ removeProcessr:r#r#r$Útest_removeProcessìs  zProcmonTests.test_removeProcesscCs |j ¡| t|jjd¡dS)zz L{ProcessMonitor.removeProcess} raises a C{KeyError} if the given process name isn't recognised. rJN)rErarfrgrpr:r#r#r$Ú!test_removeProcessUnknownKeyError÷s z.ProcmonTests.test_removeProcessUnknownKeyErrorcCsH|j ddg¡|j d¡| |jjdt¡| d|jj ¡¡dS)a When a process has been started, an instance of L{LoggingProtocol} will be added to the L{ProcessMonitor.protocols} dict and the start time of the process will be recorded in the L{ProcessMonitor.timeStarted} dictionary. rJN) rErPÚ startProcessÚassertIsInstancer`rrRÚ timeStartedrdr:r#r#r$Útest_startProcessÿs zProcmonTests.test_startProcesscCs2|j ddg¡|j d¡| |j d¡¡dS)zr L{ProcessMonitor.startProcess} silently returns if the named process is already started. rJN)rErPrsÚ assertIsNoner:r#r#r$Útest_startProcessAlreadyStarted s z,ProcmonTests.test_startProcessAlreadyStartedcCó| t|jjd¡dS)zy L{ProcessMonitor.startProcess} raises a C{KeyError} if the given process name isn't recognised. rJN)rfrgrErsr:r#r#r$Ú test_startProcessUnknownKeyErrorr_z-ProcmonTests.test_startProcessUnknownKeyErrorcCs„|j ¡|j ddg¡| d|jj¡d}|jjdj_|j |jj ¡|j  d¡|j |¡|  |j  ¡|jj d¡dS)zm L{ProcessMonitor.stopProcess} immediately sends a TERM signal to the named process. rJr N)rErarPrRr`Ú transportr)rrbrHÚ stopProcessr[Úsecondsru)rÚ timeToDier#r#r$Ú"test_stopProcessNaturalTerminations   z/ProcmonTests.test_stopProcessNaturalTerminationcCs´|j ¡|j ddg¡| d|jj¡|j |jj¡|jjdj}|jj d|_ |j  d¡|j |jj d¡|  d|jj d¡|j d¡|  |j ¡|jj d¡dS)z• L{ProcessMonitor.stopProcess} kills a process which fails to terminate naturally within L{ProcessMonitor.killTime} seconds. rJr gN)rErarPrRr`rrbrHr{ÚkillTimer)r|r[rur})rr?r#r#r$Útest_stopProcessForcedKill4s   z'ProcmonTests.test_stopProcessForcedKillcCry)zx L{ProcessMonitor.stopProcess} raises a C{KeyError} if the given process name isn't recognised. rJN)rfrgrEr|r:r#r#r$Útest_stopProcessUnknownKeyErrorJr_z,ProcmonTests.test_stopProcessUnknownKeyErrorcCs&|j ddg¡| |j d¡¡dS)zá L{ProcessMonitor.stopProcess} silently returns if the named process is already stopped. eg Process has crashed and a restart has been rescheduled, but in the meantime, the service is stopped. rJN)rErPrwr|r:r#r#r$Útest_stopProcessAlreadyStoppedQsz+ProcmonTests.test_stopProcessAlreadyStoppedcCóìg}| tj|j¡t |j¡|j ddg¡|j ¡|j  d¡|  d|jj ¡|j  |jj ¡|jj d  d¡| t|ƒd¡|dd}|dd}|dd}|dd}| |d ¡| |d ¡| |d¡| |d ¡d S) zS Getting a complete output line on stdout generates a log message. rJró hello world! r Ú log_namespaceÚstreamÚtagÚlineú%twisted.runner.procmon.ProcessMonitorÚstdoutú hello world!N)Ú addCleanuprÚremoveObserverr<Ú addObserverrErPrarrbrRr`rHÚ outReceivedÚ assertEqualsro©rÚeventsÚ namespacer‡rˆr‰r#r#r$Útest_outputReceivedCompleteLineZó$          z,ProcmonTests.test_outputReceivedCompleteLinecCr„) zS Getting a complete output line on stderr generates a log message. rJrr…r r†r‡rˆr‰rŠÚstderrrŒN)rrrŽr<rrErPrarrbrRr`rHÚ errReceivedr‘ror’r#r#r$Ú!test_ouputReceivedCompleteErrLineur–z.ProcmonTests.test_ouputReceivedCompleteErrLinecCsèg}| tj|j¡t |j¡|j ddg¡|j ¡|j  d¡|  d|jj ¡|j  |jj ¡|jj d  d¡| t|ƒd¡|d}|d}|d}|d}|d}| |d ¡| |d ¡| |d¡| |td ƒ¡d S) zN Getting invalid UTF-8 results in the repr of the raw message rJrsÿhello world! r r†r‡rˆr‰rŠr‹s ÿhello world!N)rrrŽr<rrErPrarrbrRr`rHrr‘rorQ)rr“Úmessager”r‡rˆÚoutputr#r#r$Ú*test_outputReceivedCompleteLineInvalidUTF8s&      z7ProcmonTests.test_outputReceivedCompleteLineInvalidUTF8cCsg}| tj|j¡t |j¡|j ddg¡|j ¡|j  d¡|  d|jj ¡|j  |jj ¡|jj d  d¡| t|ƒd¡|jj d ttdƒƒ¡| t|ƒd¡|dd}|dd}|dd}|dd}| |d ¡| |d ¡| |d¡| |d ¡d S) zM Getting partial line results in no events until process end rJrs hello world!r r†r‡rˆr‰rŠr‹rŒN)rrrŽr<rrErPrarrbrRr`rHrr‘ror,rrr’r#r#r$Útest_outputReceivedPartialLine¬s(          z+ProcmonTests.test_outputReceivedPartialLinecCs|j ddg¡|j ¡|j d¡| d|jj¡|j |jj¡|jjd t t dƒƒ¡|  d|jj¡|j d¡| d|jj¡dS)z§ L{ProcessMonitor.connectionLost} should immediately restart a process if it has been running longer than L{ProcessMonitor.threshold} seconds. rJrN) rErPrarrbrRr`rHr,rrrXr:r#r#r$Ú#test_connectionLostLongLivedProcessÉs   z0ProcmonTests.test_connectionLostLongLivedProcesscCs–|j ddg¡|j ¡|j d¡|j d¡| d|jj¡|jjd}| |  ¡¡|j |jj dj j ¡|  |  ¡¡| d|jj¡dS)z¡ L{ProcessMonitor.connectionLost} cancels a scheduled process killer and deletes the DelayedCall from the L{ProcessMonitor.murder} list. rJr N)rErPrarrbr|rRÚmurderÚ assertTrueÚactiver`r{r)Ú assertFalserX)rÚ delayedCallr#r#r$Útest_connectionLostMurderCancelÝs    z,ProcmonTests.test_connectionLostMurderCancelcCsj|j ¡|j ddg¡| d|jj¡|jjdj d¡|j |jjdjj ¡|  d|jj¡dS)z˜ L{ProcessMonitor.connectionLost} removes the corresponding ProcessProtocol instance from the L{ProcessMonitor.protocols} list. rJr(N) rErarPrRr`r{r2rrbr)rXr:r#r#r$Ú#test_connectionLostProtocolDeletionòs z0ProcmonTests.test_connectionLostProtocolDeletioncCsŒd|j_d|j_|j ¡|j ddg¡| |jjd|jj¡|j |jj d¡|jj d  t t dƒƒ¡| |jjd|jj¡dS)z} L{ProcessMonitor.connectionLost} will wait at least minRestartDelay s and at most maxRestartDelay s rCérJr rN)rErFrGrarPr[r0rrbrHr`r,rrr:r#r#r$Ú%test_connectionLostMinMaxRestartDelayþs z2ProcmonTests.test_connectionLostMinMaxRestartDelaycCs|j ¡|j ddg¡|j |jjd¡| d|jj¡| |jj d|jj ¡|jjd  t t dƒƒ¡| |jj d|jj d¡dS)z| L{ProcessMonitor.connectionLost} doubles the restart delay each time the process dies too quickly. rJr rrCN)rErarPrrbrHrRr`r[r0rFr,rrr:r#r#r$Ú&test_connectionLostBackoffDelayDoubless  z3ProcmonTests.test_connectionLostBackoffDelayDoublescCs:|j ddg¡|j ¡|j d¡| d|jj¡dS)zP L{ProcessMonitor.startService} starts all monitored processes. rJrN)rErPrarrbrRr`r:r#r#r$Útest_startServices  zProcmonTests.test_startServicecCs˜|j ddg¡|j ddg¡|j ¡|j |jj¡| d|jj¡| d|jj¡|j d¡|j ¡|j |jj d¡|  i|jj¡dS)zT L{ProcessMonitor.stopService} should stop all monitored processes. rJÚbarr N) rErPrarrbrHrRr`Ú stopServicer€r[r:r#r#r$Útest_stopService's   zProcmonTests.test_stopServicecCsl|j ddg¡|j ¡|j d¡|j ¡|j d¡t|jjƒ}| ¡}|  |g¡|  |j ¡dS)zR L{ProcessMonitor.restartAll} succeeds when there is one process. rJr N) rErPrarrbÚ restartAllrcr9Úpopr‘rwr*)rrZÚ myProcessr#r#r$Ú!test_restartAllRestartsOneProcess=s      z.ProcmonTests.test_restartAllRestartsOneProcesscCsš|j ddg¡|j ¡|j |jj¡| d|jj¡|j d¡|jjd t t dƒƒ¡|  |jj d  ¡¡|j ¡| |jj d  ¡¡dS)ze L{ProcessMonitor.stopService} should cancel any scheduled process restarts. rJr rN)rErPrarrbrHrRr`r,rrr Úrestartr¡r«r¢r:r#r#r$Útest_stopServiceCancelRestartsMs   z+ProcmonTests.test_stopServiceCancelRestartscCsxd|j_d|j_|j ¡|j ddg¡|j d¡|j d¡|j d¡|j ¡|j d¡|  |jj i¡dS)ze L{ProcessMonitor.stopService} should cancel all scheduled process restarts. érJr éN) rErHrFrarPrrbr|r«r[r`r:r#r#r$Ú(test_stopServiceCleanupScheduledRestartsas      z5ProcmonTests.test_stopServiceCleanupScheduledRestartsN)$r4r5r6r7rIrUrYr]r^rerhrlrnrqrrrvrxrzrrr‚rƒr•r™rœrržr¤r¥r§r¨r©r¬r°r²rµr#r#r#r$rAŒsD               rAc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ÚDeprecationTestszQ Tests that check functionality that should be deprecated is deprecated. cCstƒ|_t|jd|_dS)z5 Create reactor and process monitor. rBN)r8rr rEr:r#r#r$rIszDeprecationTests.setUpcCsx|j ddg¡|jj}| t|ƒd¡| ¡}d}|D]}| |dt¡d|dvr.d}q| |dt |ƒ›¡d S) a _Process.toTuple is deprecated. When getting the deprecated processes property, the actual data (kept in the class _Process) is converted to a tuple -- which produces a DeprecationWarning per process so converted. rJr FÚcategoryÚtoTupleršTzno tuple deprecation found:N) rErPrZr‘roÚ flushWarningsÚassertIsÚDeprecationWarningr rQ)rÚ myprocessesÚwarningsÚ foundToTupleÚwarningr#r#r$Ú test_toTuple†s €zDeprecationTests.test_toTuplecCsF|jj}| |i¡| ¡}| d¡}| |dt¡| |g¡dS)a Accessing L{ProcessMonitor.processes} results in deprecation warning Even when there are no processes, and thus no process is converted to a tuple, accessing the L{ProcessMonitor.processes} property should generate its own DeprecationWarning. rr·N)rErZr‘r¹r®rºr»)rÚ myProcessesr½Úfirstr#r#r$Útest_processes™s   zDeprecationTests.test_processescCs2t |j¡| ¡}|D] }| |dt¡q dS)zO Pickling an L{ProcessMonitor} results in deprecation warnings r·N)ÚpickleÚdumpsrEr¹rºr»)rr½r¿r#r#r$Ú test_getstate¨s ÿzDeprecationTests.test_getstateN)r4r5r6r7rIrÀrÃrÆr#r#r#r$r¶ys  r¶)r7rÄÚtwisted.internet.errorrrrÚtwisted.internet.taskrÚtwisted.loggerrÚtwisted.python.failurerÚtwisted.runner.procmonrr Útwisted.test.proto_helpersr Ú twisted.trialr r r8ÚTestCaserAÚSynchronousTestCaser¶r#r#r#r$Ús      K/p