U  W[2[ć@s¶dZddlZddlmZddlmZmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZGd d „d eƒZGd d „d ee ƒZGdd„dejƒZGdd„dejƒZdS)z& Tests for L{twisted.runner.procmon}. éN)Śunittest)ŚLoggingProtocolŚProcessMonitor)Ś ProcessDoneŚProcessTerminatedŚProcessExitedAlready)ŚClock)ŚFailure)Ślog)Ś MemoryReactorc@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Ś_reactorZ _executableZ_argsŚ _environmentŚ_pathZ_uidZ_gidZ_usePTYZ _childFDs) ŚselfŚreactorŚ executableŚargsZ environmentŚpathrŚuidŚgidŚusePTYŚchildFDs©rśB/usr/lib/python3/dist-packages/twisted/runner/test/test_procmon.pyŚ__init__"szDummyProcess.__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 )ZTERMŚKILLN)Ś_terminationDelayŚpidrrZ callLaterŚ processEndedZ_signalHandler)rZsignalIDZparamsŚdelayŚstatusrrrŚ signalProcess2s ž  ’zDummyProcess.signalProcesscCs,d|_ttdœ}|j t|||ƒƒ”dS)zC Deliver the process ended event to C{self.proto}. Nr)r"rrrr#r )rr%Z statusMaprrrr#Ls žzDummyProcess.processEnded)NNrN) Ś__name__Ś __module__Ś __qualname__Ś__doc__r"rr!rr&r#rrrrr 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 rrŚspawnedProcesses©rrrrr_s  zDummyProcessReactor.__init__rNrc Cs4t|||||||||| ƒ } | | ”|j | ”| S)zz Fake L{reactor.spawnProcess}, that logs all the process arguments and returns a L{DummyProcess}. )r ZmakeConnectionr,Śappend) rZprocessProtocolrrŚenvrrrrrŚprocrrrŚ spawnProcessfs ’  z DummyProcessReactor.spawnProcess)r'r(r)r*rr1rrrrr+Ysžr+c@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@S)AŚ 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)r+rrŚpmŚminRestartDelayŚmaxRestartDelayŚ thresholdr-rrrŚsetUp{s zProcmonTests.setUpcCsL|jjdddgddidt|jƒ}| d|”| d|”| d|”d S) z+ Repr includes all details ŚfooŚarg1Śarg2r r4©rrr/Ś1Ś2N)r6Ś addProcessŚreprŚassertIn©rZrepresentationrrrŚtest_reprLooksGood†s’   zProcmonTests.test_reprLooksGoodcCs<|jjdddgidt|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;r<r=)r/ś(ś)N)r6rArBŚ assertNotInrDrrrŚ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;r<r=r r4r>Ś processesN)r6rAŚ assertEqualŚ __getstate__r-rrrŚ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)rHr6rLr-rrrŚ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;r<r=r r4r>rN) r6rArKŚ protocolsrJŚ startServicerŚadvanceŚlistŚkeysr-rrrŚtest_addProcess±s’ ’  zProcmonTests.test_addProcessc Cs>|jjdddgddid|jt|jjdddgddiddS)z| L{ProcessMonitor.addProcess} raises a C{KeyError} if a process with the given name already exists. r;r<r=r r4r>N)r6rAŚ assertRaisesŚKeyErrorr-rrrŚ 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 r4r>rN)r6rPrArrQrKr,r)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)r6rPrArrQrKr,rr-rrrŚ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)r6rPrArKŚlenrJŚ removeProcessr-rrrŚ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)r6rPrUrVr]r-rrrŚ!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) r6rAŚ startProcessZassertIsInstancerOrrCŚ timeStartedrSr-rrrŚ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)r6rAr`Ś assertIsNoner-rrrŚtest_startProcessAlreadyStarteds 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)rUrVr6r`r-rrrŚ test_startProcessUnknownKeyErrorsz-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. r;r N)r6rPrArCrOŚ transportr!rrQr9Ś stopProcessrKŚsecondsra)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 ”|jj d”dS)z• L{ProcessMonitor.stopProcess} kills a process which fails to terminate naturally within L{ProcessMonitor.killTime} seconds. r;r gN)r6rPrArCrOrrQr9rfŚkillTimer!rgrKrarh)rr0rrrŚtest_stopProcessForcedKill3s   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)rUrVr6rgr-rrrŚtest_stopProcessUnknownKeyErrorJsz,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)r6rArcrgr-rrrŚtest_stopProcessAlreadyStoppedRsz+ProcmonTests.test_stopProcessAlreadyStoppedcCsŖg}| tj|j”t |j”|j ddg”|j ”|j  d”|  d|jj ”|j  |jj ”|jj d  d”| t|ƒd”|dd}| |tdgƒ”dS)zI Getting a complete output line generates a log message. r;rs hello world! r Śmessageś[foo] hello world!N)Ś addCleanupr ŚremoveObserverr.Ś addObserverr6rArPrrQrCrOr9Ś outReceivedŚ assertEqualsr\Śtuple©rŚeventsrnrrrŚtest_outputReceivedCompleteLine\s    z,ProcmonTests.test_outputReceivedCompleteLinecCsÜg}| tj|j”t |j”|j ddg”|j ”|j  d”|  d|jj ”|j  |jj ”|jj d  d”| t|ƒd”|dd}| t|ƒd”|d}| dd”\}}| |d”| |tdƒ”d S) zN Getting invalid UTF-8 results in the repr of the raw message r;rs’hello world! r rnś z[foo]s ’hello world!N)rpr rqr.rrr6rArPrrQrCrOr9rsrtr\ŚsplitrB)rrwZmessagesrnŚtagŚoutputrrrŚ*test_outputReceivedCompleteLineInvalidUTF8rs      z7ProcmonTests.test_outputReceivedCompleteLineInvalidUTF8cCsŌg}| tj|j”t |j”|j ddg”|j ”|j  d”|  d|jj ”|j  |jj ”|jj d  d”| t|ƒd”|jj d ttdƒƒ”| t|ƒd”|dd}| |tdgƒ”dS)zM Getting partial line results in no events until process end r;rs hello world!r rnroN)rpr rqr.rrr6rArPrrQrCrOr9rsrtr\r#r rrurvrrrŚ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) r6rArPrrQrCrOr9r#r rrHr-rrrŚ#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)r6rArPrrQrgrCZmurderŚ assertTrueŚactiverOrfr!Ś assertFalserH)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) r6rPrArCrOrfr&rrQr!rHr-rrrŚ#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 r4ér;r rN)r6r7r8rPrArKr$rrQr9rOr#r rr-rrrŚ%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 rr4N)r6rPrArrQr9rCrOrKr$r7r#r rr-rrrŚ&test_connectionLostBackoffDelayDoublesīs z3ProcmonTests.test_connectionLostBackoffDelayDoublescCs:|j ddg”|j ”|j d”| d|jj”dS)zP L{ProcessMonitor.startService} starts all monitored processes. r;rN)r6rArPrrQrCrOr-rrrŚ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;Zbarr N) r6rArPrrQr9rCrOŚ stopServicerjrKr-rrrŚ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. r;r N) r6rArPrrQZ restartAllrRr,Śpoprtrcr")rrJZ myProcessrrrŚ!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. r;r rN)r6rArPrrQr9rCrOr#r rr€Zrestartrr‰r‚r-rrrŚtest_stopServiceCancelRestarts0s   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) r6r9r7rPrArrQrgr‰rKrOr-rrrŚ(test_stopServiceCleanupScheduledRestartsEs      z5ProcmonTests.test_stopServiceCleanupScheduledRestartsN)#r'r(r)r*r:rErIrMrNrTrWrYr[r^r_rbrdrerirkrlrmrxr}r~rrƒr„r†r‡rˆrŠrŒrrrrrrr2vs@             r2c@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. r3N)r+rrr6r-rrrr:dszDeprecationTests.setUpcCsx|j ddg”|jj}| t|ƒd”| ”}d}|D]$}| |dt”d|dkr8d}q8| |d  t |ƒ””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ŚcategoryZtoTuplernTzno tuple deprecation found:{}N) r6rArJrtr\Ś flushWarningsŚassertIsŚDeprecationWarningr€ŚformatrB)rZ myprocessesŚwarningsZ foundToTupleŚwarningrrrŚ test_toTuplels  ’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)r6rJrtr“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Śdumpsr6r“r”r•)rr—r˜rrrŚ test_getstate‘s zDeprecationTests.test_getstateN)r'r(r)r*r:r™r›ržrrrrr‘^s r‘)r*rœZ twisted.trialrZtwisted.runner.procmonrrZtwisted.internet.errorrrrZtwisted.internet.taskrZtwisted.python.failurer Ztwisted.pythonr Ztwisted.test.proto_helpersr Śobjectr r+ZTestCaser2ZSynchronousTestCaser‘rrrrŚs     Ek