← Back
Editing: test_procmon.cpython-38.pyc
U W[2[ � @ s� d Z ddlZddlmZ ddlmZmZ ddlmZm Z m Z ddl m Z ddl mZ ddlmZ dd lmZ G d d � d e�ZG d d � d ee �ZG dd� dej�ZG dd� dej�ZdS )z& Tests for L{twisted.runner.procmon}. � N)�unittest)�LoggingProtocol�ProcessMonitor)� ProcessDone�ProcessTerminated�ProcessExitedAlready)�Clock)�Failure)�log)� MemoryReactorc @ s6 e Zd ZdZdZdZdZd 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 � Nr c C s@ || _ || _|| _|| _|| _|| _|| _|| _| | _| | _ d S �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__" s zDummyProcess.__init__c C sL | j dfdd�}| 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 �r r )ZTERM�KILLN)�_terminationDelay�pidr r Z callLater� processEndedZ_signalHandler)r ZsignalIDZparams�delay�statusr r r � signalProcess2 s � �zDummyProcess.signalProcessc C s, d| _ ttd�}| j�t|| |��� dS )zC Deliver the process ended event to C{self.proto}. Nr )r" r r r r# r )r r% Z statusMapr r r r# L s �zDummyProcess.processEnded)NNr N) �__name__� __module__� __qualname__�__doc__r" r r! r r& r# r r r r r s � r c @ s0 e Zd ZdZdd� Zdi dddddfdd�ZdS ) �DummyProcessReactorz� @ivar spawnedProcesses: a list that keeps track of the fake process instances built by C{spawnProcess}. @type spawnedProcesses: C{list} c C s t �| � t�| � g | _d S r )r r r �spawnedProcesses�r r r r r _ s z DummyProcessReactor.__init__r Nr c C s4 t | ||||||||| � } |�| � | j�| � | S )zz Fake L{reactor.spawnProcess}, that logs all the process arguments and returns a L{DummyProcess}. )r ZmakeConnectionr, �append) r ZprocessProtocolr r �envr r r r r �procr r r � spawnProcessf s � z DummyProcessReactor.spawnProcess)r' r( r) r* r r1 r r r r r+ Y s �r+ c @ s e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@S )A� ProcmonTestsz& Tests for L{ProcessMonitor}. c C s2 t � | _t| jd�| _d| j_d| j_d| j_dS )zL Create an L{ProcessMonitor} wrapped around a fake reactor. �r � � N)r+ r r �pm�minRestartDelay�maxRestartDelay� thresholdr- r r r �setUp{ s zProcmonTests.setUpc C sL | j jdddgddi d� t| j �}| �d|� | �d|� | �d|� d S ) z+ Repr includes all details �foo�arg1�arg2r r4 �r r r/ �1�2N)r6 � addProcess�repr�assertIn�r Zrepresentationr r r �test_reprLooksGood� s � z ProcmonTests.test_reprLooksGoodc C s< | j jdddgi d� 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)r6 rA rB � assertNotInrD r r r �test_simpleReprLooksGood� s z%ProcmonTests.test_simpleReprLooksGoodc C sD | j jdddgddi d� | �| j �� d dddgddi fi� dS ) zW The list of monitored processes must be included in the pickle state. r; r< r= r r4 r> � processesN)r6 rA � assertEqual� __getstate__r- r r r � test_getStateIncludesProcesses� s ��z+ProcmonTests.test_getStateIncludesProcessesc C s | � d| j�� � dS )z~ The private L{ProcessMonitor._reactor} instance variable should not be included in the pickle state. r N)rH r6 rL r- r r r � test_getStateExcludesReactor� s z)ProcmonTests.test_getStateExcludesReactorc C s~ | j jdddgddi d� | �| j ji � | �| j jdddgddi fi� | j �� | j�d� | �t| j j� � �dg� dS ) z� L{ProcessMonitor.addProcess} only starts the named program if L{ProcessMonitor.startService} has been called. r; r< r= r r4 r> r N) r6 rA rK � protocolsrJ � startServicer �advance�list�keysr- r r r �test_addProcess� s � � z ProcmonTests.test_addProcessc C s> | j jdddgddi d� | jt| j jdddgddi d� dS )z| L{ProcessMonitor.addProcess} raises a C{KeyError} if a process with the given name already exists. r; r< r= r r4 r> N)r6 rA � assertRaises�KeyErrorr- r r r � test_addProcessDuplicateKeyError� s � �z-ProcmonTests.test_addProcessDuplicateKeyErrorc C sP ddi}| j �� | j jddgdd|d� | j�d� | �| jjd j|� dS ) z� L{ProcessMonitor.addProcess} takes an C{env} parameter that is passed to L{IReactorProcess.spawnProcess}. ZKEY�valuer; r r4 r> r N)r6 rP rA r rQ rK r, r )r ZfakeEnvr r r �test_addProcessEnv� s �z ProcmonTests.test_addProcessEnvc C sD | j �� | j jddgdd� | j�d� | �| jjd jd� dS )z� L{ProcessMonitor.addProcess} takes an C{cwd} parameter that is passed to L{IReactorProcess.spawnProcess}. r; z /mnt/lala)�cwdr N)r6 rP rA r rQ rK r, r r- r r r �test_addProcessCwd� s �z ProcmonTests.test_addProcessCwdc C sR | j �� | j �ddg� | �t| j j�d� | j �d� | �t| j j�d� dS )zm L{ProcessMonitor.removeProcess} removes the process from the public processes list. r; r r N)r6 rP rA rK �lenrJ � removeProcessr- r r r �test_removeProcess� s z ProcmonTests.test_removeProcessc C s | j �� | �t| j jd� dS )zz L{ProcessMonitor.removeProcess} raises a C{KeyError} if the given process name isn't recognised. r; N)r6 rP rU rV r] r- r r r �!test_removeProcessUnknownKeyError� s z.ProcmonTests.test_removeProcessUnknownKeyErrorc C sH | j �ddg� | j �d� | �| j jd t� | �d| j j�� � 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) r6 rA � startProcessZassertIsInstancerO r rC � timeStartedrS r- r r r �test_startProcess� s z ProcmonTests.test_startProcessc C s2 | j �ddg� | j �d� | �| j �d�� dS )zr L{ProcessMonitor.startProcess} silently returns if the named process is already started. r; N)r6 rA r` � assertIsNoner- r r r � test_startProcessAlreadyStarted s z,ProcmonTests.test_startProcessAlreadyStartedc C s | � t| jjd� dS )zy L{ProcessMonitor.startProcess} raises a C{KeyError} if the given process name isn't recognised. r; N)rU rV r6 r` r- r r r � test_startProcessUnknownKeyError s z-ProcmonTests.test_startProcessUnknownKeyErrorc C s� | j �� | j �ddg� | �d| j j� d }| j jd j_| j�| j j � | j � d� | j�|� | � | j� � | j j d � dS )zm L{ProcessMonitor.stopProcess} immediately sends a TERM signal to the named process. r; r N)r6 rP rA rC rO � transportr! r rQ r9 � stopProcessrK �secondsra )r Z timeToDier r r �"test_stopProcessNaturalTermination s �z/ProcmonTests.test_stopProcessNaturalTerminationc C s� | j �� | j �ddg� | �d| j j� | j�| j j� | j jd j}| j j d |_ | j � d� | j�| j j d � | � d| j j d � | j�d� | � | j�� | j j d � dS )z� L{ProcessMonitor.stopProcess} kills a process which fails to terminate naturally within L{ProcessMonitor.killTime} seconds. r; r g N)r6 rP rA rC rO r rQ r9 rf �killTimer! rg rK ra rh )r r0 r r r �test_stopProcessForcedKill3 s z'ProcmonTests.test_stopProcessForcedKillc C s | � t| jjd� dS )zx L{ProcessMonitor.stopProcess} raises a C{KeyError} if the given process name isn't recognised. r; N)rU rV r6 rg r- r r r � test_stopProcessUnknownKeyErrorJ s z,ProcmonTests.test_stopProcessUnknownKeyErrorc C s& | 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)r6 rA rc rg r- r r r � test_stopProcessAlreadyStoppedR s z+ProcmonTests.test_stopProcessAlreadyStoppedc C s� g }| � tj|j� t�|j� | j�ddg� | j�� | j� d� | � d| jj � | j� | jj � | jj d � d� | �t|�d� |d d }| �|tdg�� dS )zI Getting a complete output line generates a log message. r; r s hello world! r �message�[foo] hello world!N)� addCleanupr �removeObserverr. � addObserverr6 rA rP r rQ rC rO r9 � outReceived� assertEqualsr\ �tuple�r �eventsrn r r r � test_outputReceivedCompleteLine\ s z,ProcmonTests.test_outputReceivedCompleteLinec C s� g }| � tj|j� t�|j� | j�ddg� | j�� | j� d� | � d| jj � | j� | jj � | jj d � d� | �t|�d� |d d }| �t|�d� |d }|�dd�\}}| �|d� | �|td�� d S ) zN Getting invalid UTF-8 results in the repr of the raw message r; r s �hello world! r rn � z[foo]s �hello world!N)rp r rq r. rr r6 rA rP r rQ rC rO r9 rs rt r\ �splitrB )r rw Zmessagesrn �tag�outputr r r �*test_outputReceivedCompleteLineInvalidUTF8r s z7ProcmonTests.test_outputReceivedCompleteLineInvalidUTF8c C s� g }| � tj|j� t�|j� | j�ddg� | j�� | j� d� | � d| jj � | j� | jj � | jj d � d� | �t|�d� | jj d �ttd��� | �t|�d� |d d }| �|tdg�� dS )zM Getting partial line results in no events until process end r; r s hello world!r rn ro N)rp r rq r. rr r6 rA rP r rQ rC rO r9 rs rt r\ r# r r ru rv r r r � test_outputReceivedPartialLine� s z+ProcmonTests.test_outputReceivedPartialLinec C s� | j �ddg� | j �� | j�d� | �d| j j� | j�| j j� | j jd �t t d��� | � d| j j� | j�d� | �d| j j� dS )z� L{ProcessMonitor.connectionLost} should immediately restart a process if it has been running longer than L{ProcessMonitor.threshold} seconds. r; r N) r6 rA rP r rQ rC rO r9 r# r r rH r- r r r �#test_connectionLostLongLivedProcess� s z0ProcmonTests.test_connectionLostLongLivedProcessc C s� | j �ddg� | j �� | j�d� | j �d� | �d| j j� | j jd }| �|� � � | j�| j j d j j � | � |� � � | �d| j j� dS )z� L{ProcessMonitor.connectionLost} cancels a scheduled process killer and deletes the DelayedCall from the L{ProcessMonitor.murder} list. r; r N)r6 rA rP r rQ rg rC Zmurder� assertTrue�activerO rf r! � assertFalserH )r Z delayedCallr r r � test_connectionLostMurderCancel� s �z,ProcmonTests.test_connectionLostMurderCancelc C sj | j �� | j �ddg� | �d| j j� | j jd j�d� | j�| j jd jj � | � d| j j� dS )z� L{ProcessMonitor.connectionLost} removes the corresponding ProcessProtocol instance from the L{ProcessMonitor.protocols} list. r; r N) r6 rP rA rC rO rf r& r rQ r! rH r- r r r �#test_connectionLostProtocolDeletion� s �z0ProcmonTests.test_connectionLostProtocolDeletionc C s� d| j _d| j _| j �� | j �ddg� | �| j jd | j j� | j�| j j d � | j j d � t t d��� | �| j jd | j j� dS )z} L{ProcessMonitor.connectionLost} will wait at least minRestartDelay s and at most maxRestartDelay s r4 � r; r r N)r6 r7 r8 rP rA rK r$ r rQ r9 rO r# r r r- r r r �%test_connectionLostMinMaxRestartDelay� s z2ProcmonTests.test_connectionLostMinMaxRestartDelayc C s� | j �� | j �ddg� | j�| j jd � | �d| j j� | �| j j d | j j � | j jd � t t d��� | �| j j d | j j d � dS )z| L{ProcessMonitor.connectionLost} doubles the restart delay each time the process dies too quickly. r; r r r4 N)r6 rP rA r rQ r9 rC rO rK r$ r7 r# r r r- r r r �&test_connectionLostBackoffDelayDoubles� s z3ProcmonTests.test_connectionLostBackoffDelayDoublesc C s: | j �ddg� | j �� | j�d� | �d| j j� dS )zP L{ProcessMonitor.startService} starts all monitored processes. r; r N)r6 rA rP r rQ rC rO r- r r r �test_startService� s z ProcmonTests.test_startServicec C s� | j �ddg� | j �ddg� | j �� | j�| j j� | �d| j j� | �d| j j� | j�d� | j �� | j�| j j d � | � i | j j� dS )zT L{ProcessMonitor.stopService} should stop all monitored processes. r; Zbarr N) r6 rA rP r rQ r9 rC rO � stopServicerj rK r- r r r �test_stopService s z ProcmonTests.test_stopServicec C sl | 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) r6 rA rP r rQ Z restartAllrR r, �poprt rc r" )r rJ Z myProcessr r r �!test_restartAllRestartsOneProcess s z.ProcmonTests.test_restartAllRestartsOneProcessc C s� | j �ddg� | j �� | j�| j j� | �d| j j� | j�d� | j jd �t t d��� | � | j j d � � � | j �� | �| j j d � � � dS )ze L{ProcessMonitor.stopService} should cancel any scheduled process restarts. r; r r N)r6 rA rP r rQ r9 rC rO r# r r r� Zrestartr� r� r� r- r r r � test_stopServiceCancelRestarts0 s z+ProcmonTests.test_stopServiceCancelRestartsc C sx d| j _d| j _| j �� | j �ddg� | j�d� | j �d� | j�d� | j �� | j�d� | � | j j i � dS )ze L{ProcessMonitor.stopService} should cancel all scheduled process restarts. � r; r � N) r6 r9 r7 rP rA r rQ rg r� rK rO r- r r r �(test_stopServiceCleanupScheduledRestartsE s z5ProcmonTests.test_stopServiceCleanupScheduledRestartsN)#r' r( r) r* r: rE rI rM rN rT rW rY r[ r^ r_ rb rd re ri rk rl rm rx r} r~ r r� r� r� r� r� r� r� r� r� r r r r r2 v s@ r2 c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �DeprecationTestszQ Tests that check functionality that should be deprecated is deprecated. c C s t � | _t| jd�| _dS )z5 Create reactor and process monitor. r3 N)r+ r r r6 r- r r r r: d s zDeprecationTests.setUpc C sx | j �ddg� | j j}| �t|�d� | �� }d}|D ]$}| �|d t� d|d kr8d}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�categoryZtoTuplern Tz no tuple deprecation found:{}N) r6 rA rJ rt r\ � flushWarnings�assertIs�DeprecationWarningr� �formatrB )r Z myprocesses�warningsZ foundToTuple�warningr r r � test_toTuplel s �z DeprecationTests.test_toTuplec C sF | j j}| �|i � | �� }|�d�}| �|d t� | �|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. r r� N)r6 rJ rt r� r� r� r� )r Z myProcessesr� �firstr r r �test_processes� s z DeprecationTests.test_processesc C s2 t �| j� | �� }|D ]}| �|d t� qdS )zO Pickling an L{ProcessMonitor} results in deprecation warnings r� N)�pickle�dumpsr6 r� r� r� )r r� r� r r r � test_getstate� s z DeprecationTests.test_getstateN)r' r( r) r* r: r� r� r� r r r r r� ^ s r� )r* r� Z twisted.trialr Ztwisted.runner.procmonr r Ztwisted.internet.errorr r r Ztwisted.internet.taskr Ztwisted.python.failurer Ztwisted.pythonr Ztwisted.test.proto_helpersr �objectr r+ ZTestCaser2 ZSynchronousTestCaser� r r r r �<module> s E k
Save File
Cancel