U T‚gdã@s´dZddlZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZGd d „d ƒZGd d „d e eƒZGdd„dejƒZGdd„dejƒZdS)z& Tests for L{twisted.runner.procmon}. éN)Ú ProcessDoneÚProcessExitedAlreadyÚProcessTerminated)ÚClock)Ú MemoryReactor)ÚglobalLogPublisher)ÚFailure)ÚLoggingProtocolÚProcessMonitor)Úunittestc@s6eZdZdZdZdZdZd 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Ú_pathZ_uidZ_gidZ_usePTYZ _childFDs) ÚselfÚreactorÚ executableÚargsÚ environmentÚpathrÚuidÚgidÚusePTYÚchildFDs©rúD/tmp/pip-unpacked-wheel-1_7di8ic/twisted/runner/test/test_procmon.pyÚ__init__ s zDummyProcess.__init__cCsL|jdfddœ}|jdkr tƒ‚||krH||\}}|j ||j|¡|_dS)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ÚpidrrZ callLaterÚ processEndedZ_signalHandler)rZsignalIDÚparamsÚdelayÚstatusrrr Ú signalProcess9s   ÿzDummyProcess.signalProcesscCs,d|_ttdœ}|j t|||ƒƒ¡dS)zC Deliver the process ended event to C{self.proto}. Nr")r&rrrr'r)rr*Z statusMaprrr r'Ps þzDummyProcess.processEnded)NNrN) Ú__name__Ú __module__Ú __qualname__Ú__doc__r&rr%r!r+r'rrrr 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)rr!rÚspawnedProcesses©rrrr r!cs  zDummyProcessReactor.__init__rNrc Cs4t|||||||||| ƒ } | | ¡|j | ¡| S)zz Fake L{reactor.spawnProcess}, that logs all the process arguments and returns a L{DummyProcess}. )r ZmakeConnectionr1Úappend) rZprocessProtocolrrÚenvrrrrrÚprocrrr Ú spawnProcessisö  z DummyProcessReactor.spawnProcess)r,r-r.r/r!r6rrrr r0\s ör0c@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)r0rr ÚpmÚminRestartDelayÚmaxRestartDelayÚ thresholdr2rrr ÚsetUps zProcmonTests.setUpcCsL|jjdddgddidt|jƒ}| d|¡| d|¡| d|¡d S) z+ Repr includes all details ÚfooÚarg1Úarg2r r9©rrr4Ú1Ú2N)r;Ú addProcessÚreprÚassertIn©rZrepresentationrrr Ú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. r@rArB)r4ú(ú)N)r;rFrGÚ assertNotInrIrrr Útest_simpleReprLooksGood¤s  z%ProcmonTests.test_simpleReprLooksGoodcCsD|jjdddgddid| |j ¡ddddgddifi¡dS) zW The list of monitored processes must be included in the pickle state. r@rArBr r9rCÚ processesN)r;rFÚ assertEqualÚ __getstate__r2rrr Ú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)rMr;rQr2rrr Útest_getStateExcludesReactor¹sz)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. r@rArBr r9rCrN) r;rFrPÚ protocolsrOÚ startServicerÚadvanceÚlistÚkeysr2rrr Ú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. r@rArBr r9rCN)r;rFÚ assertRaisesÚKeyErrorr2rrr Ú 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}. ZKEYÚvaluer@r r9rCrN)r;rUrFrrVrPr1r)rZfakeEnvrrr Ú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}. r@z /mnt/lala)ÚcwdrN)r;rUrFrrVrPr1rr2rrr Ú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. r@r rN)r;rUrFrPÚlenrOÚ removeProcessr2rrr Ú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. r@N)r;rUrZr[rbr2rrr Ú!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. r@N) r;rFÚ startProcessZassertIsInstancerTr rHÚ timeStartedrXr2rrr Ú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. r@N)r;rFreÚ assertIsNoner2rrr Útest_startProcessAlreadyStarted s z,ProcmonTests.test_startProcessAlreadyStartedcCs| t|jjd¡dS)zy L{ProcessMonitor.startProcess} raises a C{KeyError} if the given process name isn't recognised. r@N)rZr[r;rer2rrr Ú test_startProcessUnknownKeyErrorsz-ProcmonTests.test_startProcessUnknownKeyErrorcCs|j ¡|j ddg¡| d|jj¡d}|jjdj_|j |jj ¡|j  d¡|j |¡|j d¡|  |j  ¡|jj d¡dS)zm L{ProcessMonitor.stopProcess} immediately sends a TERM signal to the named process. r@r rN)r;rUrFrHrTÚ transportr%rrVr>Ú stopProcessrPÚsecondsrf)rZ timeToDierrr Ú"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 ddg¡|  |j ¡|jj d¡dS)z• L{ProcessMonitor.stopProcess} kills a process which fails to terminate naturally within L{ProcessMonitor.killTime} seconds. r@r grN)r;rUrFrHrTrrVr>rkÚkillTimer%rlrPrfZpumprm)rr5rrr Útest_stopProcessForcedKill8s   z'ProcmonTests.test_stopProcessForcedKillcCs| t|jjd¡dS)zx L{ProcessMonitor.stopProcess} raises a C{KeyError} if the given process name isn't recognised. r@N)rZr[r;rlr2rrr Útest_stopProcessUnknownKeyErrorUsz,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. r@N)r;rFrhrlr2rrr Útest_stopProcessAlreadyStopped\sz+ProcmonTests.test_stopProcessAlreadyStoppedcCsì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. r@ró hello world! r Ú log_namespaceÚstreamÚtagÚlineú%twisted.runner.procmon.ProcessMonitorÚstdoutú hello world!N)Ú addCleanuprÚremoveObserverr3Ú addObserverr;rFrUrrVrHrTr>Ú outReceivedÚ assertEqualsra©rÚeventsÚ namespacerurvrwrrr Útest_outputReceivedCompleteLinees$          z,ProcmonTests.test_outputReceivedCompleteLinecCsì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 stderr generates a log message. r@rrsr rtrurvrwrxÚstderrrzN)r{rr|r3r}r;rFrUrrVrHrTr>Z errReceivedrrar€rrr Ú!test_ouputReceivedCompleteErrLine€s$          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 r@rsÿhello world! r rtrurvrwrxrys ÿhello world!N)r{rr|r3r}r;rFrUrrVrHrTr>r~rrarG)rrÚmessager‚rurvÚoutputrrr Ú*test_outputReceivedCompleteLineInvalidUTF8›s&      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 r@rs hello world!r rtrurvrwrxryrzN)r{rr|r3r}r;rFrUrrVrHrTr>r~rrar'rrr€rrr Ú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. r@rN) r;rFrUrrVrHrTr>r'rrrMr2rrr Ú#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. r@r N)r;rFrUrrVrlrHZmurderÚ assertTrueÚactiverTrkr%Ú assertFalserM)rZ delayedCallrrr Ú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. r@r$N) r;rUrFrHrTrkr+rrVr%rMr2rrr Ú#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 r9ér@r rN)r;r<r=rUrFrPr)rrVr>rTr'rrr2rrr Ú%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. r@r rr9N)r;rUrFrrVr>rHrTrPr)r<r'rrr2rrr Ú&test_connectionLostBackoffDelayDoubless z3ProcmonTests.test_connectionLostBackoffDelayDoublescCs:|j ddg¡|j ¡|j d¡| d|jj¡dS)zP L{ProcessMonitor.startService} starts all monitored processes. r@rN)r;rFrUrrVrHrTr2rrr Útest_startService's  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. r@Úbarr N) r;rFrUrrVr>rHrTÚ stopServicerorPr2rrr Útest_stopService2s   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. r@r N) r;rFrUrrVZ restartAllrWr1Úpoprrhr&)rrOZ myProcessrrr Ú!test_restartAllRestartsOneProcessHs      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. r@r rN)r;rFrUrrVr>rHrTr'rrr‹ZrestartrŒr•rr2rrr Útest_stopServiceCancelRestartsXs   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. ér@r éN) r;r>r<rUrFrrVrlr•rPrTr2rrr Ú(test_stopServiceCleanupScheduledRestartsls      z5ProcmonTests.test_stopServiceCleanupScheduledRestartsN)$r,r-r.r/r?rJrNrRrSrYr\r^r`rcrdrgrirjrnrprqrrrƒr…rˆr‰rŠrŽrr‘r’r“r–r˜r™rœrrrr r7‹sB              r7c@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. r8N)r0rr r;r2rrr r?ŠszDeprecationTests.setUpcCsx|j ddg¡|jj}| t|ƒd¡| ¡}d}|D]$}| |dt¡d|dkr8d}q8| |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. r@r FÚcategoryZtoTupler†Tzno tuple deprecation found:N) r;rFrOrraÚ flushWarningsÚassertIsÚDeprecationWarningr‹rG)rZ myprocessesÚwarningsZ foundToTupleÚwarningrrr Ú 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)r;rOrrŸr—r r¡)rZ myProcessesr¢Úfirstrrr Útest_processes¤s   zDeprecationTests.test_processescCs2t |j¡| ¡}|D]}| |dt¡qdS)zO Pickling an L{ProcessMonitor} results in deprecation warnings ržN)ÚpickleÚdumpsr;rŸr r¡)rr¢r£rrr Ú test_getstate³s zDeprecationTests.test_getstateN)r,r-r.r/r?r¤r¦r©rrrr r„s r)r/r§Ztwisted.internet.errorrrrZtwisted.internet.taskrZtwisted.internet.testingrZtwisted.loggerrZtwisted.python.failurerZtwisted.runner.procmonr r Z twisted.trialr r r0ZTestCaser7ZSynchronousTestCaserrrrr Ús     J/|