File: //lib/python3/dist-packages/twisted/application/twist/test/__pycache__/test_twist.cpython-38.pyc
U
W[� � @ s� d Z ddlmZ ddlmZmZ ddlmZ ddlm Z m
Z
ddl m Z ddl
mZ dd lmZ dd
lmZ d d lmZ d d
lmZ ddlmZ ddlZG dd� dejjj �Z G dd� dejjj �Z dS )z0
Tests for L{twisted.application.twist._twist}.
� )�stdout)�LogLevel�jsonFileLogObserver)�
MemoryReactor� )�IService� MultiService)�
ExitStatus)�Runner)� DummyExit)�_twist� )� TwistOptions)�Twist)� SignalCapturingMemoryReactorNc @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�
TwistTestsz
Tests for L{Twist}.
c C s | � � d S �N)�patchInstallReactor��self� r �K/usr/lib/python3/dist-packages/twisted/application/twist/test/test_twist.py�setUp s zTwistTests.setUpc C s t � | _| �td| j� dS )zX
Patch L{_twist.exit} so we can capture usage and prevent actual exits.
�exitN)r r �patchr r r r r � patchExit" s zTwistTests.patchExitc s$ i � _ � fdd�}� �td|� dS )zo
Patch C{_options.installReactor} so we can capture usage and prevent
actual installs.
c s t � }|� j|< |S r )r �installedReactors)�_�name�reactorr r r �installReactor1 s
z6TwistTests.patchInstallReactor.<locals>.installReactorr N)r r r )r r r r r r * s z TwistTests.patchInstallReactorc s$ g � _ � fdd�}� �td|� dS )zo
Patch L{MultiService.startService} so we can capture usage and prevent
actual starts.
c s � j �| � d S r )�
serviceStarts�append)�servicer r r � startService@ s z2TwistTests.patchStartService.<locals>.startServicer$ N)r! r r )r r$ r r r �patchStartService9 s z TwistTests.patchStartServicec C s t �ddg�}| �|t� dS )zI
L{Twist.options} given valid arguments returns options.
�twist�webN)r �optionsZassertIsInstancer )r r( r r r �test_optionsValidArgumentsF s z%TwistTests.test_optionsValidArgumentsc C s\ | � � t�ddg� | �| jjtj� | �| jj �
d�� | �| jj � d� t
� ��� dS )z�
L{Twist.options} given invalid arguments exits with
L{ExitStatus.EX_USAGE} and an error/usage message.
r& z--bogus-bagelszError: z
{}N)r r r( ZassertIdenticalr Zstatusr �EX_USAGE�
assertTrue�message�
startswith�endswith�formatr r r r r � test_optionsInvalidArgumentsO s
�z'TwistTests.test_optionsInvalidArgumentsc C s6 t �ddg�}t �|jd |j�}| �t�|�� dS )z:
L{Twist.service} returns an L{IService}.
r& r' N)r r( r# �plugins�
subOptionsr+ r Z
providedBy)r r( r# r r r � test_service_ s zTwistTests.test_servicec C sv t �ddg�}|d }t j|j|j |jd�}| �� t �||� | �| j |g� | �|j
d d |j di fg� dS ) z�
L{Twist.startService} starts the service and registers a trigger to
stop the service when the reactor shuts down.
r& r' r )Zpluginr( ZbeforeZshutdownr N) r r( r# r1 Z
subCommandr2 r% r$ � assertEqualr! ZtriggersZ stopService)r r( r r# r r r �test_startServiceh s
� �z TwistTests.test_startServicec s� g � | � td� fdd�� | � tddd� � t�dddd g�}t�|� | �t� �d
� | �� d t| jd t j
t t d
�� dS )zi
L{Twist.run} runs the runner with arguments corresponding to the given
options.
�__init__c s
� � |�S r )r" )r �args�ZargsSeenr r �<lambda>� � z%TwistTests.test_run.<locals>.<lambda>�runc S s d S r r r r r r r9 � r: r& �--reactor=default�--log-format=jsonr' � r �default�r ZdefaultLogLevelZlogFile�fileLogObserverFactoryN)
r r
r r( r; r4 �len�dictr r �infor r )r Z twistOptionsr r8 r �test_run� s8
� � �
��zTwistTests.test_runc s� | � � g � G � fdd�dt�}| �td|� t�ddddg� | �t| j�d� | �t� �d� | �� d j t
| j d t j
ttd
�� | �� d jd� d S ) zy
L{Twist.main} runs the runner with arguments corresponding to the given
command line arguments.
c s e Zd Z� fdd�Zdd� ZdS )z$TwistTests.test_main.<locals>.Runnerc s || _ d| _� �| � d S )Nr )r7 �runsr" )r �kwargs�Zrunnersr r r6 � s z-TwistTests.test_main.<locals>.Runner.__init__c S s | j d7 _ d S )Nr> )rF r r r r r; � s z(TwistTests.test_main.<locals>.Runner.runN)�__name__�
__module__� __qualname__r6 r; r rH r r r
� s r
r& r< r= r' r> r r? r@ N)r% �objectr r r �mainr4 rB r! r7 rC r r rD r r rF )r r
r rH r � test_main� s, ��� zTwistTests.test_mainN)rI rJ rK �__doc__r r r r% r) r0 r3 r5 rE rN r r r r r s
r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �TwistExitTestszr
Tests to verify that the Twist script takes the expected actions related
to signals and the reactor.
c s: d� _ � fdd�}� �td|� dd� }� �td|� d S )NFc s
d� _ dS )z�
Fake to capture whether L{twisted.application._exitWithSignal
was called.
@param sig: Signal value
@type sig: C{int}
TN)�exitWithSignalCalled)Zsigr r r �fakeExitWithSignal� s z0TwistExitTests.setUp.<locals>.fakeExitWithSignalZ_exitWithSignalc S s dS )z�
Prevent Runner from adding new log observers or other
tests outside this module will fail.
@param _: Unused self param
Nr )r r r r � startLogging� s z*TwistExitTests.setUp.<locals>.startLoggingrS )rQ r r r
)r rR rS r r r r � s
zTwistExitTests.setUpc C s< t � }d|_t� }||d< t|d< t�|� | �| j� dS )zg
_exitWithSignal is not called if the reactor's _exitSignal attribute
is zero.
Nr rA )r � _exitSignalr r r r; � assertFalserQ �r r r( r r r �%test_twistReactorDoesntExitWithSignal� s
z4TwistExitTests.test_twistReactorDoesntExitWithSignalc C s6 t � }t� }||d< t|d< t�|� | �| j� dS )z�
_exitWithSignal is not called if the runner's reactor does not
implement L{twisted.internet.interfaces._ISupportsExitSignalCapturing}
r rA N)r r r r r; rU rQ rV r r r �$test_twistReactorHasNoExitSignalAttr� s
z3TwistExitTests.test_twistReactorHasNoExitSignalAttrc C s< t � }d|_t� }||d< t|d< t�|� | �| j� dS )zb
_exitWithSignal is called if the runner's reactor exits due
to a signal.
r
r rA N)r rT r r r r; r+ rQ rV r r r � test_twistReactorExitsWithSignal s
z/TwistExitTests.test_twistReactorExitsWithSignalN)rI rJ rK rO r rW rX rY r r r r rP � s
rP ) rO �sysr Ztwisted.loggerr r Ztwisted.test.proto_helpersr r# r r Z runner._exitr Zrunner._runnerr
Zrunner.test.test_runnerr r& r Z_optionsr r Ztwisted.test.test_twistdr Ztwisted.trial.unittestZtwistedZtrialZunittestZTestCaser rP r r r r �<module> s .