File: //lib/python3/dist-packages/twisted/test/__pycache__/test_tcp.cpython-38.pyc
U
s�@g � @ s� d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m
Z
ddl m Z ddl
mZmZ ddlmZmZmZmZ dd lmZ dd
lmZ dd lmZmZ dd lm Z dd
l m Z ddl m Z ddl!m"Z" dcdd�Z#G dd� dej$�Z%G dd� dej&�Z'G dd� de(�Z)G dd� de)ej&�Z*G dd� de)ej+�Z,G d d � d e j-�Z.G d d � d e j-�Z/G d d!� d!ej0�Z1G d"d#� d#e,�Z2G d$d%� d%e j-�Z3G d&d'� d'e j-�Z4G d(d)� d)ej+�Z5G d*d+� d+e j-�Z6G d,d-� d-ej$�Z7G d.d/� d/ej$�Z8G d0d1� d1ej+�Z9G d2d3� d3e j-�Z:G d4d5� d5ej$�Z;G d6d7� d7ej$�Z<G d8d9� d9ej$�Z=G d:d;� d;e=�Z>G d<d=� d=�Z?G d>d?� d?e j-e?�Z@G d@dA� dAe jA�ZBG dBdC� dCe jC�ZDG dDdE� dEe j-�ZEG dFdG� dGej$�ZFG dHdI� dIej$�ZGG dJdK� dKej+�ZHG dLdM� dMe jA�ZIG dNdO� dOe jC�ZJG dPdQ� dQe j-�ZKe
e�G dRdS� dSe"��ZLG dTdU� dUej&�ZMG dVdW� dWe j-�ZNG dXdY� dYe j-�ZOG dZd[� d[e j-�ZPG d\d]� d]e j-�ZQG d^d_� d_e j-�ZRG d`da� dae j-�ZSz ddlTZTW n eUk
�r� Y n X eT�VeTjW�d db ZXeXe@_YdS )dz.
Tests for implementations of L{IReactorTCP}.
� )�division�absolute_importN��wraps)� implementer)�unittest)�msg�err)�protocol�reactor�defer�
interfaces)�error)� IPv4Address)�IHalfCloseableProtocol�
IPullProducer)�policies)�_PY3)�platform)�AccumulatingProtocolc s\ ddl m} t�� ���fdd�}|�|�� � fdd�}��|� � �|�}|��j� �S )z�
Poor excuse for an event notification helper. This polls a condition and
calls back a Deferred when it is seen to be true.
Do not use this function.
r )�taskc s �� } | r� � | � d S �N��callback��res)�d� predicate� �7/usr/lib/python3/dist-packages/twisted/test/test_tcp.py�check% s zloopUntil.<locals>.checkc s � � � | S r )�stop)�result)�callr r r! * s zloopUntil.<locals>.stop) �twisted.internetr r �DeferredZ LoopingCall� addCallback�startZ
addErrbackZerrback)r Zintervalr r r! �d2r )r# r r r � loopUntil s
r) c @ s e Zd Zdd� Zdd� ZdS )�ClosingProtocolc C s t d� | j�� d S )N� ClosingProtocol.connectionMade�r � transport�loseConnection��selfr r r �connectionMade6 s r+ c C s t d� |�tj� d S )N� ClosingProtocol.connectionLost)r �trapr �ConnectionDone�r0 �reasonr r r �connectionLost: s r2 N��__name__�
__module__� __qualname__r1 r7 r r r r r* 4 s r* c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�ClosingFactoryz/
Factory that closes port immediately.
Nc C s | j �� | _t� S r )�port�
stopListening�
_cleanerUpperr* )r0 Zconnr r r �
buildProtocolG s z ClosingFactory.buildProtocolc C s | j dkr| j�� S | j S )zL
Clean-up for tests to wait for the port to stop listening.
N)r? r= r> r/ r r r �cleanUpL s
zClosingFactory.cleanUp)r9 r: r; �__doc__r? r@ rA r r r r r< @ s r< c @ s4 e Zd ZdZeZdZdZdZdZ dd� Z
dd� Z dS )�MyProtocolFactoryMixina
Mixin for factories which create L{AccumulatingProtocol} instances.
@type protocolFactory: no-argument callable
@ivar protocolFactory: Factory for protocols - takes the place of the
typical C{protocol} attribute of factories (but that name is used by
this class for something else).
@type protocolConnectionMade: L{None} or L{defer.Deferred}
@ivar protocolConnectionMade: When an instance of L{AccumulatingProtocol}
is connected, if this is not L{None}, the L{Deferred} will be called
back with the protocol instance and the attribute set to L{None}.
@type protocolConnectionLost: L{None} or L{defer.Deferred}
@ivar protocolConnectionLost: When an instance of L{AccumulatingProtocol}
is created, this will be set as its C{closedDeferred} attribute and
then this attribute will be set to L{None} so the L{defer.Deferred} is
not used by more than one protocol.
@ivar protocol: The most recently created L{AccumulatingProtocol} instance
which was returned from C{buildProtocol}.
@type called: C{int}
@ivar called: A counter which is incremented each time C{buildProtocol}
is called.
@ivar peerAddresses: A C{list} of the addresses passed to C{buildProtocol}.
Nr c C s
g | _ d S r )�
peerAddressesr/ r r r �__init__z s z MyProtocolFactoryMixin.__init__c C s@ | j �|� | jd7 _| �� }| |_| j|_d| _|| _|S )zi
Create a L{AccumulatingProtocol} and set it up to be able to perform
callbacks.
� N)rD �append�called�protocolFactory�factory�protocolConnectionLost�closedDeferredr
�r0 �addr�pr r r r@ ~ s z$MyProtocolFactoryMixin.buildProtocol) r9 r: r; rB r rI �protocolConnectionMaderK r
rH rE r@ r r r r rC V s rC c @ s e Zd ZdZdS )�MyServerFactoryzI
Server factory which creates L{AccumulatingProtocol} instances.
N)r9 r: r; rB r r r r rQ � s rQ c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d S ) �MyClientFactoryzI
Client factory which creates L{AccumulatingProtocol} instances.
r c C s" t �| � t�� | _t�� | _d S r )rC rE r r% �deferred� failDeferredr/ r r r rE � s
zMyClientFactory.__init__c C s d| _ || _| j�d � d S �NrF )�failedr6 rT r �r0 � connectorr6 r r r �clientConnectionFailed� s z&MyClientFactory.clientConnectionFailedc C s || _ | j�d � d S r )�
lostReasonrS r rW r r r �clientConnectionLost� s z$MyClientFactory.clientConnectionLostc C s
d| _ d S rU ��stoppedr/ r r r � stopFactory� s zMyClientFactory.stopFactoryN)
r9 r: r; rB rV r] rE rY r[ r^ r r r r rR � s rR c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�ListeningTestsc C s8 t � }tjd|dd�}| �|j� | �tj�|�� dS )zf
L{IReactorTCP.listenTCP} returns an object which provides
L{IListeningPort}.
r � 127.0.0.1�Z interfaceN) rQ r � listenTCP�
addCleanupr> �
assertTruer
ZIListeningPort�
providedBy�r0 �f�p1r r r � test_listen� s zListeningTests.test_listenc sH t � � tjd� dd�}|�� j�� �fdd�}t�|j�}|�|� |S )a
The L{IListeningPort} returned by L{IReactorTCP.listenTCP} can be
stopped with its C{stopListening} method. After the L{Deferred} it
(optionally) returns has been called back, the port number can be bound
to a new server.
r r` ra c s t j�� dd�}|�� S )Nr` ra )r rb r> )�ignoredr= �rg �nr r �cbStopListening� s z9ListeningTests.testStopListening.<locals>.cbStopListening) rQ r rb �getHostr= r �
maybeDeferredr> r& )r0 r= rm r r rk r �testStopListening� s
z ListeningTests.testStopListeningc C s t � }tjd|dd�}|�� S �Nr r` ra )rQ r rb r> rf r r r �testNumberedInterface� s z$ListeningTests.testNumberedInterfacec s^ t � }t�d|�� t� �� j����t� ����dk� � ��fdd�}t �
� j �}|� |�S )Nr ���c s �� t� ����dk� d S )Nrs )� assertFalse�repr�find�Zign�rO ZportNor0 r r �stoppedListening� s z5ListeningTests.testPortRepr.<locals>.stoppedListening)
rQ r rb �strrn r= rt ru rv r ro r> r& )r0 rg ry r r rx r � testPortRepr� s zListeningTests.testPortReprc s� t � }t�� }|_t�d|�� ��� j� t� }t�� }|_t� d� �
� j |�}��|j � � �fdd�}t�
||g��|�S )z�
Check that the repr string of the server transport get the good port
number if the server listens on 0.
r r` c sB | \}}� � � j}��t|j�d|f � |j�� |j�� d S )Nz <AccumulatingProtocol #0 on %s>)rn r= � assertEqualru r- r. )r" � serverProto� clientProto�
portNumber�r= r0 r r r � s
�
z-ListeningTests.test_serverRepr.<locals>.check)rQ r r% rP r rb rc r> rR �
connectTCPrn r= �
disconnect�
gatherResultsr& )r0 �server�serverConnMade�client�clientConnMaderX r r r� r �test_serverRepr� s � �z ListeningTests.test_serverReprc sV t � �tjd�dd�����j� � ���fdd�}dd� � t��j�}|�|� |S )z�
Stop and then try to restart a L{tcp.Port}: after a restart, the
server should be able to handle client connections.
r r` ra c sZ �� � t� }t�� �_t�� |_t�d��� j|�}�� |j
� t� �j|jg�� � �S �Nr` )
�startListeningrR r r% rP r r� rn r= rc r� r� r& �rj r� rX ��closer= r0 �
serverFactoryr r rm s
� ��z=ListeningTests.test_restartListening.<locals>.cbStopListeningc S s | \}}|j �� |j �� d S r �r- r. �r" r} r~ r r r r� s
z3ListeningTests.test_restartListening.<locals>.close)rQ r rb rc r> r ro r& )r0 rm r r r� r �test_restartListening� s
z$ListeningTests.test_restartListeningc C s4 t � }dd� }||_tjd|dd�}| �|�� t�S )z�
If the server factory raises an exception in C{stopFactory}, the
deferred returned by L{tcp.Port.stopListening} should fail with the
corresponding error.
c S s t d��d S �NzAn error�� RuntimeErrorr r r r �raiseException$ s z8ListeningTests.test_exceptInStop.<locals>.raiseExceptionr r` ra )rQ r^ r rb �
assertFailurer> r� �r0 r� r� r= r r r �test_exceptInStop s
z ListeningTests.test_exceptInStopc s` t � �dd� }|�_tjd�dd�����j� � ���fdd�}dd � � ����� t��|�S )
z�
Even if the server factory raise an exception in C{stopFactory}, the
corresponding C{tcp.Port} instance should be in a sane state and can
be restarted.
c S s t d��d S r� r� r r r r r� 3 s z>ListeningTests.test_restartAfterExcept.<locals>.raiseExceptionr r` ra c s^ �` ��� t� }t�� �_t�� |_t�d��� j |�}��
|j � t� �j|jg��
� �S r� )r^ r� rR r r% rP r r� rn r= rc r� r� r& r� r� r r rm 9 s
� ��z?ListeningTests.test_restartAfterExcept.<locals>.cbStopListeningc S s | \}}|j �� |j �� d S r r� r� r r r r� G s
z5ListeningTests.test_restartAfterExcept.<locals>.close) rQ r^ r rb rc r> r� r� r& )r0 r� rm r r� r �test_restartAfterExcept, s �z&ListeningTests.test_restartAfterExceptc sj t � }tjd|dd�}|�� j}|�d� t� � t�� |_ t�� � _ t�
d|� � � fdd�}� j � |�S )z�
If C{connectionLost} is called directly on a port object, it succeeds
(and doesn't expect the presence of a C{deferred} attribute).
C{connectionLost} is called by L{reactor.disconnectAll} at shutdown.
r r` ra Nc s � j �tj� d S r �r6 r3 r �ConnectionRefusedErrorrw �r� r r r ` s z;ListeningTests.test_directConnectionLostCall.<locals>.check)
rQ r rb rn r= r7 rR r r% rP r� rT r& )r0 r� r= r r r r� r � test_directConnectionLostCallP s
z,ListeningTests.test_directConnectionLostCallc C s8 t � }dd� }||_tjd|dd�}| �t|jd� dS )a
If C{connectionLost} is called directory on a port object and that the
server factory raises an exception in C{stopFactory}, the exception is
passed through to the caller.
C{connectionLost} is called by L{reactor.disconnectAll} at shutdown.
c S s t d��d S r� r� r r r r r� n s zFListeningTests.test_exceptInConnectionLostCall.<locals>.raiseExceptionr r` ra N)rQ r^ r rb � assertRaisesr� r7 r� r r r � test_exceptInConnectionLostCalle s
z.ListeningTests.test_exceptInConnectionLostCallN)
r9 r: r; ri rp rr r{ r� r� r� r� r� r� r r r r r_ � s $r_ c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�
LoopbackTestsz$
Test loopback connections.
c sb t � }tjd|dd���|_��|j� ��� j}t� � t�d|� � � ��fdd�}� j �
|�S )zz
A port created with L{IReactorTCP.listenTCP} can be connected to with
L{IReactorTCP.connectTCP}.
r r` ra c s, �� � jj� �� �j� � j�tj� d S r )rd r
�madeZ disconnectedrZ r3 r r4 ��x��clientFr= r0 r r r � s z<LoopbackTests.test_closePortInProtocolFactory.<locals>.check) r< r rb r= rc rA rn rR r� rS r& )r0 rg r r r r� r � test_closePortInProtocolFactoryz s
z-LoopbackTests.test_closePortInProtocolFactoryc C s t |ddd� �tj� d S )Nr3 c S s d S r r r� r r r �<lambda>� � z,LoopbackTests._trapCnxDone.<locals>.<lambda>)�getattrr r4 )r0 �objr r r � _trapCnxDone� s zLoopbackTests._trapCnxDonec
s� t � }t�� }||_tjd|dd�}| �|j� |�� j }t
� }t�� }||_t� d||� t� ||g�}� fdd�} |�
| � |S )z�
Invoke the given callback with a client protocol and a server protocol
which have been connected to each other.
r r` ra c s* | \}}� ||� |j �� |j �� d S r r� �r" �serverProtocol�clientProtocolr r r � connected� s
z>LoopbackTests._connectedClientAndServerTest.<locals>.connected)rQ r r% rP r rb rc r> rn r= rR r� r� r& )
r0 r r� r� r= r r� r� Z connsMader� r r r � _connectedClientAndServerTest� s
z+LoopbackTests._connectedClientAndServerTestc s � fdd�}� � |�S )a
The transport of a protocol connected with L{IReactorTCP.connectTCP} or
L{IReactor.TCP.listenTCP} can have its I{TCP_NODELAY} state inspected
and manipulated with L{ITCPTransport.getTcpNoDelay} and
L{ITCPTransport.setTcpNoDelay}.
c s\ | |fD ]N}|j }� �|�� d� |�d� � �|�� d� |�d� � �|�� d� qd S �Nr rF )r- r| Z
getTcpNoDelayZ
setTcpNoDelay�r� r� rO r- r/ r r r � s
z,LoopbackTests.test_tcpNoDelay.<locals>.check�r� �r0 r r r/ r �test_tcpNoDelay� s z LoopbackTests.test_tcpNoDelayc s � fdd�}� � |�S )a
The transport of a protocol connected with L{IReactorTCP.connectTCP} or
L{IReactor.TCP.listenTCP} can have its I{SO_KEEPALIVE} state inspected
and manipulated with L{ITCPTransport.getTcpKeepAlive} and
L{ITCPTransport.setTcpKeepAlive}.
c s\ | |fD ]N}|j }� �|�� d� |�d� � �|�� d� |�d� � �|�� d� qd S r� )r- r| ZgetTcpKeepAliveZsetTcpKeepAliver� r/ r r r � s
z.LoopbackTests.test_tcpKeepAlive.<locals>.checkr� r� r r/ r �test_tcpKeepAlive� s z LoopbackTests.test_tcpKeepAlivec s0 t � � tjdd� dd� � fdd�}� j�|�S )Nr` �E � )Ztimeoutc s � j �tj� d S r r� �rj �r� r r r � s z(LoopbackTests.testFailing.<locals>.check)rR r r� rT r& r� r r� r � testFailing� s zLoopbackTests.testFailingc s� g �t d�D ]*}t�� }|�d� |�d� ��|� q t��� t�t tj
�� � ���fdd���� }�� |t j
� �fdd�}|�|� �fdd �}|�|� |S )
z�
Assert that the error number of the ConnectionRefusedError is
ECONNREFUSED, and not some other socket related error.
�
�r` r rF c sF ���fdd�} �� � }|�� \}}|�� � �||�}|�| � |S )Nc s" | j �� �r�� S � �d� dS )z\
Darn. Kill it and try again, if there are any tries left.
z?Could not fail to connect - could not test errno for that case.N)r- r. Zfail��proto)r0 �
serverSockets�tryConnectFailurer r r� � s
z]LoopbackTests.test_connectionRefusedErrorNumber.<locals>.tryConnectFailure.<locals>.connected)�popZ getsocknamer� r� r& )r� � serverSocket�
serverHost�
serverPortZconnectDeferred��
clientCreatorr0 r� r� r r r� � s
zJLoopbackTests.test_connectionRefusedErrorNumber.<locals>.tryConnectFailurec s � � | jtj� d S r )r| ZosError�errnoZ ECONNREFUSED)�excr/ r r � connRefused s zDLoopbackTests.test_connectionRefusedErrorNumber.<locals>.connRefusedc s � r� � � �� q | S r )r� r� �� passthrough)r� r r �cleanup s z@LoopbackTests.test_connectionRefusedErrorNumber.<locals>.cleanup)�range�socketZbindZlistenrG �randomZshuffler
�
ClientCreatorr �Protocolr� r r� r& �addBoth)r0 �ir� ZrefusedDeferredr� r� r r� r �!test_connectionRefusedErrorNumber� s
z/LoopbackTests.test_connectionRefusedErrorNumberc C s | � tjtjddt� � dS )zu
Connecting to a named service which does not exist raises
L{error.ServiceNameUnknownError}.
r` ZthisbetternotexistN)r� r ZServiceNameUnknownErrorr r� rR r/ r r r �test_connectByServiceFail s �z'LoopbackTests.test_connectByServiceFailc s� t � �t�� }|�_tjd�dd�}��|j� |�� j � t
� }t�� }||_� fdd�}�� t d|� t�
dd|� t�||g�}��fdd �}|�|� |S )
z�
L{IReactorTCP.connectTCP} accepts the name of a service instead of a
port number and connects to the port number associated with that
service, as defined by L{socket.getservbyname}.
r r` ra c s | dkr|dkr� S dS )N�httpZtcpr� r )Z serviceNameZ protocolName)r r r �fakeGetServicePortByName5 s zELoopbackTests.test_connectByService.<locals>.fakeGetServicePortByNameZ
getservbynamer� c s. | \}}� � �jd� |j�� |j�� d S )Nz7Server factory was not called upon to build a protocol.)rd rH r- r. r� )r0 r� r r r� > s �
z6LoopbackTests.test_connectByService.<locals>.connected)rQ r r% rP r rb rc r> rn r= rR Zpatchr� r� r� r& )r0 r� r= �
clientFactoryr� r� ZconnMader� r )r r0 r� r �test_connectByService% s
z#LoopbackTests.test_connectByServiceN)
r9 r: r; rB r� r� r� r� r� r� r� r� r� r r r r r� v s B r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�StartStopFactoryr c C s | j s | jrt�d| _ d S rU ��startedr] r� r/ r r r � startFactoryN s z StartStopFactory.startFactoryc C s | j r | jrt�d| _d S rU r� r/ r r r r^ S s z StartStopFactory.stopFactoryN)r9 r: r; r� r] r� r^ r r r r r� I s r� c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �ClientStartStopFactoryr c O s t j| f|�|� t�� | _d S r )rR rE r r% � whenStopped)r0 �a�kwr r r rE ^ s z ClientStartStopFactory.__init__c C s | j s | jrt�d| _ d S rU r� r/ r r r r� b s z#ClientStartStopFactory.startFactoryc C s& | j r | jrt�d| _| j�d� d S )NrF T)r� r] r� r� r r/ r r r r^ g s z"ClientStartStopFactory.stopFactoryN)r9 r: r; r� r] rE r� r^ r r r r r� Y s
r� c @ s e Zd ZdZdd� Zdd� ZdS )� FactoryTestszTests for factories.c s� t � � tjd� dd�}��|j� ��� j� jfd� tjd� dd�}tjd� dd����� j� jfd� t� |j�}t� |j�}t�
||g�}� ��fdd�}|� |� � �fdd�}|� |� |S ) a
The factory passed to L{IReactorTCP.listenTCP} should be started only
when it transitions from being used on no ports to being used on one
port and should be stopped only when it transitions from being used on
one port to being used on no ports.
r r` ra �rF r c s �� � j� jfd� ��� S )Nr� )r| r� r] r> r� �rg Zp3r0 r r �cbClosed� s z3FactoryTests.test_serverStartStop.<locals>.cbClosedc s �� � j� jfd� d S )N)rF rF )r| r� r] r� �rg r0 r r � cbClosedAll� s z6FactoryTests.test_serverStartStop.<locals>.cbClosedAll) r� r rb rc r> r| r� r] r ro r� r& )r0 rh Zp2�d1r( rL r� r� r r� r �test_serverStartStopq s
z!FactoryTests.test_serverStartStopc sb t � }tjd|dd�}||_| �|j� |�� j}t� � t�d|� � | � � j
� t � fdd��S )z�
The factory passed to L{IReactorTCP.connectTCP} should be started when
the connection attempt starts and stopped when it is over.
r r` ra c s � j S r r\ r �rJ r r r� � r� z3FactoryTests.test_clientStartStop.<locals>.<lambda>) r< r rb r= rc rA rn r� r� rd r� r) )r0 rg rO r r r� r �test_clientStartStop� s
z!FactoryTests.test_clientStartStopN)r9 r: r; rB r� r� r r r r r� n s +r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �CannotBindTestsz]
Tests for correct behavior when a reactor cannot bind to the required TCP
port.
c C sz t � }tjd|dd�}| �|j� |�� j}|�� }| �|jd� | �|j d� | �|j|� | j
t j tj||dd� dS )z�
L{IReactorTCP.listenTCP} raises L{error.CannotListenError} if the
address to listen on is already in use.
r r` ra �TCPN)
rQ r rb rc r> rn r= r| �type�hostr� r ZCannotListenError)r0 rg rh rl �destr r r �test_cannotBind� s
�z CannotBindTests.test_cannotBindc s t ��� �fdd��}|S )zGReturns closure that when called calls f and then callbacks d.
c s �| |�}� � d� |S )N� r )�argsr� Zrtn�r rg r r �newf� s
z/CannotBindTests._fireWhenDoneFunc.<locals>.newfr )r0 r rg r� r r� r �_fireWhenDoneFunc� s z!CannotBindTests._fireWhenDoneFuncc s� t �� }t� }��||j�|_tjd|dd�����j� ���fdd�}��fdd�������fdd ���fd
d ���fd d
��� �fdd���fdd�� |� |� |S )z�
L{IReactorTCP.connectTCP} calls C{Factory.clientConnectionFailed} with
L{error.ConnectBindError} if the bind address specified is already in
use.
r r` ra c sF t �� }t� }��||j�|_tjd��� j|dd� |� � |� |S )Nr` r� �Z bindAddress)
r r% rR r� r@ r r� rn r= r& )�resultsr �cf1)�_conmaderO r0 r r � _connect1� s � z2CannotBindTests.test_clientBind.<locals>._connect1c s, t �� }��||jj�|j_|�� |� |S r )r r% r� r
r1 r& )r� r� r )�_check1connect2r0 r r r� � s � z1CannotBindTests.test_clientBind.<locals>._conmadec s� �� |jjd� t�� }t�� }|jj�� j}t� }�� ||j
�|_
�� ||j �|_ t j
d��� j|d|fd� |�� ||� |��||� t�||g�}|��||� |S )NrF r` r� )r| r
r� r r% r- rn r= rR r� rY r^ r r� r& Z DeferredList)r� r� r� r( r= �cf2Zdl)�
_check2failed�_check2stopped�_stoprO r0 r r r� � s$ ��z8CannotBindTests.test_clientBind.<locals>._check1connect2c s4 � � |jd� |j�tj� � �|j�tj�� | S rU )r| rV r6 r3 r ZConnectBindErrorrd r �r� r� r� r/ r r r� s z6CannotBindTests.test_clientBind.<locals>._check2failedc s � � |jd� | S rU �r| r] r r/ r r r s z7CannotBindTests.test_clientBind.<locals>._check2stoppedc s4 t �� }|�� |� ��||j�|_|jj�� |S r )r r% r& r� r^ r
r- r. )r� r� r� r )�_check1cleanupr0 r r r s
z.CannotBindTests.test_clientBind.<locals>._stopc s � � |jd� d S rU r )r� r� r/ r r r s z7CannotBindTests.test_clientBind.<locals>._check1cleanup)
r r% rQ r� r� r rb rc r> r& )r0 Z theDeferredZsfr� r )r r� r� r r� r rO r0 r �test_clientBind� s
z CannotBindTests.test_clientBindN)r9 r: r; rB r� r� r r r r r r� � s r� c @ s e Zd Zdd� ZdS )�MyOtherClientFactoryc C s || _ t� | _| jS r )�addressr r
)r0 r r r r r@ s z"MyOtherClientFactory.buildProtocolN)r9 r: r; r@ r r r r r s r c @ s e Zd ZdZdd� ZdS )�LocalRemoteAddressTestszo
Tests for correct getHost/getPeer values and that the correct address is
passed to buildProtocol.
c s� t � }t�� |_|j�tjd|dd�����j� ��� j }t
� � t�� }� _ t� d|� ��� ��fdd�}|�
|� ��fdd�}|�
|� |S )z
L{IListeningPort.getHost} returns the same address as a client
connection's L{ITCPTransport.getPeer}.
r r` ra c s0 �� ��� g� j� �� ��� � jj�� � d S r )r| rn rD r
r- �getPeerr� )r� r= r0 r r r 8 s
�z7LocalRemoteAddressTests.test_hostAddress.<locals>.checkc s � � � �S r )r� r� )rX �serverConnectionLostr r r� > s z9LocalRemoteAddressTests.test_hostAddress.<locals>.cleanup)rQ r r% rK r rb rc r> rn r= rR rP r� r& )r0 r� rl Z onConnectionr r� r )r� rX r= r0 r
r �test_hostAddress( s
z(LocalRemoteAddressTests.test_hostAddressN)r9 r: r; rB r r r r r r # s r c @ s e Zd Zdd� ZdS )�WriterProtocolc C s� | j �d� ddddg}| j �|� | j �� }|jdkrNtd|f � d| j_| j �� }|jdkrxtd |f � d| j_d| j_ | j �
� d S )
N� Hello Cleveland!
� Goodbye� cruel� world�
r� z#getPeer returned non-TCP socket: %srF z#getHost returned non-TCP socket: %s) r- �writeZ
writeSequencer r� r rJ �problemrn �doner. )r0 �seqZpeer�usr r r r1 K s
z WriterProtocol.connectionMadeN�r9 r: r; r1 r r r r r J s r c @ s e Zd Zdd� Zdd� ZdS )�ReaderProtocolc C s | j j|7 _d S r )rJ �data�r0 r r r r � dataReceiveda s zReaderProtocol.dataReceivedc C s d| j _d S rU �rJ r r5 r r r r7 c s z ReaderProtocol.connectionLostN�r9 r: r; r r7 r r r r r ` s r c @ s e Zd Zdd� Zdd� ZdS )�WriterClientFactoryc C s d| _ d| _d S )Nr r� )r r r/ r r r rE g s z WriterClientFactory.__init__c C s t � }| |_|| _|S r )r rJ r
rM r r r r@ j s z!WriterClientFactory.buildProtocolN�r9 r: r; rE r@ r r r r r f s r c @ s e Zd ZdZdd� Zdd� ZdS )�WriteDataTestszw
Test that connected TCP sockets can actually write data. Try to exercise
the entire ITransport interface.
c s� t �� �t�_ d�_d�_t��}tjd|dd�}��|j � |�
� j }t � � t� �}t�
d||� � ��fdd�}t�|j|jg�}|�|�S )z�
L{ITCPTransport.write} and L{ITCPTransport.writeSequence} send bytes to
the other end of the connection.
r r` ra c sX �� �jd� �� �jdkd� �� � jd� d�dddd d
g�}�� � j|kd � d S ) N�&writer didn't finish, it probably diedr zwriter indicated an error�$client didn't see connection droppedr� r
r r r r z.client didn't receive all the data it expected)rd r r �joinr )rj Zexpected�r� rg r0 r r r � s � � �z)WriteDataTests.test_writer.<locals>.check)r
�Factoryr r r � WiredFactoryr rb rc r> rn r= r r� r r� � onDisconnectr& )r0 �wrappedFrO rl �wrappedClientFr r r r$ r � test_writerv s
�zWriteDataTests.test_writerc s t jjdkrt�d��t�� �t�� �G ��fdd�dtj�}t� � }||_t j
d|dd�}| � |j � |�
� }tt�G dd � d t��� G � ��fd
d �d tj�}t� �|�_�j}�fd d
�}|�|� td|j|jf � t �|j|j�� td� | �|tj�S )a$
A TCP transport which is written to after the connection has been shut
down should notify its protocol that the connection has been lost, even
if the TCP transport is not actively being monitored for read events
(ie, pauseProducing was called on it).
Z IOCPReactora' iocpreactor does not, in fact, stop reading immediately after pauseProducing is called. This results in a bonus disconnection notification. Under some circumstances, it might be possible to not receive this notifications (specifically, pauseProducing, deliver some data, proceed with this test).c s( e Zd ZdZ� fdd�Z�fdd�ZdS )zJWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecterz�
Protocol for the server side of the connection which disconnects
itself in a callback on clientPaused and publishes notification
when its connection is actually lost.
c s" t d� � fdd�}��|� dS )z[
Set up a callback on clientPaused to lose the connection.
zDisconnector.connectionMadec s t d� � j�� t d� d S )Nz&Disconnector.connectionMade disconnectzloseConnection calledr, r� r/ r r r� � s
zmWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecter.connectionMade.<locals>.disconnectN)r r& )r0 r� )� clientPausedr/ r r1 � s zYWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecter.connectionMadec s t d� � �d� t d� dS )zt
Notify observers that the server side of the connection has
ended.
zDisconnecter.connectionLostNzserverLost called back)r r r5 )�
serverLostr r r7 � s
zYWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecter.connectionLostN)r9 r: r; rB r1 r7 r )r+ r, r r � Disconnecter� s r- r r` ra c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) zFWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinitez�
A producer which will write to its consumer as long as
resumeProducing is called.
@ivar consumer: The L{IConsumer} which will be written to.
c S s
|| _ d S r )�consumer)r0 r. r r r rE � s zOWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinite.__init__c S s t d� | j�d� t d� d S )NzInfinite.resumeProducing� xz*Infinite.resumeProducing wrote to consumer)r r. r r/ r r r �resumeProducing� s zVWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinite.resumeProducingc S s t d� d S )NzInfinite.stopProducing)r r/ r r r �
stopProducing� s zTWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinite.stopProducingN)r9 r: r; rB rE r0 r1 r r r r �Infinite� s r2 c s e Zd ZdZ� ��fdd�ZdS )zMWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.UnreadingWriterz
Trivial protocol which pauses its transport immediately and then
writes some bytes to it.
c s@ t d� � j�� ��d � t d� �� fdd�}��|� d S )Nz UnreadingWriter.connectionMadezclientPaused called backc s4 t d� � �j�}t d� �j�|d� t d� d S )Nz$UnreadingWriter.connectionMade writez5UnreadingWriter.connectionMade write created producerFz8UnreadingWriter.connectionMade write registered producer)r r- ZregisterProducer)rj Zproducer)r2 r0 r r r � s
zkWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.UnreadingWriter.connectionMade.<locals>.write)r r- �pauseProducingr r& )r0 r �r2 r+ r, r/ r r1 � s
z\WriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.UnreadingWriter.connectionMadeN)r9 r: r; rB r1 r r4 r r �UnreadingWriter� s r5 c s t d� � jS )N� cbClientLost)r rZ r� r� r r r6 s zJWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.cbClientLostzConnecting to %s:%szReturning Deferred) r � __class__r9 r ZSkipTestr r% r
r� �
ServerFactoryrb rc r> rn r r �objectrR rI rS r& r r� r= r� r� r ZConnectionLost)r0 r- r� r= rN r5 r[ r6 r )r2 r� r+ r, r �%test_writeAfterShutdownWithoutReading� s0 �
z4WriteDataTests.test_writeAfterShutdownWithoutReadingN)r9 r: r; rB r* r: r r r r r p s r c @ s e Zd Zdd� ZdS )�ConnectionLosingProtocolc C s4 | j �d� | j �� | j�� | jj�| j � d S )N� 1)r- r r. �masterZ_connectionMadeZportsrG r/ r r r r1 s
z'ConnectionLosingProtocol.connectionMadeNr r r r r r; s r; c @ s e Zd Zdd� Zdd� ZdS )� NoopProtocolc C s t �� | _| jj�| j� d S r )r r% r r= Z serverConnsrG r/ r r r r1 # s
zNoopProtocol.connectionMadec C s | j �d� d S �NT)r r r5 r r r r7 ' s zNoopProtocol.connectionLostNr8 r r r r r> " s r> c @ s e Zd ZdZdd� Zdd� ZdS )� ConnectionLostNotifyingProtocolac
Protocol which fires a Deferred which was previously passed to
its initializer when the connection is lost.
@ivar onConnectionLost: The L{Deferred} which will be fired in
C{connectionLost}.
@ivar lostConnectionReason: L{None} until the connection is lost, then a
reference to the reason passed to C{connectionLost}.
c C s d | _ || _d S r )�lostConnectionReason�onConnectionLost)r0 rB r r r rE 7 s z(ConnectionLostNotifyingProtocol.__init__c C s || _ | j�| � d S r )rA rB r r5 r r r r7 < s z.ConnectionLostNotifyingProtocol.connectionLostN)r9 r: r; rB rE r7 r r r r r@ , s
r@ c @ s e Zd ZdZdd� ZdS )�HandleSavingProtocolz�
Protocol which grabs the platform-specific socket handle and
saves it as an attribute on itself when the connection is
established.
c C s |� � | _tj�| |�S )z\
Save the platform-specific socket handle for future
introspection.
)Z getHandle�handler
r� �makeConnection)r0 r- r r r rE H s
z#HandleSavingProtocol.makeConnectionN)r9 r: r; rB rE r r r r rC B s rC c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�ProperlyCloseFilesMixinzA
Tests for platform resources properly being cleaned up.
c C s
t � �dS )z�
Bind a server port to which connections will be made. The server
should use the given protocol factory.
@return: The L{IListeningPort} for the server created.
N��NotImplementedError�r0 r r rJ r r r � createServerV s z$ProperlyCloseFilesMixin.createServerc C s
t � �dS )z�
Establish a connection to the given address using the given
L{ClientCreator} instance.
@return: A Deferred which will fire with the connected protocol instance.
NrG �r0 r r r� r r r �
connectClient` s z%ProperlyCloseFilesMixin.connectClientc C s
t � �dS )z�
Return the exception class which will be raised when an operation is
attempted on a closed platform handle.
NrG r/ r r r �getHandleExceptionTypej s z.ProperlyCloseFilesMixin.getHandleExceptionTypec C s t �� rtrtjS tjS )zn
Return the errno expected to result from writing to a closed
platform socket handle.
)r Z isWindowsr r� Z WSAENOTSOCKZEBADFr/ r r r �getHandleErrorCoder s z*ProperlyCloseFilesMixin.getHandleErrorCodec s� t �� �t�� }�fdd�|_��dd|��t �� � ��� }t�t� fdd��}��|j |j
|�}� �fdd�}|� |� �fdd �}|� |� �fd
d �}|� |� |S ) zp
Test that lost connections properly have their underlying socket
resources cleaned up.
c s t � �S r )r@ r )�onServerConnectionLostr r r� � s �zAProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.<lambda>r` r c s t � �S r )rC r )�onClientConnectionLostr r r� � r� c s$ | j �d� | j �� t�� �g�S )z�
Disconnect the client. Return a Deferred which fires when both
the client and the server have received disconnect notification.
s0 some bytes to make sure the connection is set up)r- r r. r r� r� )rP rO r r �clientConnected� s �
�zHProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.clientConnectedc s� | \}}|j �tj�s"t|j d� |j �tj�s<t|j d� � �� }� �� �� |jj d�}zH� �
|j d d d |d d � � �
|j d d d |d d � W n& t k
r� � �
|j d |� Y nX dS )zl
Verify that the underlying platform socket handle has been
cleaned up.
z,Client lost connection for unexpected reasonz,Server lost connection for unexpected reasons bytesr � N)
rA r r ZConnectionClosedr rN r� rM rD �sendr| r� � TypeError)r" r� r� ZexpectedErrorCodeZ exceptionr/ r r �clientDisconnected� s( �� �"&zKProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.clientDisconnectedc s" t ��j�}|�� fdd�� |S )zx
Shut down the server port. Return a Deferred which fires when
this has completed.
c s � S r r rw r� r r r� � r� zRProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.cleanup.<locals>.<lambda>)r ro r> r& )r� r" )r� r� r r� � s z@ProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.cleanup)
r r% r
r8 rJ rn r� r rL r� r= r& r� )r0 r� Z
serverAddrr� �clientDeferredrQ rU r� r )rP rO r0 r� r �test_properlyCloseFiles s,
� �
z/ProperlyCloseFilesMixin.test_properlyCloseFilesN) r9 r: r; rB rJ rL rM rN rW r r r r rF R s
rF c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �ProperlyCloseFilesTestsz�
Test that the sockets created by L{IReactorTCP.connectTCP} are cleaned up
when the connection they are associated with is closed.
c C s t j|||d�S )zE
Create a TCP server using L{IReactorTCP.listenTCP}.
ra )r rb rI r r r rJ � s z$ProperlyCloseFilesTests.createServerc C s |� ||�S )zF
Create a TCP client using L{IReactorTCP.connectTCP}.
)r� rK r r r rL � s z%ProperlyCloseFilesTests.connectClientc C s t jS )z�
Return L{socket.error} as the expected error type which will be
raised by a write to the low-level socket object after it has been
closed.
)r� r r/ r r r rM � s z.ProperlyCloseFilesTests.getHandleExceptionTypeN)r9 r: r; rB rJ rL rM r r r r rX � s rX c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�WiredForDeferredsc C s t j�| ||� d S r )r �ProtocolWrapperrE )r0 rJ ZwrappedProtocolr r r rE � s zWiredForDeferreds.__init__c C s t j�| � | jj�d � d S r )r rZ r1 rJ � onConnectr r/ r r r r1 � s z WiredForDeferreds.connectionMadec C s t j�| |� | jj�d � d S r )r rZ r7 rJ r' r r5 r r r r7 � s z WiredForDeferreds.connectionLostN)r9 r: r; rE r1 r7 r r r r rY � s rY c @ s e Zd ZeZdd� ZdS )r& c C s& t j�| |� t�� | _t�� | _d S r )r �WrappingFactoryrE r r% r[ r' �r0 ZwrappedFactoryr r r rE � s
zWiredFactory.__init__N)r9 r: r; rY r
rE r r r r r& � s r& c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � AddressTestszR
Tests for address-related interactions with client and server protocols.
c C s� G dd� dt j�}t� | _t�� | _| j_t�� | _| j_ || j�| _
t � | _ t�� | _
| j _t�� | _| j _ || j �| _tjd| j
dd�| _t�| j�� j| j�� j| j�| _t�| j| j
g�S )a9
Create a port and connected client/server pair which can be used
to test factory behavior related to addresses.
@return: A L{defer.Deferred} which will be called back when both the
client and server protocols have received their connection made
callback.
c @ s e Zd ZdZdd� Zdd� ZdS )z.AddressTests.setUp.<locals>.RememberingWrappera�
Simple wrapper factory which records the addresses which are
passed to its L{buildProtocol} method and delegates actual
protocol creation to another factory.
@ivar addresses: A list of the objects passed to buildProtocol.
@ivar factory: The wrapped factory to which protocol creation is
delegated.
c S s g | _ || _d S r )� addressesrJ )r0 rJ r r r rE s z7AddressTests.setUp.<locals>.RememberingWrapper.__init__c S s | j �|� | j�|�S )z�
Append the given address to C{self.addresses} and forward
the call to C{self.factory}.
)r_ rG rJ r@ )r0 rN r r r r@ s z<AddressTests.setUp.<locals>.RememberingWrapper.buildProtocolN)r9 r: r; rB rE r@ r r r r �RememberingWrapper s r` r r` ra )r
�
ClientFactoryrQ r� r r% r� rP �serverConnLostrK Z
serverWrapperrR r� r� �clientConnLost�
clientWrapperr rb r= r� rn r� rX r� )r0 r` r r r �setUp� s
�zAddressTests.setUpc C s( | j �� t�| j| jt�| jj�g�S )zf
Disconnect the client/server pair and shutdown the port created in
L{setUp}.
) rX r� r r� rb rc ro r= r> r/ r r r �tearDown4 s
�zAddressTests.tearDownc C s\ | j jj�� }| jjj�� }| �| jjt d|j
|j �g� | �| jjt d|j
|j �g� dS )ae
L{ClientFactory.buildProtocol} should be invoked with the address of
the server to which a connection has been established, which should
be the same as the address reported by the C{getHost} method of the
transport of the server protocol and as the C{getPeer} method of the
transport of the client protocol.
r� N) r� r
r- rn r� r r| rd r_ r r� r= )r0 r� Z
clientPeerr r r �test_buildProtocolClient? s ��z%AddressTests.test_buildProtocolClientN)r9 r: r; rB re rf rg r r r r r^ � s 9 r^ c @ s e Zd Zdd� ZdS )�LargeBufferWriterProtocolc C s* | j �d| jj � d| j_| j �� d S )N� XrF )r- r rJ �lenr r. r/ r r r r1 X s z(LargeBufferWriterProtocol.connectionMadeNr r r r r rh S s rh c @ s e Zd Zdd� Zdd� ZdS )�LargeBufferReaderProtocolc C s | j jt|�7 _d S r )rJ rj r r r r r _ s z&LargeBufferReaderProtocol.dataReceivedc C s d| j _d S rU r r5 r r r r7 a s z(LargeBufferReaderProtocol.connectionLostNr r r r r rk ^ s rk c @ s e Zd Zdd� Zdd� ZdS )� LargeBufferReaderClientFactoryc C s d| _ d| _d S �Nr )r rj r/ r r r rE e s z'LargeBufferReaderClientFactory.__init__c C s t � }| |_|| _|S r )rk rJ r
rM r r r r@ h s z,LargeBufferReaderClientFactory.buildProtocolNr r r r r rl d s rl c @ s e Zd ZdZdd� ZdS )� FireOnClosezaA wrapper around a protocol that makes it fire a deferred when
connectionLost is called.
c C s t j�| |� | jj�d � d S r )r rZ r7 rJ rS r r5 r r r r7 s s zFireOnClose.connectionLostN)r9 r: r; rB r7 r r r r rn o s rn c @ s e Zd ZeZdd� ZdS )�FireOnCloseFactoryc C s t j�| |� t�� | _d S r )r r\ rE r r% rS r] r r r rE { s zFireOnCloseFactory.__init__N)r9 r: r; rn r
rE r r r r ro x s ro c @ s e Zd ZdZdZdd� ZdS )�LargeBufferTestsz5Test that buffering large amounts of data works.
i �c s� t �� �t�_ d�_d�_�j�_t��}tj d|dd�}��
|j � |� � j
}t� � t� �}t�d||� t�|j|jg�}� ��fdd�}|�|�S )Nr r` ra c s@ �� �jd� �� � j�jkd� j�jf � �� � jd� d S )Nr! z9client didn't receive all the data it expected (%d != %d)r" )rd r rj �datalenr� r$ r r r � s
���z*LargeBufferTests.testWriter.<locals>.check)r
r% rh r r rq rj ro r rb rc r> rn r= rl r� r r� rS r& )r0 r( rO rl r) r r r r$ r �
testWriter� s
zLargeBufferTests.testWriterN)r9 r: r; rB rq rr r r r r rp � s rp c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� MyHCProtocolFc C s d| _ | jr| �d � d S r? )�readHalfClosed�writeHalfClosedr7 r/ r r r �readConnectionLost� s z MyHCProtocol.readConnectionLostc C s d| _ | jr| �d � d S r? )ru rt r7 r/ r r r �writeConnectionLost� s z MyHCProtocol.writeConnectionLostN)r9 r: r; rt ru rv rw r r r r rs � s rs c @ s e Zd ZdZdZdd� ZdS )� MyHCFactoryr Nc C s$ | j d7 _ t� }| |_|| _|S rU )rH rs rJ r
rM r r r r@ � s
zMyHCFactory.buildProtocol)r9 r: r; rH rP r@ r r r r rx � s rx c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�HalfCloseTestsz Test half-closing connections.c sp t � �_}tjd|dd� �_� ��� j� t� fdd��}t� tt
��_ |� � �fdd�� |� �j
� |S )Nr r` ra c s � j S r )r� r )rO r r r� � r� z&HalfCloseTests.setUp.<locals>.<lambda>c s �j �� �� j� �� j�S r )�cfr� rn r� r= ��_�rO r0 r r r� � s �)rx rg r rb rO rc r> r) r
r� rs rz r& �_setUp)r0 rg r r r} r re � s zHalfCloseTests.setUpc s: |� _ t�� � _� j _� �� j jjd� t� fdd��S )NrF c s t � jdd �d k S )Nr
)r� rg r r/ r r r� � r� z'HalfCloseTests._setUp.<locals>.<lambda>) r� r r% �clientProtoConnectionLostrL r| r- r� r) �r0 r� r r/ r r~ � s zHalfCloseTests._setUpc sL � � � jjd� � jj�� t�� jj�}|� � fdd�� |� � j
� |S )Nr c s � j S r )r rw r/ r r r� � r� z)HalfCloseTests.tearDown.<locals>.<lambda>) r| r� �closedr- r. r ro rO r> r& � _tearDown�r0 r r r/ r rf � s zHalfCloseTests.tearDownc sf �� �jjd� �� �jjjd� t�� � � �fdd�}|�jj_�jjj� � � �
�fdd�� � S )NrF r c s d�j j_� �d � d S rU )rg r
r� r )r6 �r r0 r r �_connectionLost� s
z1HalfCloseTests._tearDown.<locals>._connectionLostc s � � � jjjd�S rU )r| rg r
r� r� r/ r r r� � r� z*HalfCloseTests._tearDown.<locals>.<lambda>) r| r� r� rg r
r r% r7 r- r. r& )r0 rj r� r r� r r� � s
zHalfCloseTests._tearDownc sn �j � �j�� j���d� t�fdd��}�fdd�}� ��fdd�}� ��fdd �}|�|��|��|�S )
N� helloc s t � j�dkS rm )rj �_tempDataBufferr ��tr r r� � r� z5HalfCloseTests.testCloseWriteCloser.<locals>.<lambda>c s � � � t� fdd��S )Nc s � j S r )Z_writeDisconnectedr r� r r r� � r� zHHalfCloseTests.testCloseWriteCloser.<locals>.loseWrite.<locals>.<lambda>)�loseWriteConnectionr) r� r� r r � loseWrite� s z6HalfCloseTests.testCloseWriteCloser.<locals>.loseWritec s4 �� � j� ��� j� �� � j� t�fdd��S )Nc s � j jS r )r
rt r �rg r r r� � r� zDHalfCloseTests.testCloseWriteCloser.<locals>.check.<locals>.<lambda>)rt r� rd ru rt r) r� �r� rg r0 r r r � s z2HalfCloseTests.testCloseWriteCloser.<locals>.checkc s\ � j j}|d� |d� ��dt� j j�� ���jjd� ���jj� �� �jj
� d S )Nr s lalala fooled your r� ) r- r r| rj r� r
r rt r� rd rt )rj �wr� r r r � s z2HalfCloseTests.testCloseWriteCloser.<locals>.write)r� rg r- r r) r& )r0 r r� r r r )r� rg r0 r� r �testCloseWriteCloser� s
z#HalfCloseTests.testCloseWriteCloserc sP �j � � jj�� t�t� fdd��t�fdd��g�}|�� �fdd�� |S )Nc s � j jS r )r
ru r r� r r r�
r� z;HalfCloseTests.testWriteCloseNotification.<locals>.<lambda>c s � j jS r )r� rt r r/ r r r� r� c s �� � jjd�S )NF)r| r
rt r{ r� r r r� s �)rg r
r- r� r r� r) r& r� r r� r �testWriteCloseNotification s �z)HalfCloseTests.testWriteCloseNotificationN)
r9 r: r; rB re r~ rf r� r� r� r r r r ry � s
ry c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �0HalfCloseNoNotificationAndShutdownExceptionTestsc C s^ t � | _}t�� | j_tjd|dd� | _}t� tt
�� |� � j
|� � j�}|�| j� |S rq )rQ rg r r% rP r rb rO r
r� r r� rn r� r= r& �
_gotClient)r0 rg rO r r r r re s � z6HalfCloseNoNotificationAndShutdownExceptionTests.setUpc C s || _ | jjS r )r� rg rP r� r r r r� s z;HalfCloseNoNotificationAndShutdownExceptionTests._gotClientc C s | j j�� | j�� S r )r� r- r. rO r> r/ r r r rf % s z9HalfCloseNoNotificationAndShutdownExceptionTests.tearDownc sn � j j�d� � j j�� t�� � jj_}t�� � j _}|� � fdd�� |� � fdd�� t�
||g�S )z�
TCP protocols support half-close connections, but not all of them
support being notified of write closes. In this case, test that
half-closing the connection causes the peer's connection to be
closed.
r� c s � � � jjjd�S )Nr� )r| rg r
r r� r/ r r r� 4 s zUHalfCloseNoNotificationAndShutdownExceptionTests.testNoNotification.<locals>.<lambda>c s � � � jjj�S r �rd rg r
r� r� r/ r r r� 6 r� ) r� r- r r� r r% rg r
rL r& r� �r0 r r( r r/ r �testNoNotification) s zCHalfCloseNoNotificationAndShutdownExceptionTests.testNoNotificationc sj � j jj�� � jj�d� � jj�� t�� � j j_ }t�� � j_ }|�
� fdd�� t� ||g�S )zx
If the other side has already closed its connection,
loseWriteConnection should pass silently.
ri c s � � � jjj�S r r� r� r/ r r r� C s zXHalfCloseNoNotificationAndShutdownExceptionTests.testShutdownException.<locals>.<lambda>) rg r
r- r. r� r r� r r% rL r& r� r� r r/ r �testShutdownException9 s zFHalfCloseNoNotificationAndShutdownExceptionTests.testShutdownExceptionN)r9 r: r; re r� rf r� r� r r r r r� s
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� HalfCloseBuggyApplicationTestszI
Test half-closing connections where notification code has bugs.
c s� t � � _t�� � j_tjd� jdd�� _� �� jj � � j�
� }t � tt
�}|�|j|j�}� fdd�}|�|� t�� jj|g�S )z|
Set up a server and connect a client to it. Return a Deferred which
only fires once this is done.
r r` ra c s
| � _ d S r �r� r� r/ r r � setClientZ s z7HalfCloseBuggyApplicationTests.setUp.<locals>.setClient)rx r� r r% rP r rb r= rc r> rn r
r� rs r� r� r& r� )r0 rN ZcreatorrV r� r r/ r re M s" �
�z$HalfCloseBuggyApplicationTests.setUpc G s t d��dS )z`
Fake implementation of a callback which illegally raises an
exception.
zONO I AM BUGGY CODENr� )r0 r� r r r �aBugb s z#HalfCloseBuggyApplicationTests.aBugc s6 t �� }� j_� jj�� � fdd�}|�|� |S )z~
Helper for testing that an exception is logged by the time the
client protocol loses its connection.
c s � � t�}� �t|�d� d S rU )ZflushLoggedErrorsr� r| rj )rj �errorsr/ r r r q s
zEHalfCloseBuggyApplicationTests._notificationRaisesTest.<locals>.check)r r% r� rL r- r� r& )r0 r� r r r/ r �_notificationRaisesTestj s
z6HalfCloseBuggyApplicationTests._notificationRaisesTestc C s | j | jj_| �� S )z�
If C{readConnectionLost} raises an exception when the transport
calls it to notify the protocol of that event, the exception should
be logged and the protocol should be disconnected completely.
)r� r� r
rv r� r/ r r r �test_readNotificationRaisesx s z:HalfCloseBuggyApplicationTests.test_readNotificationRaisesc C s | j | j_| �� S )z�
If C{writeConnectionLost} raises an exception when the transport
calls it to notify the protocol of that event, the exception should
be logged and the protocol should be disconnected completely.
)r� r� rw r� r/ r r r � test_writeNotificationRaises� s
z;HalfCloseBuggyApplicationTests.test_writeNotificationRaisesN) r9 r: r; rB re r� r� r� r� r r r r r� H s
r� c @ s e Zd ZdZdd� ZdS )�LogTestsz4
Test logging facility of TCP base classes.
c s� t � }t� }t�� |_tjd|dd�}��|j� t� |�
� j |�
� j |�� ��� j
� ��� jjd� � �fdd�}|j�|� |jS )zx
Check that the log customization of the client transport happens
once the client is connected.
r r` ra Z
Uninitializedc s �� � jjd� d S )NzAccumulatingProtocol,client)r| r- �logstrrw �rX r0 r r �cb� s
�z+LogTests.test_logstrClientSetup.<locals>.cb)rQ rR r r% rP r rb rc r> r� rn r� r= r� r| r- r� r& )r0 r� r� r= r� r r� r �test_logstrClientSetup� s"
�
� z LogTests.test_logstrClientSetupN)r9 r: r; rB r� r r r r r� � s r� c @ s* e Zd ZdZdd� Zej�e�s&de_ dS )�PauseProducingTestszG
Test some behaviors of pausing the production of a transport.
c s� t � }t� }t�� |_tjd|dd�}��|j� t� |�
� j |�
� j |�}��|j
� � �fdd�}�fdd�� |j�|� |jS )z�
In C{connectionMade} of a client protocol, C{pauseProducing} used to be
ignored: this test is here to ensure it's not ignored.
r r` ra c sf | j }��|t�� t�� � | j �� ��|t�� t�� � t�� }|� � � t�
d|j | � |S rm ) r- ZassertInr �
getReaders�
getWritersr3 � assertNotInr r% r& � callLaterr )r� �trr ��checkAfterConnectionMader0 r r �checkInConnectionMade� s �
�
zVPauseProducingTests.test_pauseProducingInConnectionMade.<locals>.checkInConnectionMadec s" | j }� �|t�� t�� � d S r )r- r� r r� r� )r� r� r/ r r r� � s �zYPauseProducingTests.test_pauseProducingInConnectionMade.<locals>.checkAfterConnectionMade)rQ rR r r% rP r rb rc r> r� rn r� r= r� r& )r0 r� r� r= rX r� r r� r �#test_pauseProducingInConnectionMade� s
� z7PauseProducingTests.test_pauseProducingInConnectionMadez#Reactor not providing IReactorFDSetN)
r9 r: r; rB r� r
Z
IReactorFDSetre r �skipr r r r r� � s % r� c @ s e Zd ZdZdd� ZdS )�CallBackOrderTestsz-
Test the order of reactor callbacks
c s� t � }t�� ��fdd��|_t� � t�� � _t�� � _� fdd�}� j�|� tjd|dd�}�� |j
� t� |� � j
|� � j� �}�� |j� dd � }d
d � }�fd d
�}t�� j�|�� j�|�g�}|�|�S )zl
Check that Protocol.connectionLost is called before factory's
clientConnectionLost
c s � � | jj�S r )rc r- r. r� r/ r r r� � s �z3CallBackOrderTests.test_loseOrder.<locals>.<lambda>c s t �d� jjj� dS )z:
protocol.connectionMade callback
r N)r r� r
r- r. r r� r r �_cbCM� s z0CallBackOrderTests.test_loseOrder.<locals>._cbCMr r` ra c S s dS )z?
factory.clientConnectionLost callback
�CCLr r r r r �_cbCCL s z1CallBackOrderTests.test_loseOrder.<locals>._cbCCLc S s dS )z:
protocol.connectionLost callback
�CLr r r r r �_cbCL s z0CallBackOrderTests.test_loseOrder.<locals>._cbCLc s � � | ddg� d S )Nr� r� )r| r r/ r r � _cbGather s z4CallBackOrderTests.test_loseOrder.<locals>._cbGather)rQ r r% r& rP rR rK r rb rc r> r� rn r� r= r� r� rS ) r0 r� r� r= rX r� r� r� r r )r� r0 r �test_loseOrder� s2
�
�
�z!CallBackOrderTests.test_loseOrderN)r9 r: r; rB r� r r r r r� � s r� r� )r )ZrB Z
__future__r r r� r� r� � functoolsr Zzope.interfacer Z
twisted.trialr Ztwisted.python.logr r r$ r
r r r
r Ztwisted.internet.addressr Ztwisted.internet.interfacesr r Ztwisted.protocolsr Ztwisted.python.compatr Ztwisted.python.runtimer Ztwisted.test.proto_helpersr r) r� r* r8 r< r9 rC rQ ra rR ZTestCaser_ r� r% r� r� r� r� r r r r r r r; r> r@ rC rF rX rZ rY r\ r& r^ rh rk rl rn ro rp rs rx ry r� r� r� r� r� �resource� ImportErrorZ getrlimitZ
RLIMIT_NOFILEZ numRoundsZ numberRoundsr r r r �<module> s�
8 H T@m'
*
p
\
R7E"/8