File: //lib/python3/dist-packages/twisted/application/twist/test/__pycache__/test_twist.cpython-310.pyc
o
�b� � @ s� d Z ddlmZ ddlmZmZmZ ddlZddl m
Z
ddl m Z m
Z
ddlmZ ddlmZ d d
lmZ d d lmZ d d lmZ d d
lmZmZ d dlm Z ddl m Z ddl m Z G dd� dej j!j"�Z#G dd� dej j!j"�Z$dS )z0
Tests for L{twisted.application.twist._twist}.
� )�stdout)�Any�Dict�ListN)� IReactorCore)�LogLevel�jsonFileLogObserver)�
MemoryReactor)� SignalCapturingMemoryReactor� )�
ExitStatus)�Runner)� DummyExit)�IService� MultiService)�_twist� )� TwistOptions)�Twistc @ st e Zd ZdZddd�Zddd�Zddd �Zdd
d �Zdd d
�Zddd�Z ddd�Z
ddd�Z ddd�Z ddd�Z
dS )�
TwistTestsz
Tests for L{Twist}.
�returnNc 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 s2 i � _ dtdtdtf� fdd� }� �td|� dS )zo
Patch C{_options.installReactor} so we can capture usage and prevent
actual installs.
�_�namer c s t � }|� j|< |S r )r �installedReactors)r! r" �reactorr r r �installReactor/ s
z6TwistTests.patchInstallReactor.<locals>.installReactorr% N)r# r �strr r )r r% r r r r ( s z TwistTests.patchInstallReactorc s. g � _ dtddf� fdd� }� �td|� dS )zo
Patch L{MultiService.startService} so we can capture usage and prevent
actual starts.
�servicer Nc s � j �| � d S r )�
serviceStarts�append)r' r r r � startService= s z2TwistTests.patchStartService.<locals>.startServicer* )r( r r r )r r* r r r �patchStartService6 s z TwistTests.patchStartServicec C s t �ddg�}| �|t� dS )zI
L{Twist.options} given valid arguments returns options.
�twist�webN)r �options�assertIsInstancer )r r. r r r �test_optionsValidArgumentsB s z%TwistTests.test_optionsValidArgumentsc C sj | � � t�ddg� | �| jjtj� | �| jj � | �
| jj � d�� | �
| jj � dt
� � ��� 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. �assertIdenticalr �statusr �EX_USAGE�assertIsNotNone�message�
assertTrue�
startswith�endswithr r r r r � test_optionsInvalidArgumentsJ 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�
subOptionsr6 r �
providedBy)r r. r' r r r � test_service^ s zTwistTests.test_servicec C s� t �ddg�}|d }|j}|dusJ �t 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$ N)�pluginr. �before�shutdownr ) r r. �
subCommandr' r: r; r+ r* � assertEqualr( �triggers� stopService)r r. r$ rA r' r r r �test_startServicef s � �z TwistTests.test_startServicec s| g � | � td� fdd�� | � tddd� � t�g d��}t�|� | �t� �d� | �� d t| jd t j
t t d
�� d S ) zi
L{Twist.run} runs the runner with arguments corresponding to the given
options.
�__init__c s
� � |�S r )r) )r �args��argsSeenr r �<lambda>� s
z%TwistTests.test_run.<locals>.<lambda>�runc S s d S r r r r r r rJ � � �r, z--reactor=defaultz--log-format=jsonr- � r �default�r$ �defaultLogLevel�logFile�fileLogObserverFactoryN)
r r
r r. rK rB �len�dictr# r �infor r )r � twistOptionsr rH r �test_run~ s" �
��zTwistTests.test_runc s� | � � g � G � fdd�d�}| �td|� t�g d�� | �t| j�d� | �t� �d� | �� d jt | j
d t j t
td�� | �� d jd� dS ) zy
L{Twist.main} runs the runner with arguments corresponding to the given
command line arguments.
c s, e Zd Zdeddf� fdd� Zddd�ZdS ) z$TwistTests.test_main.<locals>.Runner�kwargsr Nc s || _ d| _� �| � d S )Nr )rG �runsr) )r rY ��runnersr r rF � s z-TwistTests.test_main.<locals>.Runner.__init__c S s | j d7 _ d S )NrN )rZ r r r r rK � s z(TwistTests.test_main.<locals>.Runner.run�r N)�__name__�
__module__� __qualname__r rF rK r r[ r r r
� s r
rM rN r rO rP N)r+ r r r �mainrB rT r( rG rU r# r rV r r rZ )r r
r r[ r � test_main� s" �� zTwistTests.test_mainr] )r^ r_ r` �__doc__r r r r+ r0 r9 r= rE rX rb r r r r r s
r c @ s8 e Zd ZdZd dd�Zd dd�Zd dd �Zd d
d �ZdS )
�TwistExitTestszr
Tests to verify that the Twist script takes the expected actions related
to signals and the reactor.
r Nc sN d� _ dtdd f� fdd� }� �td|� dtdd fdd �}� �td
|� d S ) NF�sigr c s
d� _ dS )z�
Fake to capture whether L{twisted.application._exitWithSignal
was called.
@param sig: Signal value
@type sig: C{int}
TN)�exitWithSignalCalled)re r r r �fakeExitWithSignal� s
z0TwistExitTests.setUp.<locals>.fakeExitWithSignal�_exitWithSignalr! c 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� rL z*TwistExitTests.setUp.<locals>.startLoggingri )rf �intr r r
)r rg ri 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$ rS )r
� _exitSignalr r r rK � assertFalserf �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$ rS N)r r r r rK rl rf rm 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$ rS N)r
rk r r r rK r6 rf rm r r r � test_twistReactorExitsWithSignal� s
z/TwistExitTests.test_twistReactorExitsWithSignalr] )r^ r_ r` rc r rn ro rp r r r r rd � s
rd )%rc �sysr �typingr r r �twisted.trial.unittest�twisted�twisted.internet.interfacesr �twisted.loggerr r �twisted.test.proto_helpersr �twisted.test.test_twistdr
� runner._exitr �runner._runnerr
�runner.test.test_runnerr r' r r r, r �_optionsr r �trial�unittest�TestCaser rd r r r r �<module> s$ $