File: //lib/python3/dist-packages/twisted/test/__pycache__/test_tcp.cpython-310.pyc
o
�b � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
ddl m Z ddl
Z
ddlmZmZmZmZmZ ddlmZ dd lmZmZ dd
lmZ dd lm Z dd l m Z m Z dd
l m!Z! ddl"m#Z# ddl$m%Z%m&Z& dddd�Z'G dd� dej�Z(G dd� dej)�Z*G dd� d�Z+G dd� de+ej)�Z,G dd� de+ej-�Z.G d d � d e&�Z/G d d � d e&�Z0G d d!� d!ej1�Z2G d"d#� d#e.�Z3G d$d%� d%e&�Z4G d&d'� d'e&�Z5G d(d)� d)ej-�Z6G d*d+� d+e&�Z7G d,d-� d-ej�Z8G d.d/� d/ej�Z9G d0d1� d1ej-�Z:G d2d3� d3e&�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&e@�ZAG d@dA� dAe jB�ZCG dBdC� dCe jD�ZEG dDdE� dEe&�ZFG dFdG� dGej�ZGG dHdI� dIej�ZHG dJdK� dKej-�ZIG dLdM� dMe jB�ZJG dNdO� dOe jD�ZKG dPdQ� dQe&�ZLe e�G dRdS� dSe#��ZMG dTdU� dUej)�ZNG dVdW� dWe&�ZOG dXdY� dYe&�ZPG dZd[� d[e&�ZQG d\d]� d]e&�ZRG d^d_� d_e&�ZSG d`da� dae&�ZTzddlUZUW n eV�y� Y dS w eU�WeUjX�d db ZYeZeAdceY� dS )ez.
Tests for implementations of L{IReactorTCP}.
� N��wraps)�Callable�Optional)�skipIf)� implementer)�defer�error�
interfaces�protocol�reactor)� IPv4Address)�IHalfCloseableProtocol�
IPullProducer)�Protocol)�policies)�err�msg)�platform)�AccumulatingProtocol)�SkipTest�TestCasec 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 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# 1 s zloopUntil.<locals>.stop) �twisted.internetr r �Deferred� LoopingCall� addCallback�start�
addErrback�errback)r �intervalr r" r# �d2r )r% r r r! � loopUntil s
r/ c @ � e Zd Zdd� Zdd� ZdS )�ClosingProtocolc C s t d� | j�� d S )N� ClosingProtocol.connectionMade�r � transport�loseConnection��selfr r r! �connectionMade< s r2 c C s t d� |�tj� d S )N� ClosingProtocol.connectionLost)r �trapr �ConnectionDone�r7 �reasonr r r! �connectionLost@ � r9 N��__name__�
__module__� __qualname__r8 r> r r r r! r1 ; � r1 c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�ClosingFactoryz/
Factory that closes port immediately.
Nc C s | j �� | _t� S r )�port�
stopListening�
_cleanerUpperr1 )r7 �connr r r! �
buildProtocolL s z ClosingFactory.buildProtocolc C s | j du r
| j�� S | j S )zL
Clean-up for tests to wait for the port to stop listening.
N)rH rF rG r6 r r r! �cleanUpP s
zClosingFactory.cleanUp)rA rB rC �__doc__rH rJ rK r r r r! rE E s
rE c @ sJ e Zd ZU dZeZdZdZdZe e
g e f e d< dZ
dd� Zdd� ZdS ) �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 r c C s
g | _ d S r )�
peerAddressesr6 r r r! �__init__~ �
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)rN �append�called�protocolFactory�factory�protocolConnectionLost�closedDeferredr �r7 �addr�pr r r! rJ � s z$MyProtocolFactoryMixin.buildProtocol)rA rB rC rL r rT �protocolConnectionMaderV r r r r �__annotations__rS rO rJ r r r r! rM Y s
rM c @ s e Zd ZdZdS )�MyServerFactoryzI
Server factory which creates L{AccumulatingProtocol} instances.
N)rA rB rC rL r r r r! r] � s r] 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 )rM rO r r' �deferred� failDeferredr6 r r r! rO � s
zMyClientFactory.__init__c C s d| _ || _| j�d � d S �NrQ )�failedr= r` r �r7 � connectorr= r r r! �clientConnectionFailed� s z&MyClientFactory.clientConnectionFailedc C s || _ | j�d � d S r )�
lostReasonr_ r rc r r r! �clientConnectionLost� � z$MyClientFactory.clientConnectionLostc C s
d| _ d S ra ��stoppedr6 r r r! � stopFactory� rP zMyClientFactory.stopFactoryN)
rA rB rC rL rb rj rO re rg rk r r r r! r^ � s r^ 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�� interfaceN) r] r � listenTCP�
addCleanuprG �
assertTruer
�IListeningPort�
providedBy�r7 �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 rm rn c s t j�� dd�}|�� S )Nrm rn )r rp rG )�ignoredrF �rv �nr r! �cbStopListening� s z9ListeningTests.testStopListening.<locals>.cbStopListening) r] r rp �getHostrF r �
maybeDeferredrG r) )r7 rF r| r r rz r! �testStopListening� s
z ListeningTests.testStopListeningc C s t � }tjd|dd�}|�� S �Nr rm rn )r] r rp rG ru 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 )Nr� )� assertFalse�repr�find��ign�rZ �portNor7 r r! �stoppedListening� s z5ListeningTests.testPortRepr.<locals>.stoppedListening)
r] r rp �strr} rF r� r� r� r r~ rG r) )r7 rv r� r r r� 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 rm c sD | \}}� � � j}��t|j�d|� d�� |j�� |j�� d S )Nz <AccumulatingProtocol #0 on �>)r} rF � assertEqualr� r4 r5 )r$ � serverProto� clientProto�
portNumber�rF r7 r r! r" � s
�
z-ListeningTests.test_serverRepr.<locals>.check)r] r r' r[ r rp rq rG r^ �
connectTCPr} rF �
disconnect�
gatherResultsr) )r7 �server�serverConnMade�client�clientConnMaderd 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 rm rn c sZ �� � t� }t�� �_t�� |_t�d��� j|�}�� |j
� t� �j|jg�� � �S �Nrm )
�startListeningr^ r r' r[ r r� r} rF rq r� r� r) �ry r� rd ��closerF r7 �
serverFactoryr r! r| s
��z=ListeningTests.test_restartListening.<locals>.cbStopListeningc S � | \}}|j �� |j �� d S r �r4 r5 �r$ r� r� r r r! r� �
z3ListeningTests.test_restartListening.<locals>.close)r] r rp rq rG r r~ r) )r7 r| 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 � t d���NzAn error�� RuntimeErrorr r r r! �raiseException" � z8ListeningTests.test_exceptInStop.<locals>.raiseExceptionr rm rn )r] rk r rp �
assertFailurerG r� �r7 r� r� rF 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 r� r� r� r r r r! r� 2 r� z>ListeningTests.test_restartAfterExcept.<locals>.raiseExceptionr rm rn c s^ �` ��� t� }t�� �_t�� |_t�d��� j |�}��
|j � t� �j|jg��
� �S r� )rk r� r^ r r' r[ r r� r} rF rq r� r� r) r� r� r r! r| 9 s
��z?ListeningTests.test_restartAfterExcept.<locals>.cbStopListeningc S r� r r� r� r r r! r� F r� z5ListeningTests.test_restartAfterExcept.<locals>.close) r] rk r rp rq rG r� r� r) )r7 r� r| 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 rm rn Nc � � j �tj� d S r �r= r: r �ConnectionRefusedErrorr� �r� r r! r" ` � z;ListeningTests.test_directConnectionLostCall.<locals>.check)
r] r rp r} rF r> r^ r r' r[ r� r` r) )r7 r� rF r� r" r r� r! � test_directConnectionLostCallO 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 r� r� r� r r r r! r� o r� zFListeningTests.test_exceptInConnectionLostCall.<locals>.raiseExceptionr rm rn N)r] rk r rp � assertRaisesr� r> r� r r r! � test_exceptInConnectionLostCalle s
z.ListeningTests.test_exceptInConnectionLostCallN)
rA rB rC rx r r� r� r� r� r� r� r� r� r r r r! rl � s
% rl 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 rm rn c s, �� � jj� �� �j� � j�tj� d S r )rr r �made� disconnectedrf r: r r; ��x��clientFrF r7 r r! r" � s z<LoopbackTests.test_closePortInProtocolFactory.<locals>.check) rE r rp rF rq rK r} r^ r� r_ r) )r7 rv r� r" r r� r! � test_closePortInProtocolFactory| s
z-LoopbackTests.test_closePortInProtocolFactoryc C s t |ddd� �tj� d S )Nr: c S s d S r r r� r r r! �<lambda>� � z,LoopbackTests._trapCnxDone.<locals>.<lambda>)�getattrr r; )r7 �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 rm rn c s* | \}}� ||� |j �� |j �� d S r r� �r$ �serverProtocol�clientProtocolr r r! � connected� s
z>LoopbackTests._connectedClientAndServerTest.<locals>.connected)r] r r' r[ r rp rq rG r} rF r^ r� r� r) )
r7 r r� r� rF r� r� r� � connsMader� r r r! � _connectedClientAndServerTest� s
z+LoopbackTests._connectedClientAndServerTestc � � 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 �\ | |fD ]'}|j }� �|�� d� |�d� � �|�� d� |�d� � �|�� d� qd S �Nr rQ )r4 r� �
getTcpNoDelay�
setTcpNoDelay�r� r� rZ r4 r6 r r! r" � �
�z,LoopbackTests.test_tcpNoDelay.<locals>.check�r� �r7 r" r r6 r! �test_tcpNoDelay� �
z LoopbackTests.test_tcpNoDelayc r� )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 r� r� )r4 r� �getTcpKeepAlive�setTcpKeepAliver� r6 r r! r" � r� z.LoopbackTests.test_tcpKeepAlive.<locals>.checkr� r� r r6 r! �test_tcpKeepAlive� r� z LoopbackTests.test_tcpKeepAlivec s0 t � � tjdd� dd� � fdd�}� j�|�S )Nrm �E � )�timeoutc r� r r� �ry �r� r r! r" � r� z(LoopbackTests.testFailing.<locals>.check)r^ r r� r` r) r� r r� r! � testFailing� s zLoopbackTests.testFailingc s� g �t d�D ]}t�� }|�d� |�d� ��|� qt��� 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.
�
�rm r rQ 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)r4 r5 �fail��proto)r7 �
serverSockets�tryConnectFailurer r! r� s
�z]LoopbackTests.test_connectionRefusedErrorNumber.<locals>.tryConnectFailure.<locals>.connected)�pop� getsocknamer� r� r) )r� � serverSocket�
serverHost�
serverPort�connectDeferred��
clientCreatorr7 r� r� r r! r� s
zJLoopbackTests.test_connectionRefusedErrorNumber.<locals>.tryConnectFailurec s � � | jtj� d S r )r� �osError�errno� ECONNREFUSED)�excr6 r r! � connRefused � zDLoopbackTests.test_connectionRefusedErrorNumber.<locals>.connRefusedc s � r
� � � �� � s| S r )r� r� �� passthrough)r� r r! �cleanup s �z@LoopbackTests.test_connectionRefusedErrorNumber.<locals>.cleanup)�range�socket�bind�listenrR �random�shuffler �
ClientCreatorr r r� r r� r) �addBoth)r7 �ir� �refusedDeferredr� 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}.
rm �thisbetternotexistN)r� r �ServiceNameUnknownErrorr r� r^ r6 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 rm rn c s | dkr
|dkr
� S dS )N�http�tcpr� r )� serviceName� protocolName)r� r r! �fakeGetServicePortByNameC s zELoopbackTests.test_connectByService.<locals>.fakeGetServicePortByName�
getservbynamer c s. | \}}� � �jd� |j�� |j�� d S )Nz7Server factory was not called upon to build a protocol.)rr rS r4 r5 r� )r7 r� r r! r� N s �
z6LoopbackTests.test_connectByService.<locals>.connected)r] r r' r[ r rp rq rG r} rF r^ �patchr� r� r� r) )r7 r� rF �
clientFactoryr� r �connMader� r )r� r7 r� r! �test_connectByService3 s
z#LoopbackTests.test_connectByServiceN)
rA rB rC rL r� r� r� r� r� r� r r r r r r r! r� w s
G
r� c @ �$ e Zd ZdZdZdd� Zdd� ZdS )�StartStopFactoryr c C � | j s| jrt�d| _ d S ra ��startedrj r� r6 r r r! � startFactory` �
z StartStopFactory.startFactoryc C s | j r| jrt�d| _d S ra r r6 r r r! rk e r z StartStopFactory.stopFactoryN)rA rB rC r rj r rk r r r r! r [ s
r c @ s, e Zd ZdZdZdd� Zdd� Zdd� ZdS ) �ClientStartStopFactoryr c O s& t j| g|�R i |�� t�� | _d S r )r^ rO r r' � whenStopped)r7 �a�kwr r r! rO p s z ClientStartStopFactory.__init__c C r ra r r6 r r r! r t r z#ClientStartStopFactory.startFactoryc C s& | j r| jrt�d| _| j�d� d S )NrQ T)r rj r� r r r6 r r r! rk y s z"ClientStartStopFactory.stopFactoryN)rA rB rC r rj rO r rk r r r r! r k s r c @ � 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 rm rn �rQ r c s �� � j� jfd� ��� S )Nr )r� r rj rG r� �rv �p3r7 r r! �cbClosed� s z3FactoryTests.test_serverStartStop.<locals>.cbClosedc s �� � j� jfd� d S )N)rQ rQ )r� r rj r� �rv r7 r r! � cbClosedAll� � z6FactoryTests.test_serverStartStop.<locals>.cbClosedAll) r r rp rq rG r� r rj r r~ r� r) )r7 rw �p2�d1r. rW r r! r r r! �test_serverStartStop� 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 rm rn c � � j S r ri r �rU r r! r� � � z3FactoryTests.test_clientStartStop.<locals>.<lambda>) rE r rp rF rq rK r} r r� rr r r/ )r7 rv rZ r� r r' r! �test_clientStartStop� s
z!FactoryTests.test_clientStartStopN)rA rB rC rL r% r) r r r r! r � s -r c @ �( 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 rm rn �TCPN)
r] r rp rq rG r} rF r� �type�hostr� r �CannotListenError)r7 rv rw r{ �destr r r! �test_cannotBind� s
�z CannotBindTests.test_cannotBindc s t ��� �fdd��}|S )z>Returns closure that when called calls f and then callbacks d.c s �| i |��}� � d� |S )N� r )�argsr �rtn�r rv r r! �newf� s
z/CannotBindTests._fireWhenDoneFunc.<locals>.newfr )r7 r rv r6 r r5 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 rm rn c sF t �� }t� }��||j�|_tjd��� j|dd� |� � |� |S )Nrm r� �� bindAddress)
r r' r^ r7 rJ r r� r} rF r) )�resultsr �cf1)�_conmaderZ r7 r r! � _connect1� s � z2CannotBindTests.test_clientBind.<locals>._connect1c s, t �� }��||jj�|j_|�� |� |S r )r r' r7 r r8 r) )r: r; r )�_check1connect2r7 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 )NrQ rm r8 )r� r r� r r' r4 r} rF r^ r7 re rk r r� r) � DeferredList)r: r; r$ r. rF �cf2�dl)�
_check2failed�_check2stopped�_stoprZ r7 r r! r> s" ��z8CannotBindTests.test_clientBind.<locals>._check1connect2c s4 � � |jd� |j�tj� � �|j�tj�� | S ra )r� rb r= r: r �ConnectBindErrorrr r" �r: r; r@ r6 r r! rB s z6CannotBindTests.test_clientBind.<locals>._check2failedc s � � |jd� | S ra �r� rj rF r6 r r! rC s z7CannotBindTests.test_clientBind.<locals>._check2stoppedc s4 t �� }|�� |� ��||j�|_|jj�� |S r )r r' r) r7 rk r r4 r5 )r: r; r@ r )�_check1cleanupr7 r r! rD " s
z.CannotBindTests.test_clientBind.<locals>._stopc s � � |jd� d S ra rG )r: r; r6 r r! rH ) r� z7CannotBindTests.test_clientBind.<locals>._check1cleanup)
r r' r] r7 r r rp rq rG r) )r7 � theDeferred�sfr= r )rH r> rB rC r< rD rZ r7 r! �test_clientBind� s
z CannotBindTests.test_clientBindN)rA rB rC rL r1 r7 rK r r r r! r+ � s
r+ c @ � e Zd Zdd� ZdS )�MyOtherClientFactoryc C s || _ t� | _| jS r )�addressr r )r7 rN r r r! rJ 1 s z"MyOtherClientFactory.buildProtocolN)rA rB rC rJ r r r r! rM 0 � rM c @ � 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 rm rn c s0 �� ��� g� j� �� ��� � jj�� � d S r )r� r} rN r r4 �getPeerr� )r rF r7 r r! r" M s z7LocalRemoteAddressTests.test_hostAddress.<locals>.checkc s � � � �S r )r� r� )rd �serverConnectionLostr r! r� S s z9LocalRemoteAddressTests.test_hostAddress.<locals>.cleanup)r] r r' rV r rp rq rG r} rF r^ r[ r� r) )r7 r� r{ � onConnectionr" r� r )r rd rF r7 rS r! �test_hostAddress= s
z(LocalRemoteAddressTests.test_hostAddressN)rA rB rC rL rU r r r r! rQ 7 s rQ c @ rL )�WriterProtocolc C s� | j �d� g d�}| j �|� | j �� }|jdkr%td|� �� d| j_| j �� }|jdkr:td|� �� d| j_d| j_ | j �
� d S )N� Hello Cleveland!
)� Goodbye� cruel� world�
r, z!getPeer returned non-TCP socket: rQ z!getHost returned non-TCP socket: ) r4 �write�
writeSequencerR r- r rU �problemr} �doner5 )r7 �seq�peer�usr r r! r8 ` s
z WriterProtocol.connectionMadeN�rA rB rC r8 r r r r! rV _ rO rV c @ r0 )�ReaderProtocolc C s | j j|7 _d S r )rU �data�r7 re r r r! � dataReceivedw r� zReaderProtocol.dataReceivedc C � d| j _d S ra �rU r_ r< r r r! r> z � z ReaderProtocol.connectionLostN�rA rB rC rg r> r r r r! rd v � rd c @ r0 )�WriterClientFactoryc C s d| _ d| _d S )Nr � )r_ re r6 r r r! rO �
z WriterClientFactory.__init__c C � t � }| |_|| _|S r )rd rU r rX r r r! rJ � � z!WriterClientFactory.buildProtocolN�rA rB rC rO rJ r r r r! rm ~ rD rm c @ r )�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 rm rn c sR �� �jd� �� �jdkd� �� � jd� d�g d��}�� � j|kd� d S )N�&writer didn't finish, it probably diedr zwriter indicated an error�$client didn't see connection droppedrn )rW rX rY rZ r[ z.client didn't receive all the data it expected)rr r_ r^ �joinre )ry �expected�r� rv r7 r r! r" � s ��z)WriteDataTests.test_writer.<locals>.check)r �FactoryrV r_ r^ � WiredFactoryr rp rq rG r} rF rm r� r r� � onDisconnectr) )r7 �wrappedFrZ r{ �wrappedClientFr" r r rx r! � test_writer� s
zWriteDataTests.test_writerc s t jjdkr
td��t�� �t�� �G ��fdd�dtj�}t�� }||_t j d|dd�}| �
|j � |� � }t
t�G dd � d ��� G � ��fd
d �d tj�}t� �|�_�j}�fd d
�}|�|� td|j� d|j� �� 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).
� 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 calledr3 r� r6 r r! r� � s
zmWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecter.connectionMade.<locals>.disconnectN)r r) )r7 r� )� clientPausedr6 r! r8 � 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 r< )�
serverLostr r! r> � s
zYWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Disconnecter.connectionLostN)rA rB rC rL r8 r> r )r� r� r r! � Disconnecter� s
r� r rm rn c @ r* ) 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)r7 r� r r r! rO � rP 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\ r6 r r r! �resumeProducing s zVWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinite.resumeProducingc S s t d� d S )NzInfinite.stopProducing)r r6 r r r! �
stopProducing rj zTWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.Infinite.stopProducingN)rA rB rC rL rO r� r� r r r r! �Infinite� s
r� 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 r4 �registerProducer)ry �producer)r� r7 r r! r\ s
zkWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.UnreadingWriter.connectionMade.<locals>.write)r r4 �pauseProducingr r) )r7 r\ �r� r� r� r6 r! r8 s
z\WriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.UnreadingWriter.connectionMadeN)rA rB rC rL r8 r r� r r! �UnreadingWriter s r� c s t d� � jS )N� cbClientLost)r rf r� r� r r! r� . s zJWriteDataTests.test_writeAfterShutdownWithoutReading.<locals>.cbClientLostzConnecting to �:zReturning Deferred)r � __class__rA r r r' r r �
ServerFactoryrp rq rG r} r r r^ rT r_ r) r r. rF r� r� r �ConnectionLost)r7 r� r� rF rY r� rg r� r )r� r� r� r� r! �%test_writeAfterShutdownWithoutReading� s0 � !
z4WriteDataTests.test_writeAfterShutdownWithoutReadingN)rA rB rC rL r~ r� r r r r! rs � s rs c @ rL )�ConnectionLosingProtocolc C s4 | j �d� | j �� | j�� | jj�| j � d S )N� 1)r4 r\ r5 �master�_connectionMade�portsrR r6 r r r! r8 = s
z'ConnectionLosingProtocol.connectionMadeNrc r r r r! r� < rO r� c @ r0 )� NoopProtocolc C s t �� | _| jj�| j� d S r )r r' r r� � serverConnsrR r6 r r r! r8 E �
zNoopProtocol.connectionMadec C s | j �d� d S �NT)r r r< r r r! r> I s zNoopProtocol.connectionLostNr@ r r r r! r� D rD r� c @ r )� 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)r7 r� r r r! rO Y ro z(ConnectionLostNotifyingProtocol.__init__c C s || _ | j�| � d S r )r� r� r r< r r r! r> ] rh z.ConnectionLostNotifyingProtocol.connectionLostN)rA rB rC rL rO r> r r r r! r� M s r� c @ rP )�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.
)� getHandle�handler r �makeConnection)r7 r4 r r r! r� i s
z#HandleSavingProtocol.makeConnectionN)rA rB rC rL r� r r r r! r� b s r� c @ �8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�ProperlyCloseFilesMixinzA
Tests for platform resources properly being cleaned up.
c C � t � �)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.
��NotImplementedError�r7 rN r� rU r r r! � createServerw � z$ProperlyCloseFilesMixin.createServerc C r� )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.
r� �r7 rN r� r� r r r! �
connectClient� r� z%ProperlyCloseFilesMixin.connectClientc C r� )z�
Return the exception class which will be raised when an operation is
attempted on a closed platform handle.
r� r6 r r r! �getHandleExceptionType� s z.ProperlyCloseFilesMixin.getHandleExceptionTypec C s t �� r
t�tj�S t�tj�S )z�
Return a L{hamcrest.core.matcher.Matcher} that matches the
errno expected to result from writing to a closed platform
socket handle.
)r � isWindows�hamcrest�equal_tor� � WSAENOTSOCK�EBADFr6 r r r! �getHandleErrorCodeMatcher� s z1ProperlyCloseFilesMixin.getHandleErrorCodeMatcherc 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 � t � �S r )r� r )�onServerConnectionLostr r! r� � s �zAProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.<lambda>rm r c r� r )r� r )�onClientConnectionLostr 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)r4 r\ r5 r r� r� )r� r� r r! �clientConnected� s
zHProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.clientConnectedc sp | \}}|j �tj�st|j d� |j �tj�s t|j d� � �� }� �� �� |jj d�}t
� |j d |� 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)
r� r" r �ConnectionClosedr r� r� r� r� �sendr� � assert_thatr3 )r$ r� r� �errorCodeMatcher� exceptionr6 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 r� r� r r! r� � r� zRProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.cleanup.<locals>.<lambda>)r r~ rG r) )r� r$ )r� r� r! r� � s z@ProperlyCloseFilesMixin.test_properlyCloseFiles.<locals>.cleanup)
r r' r r� r� r} r� r r� r. rF r) r� )r7 r� �
serverAddrr� �clientDeferredr� r� r� r )r� r� r7 r� r! �test_properlyCloseFiles� s& �
�
z/ProperlyCloseFilesMixin.test_properlyCloseFilesN) rA rB rC rL r� r� r� r� r� r r r r! r� r s r� c @ r* ) �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}.
rn )r rp r� r r r! r� � s z$ProperlyCloseFilesTests.createServerc C s |� ||�S )zF
Create a TCP client using L{IReactorTCP.connectTCP}.
)r� r� r r r! r� � 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 r6 r r r! r� � s z.ProperlyCloseFilesTests.getHandleExceptionTypeN)rA rB rC rL r� r� r� r r r r! r� � s
r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�WiredForDeferredsc C s t j�| ||� d S r )r �ProtocolWrapperrO )r7 rU �wrappedProtocolr r r! rO r� zWiredForDeferreds.__init__c C s t j�| � | jj�d � d S r )r r� r8 rU � onConnectr r6 r r r! r8 s z WiredForDeferreds.connectionMadec C � t j�| |� | jj�d � d S r )r r� r> rU r{ r r< r r r! r> � z WiredForDeferreds.connectionLostN)rA rB rC rO r8 r> r r r r! r� s r� c @ � e Zd ZeZdd� ZdS )rz c C s& t j�| |� t�� | _t�� | _d S r )r �WrappingFactoryrO r r' r� r{ �r7 �wrappedFactoryr r r! rO s
zWiredFactory.__init__N)rA rB rC r� r rO r r r r! rz
� rz c @ r* ) � 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 @ r )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 )� addressesrU )r7 rU r r r! rO 0 ro 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� rR rU rJ )r7 rY r r r! rJ 7 s z<AddressTests.setUp.<locals>.RememberingWrapper.buildProtocolN)rA rB rC rL rO rJ r r r r! �RememberingWrapper% s
r� r rm rn )r �
ClientFactoryr] r� r r' r� r[ �serverConnLostrV �
serverWrapperr^ r� r� �clientConnLost�
clientWrapperr rp rF r� r} r. rd r� )r7 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}.
) rd r� r r� r� r� r~ rF rG r6 r r r! �tearDownV 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 r4 r} r� rR r� r� r� r
r. rF )r7 r� �
clientPeerr r r! �test_buildProtocolClientd s ��z%AddressTests.test_buildProtocolClientN)rA rB rC rL r� r� r� r r r r! r� s
; r� c @ rL )�LargeBufferWriterProtocolc s4 � j �d� jjd � � fdd�}t�d|� d S )N� XrQ c s" � j �d� d� j_� j �� d S )Nr� rQ )r4 r\ rU r_ r5 r r6 r r! �finish� s z8LargeBufferWriterProtocol.connectionMade.<locals>.finishg����MbP?)r4 r\ rU �lenr � callLater)r7 r� r r6 r! r8 s z(LargeBufferWriterProtocol.connectionMadeNrc r r r r! r� y s r� c @ r0 )�LargeBufferReaderProtocolc C s | j jt|�7 _d S r )rU r� rf r r r! rg � r" z&LargeBufferReaderProtocol.dataReceivedc C rh ra ri r< r r r! r> � rj z(LargeBufferReaderProtocol.connectionLostNrk r r r r! r� � rl r� c @ r0 )� LargeBufferReaderClientFactoryc C s d| _ d| _d S �Nr )r_ r� r6 r r r! rO � ro z'LargeBufferReaderClientFactory.__init__c C rp r )r� rU r rX r r r! rJ � rq z,LargeBufferReaderClientFactory.buildProtocolNrr r r r r! r� � rD r� c @ rP )� FireOnClosezaA wrapper around a protocol that makes it fire a deferred when
connectionLost is called.
c C r� r )r r� r> rU r_ r r< r r r! r> � r� zFireOnClose.connectionLostN)rA rB rC rL r> r r r r! r� � � r� c @ r� )�FireOnCloseFactoryc C s t j�| |� t�� | _d S r )r r� rO r r' r_ r� r r r! rO � s zFireOnCloseFactory.__init__N)rA rB rC r� r rO r r r r! r� � r� r� c @ s e Zd ZdZdZdd� ZdS )�LargeBufferTestsz0Test 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 rm rn c s` �� �jd� �� � j�jkd� j�jf � �� � j�jkd� j�jf � �� � jd� d S )Nrt z9client didn't receive all the data it expected (%d != %d)z8client did receive more data than it expected (%d != %d)ru )rr r_ r� �datalenr� rx r r! r" � s
��
��z*LargeBufferTests.testWriter.<locals>.check)r ry r� r_ r^ r� r� r� r rp rq rG r} rF r� r� r r� r_ r) )r7 r| rZ r{ r} r r" r rx r! �
testWriter� s
zLargeBufferTests.testWriterN)rA rB rC rL r� r� r r r r! r� � s r� c @ r )� MyHCProtocolFc C � d| _ | jr
| �d � d S d S r� )�readHalfClosed�writeHalfClosedr> r6 r r r! �readConnectionLost� � �z MyHCProtocol.readConnectionLostc C r� r� )r� r� r> r6 r r r! �writeConnectionLost� r� z MyHCProtocol.writeConnectionLostN)rA rB rC r� r� r� r� r r r r! r� � s
r� c @ s e Zd ZdZdZdd� ZdS )� MyHCFactoryr Nc C s$ | j d7 _ t� }| |_|| _|S ra )rS r� rU r rX r r r! rJ � s
zMyHCFactory.buildProtocol)rA rB rC rS r[ rJ r r r r! r� � s r� 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 rm rn c r& r )r� r )rZ r r! r� � r( z&HalfCloseTests.setUp.<locals>.<lambda>c s �j �� �� j� �� j�S r )�cfr� r} r. rF ��_�rZ r7 r r! r� s )r� rv r rp rZ rq rG r/ r r� r� r� r) �_setUp)r7 rv r r r� r! r� � s zHalfCloseTests.setUpc s: |� _ t�� � _� j _� �� j jjd� t� fdd��S )NrQ c s t � jdd �d uS )Nr )r� rv r r6 r r! r�
� z'HalfCloseTests._setUp.<locals>.<lambda>) r� r r' �clientProtoConnectionLostrW r� r4 r� r/ �r7 r� r r6 r! r� s zHalfCloseTests._setUpc sL � � � jjd� � jj�� t�� jj�}|� � fdd�� |� � j
� |S )Nr c r& r )r� r� r6 r r! r� r( z)HalfCloseTests.tearDown.<locals>.<lambda>) r� r� �closedr4 r5 r r~ rZ rG r) � _tearDown�r7 r r r6 r! r� s zHalfCloseTests.tearDownc sf �� �jjd� �� �jjjd� t�� � � �fdd�}|�jj_�jjj� � � �
�fdd�� � S )NrQ r c s d�j j_� �d � d S ra )rv r r� r )r= �r r7 r r! �_connectionLost s
z1HalfCloseTests._tearDown.<locals>._connectionLostc � � � � jjjd�S ra )r� rv r r� r� r6 r r! r� ! r� z*HalfCloseTests._tearDown.<locals>.<lambda>) r� r� r� rv r r r' r> r4 r5 r) )r7 ry 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 r� )r� �_tempDataBufferr ��tr r! r� * � z5HalfCloseTests.testCloseWriteCloser.<locals>.<lambda>c s � � � t� fdd��S )Nc r& r )�_writeDisconnectedr r r r! r� . r( zHHalfCloseTests.testCloseWriteCloser.<locals>.loseWrite.<locals>.<lambda>)�loseWriteConnectionr/ r� r r r! � loseWrite, r? z6HalfCloseTests.testCloseWriteCloser.<locals>.loseWritec s4 �� � j� ��� j� �� � j� t�fdd��S )Nc � � j jS r )r r� r �rv r r! r� 4 r� zDHalfCloseTests.testCloseWriteCloser.<locals>.check.<locals>.<lambda>)r� r� rr r� r� r/ r� �r� rv r7 r r! r" 0 s z2HalfCloseTests.testCloseWriteCloser.<locals>.checkc s\ � j j}|d� |d� ��dt� j j�� ���jjd� ���jj� �� �jj
� d S )NrZ s lalala fooled your r� ) r4 r\ r� r� r r re r� r� rr r� )ry �wr r r! r\ 6 s z2HalfCloseTests.testCloseWriteCloser.<locals>.write)r� rv r4 r\ r/ r) )r7 r r r" r\ r )r� rv r7 r r! �testCloseWriteCloser$ s
z#HalfCloseTests.testCloseWriteCloserc sP �j � � jj�� t�t� fdd��t�fdd��g�}|�� �fdd�� |S )Nc r r )r r� r r r r! r� G r� z;HalfCloseTests.testWriteCloseNotification.<locals>.<lambda>c r r )r� r� r r6 r r! r� H r� c s �� � jjd�S )NF)r� r r� r� r r r! r� K � )rv r r4 r r r� r/ r) r� r r r! �testWriteCloseNotificationA s ��z)HalfCloseTests.testWriteCloseNotificationN)
rA rB rC rL r� r� r� r� r r
r r r r! r� � s r� 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 r� )r] rv r r' r[ r rp rZ r r� r r� r} r. rF r) �
_gotClient)r7 rv rZ r r r r! r� P s � z6HalfCloseNoNotificationAndShutdownExceptionTests.setUpc C s || _ | jjS r )r� rv r[ r� r r r! r \ s z;HalfCloseNoNotificationAndShutdownExceptionTests._gotClientc C s | j j�� | j�� S r )r� r4 r5 rZ rG r6 r r r! r� c 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 r� )Nr� )r� rv r re r� r6 r r! r� r r� zUHalfCloseNoNotificationAndShutdownExceptionTests.testNoNotification.<locals>.<lambda>c � � � � jjj�S r �rr rv r r� r� r6 r r! r� s r ) r� r4 r\ r r r' rv r rW r) r� �r7 r r. r r6 r! �testNoNotificationg 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.
r� c r r r r� r6 r r! r� � r zXHalfCloseNoNotificationAndShutdownExceptionTests.testShutdownException.<locals>.<lambda>) rv r r4 r5 r� r\ r r r' rW r) r� r r r6 r! �testShutdownExceptionv s zFHalfCloseNoNotificationAndShutdownExceptionTests.testShutdownExceptionN)rA rB rC r� r r� r r r r r r! r O s r c @ r� )
� 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 rm rn c s
| � _ d S r �r� r r6 r r! � setClient� rP z7HalfCloseBuggyApplicationTests.setUp.<locals>.setClient)r� r� r r' r[ r rp rF rq rG r} r r� r� r� r. r) r� )r7 rY �creatorr� r r r6 r! r� � s
�z$HalfCloseBuggyApplicationTests.setUpc G r� )z`
Fake implementation of a callback which illegally raises an
exception.
zONO I AM BUGGY CODEr� )r7 r3 r r r! �aBug� 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 ra )�flushLoggedErrorsr� r� r� )ry �errorsr6 r r! r" � r� zEHalfCloseBuggyApplicationTests._notificationRaisesTest.<locals>.check)r r' r� rW r4 r r) )r7 r� r" r r6 r! �_notificationRaisesTest� 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 r� r r6 r r r! �test_readNotificationRaises� 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� r� r r6 r r r! � test_writeNotificationRaises� s
z;HalfCloseBuggyApplicationTests.test_writeNotificationRaisesN) rA rB rC rL r� r r r r r r r r! r � s r c @ rP )�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 rm rn �
Uninitializedc s �� � jjd� d S )NzAccumulatingProtocol,client)r� r4 �logstrr� �rd r7 r r! �cb� r� z+LogTests.test_logstrClientSetup.<locals>.cb)r] r^ r r' r[ r rp rq rG r� r} r. rF r� r� r4 r! r) )r7 r� r� rF r# r r" r! �test_logstrClientSetup� s
z LogTests.test_logstrClientSetupN)rA rB rC rL r$ r r r r! r � r� r c @ s, e Zd ZdZeej�e� d�dd� �Z dS )�PauseProducingTestszG
Test some behaviors of pausing the production of a transport.
z#Reactor not providing IReactorFDSetc 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 rm rn c sf | j }��|t�� t�� � | j �� ��|t�� t�� � t�� }|� � � t�
d|j | � |S r� ) r4 �assertInr �
getReaders�
getWritersr� � assertNotInr r' r) r� r )r� �trr ��checkAfterConnectionMader7 r r! �checkInConnectionMade� s
zVPauseProducingTests.test_pauseProducingInConnectionMade.<locals>.checkInConnectionMadec s" | j }� �|t�� t�� � d S r )r4 r) r r' r( )r� r* r6 r r! r,
s zYPauseProducingTests.test_pauseProducingInConnectionMade.<locals>.checkAfterConnectionMade)r] r^ r r' r[ r rp rq rG r� r} r. rF r� r) )r7 r� r� rF rd r- r r+ r! �#test_pauseProducingInConnectionMade� s
z7PauseProducingTests.test_pauseProducingInConnectionMadeN)
rA rB rC rL r r
�
IReactorFDSetrt r r. r r r r! r% � s �r% c @ rP )�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 )rq r4 r5 r� r6 r r! r� r z3CallBackOrderTests.test_loseOrder.<locals>.<lambda>c s t �d� jjj� dS )z:
protocol.connectionMade callback
r N)r r� r r4 r5 r r� r r! �_cbCM& s z0CallBackOrderTests.test_loseOrder.<locals>._cbCMr rm rn c S � dS )z?
factory.clientConnectionLost callback
�CCLr r r r r! �_cbCCL4 � z1CallBackOrderTests.test_loseOrder.<locals>._cbCCLc S r2 )z:
protocol.connectionLost callback
�CLr r r r r! �_cbCL: r5 z0CallBackOrderTests.test_loseOrder.<locals>._cbCLc s � � | ddg� d S )Nr6 r3 )r� r r6 r r! � _cbGather@ r� z4CallBackOrderTests.test_loseOrder.<locals>._cbGather)r] r r' r) r[ r^ rV r rp rq rG r� r} r. rF r� r� r_ ) r7 r� r1 rF rd r4 r7 r8 r r )r� r7 r! �test_loseOrder s,
�
��
z!CallBackOrderTests.test_loseOrderN)rA rB rC rL r9 r r r r! r0 r� r0 r� � numberRounds)r )[rL r� r� r� � functoolsr �typingr r �unittestr �zope.interfacer r� r& r r r
r r �twisted.internet.addressr
�twisted.internet.interfacesr r �twisted.internet.protocolr �twisted.protocolsr �twisted.python.logr r �twisted.python.runtimer �twisted.test.proto_helpersr �twisted.trial.unittestr r r/ r1 r� rE rM r] r� r^ rl r� ry r r r r+ rM rQ rV rd rm rs r� r� r� r� r� r� r� r� r� rz r� r� r� r� r� r� r� r� r� r� r r r r% r0 �resource� ImportError� getrlimit�
RLIMIT_NOFILE� numRounds�setattrr r r r! �<module> s�
7 H eAo( 3 s
c
&
Y5C -9 �