File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_application.cpython-38.pyc
U
W[0� � @ s2 d Z ddlmZmZ ddlZddlZddlZz ddlZW n ek
rP dZY nX ddl m
Z
m Z m Z m
Z
ddlmZmZmZmZ ddlmZ ddlmZ ddlmZmZ dd lmZ dd
l m Z dd l m Z dd l m!Z! dd
l"m#Z# ddl$m%Z% G dd� d�Z&G dd� de%j'�Z(e)ed��r0e�*� Z+e�,� Z-nd Z+Z-G dd� de%j'�Z.G dd� de%j'�Z/G dd� de%j'�Z0G dd� de%j'�Z1G d d � d e%j'�Z2G d d � d ej3�Z4G d d!� d!�Z5G d"d#� d#�Z6G d$d%� d%ej7�Z8G d&d'� d'e%j'�Z9G d(d)� d)e%j'�Z:G d*d+� d+e
j;�Z<G d,d-� d-e!e%j'�Z=G d.d/� d/e%j'�Z>dS )0zU
Tests for L{twisted.application} and its interaction with
L{twisted.persisted.sob}.
� )�absolute_import�divisionN)�service�internet�app�reactors)�
interfaces�defer�protocol�reactor)�sob)�twisted_reactors)�wire�basic)�usage)�NativeStringIO)� platformType)�TwistedModulesMixin)�
MemoryReactor)�unittestc @ s e Zd ZdZdS )�DummyN)�__name__�
__module__� __qualname__� processName� r r �?/usr/lib/python3/dist-packages/twisted/test/test_application.pyr " s r c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) � ServiceTestsc C s$ t �� }|�d� | �|jd� d S �N�hello)r �Service�setName� assertEqual�name��self�sr r r �testName' s
zServiceTests.testNamec C s> t �� }t �� }|�|� | �t|�|g� | �|j|� d S �N)r r � MultiService�setServiceParentr" �list�parent�r% r& �pr r r �
testParent, s
zServiceTests.testParentc C sL t �� }t �d�}|�|� | �tt �|��|g� | �|jt �|�� d S )N� )r r � Applicationr* r" r+ �IServiceCollectionr, r- r r r �testApplicationAsParent3 s
z$ServiceTests.testApplicationAsParentc C sZ t �� }t �� }|�d� |�|� | �t|�|g� | �|j|� | �|�d�|� d S r ) r r r) r! r* r" r+ r, ZgetServiceNamedr- r r r �testNamedChild: s
zServiceTests.testNamedChildc C s8 t �� }t �� }|�d� |�|� | �t|jd� d S )Nr �lala�r r r) r! r* � assertRaises� RuntimeErrorr- r r r �testDoublyNamedChildC s
z!ServiceTests.testDoublyNamedChildc C sJ t �� }t �� }|�d� |�|� t �� }|�d� | �t|j|� d S r r6 r- r r r �testDuplicateNamedChildJ s
z$ServiceTests.testDuplicateNamedChildc C sb t �� }t �� }|�|� | �t|�|g� | �|j|� |�� | �t|�g � | �|j� d S r( ) r r r) r* r" r+ r, �disownServiceParent� assertIsNoner- r r r �
testDisowningS s
zServiceTests.testDisowningc C s@ t �� }| �|j� |�� | �|j� |�� | �|j� d S r( )r r � assertFalse�running� startService�
assertTrue� stopServicer$ r r r � testRunning] s zServiceTests.testRunningc C sv t �� }t �� }|�|� | �|j� | �|j� |�� | �|j� | �|j� |�� | �|j� | �|j� d S r( ) r r r) r* r> r? r@ rA rB r- r r r �testRunningChildren1e s
z!ServiceTests.testRunningChildren1c sZ t �� � � �fdd�}t �� }||_||_t �� }� �|� |�|� |�� |�� d S )Nc s �� � j� d S r( )rA r? r �r& r% r r � checkRunningt s z7ServiceTests.testRunningChildren2.<locals>.checkRunning)r r rB r@ r) r* )r% rF �tr. r rE r �testRunningChildren2r s
z!ServiceTests.testRunningChildren2c C sR t �� }|�� t �� }| �|j� |�|� | �|j� |�� | �|j� d S r( ) r r) r@ r r> r? r* rA r; )r% r. r& r r r �testAddingIntoRunning s
z"ServiceTests.testAddingIntoRunningc sV t �� � � fdd�}|� _t �� }t �� }� �|� |�|� |�� | �� j� d S )Nc s
d� _ d S )N� )�privilegedStartedr �r& r r �pss� s z(ServiceTests.testPrivileged.<locals>.pss)r r �privilegedStartServicer) r* rA rK )r% rM �s1r. r rL r �testPrivileged� s
zServiceTests.testPrivilegedc C s6 t �� }|�� t�|�}| �|j� | �|j� d S r( )r r r@ �copyr> r? rA )r% r& rO r r r � testCopying� s
zServiceTests.testCopyingN)r r r r' r/ r3 r4 r9 r: r= rC rD rH rI rP rR r r r r r % s
r �getuidc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ProcessTestsc C s, t �dd�}| �|jd� | �|jd� d S )N� � )r �Processr" �uid�gid�r% r. r r r �testID� s zProcessTests.testIDc C sn t �d�}| �|jd� | �|j� t jdd�}| �|j� | �|jd� t �� }| �|j� | �|j� d S )NrU )rY )r rW r" rX r< rY rZ r r r � testDefaults� s
zProcessTests.testDefaultsc C s, t �� }| �|j� d|_| �|jd� d S r )r rW r< r r" rZ r r r �testProcessName� s z ProcessTests.testProcessNameN)r r r r[ r\ r] r r r r rT � s rT c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�InterfacesTestsc C s | � tj�t�� �� d S r( )rA r �IService�
providedByr �r% r r r � testService� s zInterfacesTests.testServicec C s0 | � tj�t�� �� | � tj�t�� �� d S r( )rA r r_ r` r) r2 ra r r r �testMultiService� s
�z InterfacesTests.testMultiServicec C s | � tj�t�� �� d S r( )rA r �IProcessr` rW ra r r r � testProcess� s zInterfacesTests.testProcessN)r r r rb rc re r r r r r^ � s r^ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ApplicationTestsc C s( t �d� t �dd� t �ddd� d S �Nr rU rV )r r1 ra r r r �testConstructor� s
z ApplicationTests.testConstructorc C s� t �d�}| �t �|�j� | �t �|�j� t �dd�}| �t �|�jd� | �t �|�j� t �ddd�}| �t �|�jd� | �t �|�jd� d S rg )r r1 r< rd rX rY r" �r% �ar r r �testProcessComponent� s
z%ApplicationTests.testProcessComponentc C sL t �d�}| �t �|�t �|�� | �t �|�jd� | �t �|�j� d S r ) r r1 �assertIsr_ r2 r" r# r< r, ri r r r �testServiceComponent� s
z%ApplicationTests.testServiceComponentc C sB t �d�}t�|�}| �|jd� | �|jd� | �|j|� d S )Nr �pickle) r r1 r � IPersistabler" �styler# rl Zoriginal)r% rj r. r r r �testPersistableComponent� s
z)ApplicationTests.testPersistableComponentN)r r r rh rk rm rq r r r r rf � s rf c @ s e Zd Zdd� ZdS )� LoadingTestsc C s� t �d�}t�|�}d�� D ]>}|�|� |�� t �d|d |�}| �t � |�j
d� q t dd��}|� ddg� W 5 Q R X t �dd �}| �t � |�j
d� d S )
Nr �
source picklezhello.tar z hello.tac�w�(from twisted.application import service
�+application = service.Application('hello')
�python)
r r1 r ro �split�setStyle�save�loadApplicationr" r_ r# �open�
writelines)r% rj r. rp �a1�fr r r �test_simpleStoreAndLoad� s
� z$LoadingTests.test_simpleStoreAndLoadN)r r r r� r r r r rr � s rr c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�AppSupportTestsc C s | � t�d�� d S )Nr )r< r Z
getPassphrasera r r r �testPassphrase s z AppSupportTests.testPassphrasec C s� t �d�}dddd�}d�� D ]b}|�� }d|ddi�||�< t�|��|� t�|�jdd� t �
|d�}| � t � |�j
d� q |�� }d|d < tdd
��}|�d d g� W 5 Q R X t �
|d�}| � t � |�j
d� dS )
zL
Test loading an application file in different dump format.
r N)�file�sourcerw rs Zhelloapplicationrn r� ��filenamerw rt ru rv )r r1 rx rQ �getr ro ry rz r ZgetApplicationr" r_ r# r| r} )r% rj Z
baseconfigrp Zconfigr~ r r r r �testLoadApplication s$
� z#AppSupportTests.testLoadApplicationc
C s� t �d�}d�� D ]h}d�� D ]Z}t�|��|� t�|�jdd� t�d|d d|d� t � d|�}| �
t � |�j d� q qd S )Nr5 rs Z converttestr� zconverttest.outr )
r r1 rx r ro ry rz r Z convertStyler{ r" r_ r# )r% �applZinstyleZoutstyleZappl2r r r �test_convertStyle s
� z!AppSupportTests.test_convertStylec C s, t �d�}t�|d� | �t �|�j� d S )Nr5 r )r r1 r ZstartApplicationrA r_ r? )r% r� r r r �test_startApplication& s
z%AppSupportTests.test_startApplicationN)r r r r� r� r� r� r r r r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Fooc C s | j �d� d S )Ns lalala
)� transport�writera r r r �connectionMade- s zFoo.connectionMadec C s || j _| j�� d S r( )�factory�liner� ZloseConnection)r% r� r r r � lineReceived/ s zFoo.lineReceivedc C s | j j�| j j� d S r( )r� �d�callbackr� �r% �reasonr r r �connectionLost2 s zFoo.connectionLostN)r r r r� r� r� r r r r r� , s r� c @ s e Zd ZdZdd� Zdd� ZdS )�DummyAppNc C s || j |j< d S r( ��servicesr# �r% r r r r �
addService8 s zDummyApp.addServicec C s | j |j= d S r( r� r� r r r �
removeService: s zDummyApp.removeService)r r r r r� r� r r r r r� 6 s r� c @ s e Zd Zdd� Zdd� ZdS )� TimerTargetc C s
g | _ d S r( ��lra r r r �__init__? s zTimerTarget.__init__c C s | j �|� d S r( )r� �append)r% Zwhatr r r r� A s zTimerTarget.appendN)r r r r� r� r r r r r� > s r� c @ s e Zd Zdd� ZdS )�TestEchoc C s | j �d� d S �NT)r� r� r� r r r r� G s zTestEcho.connectionLostN)r r r r� r r r r r� F s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
e � e
d�sjdZee_ee _ee
_dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� ZdS )#�
InternetTestsc s� t �� � � �� t�� }t|_t�� t_t �
d|�}|� � � |j �
� j}t�� }t�� |_t|_d |_t �d||�� � � |j�| jd� |j�� fdd�� |j�dd� � |jS )Nr � 127.0.0.1� lalalac s � � � S r( �rB ��xrL r r �<lambda>] � z'InternetTests.testTCP.<locals>.<lambda>c S s t jS r( �r� r� r� r r r r� ^ r� )r r) r@ r
�
ServerFactoryr� r �Deferredr� r � TCPServerr* �_port�getHost�port�
ClientFactoryr� r� � TCPClient� addCallbackr" )r% r� rG �numr rL r �testTCPN s"
zInternetTests.testTCPc sj t �td�st�d��t�� �t�d��}|� � |j
� � j � | �
� d� � �fdd�}t�|j��|�S )z�
Test L{internet.UDPServer} with a random port: starting the service
should give it valid port, and stopService should free it so that we
can start a server on the same port again.
Nz)This reactor does not support UDP socketsr c s t �� ��}|�� |�� S r( )r � UDPServerr@ rB )�ignoredrG �r� r. r r �onStopo s z&InternetTests.test_UDP.<locals>.onStop)r Z IReactorUDPr r ZSkipTestr
ZDatagramProtocolr r� r@ r� r� r� ZassertNotEqualr Z
maybeDeferredrB r� )r% rG r� r r� r �test_UDPb s
zInternetTests.test_UDPc s� t �� }t|_ t�� t_t�d|��d�_�� � �j
� � j }t �
� }t�� |_t|_ d |_t�d||�� � �� |j�| jd� |j�� fdd�� |j��fdd�� |j�dd� � |jS ) Nr rJ r� r� c s � � � S r( r� r� )�cr r r� � r� z.InternetTests.testPrivileged.<locals>.<lambda>c s � � � S r( r� r� �rG r r r� � r� c S s t jS r( r� r� r r r r� � r� )r
r� r� r r� r� r r� Z
privilegedrN r� r� r� r� r� r� r� r@ r� r" )r% r� r� r )r� rG r rP v s$
z InternetTests.testPrivilegedc sp t �� }tj|_ t�d|�}|�� |j�� j }|�
� t � � � t �
� }� fdd�|_t�d||�}|�� � S )Nr c s
� � d �S r( �r� ��args�r� r r r� � r� z<InternetTests.testConnectionGettingRefused.<locals>.<lambda>r� )r
r� r �Echor r� r@ r� r� r� rB r r� r� �clientConnectionFailedr� )r% r� rG r� r� r r� r � testConnectionGettingRefused� s z*InternetTests.testConnectionGettingRefusedc s� t �� � � �� t�� }t|_t�� t_t �
d|�}|� � � t� � }t
|_t�� |_d |_t �d|�� � � |j�| jd� |j�� fdd�� |j�dd� � |j�| j|� � |jS )N�echo.sktr� c s � � � S r( r� r� rL r r r� � r� z(InternetTests.testUNIX.<locals>.<lambda>c S s t jS r( r� r� r r r r� � r� )r r) r@ r
r� r� r r� r� r �
UNIXServerr* r� r� r� �
UNIXClientr� r" � _cbTestUnix�r% r� rG r rL r �testUNIX� s"
zInternetTests.testUNIXc s\ t �� t_d |_t �� |_� �� |j�| jd� |j�� fdd�� |j�dd� � |jS )Nr� c s � � � S r( r� r� rL r r r� � r� z+InternetTests._cbTestUnix.<locals>.<lambda>c S s t jS r( r� r� r r r r� � r� )r r� r� r� r� r@ r� r" )r% r� r� r& r rL r r� � s
zInternetTests._cbTestUnixc C s� t �� }tj|_ t�d|�}|�� | �|jd � t � |�}| �
|j� |� � | �
|j� | � |j
� t �� }tj|_ t�d|�}|�� | �|jd � t � |�}| �
|j� |� � | �
|j� | � |j
� d S )Nr� )r
r� r r� r r� r@ ZfailIfIdenticalr� rQ r< rB r> r? r� r� Z _connection)r% r� rG �t1r r r � testVolatile� s(
zInternetTests.testVolatilec sf t �� }tj|_ t�d|�}|�� |�� | �|j � t �
� }t � � � � fdd�|_
t�d|� � S )Nr� c s
� � d �S r( r� r� r� r r r� � r� z2InternetTests.testStoppingServer.<locals>.<lambda>)r
r� r r� r r� r@ rB r> r? r� r r� r� r Z connectUNIXr� r r� r �testStoppingServer� s z InternetTests.testStoppingServerNz1This reactor does not support UNIX domain socketsc C sJ t � }t�d|jd�}|�� t�|�}|�� t�|�}| � |j
� d S )NrJ r ) r� r � TimerServicer� r@ rn �dumpsrB �loadsr> r? )r% �targetZt0r& rG r r r �testPickledTimer� s
z InternetTests.testPickledTimerc sb t �� � t�ddd� ���j�� �fdd�}|�_��� � ��fdd�� � ��fdd�� � S )NrJ c S s dd S )NrJ r r r r r r r� � r� z/InternetTests.testBrokenTimer.<locals>.<lambda>c s �| � � � d � d S r( r� )Zwhy)r� � oldFailedr r �_failed� s z.InternetTests.testBrokenTimer.<locals>._failedc s � j S r( r� r� r� r r r� � r� c s � � tgdd� � �t�D ��S )Nc S s g | ] }|j j�qS r )�value� __class__)�.0�or r r �
<listcomp>� s zCInternetTests.testBrokenTimer.<locals>.<lambda>.<locals>.<listcomp>)r" �ZeroDivisionErrorZflushLoggedErrorsr� ra r r r� � s �)r r� r r� r� r@ r� )r% r� r )r� r� r% rG r �testBrokenTimer� s z InternetTests.testBrokenTimerc C s� d� � }|dd� D ]"}ttd| �td�s|�|� q|D ]�}d� � D ]�}|dkr^|dkr^qH|dkrp|dkrpqH| �tt|| �� tt|| �j}dd d
�| }| �tt|| �p�|d ko�|dk� tt|| �� }| � t
� |�|� qHq<dS ) z�
L{twisted.application.internet} dynamically defines a set of
L{service.Service} subclasses that in general have corresponding
reactor.listenXXX or reactor.connectXXX calls.
z'TCP UNIX SSL UDP UNIXDatagram MulticastNZIReactorz
Server ClientZ Multicast�ClientZUDPZlistenZconnect)ZServerr� ) rx �getattrr r �removerA �hasattrr �methodr" r r_ )r% ZtransZtranZsider� �prefixr� r r r �test_everythingThere� s$ �z"InternetTests.test_everythingTherec C s( t jD ] }| �tt |�d|f � qdS )z�
L{twisted.application.internet} dynamically defines L{service.Service}
subclasses. This test ensures that the subclasses exposed by C{__all__}
are valid attributes of the module.
z3%s not importable from twisted.application.internetN)r �__all__rA r� )r% �clsr r r �test_importAll s
�z InternetTests.test_importAllc C sF t � }t� }tjd||d�}|�� | �|j�� dd� d|f� dS )z�
L{internet._AbstractServer} supports a C{reactor} keyword argument
that can be used to parametrize the reactor used to listen for
connections.
�� �r N� )r �objectr r� r@ r" �
tcpServers�pop�r% r r� rG r r r �#test_reactorParametrizationInServer" s
z1InternetTests.test_reactorParametrizationInServerc C sL t � }t�� }tjdd||d�}|�� | �|j�� dd� dd|f� dS )z�
L{internet._AbstractClient} supports a C{reactor} keyword arguments
that can be used to parametrize the reactor used to create new client
connections.
r� r� r� N� ) r r
r� r r� r@ r" �
tcpClientsr� r� r r r �#test_reactorParametrizationInClient0 s �z1InternetTests.test_reactorParametrizationInClientc C sv t � }t�� }tjd||d�}|�� | �|j�� dd� d|f� |� � |�� | �|j�� dd� d|f� dS )z�
Like L{test_reactorParametrizationInServer}, but stop and restart the
service and check that the given reactor is still used.
r� r� Nr� )
r r
ZFactoryr r� r@ r" r� r� rB r� r r r �0test_reactorParametrizationInServerMultipleStart? s z>InternetTests.test_reactorParametrizationInServerMultipleStartc C s| t � }t�� }tjdd||d�}|�� | �|j�� dd� dd|f� |� � |�� | �|j�� dd� dd|f� dS )z�
Like L{test_reactorParametrizationInClient}, but stop and restart the
service and check that the given reactor is still used.
r� r� r� Nr� )
r r
r� r r� r@ r" r� r� rB r� r r r �0test_reactorParametrizationInClientMultipleStartO s � �z>InternetTests.test_reactorParametrizationInClientMultipleStart)r r r r� r� rP r� r� r� r� r� r Z IReactorUNIXr Z_skipMsg�skipr� r� r� r� r� r� r� r� r r r r r� L s*
r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TimerBasicTestsc sZ t �� }t�d|jd�� _� j�� |�� jd� |�� fdd�� |�� fdd�� |S )NrJ r c s
� j �� S r( �rG rB r� ra r r r� i r� z/TimerBasicTests.testTimerRuns.<locals>.<lambda>c s � � � jj�S r( )r> rG r? r� ra r r r� j r� ) r r� r r� r� rG r@ r� r" )r% r� r ra r �
testTimerRunsd s
z TimerBasicTests.testTimerRunsc C s
| j �� S r( r� ra r r r �tearDownm s zTimerBasicTests.tearDownc s� t �� }t �� � � df|dfg��fdd�}t�d|��_�j�� �fdd�}� �fdd �}�fd
d �}|�|� |�|� |�|� |S ) N�bar�fooc s � � � \} }| �|� d S r( )r� r� )r� �arg)�workr r �triggeru s z1TimerBasicTests.testTimerRestart.<locals>.triggerrJ c s � � | d� � j�� S )Nr� �r" rG rB ��resultra r r �
onFirstResultz s z7TimerBasicTests.testTimerRestart.<locals>.onFirstResultc s �� �jj� �j�� � S r( )r> rG r? r@ )r� )�d2r% r r � onFirstStop} s
z5TimerBasicTests.testTimerRestart.<locals>.onFirstStopc s � � | d� � j�� d S )Nr� r� r ra r r �onSecondResult� s z8TimerBasicTests.testTimerRestart.<locals>.onSecondResult)r r� r r� rG r@ r� )r% Zd1r� r r r r )r r% r� r �testTimerRestartp s
z TimerBasicTests.testTimerRestartc s` g � � fdd�}t �� }t�d|dd|��_�j�� |��jdgd � |��fdd�� |S )Nc s$ � � | � t� �|kr |�� � d S r( )r� �lenr� )�dataZnumberr� r� r r r� � s
z/TimerBasicTests.testTimerLoops.<locals>.triggerg{�G�z�?r �
c s
� j �� S r( r� r� ra r r r� � r� z0TimerBasicTests.testTimerLoops.<locals>.<lambda>)r r� r r� rG r@ r� r" )r% r� r� r )r� r% r �testTimerLoops� s
z TimerBasicTests.testTimerLoopsN)r r r r� r� r r
r r r r r� b s r� c @ s e Zd ZdZdd� ZdS )� FakeReactorz6
A fake reactor with a hooked install method.
c O s t jj| f|�|� || _dS )zv
@param install: any callable that will be used as install method.
@type install: C{callable}
N)r �Reactorr� �install)r% r
r� �kwargsr r r r� � s zFakeReactor.__init__N)r r r �__doc__r� r r r r r � s r c @ sz e Zd ZdZdd� Zdd� Zd dd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � ZdS ) �PluggableReactorTestsz:
Tests for the reactor discovery/inspection APIs.
c C s g | _ g | _tj| _| jt_dS )aI
Override the L{reactors.getPlugins} function, normally bound to
L{twisted.plugin.getPlugins}, in order to control which
L{IReactorInstaller} plugins are seen as available.
C{self.pluginResults} can be customized and will be used as the
result of calls to C{reactors.getPlugins}.
N)� pluginCalls�
pluginResultsr �
getPlugins�originalFunction� _getPluginsra r r r �setUp� s zPluggableReactorTests.setUpc C s | j t_dS )z>
Restore the original L{reactors.getPlugins}.
N)r r r ra r r r r� � s z PluggableReactorTests.tearDownNc C s | j �||f� t| j�S )zy
Stand-in for the real getPlugins method which records its arguments
and returns a fixed result.
)r r� r+ r )r% Z interface�packager r r r � s z!PluggableReactorTests._getPluginsc C st d}t d }d}t�|||�g| _t�� }| �| jtjdfg� |D ] }|j|krD| �|j |� qpqD| �
d� dS )zP
Test that reactor plugins are returned from L{getReactorTypes}
�fakereactortestz .fakereactor� descriptionNz6Reactor plugin not present in getReactorTypes() result) r r r r �getReactorTypesr" r ZIReactorInstaller� shortNamer Zfail)r% r# r r Z reactorTypes�rr r r �test_getPluginReactorTypes� s
�
z0PluggableReactorTests.test_getPluginReactorTypesc sX g � � fdd�}t |dtd�}d|i}| �|� t�ddd�}|�� | �� dg� dS )zy
Test that L{reactors.Reactor.install} loads the correct module and
calls its install attribute.
c s � � d� d S r� �r� r �� installedr r r
� s z?PluggableReactorTests.test_reactorInstallation.<locals>.installr � describedZ fakereactorTN)r r ZreplaceSysModulesr r r
r" )r% r
� fakeReactor�modulesZ installerr r r �test_reactorInstallation� s �
z.PluggableReactorTests.test_reactorInstallationc sJ g � � fdd�}d}t }d}t||||�g| _t�|� | �� dg� dS )zu
Test that the L{reactors.installReactor} function correctly installs
the specified reactor.
c s � � d� d S r� r r r r r r
� s z:PluggableReactorTests.test_installReactor.<locals>.installr r TN�r r r r �installReactorr" �r% r
r# r r r r r �test_installReactor� s
z)PluggableReactorTests.test_installReactorc sL t � � � �fdd�}d}t}d}t||||�g�_t�|�}��|� � dS )zt
Test that the L{reactors.installReactor} function correctly returns
the installed reactor.
c s ddl m} ��| d� � d S )Nr �r r )Ztwistedr Zpatchr) �r r% r r r
s zHPluggableReactorTests.test_installReactorReturnsReactor.<locals>.installr r N)r� r r r r r& rl )r% r
r# r r r r r* r �!test_installReactorReturnsReactor s
z7PluggableReactorTests.test_installReactorReturnsReactorc sb g � � fdd�}d}t }d}t||||�}tdd� d||�}||g| _t�|� | �� dg� d S )
z�
Test that the L{reactors.installReactor} function correctly installs
the specified reactor when there are multiple reactor plugins.
c s � � d� d S r� r r r r r r
s zIPluggableReactorTests.test_installReactorMultiplePlugins.<locals>.installr r c S s d S r( r r r r r r� r� zJPluggableReactorTests.test_installReactorMultiplePlugins.<locals>.<lambda>Z otherreactorTNr% )r% r
r# r r r" Z otherReactorr r r �"test_installReactorMultiplePlugins s �
z8PluggableReactorTests.test_installReactorMultiplePluginsc C s g | _ | �tjtjd� dS )z�
Test that L{reactors.installReactor} raises L{reactors.NoSuchReactor}
when asked to install a reactor which it cannot find.
Z somereactorN)r r7 r Z
NoSuchReactorr& ra r r r � test_installNonExistentReactor% s �z4PluggableReactorTests.test_installNonExistentReactorc C s: dd� }d}t }d}t||||�g| _| �ttj|� dS )z�
Test that L{reactors.installReactor} raises an exception when asked to
install a reactor which doesn't work in this environment.
c S s t d��d S )N�Missing foo bar�� ImportErrorr r r r r
5 s zFPluggableReactorTests.test_installNotAvailableReactor.<locals>.installr r N)r r r r7 r0 r r& r' r r r � test_installNotAvailableReactor0 s z5PluggableReactorTests.test_installNotAvailableReactorc s� g �d� d�G ��fdd�dt jtj�}� �fdd�}t|dtd�g| _|� }|�d dd
g� | ��d � � | ��� � �d � | �|d
d� dS )zx
Test that the reactor selected is installed as soon as possible, ie
when the option is parsed.
zreactor installedzsubcommands loadedc s" e Zd Z� �fdd�Zee�ZdS )zQPluggableReactorTests.test_reactorSelectionMixin.<locals>.ReactorSelectionOptionsc s �� �� dd � fdd�dfgS )N�
subcommandc s � S r( r r ra r r r� J r� zoPluggableReactorTests.test_reactorSelectionMixin.<locals>.ReactorSelectionOptions.subCommands.<locals>.<lambda>ztest subcommandr ra ��SUBCOMMAND_EVENT�executedra r � subCommandsH s
z]PluggableReactorTests.test_reactorSelectionMixin.<locals>.ReactorSelectionOptions.subCommandsN)r r r r6 �propertyr r3 r r �ReactorSelectionOptionsG s r8 c s �� � � d S r( r r )�
INSTALL_EVENTr5 r r r
M s zAPluggableReactorTests.test_reactorSelectionMixin.<locals>.installr r! � --reactorr2 r rJ r N)
r �Optionsr �ReactorSelectionMixinr r r � parseOptionsr" �count)r% r8 r
�optionsr )r9 r4 r5 r �test_reactorSelectionMixin> s �z0PluggableReactorTests.test_reactorSelectionMixinc C sj G dd� dt jtj�}g | _|� }t� |_| �t j|j dddg�}| �
d|j d � | �
d|j d � dS ) z�
Test that the usage mixin exits when trying to use a non existent
reactor (the name not matching to any reactor), giving an error
message.
c @ s e Zd ZdS )z\PluggableReactorTests.test_reactorSelectionMixinNonExistent.<locals>.ReactorSelectionOptionsN�r r r r r r r r8 ` s r8 r: r r2 r �
help-reactorsN) r r; r r< r r �
messageOutputr7 �
UsageErrorr= �assertInr� )r% r8 r? �er r r �%test_reactorSelectionMixinNonExistentZ s �z;PluggableReactorTests.test_reactorSelectionMixinNonExistentc s� G dd� dt jtj�}d� � fdd�}d}t}d}t||||�g| _|� }t� |_| � t j
|j ddd g�}| � � |j
d
� | � d |j
d
� d S )
z�
Test that the usage mixin exits when trying to use a reactor not
available (the reactor raises an error at installation), giving an
error message.
c @ s e Zd ZdS )z]PluggableReactorTests.test_reactorSelectionMixinNotAvailable.<locals>.ReactorSelectionOptionsNrA r r r r r8 r s r8 r. c s t � ��d S r( r/ r ��messager r r
u s zMPluggableReactorTests.test_reactorSelectionMixinNotAvailable.<locals>.installr r r: r2 r rB N)r r; r r< r r r r rC r7 rD r= rE r� )r% r8 r
r# r r r? rF r rH r �&test_reactorSelectionMixinNotAvailablel s �z<PluggableReactorTests.test_reactorSelectionMixinNotAvailable)N)r r r r r r� r r r$ r( r+ r, r- r1 r@ rG rJ r r r r r � s
r c @ sp e Zd ZdZdd� Zdd� Zer&de_dd� Zes8d e_d
d � Z e
d krNd
e _dd� Z e
d krdd
e _dd� Z dS )�HelpReactorsTestsz6
--help-reactors lists the available reactors
c C s6 t �� | _t� | j_| �t| jj� | jj�� | _ dS )z3
Get the text from --help-reactors
N)
r r< r? r rC r7 �
SystemExit�opt_help_reactors�getvaluerI ra r r r r � s
zHelpReactorsTests.setUpc C s, | � tjj| j� | � dtjj | j� dS )zX
--help-reactors should NOT display the asyncio reactor on Python < 3.4
�!N)rE r
�asyncior rI r ra r r r �test_lacksAsyncIO� s z#HelpReactorsTests.test_lacksAsyncIOz$Not applicable, asyncio is availablec C s, | � tjj| j� | �dtjj | j� dS )zU
--help-reactors should display the asyncio reactor on Python >= 3.4
rO N)rE r
rP r rI � assertNotInr ra r r r �test_hasAsyncIO� s
�z!HelpReactorsTests.test_hasAsyncIOz asyncio library not availablec C s, | � tjj| j� | �dtjj | j� dS )zL
--help-reactors should display the iocp reactor on Windows
rO N)rE r
�iocpr rI rR r ra r r r �test_iocpWin32� s z HelpReactorsTests.test_iocpWin32Zwin32z Test only applicable on Windowsc C s, | � tjj| j� | � dtjj | j� dS )zP
--help-reactors should NOT display the iocp reactor on Windows
rO N)rE r
rT r rI r ra r r r �test_iocpNotWin32� s z#HelpReactorsTests.test_iocpNotWin32c C sF dd� }t �� }||_t� |_| �t|j� |j�� }| � d|� dS )z>
--help-reactors with only supported reactors
c s s t jV d S r( )r
�defaultr r r r r � s zEHelpReactorsTests.test_onlySupportedReactors.<locals>.getReactorTypeszreactors not availableN)
r r< Z_getReactorTypesr rC r7 rL rM rN rR )r% r r? rI r r r �test_onlySupportedReactors� s
z,HelpReactorsTests.test_onlySupportedReactorsN)
r r r r r rQ rP r� rS rU r rV rX r r r r rK � s
rK )?r Z
__future__r r rQ �osrn rP r0 Ztwisted.applicationr r r r Ztwisted.internetr r r
r Ztwisted.persistedr Ztwisted.pluginsr
Ztwisted.protocolsr r Ztwisted.pythonr Ztwisted.python.compatr Ztwisted.python.runtimer Z#twisted.python.test.modules_helpersr Ztwisted.test.proto_helpersr Z
twisted.trialr r ZTestCaser r� rS Zcuruid�getgidZcurgidrT r^ rf rr r� Z LineReceiverr� r� r� r� r� r� r� r r r rK r r r r �<module> sT
x
.
5 a