File: //lib/python3/dist-packages/twisted/application/runner/test/__pycache__/test_runner.cpython-38.pyc
U
W[�3 � @ s0 d Z ddlmZ ddlmZ ddlZddlmZmZm Z ddl
m Z m Z m
Z
mZmZmZ ddlmZ dd lmZ d
d lmZ d
d lmZmZ d
d
lmZ ddlm Z ddl Z G dd� de j j j!�Z"edd�G dd� de��Z#G dd� de�Z$G dd� de%�Z&G dd� de%�Z'G d d � d e%�Z(G d d � d e%�Z)dS ) z2
Tests for L{twisted.application.runner._runner}.
� )�SIGTERM)�BytesION)�attrib�attrs�Factory)�LogLevel� LogPublisher� LogBeginner�FileLogObserver�FilteringLogObserver�LogLevelFilterPredicate)�
MemoryReactor� )�_runner� )�
ExitStatus)�PIDFile� NonePIDFile)�Runner� )�
DummyFilePathc @ 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!d"�Zd S )$� RunnerTestsz
Tests for L{Runner}.
c C s� t � | _t� | _| �td| j� | �td| j� d| _d�| j��d�| _ t
� | _ t
� | _ t
| j | j �| _t� | _t� | _t| j| jj | j| j�| _| �td| j � | �td| j� d S )N�exit�killi9 z{}
zutf-8�stderr�globalLogBeginner)� DummyExitr � DummyKillr �patchr �pid�format�encode�pidFileContentr �stdoutr �DummyStandardIOZstdio�DummyWarningsModule�warningsr ZglobalLogPublisherr r ��self� r) �M/usr/lib/python3/dist-packages/twisted/application/runner/test/test_runner.py�setUp# s& �zRunnerTests.setUpc C s. t t� d�}|�� | �|jddddg� dS )zD
L{Runner.run} calls the expected methods in order.
��reactor�killIfRequested� startLogging� startReactor�
reactorExitedN)� DummyRunnerr
�run� assertEqual�
calledMethods�r( �runnerr) r) r* �test_runInOrderE s ��zRunnerTests.test_runInOrderc C sP t � }tt� |d�}| �|j� | �|j� |�� | �|j� | �|j� dS )z;
L{Runner.run} uses the provided PID file.
�r- �pidFileN)� DummyPIDFiler r
� assertFalse�entered�exitedr3 �
assertTrue�r( r: r7 r) r) r* �test_runUsesPIDFileW s z RunnerTests.test_runUsesPIDFilec C sT t t| j��}dd� |_tt� |d�}|�� | �| jj t
j � | �| jj d� dS )z�
L{Runner.run} exits with L{ExitStatus.EX_USAGE} and the expected
message if a process is already running that corresponds to the given
PID file.
c S s dS �NTr) r) r) r) r* �<lambda>o � z4RunnerTests.test_runAlreadyRunning.<locals>.<lambda>r9 zAlready running.N)
r r r" Z isRunningr r
r3 r4 r �statusr � EX_CONFIG�messager@ r) r) r* �test_runAlreadyRunningh s
z"RunnerTests.test_runAlreadyRunningc C s6 t t� d�}|�� | �| jjg � | �| jj� dS )z�
L{Runner.killIfRequested} when C{kill} is false doesn't exit and
doesn't indiscriminately murder anyone.
r, N) r r
r. r4 r �callsr< r r> r6 r) r) r* �test_killNotRequestedx s z!RunnerTests.test_killNotRequestedc C sL t t� dd�}|�� | �| jjg � | �| jjtj � | �| jj
d� dS )z�
L{Runner.killIfRequested} when C{kill} is true but C{pidFile} is
L{nonePIDFile} exits with L{ExitStatus.EX_USAGE} and the expected
message; and also doesn't indiscriminately murder anyone.
T)r- r zNo PID file specified.N) r r
r. r4 r rI r rE r �EX_USAGErG r6 r) r) r* � test_killRequestedWithoutPIDFile� s
z,RunnerTests.test_killRequestedWithoutPIDFilec C sd t t| j��}tt� d|d�}|�� | �| jj| j t
fg� | �| j j t
j� | �| j jd� dS )z�
L{Runner.killIfRequested} when C{kill} is true and given a C{pidFile}
performs a targeted killing of the appropriate process.
T�r- r r: N)r r r" r r
r. r4 r rI r r r rE r �EX_OK�assertIdenticalrG r@ r) r) r* � test_killRequestedWithPIDFile� s z)RunnerTests.test_killRequestedWithPIDFilec C sX t td��}dd� }||_tt� d|d�}|�� | �| jjt j
� | �| jj d� dS )z�
L{Runner.killIfRequested} when C{kill} is true and given a C{pidFile}
that it can't read exits with L{ExitStatus.EX_IOERR}.
Nc S s t tjd��d S )NzPermission denied)�OSError�errnoZEACCESr) r) r) r* �read� s z?RunnerTests.test_killRequestedWithPIDFileCantRead.<locals>.readTrM zUnable to read PID file.) r r rS r r
r. r4 r rE r �EX_IOERRrG )r( r: rS r7 r) r) r* �%test_killRequestedWithPIDFileCantRead� s z1RunnerTests.test_killRequestedWithPIDFileCantReadc C sJ t td��}tt� d|d�}|�� | �| jjtj � | �| jj
d� dS )z�
L{Runner.killIfRequested} when C{kill} is true and given a C{pidFile}
containing no value exits with L{ExitStatus.EX_DATAERR}.
rD TrM �Invalid PID file.N� r r r r
r. r4 r rE r �
EX_DATAERRrG r@ r) r) r* �"test_killRequestedWithPIDFileEmpty� s
z.RunnerTests.test_killRequestedWithPIDFileEmptyc C sJ t td��}tt� d|d�}|�� | �| jjtj � | �| jj
d� dS )z�
L{Runner.killIfRequested} when C{kill} is true and given a C{pidFile}
containing a non-integer value exits with L{ExitStatus.EX_DATAERR}.
s ** totally not a number, dude **TrM rV NrW r@ r) r) r* �%test_killRequestedWithPIDFileNotAnInt� s
z1RunnerTests.test_killRequestedWithPIDFileNotAnIntc s� t � }G � fdd�dt�� | �td� � � G �fdd�dt��| �td�� G �fdd�dt��tt� tj |�d �}|�
� | � t � j
�d
� | �� j
d t� | � t �j�d
� | ��jd t� | ��jd jtj � | ��j�� | ��jj|� d S )
z�
L{Runner.startLogging} sets up a filtering observer with a log level
predicate set to the given log level that contains a file observer of
the given type which writes to the given file.
c s e Zd Z� fdd�ZdS )z2RunnerTests.test_startLogging.<locals>.LogBeginnerc s
|� _ d S �N)� observers)r( r\ �r r) r* �beginLoggingTo� s zARunnerTests.test_startLogging.<locals>.LogBeginner.beginLoggingToN)�__name__�
__module__� __qualname__r^ r) r] r) r* r � s r r c s e Zd Zdd� f� fdd� ZdS )z?RunnerTests.test_startLogging.<locals>.MockFilteringLogObserverc S s d S r[ r) )Zeventr) r) r* rC � rD zHRunnerTests.test_startLogging.<locals>.MockFilteringLogObserver.<lambda>c s |� _ |� _t�| |||� d S r[ )�observer�
predicatesr �__init__)r( rb rc ZnegativeObserver��MockFilteringLogObserverr) r* rd � s �zHRunnerTests.test_startLogging.<locals>.MockFilteringLogObserver.__init__N�r_ r` ra rd r) re r) r* rf � s �rf r c s e Zd Z� fdd�ZdS )z:RunnerTests.test_startLogging.<locals>.MockFileLogObserverc s |� _ t�| |t� d S r[ )�outFiler
rd �str)r( rh ��MockFileLogObserverr) r* rd � s zCRunnerTests.test_startLogging.<locals>.MockFileLogObserver.__init__Nrg r) rj r) r* rk � s rk )r- �defaultLogLevel�logFileZfileLogObserverFactoryr r N)r �objectr r r r
r r
r Zcriticalr/ r4 �lenr\ ZassertIsInstancerc r rO rl rb rh )r( rm r7 r) )r rk rf r* �test_startLogging� s@ ��
� � �z RunnerTests.test_startLoggingc C s( t � }t|d�}|�� | �|j� dS )ze
L{Runner.startReactor} with the C{reactor} argument runs the given
reactor.
r, N)r
r r0 r? ZhasRun)r( r- r7 r) r) r* � test_startReactorWithReactor s
z(RunnerTests.test_startReactorWithReactorc C s | � dd� dS )z�
L{Runner.startReactor} ensures that C{whenRunning} is called with
C{whenRunningArguments} when the reactor is running.
� whenRunningr0 N�� _testHookr' r) r) r* � test_startReactorWhenRunning# s z(RunnerTests.test_startReactorWhenRunningc C s | � d� dS )zb
L{Runner.whenRunning} calls C{whenRunning} with
C{whenRunningArguments}.
rr Nrs r' r) r) r* � test_whenRunningWithArguments+ s z)RunnerTests.test_whenRunningWithArgumentsc C s | � d� dS )zf
L{Runner.whenRunning} calls C{reactorExited} with
C{reactorExitedArguments}.
r1 Nrs r' r) r) r* � test_reactorExitedWithArguments3 s z+RunnerTests.test_reactorExitedWithArgumentsNc s� |dkr |}t t� t� t� d�}g � � fdd�}||d�|�|�� i}tf dt� i|��}t||�}|� | �t� �d� | �� d |� dS ) a
Verify that the named hook is run with the expected arguments as
specified by the arguments used to create the L{Runner}, when the
specified caller is invoked.
@param methodName: The name of the hook to verify.
@type methodName: L{str}
@param callerName: The name of the method that is expected to cause the
hook to be called.
If C{None}, use the L{Runner} method with the same name as the
hook.
@type callerName: L{str}
N)�a�b�cc s � � | � d S r[ )�append)� arguments�Z
argumentsSeenr) r* �hookP s z#RunnerTests._testHook.<locals>.hookz {}Argumentsr- r r ) �dictrn r �copyr r
�getattrr4 ro )r( Z
methodNameZ
callerNamer| r~ ZrunnerArgumentsr7 Z
hookCallerr) r} r* rt ; s �
zRunnerTests._testHook)N)r_ r` ra �__doc__r+ r8 rA rH rJ rL rP rU rY rZ rp rq ru rv rw rt r) r) r) r* r s" "
J r T)�frozenc @ s> e Zd ZdZeee�d�Zdd� Zdd� Z dd� Z
d d
� Z d S ) r2 zg
Stub for L{Runner}.
Keep track of calls to some methods without actually doing anything.
)�defaultc C s | j �d� d S )Nr. �r5 r{ r' r) r) r* r. l s zDummyRunner.killIfRequestedc C s | j �d� d S )Nr/ r� r' r) r) r* r/ p s zDummyRunner.startLoggingc C s | j �d� d S )Nr0 r� r' r) r) r* r0 t s zDummyRunner.startReactorc C s | j �d� d S )Nr1 r� r' r) r) r* r1 x s zDummyRunner.reactorExitedN) r_ r` ra r� r r �listr5 r. r/ r0 r1 r) r) r) r* r2 a s r2 c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) r; z]
Stub for L{PIDFile}.
Tracks context manager entry/exit without doing anything.
c C s t �| � d| _d| _d S �NF)r rd r= r> r' r) r) r* rd � s
zDummyPIDFile.__init__c C s
d| _ | S rB )r= r' r) r) r* � __enter__� s zDummyPIDFile.__enter__c C s
d| _ d S rB �r> )r( ZexcTypeZexcValue� tracebackr) r) r* �__exit__� s zDummyPIDFile.__exit__N)r_ r` ra r� rd r� r� r) r) r) r* r; } s r; c @ s" e Zd ZdZdd� Zddd�ZdS )r zs
Stub for L{exit} that remembers whether it's been called and, if it has,
what arguments it was given.
c C s
d| _ d S r� r� r' r) r) r* rd � s zDummyExit.__init__Nc C s | j r
t�|| _|| _d| _ d S rB )r> �AssertionErrorrE rG )r( rE rG r) r) r* �__call__� s
zDummyExit.__call__)N�r_ r` ra r� rd r� r) r) r) r* r � s r c @ s e Zd ZdZdd� Zdd� ZdS )r zv
Stub for L{os.kill} that remembers whether it's been called and, if it has,
what arguments it was given.
c C s
g | _ d S r[ )rI r' r) r) r* rd � s zDummyKill.__init__c C s | j �||f� d S r[ )rI r{ )r( r Zsigr) r) r* r� � s zDummyKill.__call__Nr� r) r) r) r* r � s r c @ s e Zd ZdZdd� ZdS )r$ zQ
Stub for L{sys} which provides L{BytesIO} streams as stdout and stderr.
c C s || _ || _d S r[ )r# r )r( r# r r) r) r* rd � s zDummyStandardIO.__init__N)r_ r` ra r� rd r) r) r) r* r$ � s r$ c @ s e Zd ZdZdd� ZdS )r% zV
Stub for L{warnings} which provides a C{showwarning} method that is a no-op.
c O s dS )z\
Do nothing.
@param args: ignored.
@param kwargs: ignored.
Nr) )�args�kwargsr) r) r* � showwarning� s z DummyWarningsModule.showwarningN)r_ r` ra r� r� r) r) r) r* r% � s r% )*r� �signalr �ior rR �attrr r r Ztwisted.loggerr r r r
r r Ztwisted.test.proto_helpersr
r7 r �_exitr Z_pidfiler r r Z test_pidfiler Ztwisted.trial.unittestZtwistedZtrialZunittestZTestCaser r2 r; rn r r r$ r% r) r) r) r* �<module> s, E