File: //usr/lib/python3/dist-packages/twisted/runner/test/__pycache__/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