File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_tcp.cpython-310.pyc
o
�b}� �
@ s� d Z ddlZddlZddlZddlZddlZddlmZ ddlm Z m
Z
m Z m Z m
Z
mZ ddlmZ ddlmZmZ ddlmZmZ ddlZddlmZmZ dd lm Z m Z m Z m Z m Z m!Z! dd
l"m#Z#m$Z$ dd l%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z. dd l/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9 dd
l:m;Z;m<Z<m=Z= ddl>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZE ddlFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNmOZO ddlPmQZQmRZRmSZS ddlTmUZU ddlVmWZW ddlXmYZY ddlZm[Z[ ddl\m]Z]m^Z^ ddl_m`Z`maZambZbmcZc ddldmeZemfZfmgZg zddlhmiZi W n ej�y dZkY n w ddllmmZm dZkdZnze�ejoejp�Znen�qd � W n er�yN Zs z
dZtd es� �ZuW Y dZs[sn dZs[sww dZtd Zuendu�r\en�v� e[�w� �rmdd lxmyZy eyjzZ{dZ|n zdd lxm}Z} W n ej�y� d!d"� Z{Y nw e}j~Z{dZ|d#d$� Zd%d&� Z�G d'd(� d(�Z�G d)d*� d*eg�Z�G d+d,� d,e<�Z�ee5�G d-d.� d.��Z�ee5e�� G d/d0� d0eg�Z�G d1d2� d2eg�Z�G d3d4� d4eJ�Z�G d5d6� d6e��Z�ee8�G d7d8� d8��Z�G d9d:� d:eQeIeM�Z�G d;d<� d<e��Z�eeteu�G d=d>� d>e���Z�G d?d@� d@eQ�Z�G dAdB� dBe��Z�eeteu�G dCdD� dDe���Z�dEdF� Z�G dGdH� dHe�Z�ee��ejnddI�G dJdK� dK���Z�ee|dL�G dMdN� dNef��Z�dOdP� Z�ee|dL�G dQdR� dRef��Z�G dSdT� dTeK�Z�G dUdV� dV�Z�G dWdX� dXe��Z�G dYdZ� dZe��Z�G d[d\� d\�Z�G d]d^� d^eQe�e�e��Z�G d_d`� d`eQe�e�e��Z�G dadb� dbe<�Z�dcdd� Z�dedf� Z�dgdh� Z�G didj� djeQ�Z�G dkdl� dl�Z�G dmdn� dn�Z�G dodp� dpe�e�eQ�Z�G dqdr� dre�e�eQ�Z�e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � e�� ��e���� � G dsdt� dteH�Z�G dudv� dveH�Z�G dwdx� dxeH�Z�G dydz� dze��Z�G d{d|� d|eH�Z�G d}d~� d~eH�Z�G dd�� d�eH�Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d�eH�Z�G d�d�� d�eH�Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d�e��Z�G d�d�� d��Z�G d�d�� d�eQe��Z�e�� ��e���� � eeteu�G d�d�� d�eg��Z�G d�d�� d�ef�Z�ee|dL�G d�d�� d�ef��Z�G d�d�� d�ef�Z�dS )�zU
Tests for implementations of L{IReactorTCP} and the TCP parts of
L{IReactorSocket}.
� N��wraps)�Callable�ClassVar�List�Optional�Sequence�Type)�skipIf)� Interface� implementer)� verifyClass� verifyObject)� IPv4Address� IPv6Address)�Deferred� DeferredList�fail�
gatherResults�
maybeDeferred�succeed)�TCP4ClientEndpoint�TCP4ServerEndpoint) �ConnectBindError�ConnectionAborted�ConnectionClosed�ConnectionDone�ConnectionLost�ConnectionRefusedError�DNSLookupError�
NoProtocol� UserError)
�
IConnector�IHalfCloseableProtocol�ILoggingContext�
IPullProducer�
IPushProducer�
IReactorFDSet�IReactorSocket� IReactorTCP�IResolverSimple�
ITLSTransport)�
ClientFactory�Protocol�
ServerFactory)�
Connection�Server� _BuffersLogs�_FileDescriptorReservation�_IFileDescriptorReservation� _NullFileDescriptorReservation� _resolveIPv6) �BrokenContextFactory�ConnectableProtocol�ConnectionTestsMixin�EndpointCreator�LogObserverMixin�Stop�StreamClientTestsMixin� findFreePort�runProtocolsWithReactor)�ReactorBuilder�needsRunningReactor� stopOnError)�Logger)�log)�Failure)�platform)�
MemoryReactor�StringTransport)�ClientStartStopFactory�ClosingFactory�MyClientFactory�MyServerFactory)�SkipTest�SynchronousTestCase�TestCase)�SSLF)�ClientContextFactoryT)�::1r zIPv6 not available. � )� _win32ifaces)� _posixifacesc C s g S �N� rV rV rV �@/usr/lib/python3/dist-packages/twisted/internet/test/test_tcp.py�<lambda>� � rX c C s t � } | r | d S td��)al
Find and return a configured link local IPv6 address including a scope
identifier using the % separation syntax. If the system has no link local
IPv6 addresses, raise L{SkipTest} instead.
@raise SkipTest: if no link local address can be found or if the
C{netifaces} module is not available.
@return: a C{str} giving the address
r z#Link local IPv6 address unavailable)�getLinkLocalIPv6AddressesrL )� addressesrV rV rW �getLinkLocalIPv6Address� s r\ c C sD |\}}d|v s d|v rt �||�d d }n||f}| �|� dS )a7
Connect a socket to the given destination.
@param client: A C{socket.socket}.
@param destination: A tuple of (host, port). The host is a C{str}, the
port a C{int}. If the C{host} is an IPv6 IP, the address is resolved
using C{getaddrinfo} and the first version found is used.
�%�:r � N)�socket� getaddrinfo�connect)�client� destination�host�port�addressrV rV rW rb � s
rb c @ �P e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�
FakeSocketz�
A fake for L{socket.socket} objects.
@ivar data: A C{str} giving the data which will be returned from
L{FakeSocket.recv}.
@ivar sendBuffer: A C{list} of the objects passed to L{FakeSocket.send}.
c C s || _ g | _d S rU )�data�
sendBuffer��selfrj rV rV rW �__init__� s
zFakeSocket.__init__c C �
|| _ d S rU )�blocking)rm rp rV rV rW � setblocking� �
zFakeSocket.setblockingc C s | j S rU �rj )rm �sizerV rV rW �recv� s zFakeSocket.recvc C s | j �|� t|�S )z�
I{Send} all of C{bytes} by accumulating it into C{self.sendBuffer}.
@return: The length of C{bytes}, indicating all the data has been
accepted.
)rk �append�len�rm �bytesrV rV rW �send� s zFakeSocket.sendc C � dS )z�
Shutdown is not implemented. The method is provided since real sockets
have it and some code expects it. No behavior of L{FakeSocket} is
affected by a call to it.
NrV )rm �howrV rV rW �shutdown� rY zFakeSocket.shutdownc C r{ )z�
Close is not implemented. The method is provided since real sockets
have it and some code expects it. No behavior of L{FakeSocket} is
affected by a call to it.
NrV �rm rV rV rW �close� rY zFakeSocket.closec G r{ )z�
Setsockopt is not implemented. The method is provided since
real sockets have it and some code expects it. No behavior of
L{FakeSocket} is affected by a call to it.
NrV )rm �argsrV rV rW �
setsockopt� rY zFakeSocket.setsockoptc C r{ )z�
Return a fake file descriptor. If actually used, this will have no
connection to this L{FakeSocket} and will probably cause surprising
results.
� rV r~ rV rV rW �fileno� s zFakeSocket.filenoN) �__name__�
__module__� __qualname__�__doc__rn rq ru rz r} r r� r� rV rV rV rW ri � s
ri c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FakeSocketTestszT
Test that the FakeSocket can be used by the doRead method of L{Connection}
c C s$ t d�}|�d� | �|jd� d S )N� someDatar )ri rq � assertEqualrp �rm �sktrV rV rW �
test_blocking� s
z FakeSocketTests.test_blockingc C s t d�}| �|�d�d� d S )Nr� �
)ri r� ru r� rV rV rW � test_recv� s zFakeSocketTests.test_recvc C s2 t d�}|�d�}| �|d� | �|jdg� dS )z�
L{FakeSocket.send} accepts the entire string passed to it, adds it to
its send buffer, and returns its length.
� s foo� N)ri rz r� rk )rm r� �countrV rV rW � test_send� s
zFakeSocketTests.test_sendN)r� r� r� r� r� r� r� rV rV rV rW r� � s
r� c @ � e Zd ZdZdd� ZdS )� FakeProtocolzL
An L{IProtocol} that returns a value from its dataReceived method.
c C r{ )zq
Return something other than L{None} to trigger a deprecation warning for
that behavior.
rV rV rl rV rV rW � dataReceived s zFakeProtocol.dataReceivedN�r� r� r� r� r� rV rV rV rW r� � r� c @ rh )�_FakeFDSetReactora�
An in-memory implementation of L{IReactorFDSet}, which records the current
sets of active L{IReadDescriptor} and L{IWriteDescriptor}s.
@ivar _readers: The set of L{IReadDescriptor}s active on this
L{_FakeFDSetReactor}
@type _readers: L{set}
@ivar _writers: The set of L{IWriteDescriptor}s active on this
L{_FakeFDSetReactor}
@ivar _writers: L{set}
c C s t � | _t � | _d S rU )�set�_readers�_writersr~ rV rV rW rn � z_FakeFDSetReactor.__init__c C � | j �|� d S rU )r� �add�rm �readerrV rV rW � addReader � z_FakeFDSetReactor.addReaderc C � || j v r
| j �|� d S d S rU )r� �remover� rV rV rW � removeReader# �
�z _FakeFDSetReactor.removeReaderc C r� rU )r� r� �rm �writerrV rV rW � addWriter' r� z_FakeFDSetReactor.addWriterc C r� rU )r� r� r� rV rV rW � removeWriter* r� z _FakeFDSetReactor.removeWriterc C s | � � | �� }| �� |S rU )�
getReaders�
getWritersrn �rm �resultrV rV rW � removeAll. s z_FakeFDSetReactor.removeAllc C �
t | j�S rU )�listr� r~ rV rV rW r� 3 rr z _FakeFDSetReactor.getReadersc C r� rU )r� r� r~ rV rV rW r� 6 rr z _FakeFDSetReactor.getWritersN) r� r� r� r� rn r� r� r� r� r� r� r� rV rV rV rW r�
s
r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�TCPServerTestsz<
Whitebox tests for L{twisted.internet.tcp.Server}.
c C sH t � | _G dd� d�}td�| _t� | _t| j| jd|� d | j�| _d S )Nc @ � e Zd ZdZdS )z&TCPServerTests.setUp.<locals>.FakePortr� N)r� r� r� �_realPortNumberrV rV rV rW �FakePortE � r� r� )rR r )r� �reactorri r� r- �protocolr0 �server)rm r� rV rV rW �setUpB s
�zTCPServerTests.setUpc C s4 | j �ttd��� | j �d� | �| jjg � dS )zq
L{Server.write} discards bytes passed to it if called after it has lost
its connection.
�Simulated lost connection� hello worldN)r� �connectionLostrD � Exception�writer� r� rk r~ rV rV rW �test_writeAfterDisconnectN s z(TCPServerTests.test_writeAfterDisconnectc C � d| j _| �� dS )z�
L{Server.write} discards bytes passed to it if called after it has lost
its connection when the connection had started TLS.
TN)r� �TLSr� r~ rV rV rW �!test_writeAfterDisconnectAfterTLSW � z0TCPServerTests.test_writeAfterDisconnectAfterTLSc C s6 | j �ttd��� | j �dg� | �| jjg � dS )zy
L{Server.writeSequence} discards bytes passed to it if called after it
has lost its connection.
r� r� N)r� r� rD r� �
writeSequencer� r� rk r~ rV rV rW �!test_writeSequenceAfterDisconnect_ s z0TCPServerTests.test_writeSequenceAfterDisconnectc C r� )z�
L{Server.writeSequence} discards bytes passed to it if called after it
has lost its connection when the connection had started TLS.
TN)r� r� r� r~ rV rV rW �)test_writeSequenceAfterDisconnectAfterTLSh r� z8TCPServerTests.test_writeSequenceAfterDisconnectAfterTLSN) r� r� r� r� r� r� r� r� r� rV rV rV rW r� = s r� c @ s4 e Zd ZdZdd� Zdd� Zee d�dd� �Zd S )
�TCPConnectionTestsz@
Whitebox tests for L{twisted.internet.tcp.Connection}.
c C sj t d�}t� }t||�}|�� | �tjg�}| �|d d t� | �|d d d� | �t|�d� dS )z�
When an L{IProtocol} implementation that returns a value from its
C{dataReceived} method, a deprecated warning is emitted.
r� r �category�messagez�Returning a value other than None from twisted.internet.test.test_tcp.FakeProtocol.dataReceived is deprecated since Twisted 11.0.0.r� N) ri r� r/ �doRead�
flushWarningsr� r� �DeprecationWarningrw )rm r� r� �conn�warningsrV rV rW �test_doReadWarningIsRaisedv s
�z-TCPConnectionTests.test_doReadWarningIsRaisedc C s( t d�}t� }t||�}| �|j� dS )z
The C{TLS} attribute of a L{Connection} instance is C{False} before
L{Connection.startTLS} is called.
r� N)ri r� r/ � assertFalser� �rm r� r� r� rV rV rW �test_noTLSBeforeStartTLS� s
z+TCPConnectionTests.test_noTLSBeforeStartTLSzNo SSL support availablec C sB t d�}t� }t||t� d�}d|_|�t� d� | �|j� dS )z}
The C{TLS} attribute of a L{Connection} instance is C{True} after
L{Connection.startTLS} is called.
r� �r� TN) ri r� r/ r� �_tlsClientDefault�startTLSrP �
assertTruer� r� rV rV rW �test_tlsAfterStartTLS� s z(TCPConnectionTests.test_tlsAfterStartTLSN) r� r� r� r� r� r� r
�useSSLr� rV rV rV rW r� q s
r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
TCPCreatorzO
Create IPv4 TCP endpoints for L{runProtocolsWithReactor}-based tests.
� 127.0.0.1c C s t |d| jd�S )z4
Create a server-side TCP endpoint.
r �� interface)r r� )rm r� rV rV rW r� � � zTCPCreator.serverc C s t || j|j�S )z�
Create a client end point that will connect to the given address.
@type serverAddress: L{IPv4Address}
)r r� rf )rm r� �
serverAddressrV rV rW rc � � zTCPCreator.clientN)r� r� r� r� r� r� rc rV rV rV rW r� � s
r� c @ r� )� TCP6CreatoraV
Create IPv6 TCP endpoints for
C{ReactorBuilder.runProtocolsWithReactor}-based tests.
The endpoint types in question here are still the TCP4 variety, since
these simply pass through IPv6 address literals to the reactor, and we are
only testing address literals, not name resolution (as name resolution has
not yet been implemented). See http://twistedmatrix.com/trac/ticket/4470
for more specific information about new endpoint classes. The naming is
slightly misleading, but presumably if you're passing an IPv6 literal, you
know what you're asking for.
c C � t � | _d S rU �r\ r� r~ rV rV rW rn � � zTCP6Creator.__init__N)r� r� r� r� rn rV rV rV rW r� � s
r� c @ � e Zd ZdZdd� Zdd� ZdS )� FakeResolverzR
A resolver implementation based on a C{dict} mapping names to addresses.
c C ro rU )�names)rm r� rV rV rW rn � rr zFakeResolver.__init__c C s4 zt | j| �W S ty ttd| �� Y S w )z�
Return the address mapped to C{name} if it exists, or raise a
C{DNSLookupError}.
@param name: The name to resolve.
@param timeout: The lookup timeout, ignore here.
zFakeResolver couldn't find )r r� �KeyErrorr r )rm �name�timeoutrV rV rW �
getHostByName� s
�zFakeResolver.getHostByNameN)r� r� r� r� rn r� rV rV rV rW r� � s r� c @ sZ e Zd ZdZefZdZedd� �Zedd� �Z dd� Z
d d
� Z d d � Z d
d� Z
dd� ZdS )�TCPClientTestsBasea�
Base class for builders defining tests related to
L{IReactorTCP.connectTCP}. Classes which uses this in must provide all of
the documented instance variables in order to specify how the test works.
These are documented as instance variables rather than declared as methods
due to some peculiar inheritance ordering concerns, but they are
effectively abstract methods.
@ivar endpoints: A client/server endpoint creator appropriate to the
address family being tested.
@type endpoints: L{twisted.internet.test.connectionmixins.EndpointCreator}
@ivar interface: An IP address literal to locally bind a socket to as well
as to connect to. This can be any valid interface for the local host.
@type interface: C{str}
@ivar port: An unused local listening port to listen on and connect to.
This will be used in conjunction with the C{interface}. (Depending on
what they're testing, some tests will locate their own port with
L{findFreePort} instead.)
@type port: C{int}
@ivar family: an address family constant, such as L{socket.AF_INET},
L{socket.AF_INET6}, or L{socket.AF_UNIX}, which indicates the address
family of the transport type under test.
@type family: C{int}
@ivar addressClass: the L{twisted.internet.interfaces.IAddress} implementor
associated with the transport type under test. Must also be a
3-argument callable which produces an instance of same.
@type addressClass: C{type}
@ivar fakeDomainName: A fake domain name to use, to simulate hostname
resolution and to distinguish between hostnames and IP addresses where
necessary.
@type fakeDomainName: C{str}
Nc C s( | j dur | j �� jS t| j| j�d S )z�
Return the port number to connect to, using C{self._port} set up by
C{listen} if available.
@return: The port number to connect to.
@rtype: C{int}
Nr� )�_port�getHostrf r= r� �familyr~ rV rV rW rf s
zTCPClientTestsBase.portc C s | j jS )zK
Return the interface attribute from the endpoints object.
)� endpointsr� r~ rV rV rW r� s z TCPClientTestsBase.interfacec C s |j d|| jd�| _| jS )z�
Start a TCP server with the given C{factory}.
@param reactor: The reactor to create the TCP port in.
@param factory: The server factory.
@return: A TCP port instance.
r r� )� listenTCPr� r� �rm r� �factoryrV rV rW �listen s
zTCPClientTestsBase.listenc C s |� | j| j|�S )z�
Start a TCP client with the given C{factory}.
@param reactor: The reactor to create the connection in.
@param factory: The client factory.
@return: A TCP connector instance.
)�
connectTCPr� rf r� rV rV rW rb - s
zTCPClientTestsBase.connectc
s� t � ���� �t� }�|_t��� G dd� dt�� �j����|�}� ��fdd�}|� |�}�fdd�}dd� }|�
||� �fd d
�}|� |�}|� t j
� �fd d �} |� | � ���� d
S )zr
When the factory's C{buildProtocol} returns L{None} the connection is
gracefully closed.
c @ � e Zd Zdd� ZdS )zETCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.NoneFactoryc S � d S rU rV �rm rg rV rV rW �
buildProtocolG � zSTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.NoneFactory.buildProtocolN�r� r� r� r� rV rV rV rW � NoneFactoryF � r� c s" � � }�j ��| �� �}|�|�S rU )r� rc r� rb )rf �
clientFactory�endpoint)r� r� rm rV rW �listenedL s
zBTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.listenedc s � � d| f � d S )NzUStream client endpoint connect succeeded with %r, should have failed with NoProtocol.)r �r� r~ rV rW �connectSucceededS s
��zJTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.connectSucceededc S s | � t� d S rU )�trapr )�reasonrV rV rW �
connectFailedY � zGTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.connectFailedc s � S rU rV ��ignored)r� rV rW � connected^ s zCTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.connectedc s � � � d S rU ��stopr r� rV rW � disconnectedi � zFTCPClientTestsBase.test_buildProtocolReturnsNone.<locals>.disconnectedN)r � buildReactorrK �protocolConnectionLostrA r. r� r� r� � addCallback� addCallbacks�
addErrbackrC �err�
runReactor)
rm �
serverFactory� listeningr �
connectingr r r �
disconnectingr rV )r� r� r� rm rW � test_buildProtocolReturnsNone9 s$
z0TCPClientTestsBase.test_buildProtocolReturnsNonec s� t �j�j�dd� \}}��� ��j���t��ji�� �jdt� t
�|d���� � }dddd��G ��fdd�dt
�}t ��� |� _
� ����fdd �}t�|� ���� � jrd��� j�� � d
��d j�d jt�d ��}|gtt��j� j�d d d
d� � }|gtt��j|j�d d d
d� � }�� �d �j dg|�R � � �� �d �j dg|�R � � �� t �d �|� dS )a�
A client's transport's C{getHost} and C{getPeer} return L{IPv4Address}
instances which have the dotted-quad string form of the resolved
address of the local and remote endpoints of the connection
respectively as their C{host} attribute, not the hostname originally
passed in to
L{connectTCP<twisted.internet.interfaces.IReactorTCP.connectTCP>}, if a
hostname was used.
N� r r� )re �peer�instancec � e Zd Z� �fdd�ZdS )z7TCPClientTestsBase.test_addresses.<locals>.CheckAddressc s, |� � �d< |�� �d< |�d< � �� d S )Nre r r )r� �getPeerr
�rm � transport�r� �
transportDatarV rW �makeConnection� s zFTCPClientTestsBase.test_addresses.<locals>.CheckAddress.makeConnectionN�r� r� r� r% rV r# rV rW � CheckAddress� � r' c sl t �j�j�} �j| d f}t�d|� �� z�j���� j� |d� W n ty. Y q w | d � _ d S )NTr� z!Connect attempt with bindAddress )� bindAddress)
r= r� r� rC �msgr� r� rf r � boundPort)rf r) )r �
fakeDomainr� rm r� rV rW � connectMe� s
� �
z4TCPClientTestsBase.test_addresses.<locals>.connectMez<{} to {} at {:x}>r ���r� re �TCPr ) r= r� r� r �fakeDomainName�installResolverr� r� r. � forProtocolr- r� r; r� r@ r �
failReasonr � getTraceback�format� __class__�addr�idr� r` ra r+ rf r� � addressClass�repr) rm re r
r� r' r- �
transportReprr+ �
serverPortrV )r r, r� rm r� r$ rW �test_addressest s>
�
� � � z!TCPClientTestsBase.test_addressesc s� �� � �t� � g �t�t�}�jd|�jd�}�j��|� � �}t
� }t|_ |� |��� ��fdd���fdd������fdd�}t
�|� ���� ��t��d d
�� �� t�d t�re�d �� �d d krotd ����tjt�d �� d
S )z�
If the context factory passed to L{ITCPTransport.startTLS} raises an
exception from its C{getContext} method, that exception is raised by
L{ITCPTransport.startTLS}.
r r� c s6 t �| j�s
��d� d S ����t| jj� �� d S )N�skip)r+ �
providedByr" rv � assertRaises�
ValueErrorr� r )�
brokenFactory�resultsrm rV rW r � s
��z5TCPClientTestsBase.test_badContext.<locals>.connectedc s � � | � d S rU �rv )�failure)rC rV rW r � r z9TCPClientTestsBase.test_badContext.<locals>.connectFailedc s* � � �� � ��� � ��fdd�� d S )Nc � � � � S rU r ��ignr� rV rW rX � � zETCPClientTestsBase.test_badContext.<locals>.whenRun.<locals>.<lambda>)r r �addBothrV )�connectDeferredr r r� rV rW �whenRun� s
z3TCPClientTestsBase.test_badContext.<locals>.whenRunr� z more than one callback result: r> z&Reactor does not support ITLSTransportN)r r6 r. r2 r- r� r� r� rc r� r, r� rb r@ r r� rw �
isinstancerD �raiseExceptionrL r� �str)rm r rf r r rL rV )rB rK r r r� rC rm rW �test_badContext� s(
z"TCPClientTestsBase.test_badContext)r� r� r� r� r) �requiredInterfacesr� �propertyrf r� r� rb r r= rP rV rV rV rW r� � s &
; Jr� c @ s$ e Zd ZdZdZejZeZ e
� Z dS )�TCP4ClientTestsBuilderze
Builder configured with IPv4 parameters for tests related to
L{IReactorTCP.connectTCP}.
z some-fake.domain.example.comN) r� r� r� r� r0 r` �AF_INETr� r r9 r� r� rV rV rV rW rS � s
rS c @ s" e Zd ZdZejZeZdd� Z dS )�TCP6ClientTestsBuilderze
Builder configured with IPv6 parameters for tests related to
L{IReactorTCP.connectTCP}.
c C s t � | _| jj| _d S rU )r� r� r� r0 r~ rV rV rW r� s z TCP6ClientTestsBuilder.setUpN)
r� r� r� r� r` �AF_INET6r� r r9 r� rV rV rV rW rU � s
rU c @ s. e Zd ZdZefZdd� Zdd� Zdd� ZdS ) �TCPConnectorTestsBuilderzU
Tests for the L{IConnector} provider returned by L{IReactorTCP.connectTCP}.
c s� t � }| �� � � jd|| jd�}||_|�� j}g �g �t� }��fdd�|_�j|_ � �
| j||�}| � t �
|�� |�� }| �|jd� | �|j| j� | �|j|� |j�� fdd�� | �� � �d �t� | ��||g� dS )aU
L{IReactorTCP.connectTCP} returns an object which provides
L{IConnector}. The destination of the connector is the address which
was passed to C{connectTCP}. The same connector object is passed to
the factory's C{startedConnecting} method as to the factory's
C{clientConnectionLost} method.
r r� c s � � | ��� |�fS rU rD �� connectorr )�seenConnectors� seenFailuresrV rW rX . s �zATCPConnectorTestsBuilder.test_connectorIdentity.<locals>.<lambda>r/ c rF rU r ��_r� rV rW rX ; rI N)rI r r� r� rf r� rH �clientConnectionLostrv �startedConnectingr� r� r" r? �getDestinationr� �typere � whenStoppedrJ r r r )rm r �tcpPort�
portNumberr rY �destrV )r� rZ r[ rW �test_connectorIdentity s(
z/TCPConnectorTestsBuilder.test_connectorIdentityc s� t � }��� ��jd|�jd�}|�� j�g ���fdd�}t� � |� _� j� �fdd�� ��
� ���fdd�� �� �� �rJ�� �d �
� � � j�t� ��� jd� d S )
z�
Calling L{IConnector.stopConnecting} in C{Factory.startedConnecting}
results in C{Factory.clientConnectionFailed} being called with
L{error.UserError} as the reason.
r r� c s8 z| � � W d S ty � �t� � ��� Y d S w rU )�stopConnectingr� rv rD r
)rY )� fatalErrorsr� rV rW r_ O s �zATCPConnectorTestsBuilder.test_userFail.<locals>.startedConnectingc rF rU r r\ r� rV rW rX Y rI z8TCPConnectorTestsBuilder.test_userFail.<locals>.<lambda>c s �� �j�� �S rU )r� r� rV )r rd r� rm rV rW rX \ s r� N)rK r r� r� r� rf rH r_ rb rJ �callWhenRunningr r r4 r r r! r� �failed)rm r rc r_ rV )r rh rd r� rm rW �
test_userFailB s"
�
z&TCPConnectorTestsBuilder.test_userFailc s� t � }| �� ��jd|| jd�}||_|�� j}t� � dd� }|� _��| j|� � g �� ��fdd�}� j �
|� | � �� � j �
t� | ��dg� dS ) z�
Calling L{IConnector.connect} in C{Factory.clientConnectionLost} causes
a new connection attempt to be made.
r r� c S � | � � d S rU )rb rX rV rV rW r^ s r� zETCPConnectorTestsBuilder.test_reconnect.<locals>.clientConnectionLostc s$ � j }��|j|jf� ��� d S rU )r� rv �made�closedr
)r
�p�r �protocolMadeAndClosedr� rV rW �reconnectFailed{ s z@TCPConnectorTestsBuilder.test_reconnect.<locals>.reconnectFailed)r� r� N)rI r r� r� rf r� rJ r^ r� � failDeferredr r r r r r� )rm r rc rd r^ rr rV rp rW �test_reconnectf s
z'TCPConnectorTestsBuilder.test_reconnectN) r� r� r� r� r) rQ rf rk rt rV rV rV rW rW s & $rW c @ s e Zd ZdZejZeZdS )�TCP4ConnectorTestsBuilderr� N) r� r� r� r� r` rT r� r r9 rV rV rV rW ru � s ru c @ s e Zd ZejZeZdd� ZdS )�TCP6ConnectorTestsBuilderc C r� rU r� r~ rV rV rW r� � r� z TCP6ConnectorTestsBuilder.setUpN) r� r� r� r` rV r� r r9 r� rV rV rV rW rv � s rv c C s t � ||�}| �|j� |S )z�
Create a socket for the duration of the given test.
@param test: the test to add cleanup to.
@param addressFamily: an C{AF_*} constant
@param socketType: a C{SOCK_*} constant.
@return: a socket object.
)r` �
addCleanupr )�test�
addressFamily�
socketTyper� rV rV rW �createTestSocket� s r{ c @ r� ) �_IExhaustsFileDescriptorsz%
A way to trigger C{EMFILE}.
c C r{ )z�
Open file descriptors until C{EMFILE} is reached.
This can raise any exception except an L{OSError} whose
C{errno} is C{EMFILE}. Any exception raised to the caller
implies L{release}.
NrV rV rV rV rW �exhaust� rY z!_IExhaustsFileDescriptors.exhaustc C r{ �zD
Release all file descriptors opened by L{exhaust}.
NrV rV rV rV rW �release� rY z!_IExhaustsFileDescriptors.releasec C r{ )��
Return the number of opened file descriptors.
@return: The number of opened file descriptors; this will be
zero if this instance has not opened any.
@rtype: L{int}
NrV rV rV rV rW r� � rY z _IExhaustsFileDescriptors.countN�r� r� r� r� r} r r� rV rV rV rW r| � s
r| )� auto_attribsc @ s� e Zd ZU dZe� Zee ed< ej dd� dd�Z
e g e f ed< ej e
jdd�Ze e gdf ed < ej e�e�ddd
�Zee ed < d d
� Zdd� Zdd� ZdS )�_ExhaustsFileDescriptorsa_
A class that triggers C{EMFILE} by creating as many file
descriptors as necessary.
@ivar fileDescriptorFactory: A factory that creates a new file
descriptor.
@type fileDescriptorFactory: A L{callable} that accepts no
arguments and returns an integral file descriptor, suitable
for passing to L{os.close}.
�_logc C s
t �d�S �Nr )�os�duprV rV rV rW rX � s
z!_ExhaustsFileDescriptors.<lambda>F)�defaultr: �_fileDescriptorFactoryN�_close)r� �initr: �_fileDescriptorsc
C s� t �� z* z| �� }W n ty% } z
|jtjkr W Y d}~n
� d}~ww | j�|� qW n ty: | � � � w | j
j d| � � d� dS )zC
Open file descriptors until C{EMFILE} is reached.
TNzCEMFILE reached by opening {openedFileDescriptors} file descriptors.)�openedFileDescriptors)
�gc�collectr� �OSError�errno�EMFILEr� rv r� r r� �infor� �rm �fd�erV rV rW r} � s, �� � �
�z _ExhaustsFileDescriptors.exhaustc
C s` | j r.| j �� }z| �|� W n ty( } z
|jtjkr#W Y d}~q � d}~ww | j sdS dS r~ )r� �popr� r� r� �EBADFr� rV rV rW r � s
���z _ExhaustsFileDescriptors.releasec C r� )r� )rw r� r~ rV rV rW r� s
z _ExhaustsFileDescriptors.count)r� r� r� r� rB r� r �__annotations__�attr�ibr� r �intr� r r� �Factoryr� r� r r} r r� rV rV rV rW r� � s
�" �
r� �9Reserved EMFILE file descriptor not supported on Windows.c @ �` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� ExhaustsFileDescriptorsTestsz0
Tests for L{_ExhaustsFileDescriptors}.
c C s t � | _| �| jj� d S rU )r� � exhausterrw r r~ rV rV rW r� s z"ExhaustsFileDescriptorsTests.setUpc C s t tj��� dS )z`
Attempt to open a file; if successful, the file is immediately
closed.
N)�openr� �devnullr r~ rV rV rW � openAFile � z&ExhaustsFileDescriptorsTests.openAFilec C � t t| j� dS )ze
L{_ExhaustsFileDescriptors} instances provide
L{_IExhaustsFileDescriptors}.
N)r r| r� r~ rV rV rW �test_providesInterface! � z3ExhaustsFileDescriptorsTests.test_providesInterfacec C sN | � | j�� d� | j�� | �| j�� d� | j�� | � | j�� d� dS )zh
L{_ExhaustsFileDescriptors.count} returns the number of open
file descriptors.
r N)r� r� r� r} �
assertGreaterr r~ rV rV rW �
test_count( s
z'ExhaustsFileDescriptorsTests.test_countc C s: | � | jj� | j�� | �t| j�}| �|jtj � dS )z{
L{_ExhaustsFileDescriptors.exhaust} causes the process to
exhaust its available file descriptors.
N)
rw r� r r} r@ �IOErrorr� r� r� r� )rm � exceptionrV rV rW �test_exhaustTriggersEMFILE3 s
z7ExhaustsFileDescriptorsTests.test_exhaustTriggersEMFILEc C s" dd� }t |�}| �t|j� dS )z�
An L{OSError} raised within
L{_ExhaustsFileDescriptors.exhaust} with an C{errno} other
than C{EMFILE} is reraised to the caller.
c S s t tjd d��)Nr� z
Not EMFILE)r� r� r� rV rV rV rW � raiseOSErrorD r� zLExhaustsFileDescriptorsTests.test_exhaustRaisesOSError.<locals>.raiseOSErrorN)r� r@ r� r} )rm r� r� rV rV rW �test_exhaustRaisesOSError= s z6ExhaustsFileDescriptorsTests.test_exhaustRaisesOSErrorc C s | j �� | j �� | �� dS )zc
L{_ExhaustsFileDescriptors.release} releases all opened
file descriptors.
N)r� r} r r� r~ rV rV rW � test_releaseJ s
z)ExhaustsFileDescriptorsTests.test_releasec s� g � � fdd�}t |�}| �|j� | �t|j� | �t� �d� | �|�� d� � D ]}| �t t
j |�}| �|j t j
� q+dS )z�
L{_ExhaustsFileDescriptors.exhaust} closes any opened file
descriptors if an exception occurs during its exhaustion loop.
c s, t � �dkr
td��t�d�} � �| � | S )Nr� z9test_fileDescriptorsReleasedOnFailure fake open exceptionr )rw rA r� r� rv �r� ��fileDescriptorsrV rW �failsAfterThree[ s �
z[ExhaustsFileDescriptorsTests.test_fileDescriptorsReleasedOnFailure.<locals>.failsAfterThreer� r N)r� rw r r@ rA r} r� rw r� r� r� �fstatr� r� )rm r� r� r� r� rV r� rW �%test_fileDescriptorsReleasedOnFailureT s
�zBExhaustsFileDescriptorsTests.test_fileDescriptorsReleasedOnFailurec sf g � � fdd�}t |�}| �|j� |�� | �|�� d� t�� d � |�� | �|�� d� dS )z�
L{_ExhaustsFileDescriptors.release} continues to close opened
file descriptors even when closing one fails with C{EBADF}.
c s t �d�} � �| � | S r� )r� r� rv r� r� rV rW �recordFileDescriptorsw s
zTExhaustsFileDescriptorsTests.test_releaseIgnoresEBADF.<locals>.recordFileDescriptorsr N) r� rw r r} r� r� r� r r� )rm r� r� rV r� rW �test_releaseIgnoresEBADFp s z5ExhaustsFileDescriptorsTests.test_releaseIgnoresEBADFc s^ g � � fdd�}dd� }t ||d�}| �|�� d� |�� | �|�� d� | �t|j� dS )z�
An L{OSError} raised within
L{_ExhaustsFileDescriptors.release} with an C{errno} other than
C{EBADF} is reraised to the caller.
c s* t � �dkr ttjd��� �d� � d S )Nr� zToo many filesr. )rw r� r� r� rv rV ��fakeFileDescriptorsrV rW �
opensThree� s
zJExhaustsFileDescriptorsTests.test_releaseRaisesOSError.<locals>.opensThreec S s
t dd��)N� z&test_releaseRaisesOSError fake OSError)r� r� rV rV rW � failingClose� rr zLExhaustsFileDescriptorsTests.test_releaseRaisesOSError.<locals>.failingClose)r r N)r� r� r� r} r� r@ r� r )rm r� r� r� rV r� rW �test_releaseRaisesOSError� s z6ExhaustsFileDescriptorsTests.test_releaseRaisesOSErrorN)r� r� r� r� r� r� r� r� r� r� r� r� r� r� rV rV rV rW r� s
r� c s� | � � j� t� }t� }||_dg���fdd�}|�|� |�|�} | �� }
t� } |�|
| � ��� j � � �fdd�} | j
� | � | j � | � |�� g }
|� |
j
� | �|
d� | �| j � | �| j
� | �t| jj� dS )a#
Assert that an L{IListeningPort} immediately closes an accepted
peer socket when the number of open file descriptors exceeds the
soft resource limit.
@param testCase: The test case under which to run this assertion.
@type testCase: L{trial.unittest.SynchronousTestCase}
@param exhauster: The file descriptor exhauster.
@type exhauster: L{_ExhaustsFileDescriptors}
@param reactor: The reactor under test.
@param runReactor: A callable that will synchronously run the
provided reactor.
@param listen: A callback to bind to a port.
@type listen: A L{callable} that accepts two arguments: the
provided C{reactor}; and a L{ServerFactory}. It must return
an L{IListeningPort} provider.
@param connect: A callback to connect a client to the listening
port.
@type connect: A L{callable} that accepts three arguments: the
provided C{reactor}; the address returned by
L{IListeningPort.getHost}; and a L{ClientFactory}. Its return
value is ignored.
Fc s � � � d�d< d S )NTr r r\ )r� �serverConnectionCompletedrV rW �stopReactorIfServerAccepted� r� z=assertPeerClosedOnEMFILE.<locals>.stopReactorIfServerAcceptedc s � � � ��� | S rU )r r
�r� )r� r� rV rW �"stopReactorAndCloseFileDescriptors� s zDassertPeerClosedOnEMFILE.<locals>.stopReactorAndCloseFileDescriptorsz1Server accepted connection; EMFILE not triggered.N)rw r rK r �protocolConnectionMader r� rJ ri r} �deferredrJ rs rv r� �assertNoResult�successResultOfr@ r �
lostReasonrN )�testCaser� r� r r� rb r �serverConnectionMader� rf �
listeningHostr r� �noResultrV )r� r� r� rW �assertPeerClosedOnEMFILE� s2 $
�r� c @ s> e Zd ZdZee�G dd� d��Zdd� Zdd� Zdd � Z d
S ) � AssertPeerClosedOnEMFILETestsz0
Tests for L{assertPeerClosedOnEMFILE}.
c @ r� ) z+AssertPeerClosedOnEMFILETests.NullExhausterz1
An exhauster that does nothing.
c C r{ )zB
See L{_IExhaustsFileDescriptors.exhaust}
NrV r~ rV rV rW r} � rY z3AssertPeerClosedOnEMFILETests.NullExhauster.exhaustc C r{ )zB
See L{_IExhaustsFileDescriptors.release}
NrV r~ rV rV rW r � rY z3AssertPeerClosedOnEMFILETests.NullExhauster.releasec C r{ )z@
See L{_IExhaustsFileDescriptors.count}
NrV r~ rV rV rW r� rY z1AssertPeerClosedOnEMFILETests.NullExhauster.countNr� rV rV rV rW �
NullExhauster� s
r� c C s t � | _t� | _d S rU )rF r� rM r� r~ rV rV rW r� r� z#AssertPeerClosedOnEMFILETests.setUpc C s t t| �� � dS )zZ
L{NullExhauster} instances provide
L{_IExhaustsFileDescriptors}.
N)r r| r� r~ rV rV rW �#test_nullExhausterProvidesInterface r� zAAssertPeerClosedOnEMFILETests.test_nullExhausterProvidesInterfacec
sr | � � }dg� � fdd�}� fdd�}dd� }| j| jjt| j|| j|||d�}| �d t|�� | �| jj � dS )
z�
If the exhauster fails to trigger C{EMFILE} and a connection
reaches the server, the reactor is stopped and the test fails.
Nc s. | � � � d �tddd��}|�t� � d S )Nr r/ r� i� )�runr� r r% rG )r� �proto�r rV rW r s
�z[AssertPeerClosedOnEMFILETests.test_reactorStoppedOnSuccessfulConnection.<locals>.runReactorc s" | � dd|�}|�� |� d< |S )Nr� i� r )r� �doStart)r� r� rf r� rV rW r� " s zWAssertPeerClosedOnEMFILETests.test_reactorStoppedOnSuccessfulConnection.<locals>.listenc S s | � dd|� d S )Nr� r )r� )r� rg r� rV rV rW rb ( s zXAssertPeerClosedOnEMFILETests.test_reactorStoppedOnSuccessfulConnection.<locals>.connect�r� r� r� r r� rb r� )
r� r@ r� �failureExceptionr� r� �assertInrO r� �running)rm r� r r� rb r� rV r� rW �)test_reactorStoppedOnSuccessfulConnection s" � zGAssertPeerClosedOnEMFILETests.test_reactorStoppedOnSuccessfulConnectionN)
r� r� r� r� r r| r� r� r� r� rV rV rV rW r� � s r� c @ s2 e Zd ZdZdd� Zdd� Zeed�dd� �Zd S )
�StreamTransportTestsMixinzR
Mixin defining tests which apply to any port/connection based transport.
c C s` | � � }| �� }tt�G dd� dt��}|� }| �||�}| �|d�}| �|f|d d � dS )zt
When a port starts, a message including a description of the associated
factory is logged.
c @ r� )zNStreamTransportTestsMixin.test_startedListeningLogMessage.<locals>.SomeFactoryc S r{ )N�
Crazy FactoryrV r~ rV rV rW � logPrefixI r� zXStreamTransportTestsMixin.test_startedListeningLogMessage.<locals>.SomeFactory.logPrefixN)r� r� r� r� rV rV rV rW � SomeFactoryG s r� r� r r� N)�observer r r$ r. �getListeningPort�#getExpectedStartListeningLogMessager� )rm �loggedMessagesr� r� r� ro �expectedMessagerV rV rW � test_startedListeningLogMessage? s z9StreamTransportTestsMixin.test_startedListeningLogMessagec sz g ��fdd�� | � � �| ��t� ��| ���}t�� � � �fdd��� ��fdd�}��|� ��� | �|�� dS )z�
When a connection is lost, an informative message should be logged
(see L{getExpectedConnectionLostLogMsg}): an address identifying
the port and the fact that it was closed.
c s � � t�| �� d S rU )rv rC �textFromEventDict)� eventDict)r� rV rW �logConnectionLostMsgZ s zQStreamTransportTestsMixin.test_connectionLostLogMsg.<locals>.logConnectionLostMsgc s t �� � ��� d S rU )rC �removeObserverr
r )r� r� rV rW � stopReactorb s
zHStreamTransportTestsMixin.test_connectionLostLogMsg.<locals>.stopReactorc s t �� � t�j���� d S rU )rC � addObserverr �
stopListeningr rV )r� ro r� rV rW �doStopListeningf s
zLStreamTransportTestsMixin.test_connectionLostLogMsg.<locals>.doStopListeningN) r r� r. � getExpectedConnectionLostLogMsgrC r� ri r� r� )rm r� r� rV )r� r� ro r� r� rW �test_connectionLostLogMsgQ s
z3StreamTransportTestsMixin.test_connectionLostLogMsgr� c C s$ t | t� | �� | j| j| jd� dS )z2
See L{assertPeerClosedOnEMFILE}.
r� N)r� r� r r r� �connectToListenerr~ rV rV rW �test_closePeerOnEMFILEo s
�z0StreamTransportTestsMixin.test_closePeerOnEMFILEN) r� r� r� r� r� r� r
� SKIP_EMFILEr� rV rV rV rW r� : s r� c @ � e Zd ZdZdZdd� ZdS )�ConnectToTCPListenerMixina^
Provides L{connectToListener} for TCP transports.
@ivar LISTENER_HOST: The host on which the port is expected to be
listening. This is specific to avoid compatibility issues
with Windows, which cannot connect to the wildcard host.
@type LISTENER_HOST: L{str}
@see: U{http://twistedmatrix.com/trac/ticket/1472}
r� c C s |� | j|j|�S )a�
Connect to the given listening TCP port.
@param reactor: The reactor under test.
@type reactor: L{IReactorTCP}
@param address: The listening port's address. Only the
C{port} component is used; see L{LISTENER_HOST}.
@type address: L{IPv4Address} or L{IPv6Address}
@param factory: The client factory.
@type factory: L{ClientFactory}
@return: The connector
)r� �
LISTENER_HOSTrf )rm r� rg r� rV rV rW r� � s z+ConnectToTCPListenerMixin.connectToListenerN)r� r� r� r� r� r� rV rV rV rW r� ~ s r� c @ � e Zd ZdZddd�ZdS )�ListenTCPMixinzT
Mixin which uses L{IReactorTCP.listenTCP} to hand out listening TCP ports.
r rR c C s |j |||d�S )z0
Get a TCP port from a reactor.
r� )r� )rm r� r� rf r� rV rV rW r� � r� z ListenTCPMixin.getListeningPortN�r rR �r� r� r� r� r� rV rV rV rW r� � s r� c @ r� )�SocketTCPMixinza
Mixin which uses L{IReactorSocket.adoptStreamPort} to hand out
listening TCP ports.
r rR c C s� t �|�rOd|v rtj}t�||�d d }ntj}||f}t�|�}|�|� |�d� |�d� z|� |�
� |j |�W |�
� |� � S |�
� |� � w t
d��)zi
Get a TCP port from a reactor, wrapping an already-initialized file
descriptor.
r^ r r_ r� Fz'Reactor does not provide IReactorSocket)r( r? r` rV ra rT �bindr� rq �adoptStreamPortr� r� r rL )rm r� r� rf r� �domainrg �portSockrV rV rW r� � s&
�
�
z SocketTCPMixin.getListeningPortNr� r� rV rV rV rW r� � s r� c @ s� e Zd ZU dZefZeeee e
d< dd� Z dd� Z dd� Z
eee�d d
� �Zeee�d d � �Zd
d� Zdd� Zeee�dd� �Zeee�dd� �Zdd� Zdd� Zeee�dd� �Zeee�dd � �Zd d � Zeee�d d � �Z eee�d!d"� �Z d#S )$�TCPPortTestsMixinz,
Tests for L{IReactorTCP.listenTCP}
rQ c C s d||� � jf S )zY
Get the message expected to be logged when a TCP port starts listening.
z%s starting on %d�r� rf )rm rf r� rV rV rW r� � � z5TCPPortTestsMixin.getExpectedStartListeningLogMessagec C s d|� � j� d�S )zJ
Get the expected connection lost message for a TCP port.
z
(TCP Port z Closed)r� )rm rf rV rV rW r� � r� z1TCPPortTestsMixin.getExpectedConnectionLostLogMsgc C s. | � � }| �|t� �}|�� }| �|t� dS )z�
When no interface is passed to L{IReactorTCP.listenTCP}, the returned
listening port listens on an IPv4 address.
N)r r� r. r� �assertIsInstancer )rm r� rf rg rV rV rW �test_portGetHostOnIPv4� s z(TCPPortTestsMixin.test_portGetHostOnIPv4c C sh | � � }ttjdd�dd� \}}| �|t� ||�}|�� }| �|t� | � d|j
� | � ||j � dS )z�
When listening on an IPv6 address, L{IListeningPort.getHost} returns
an L{IPv6Address} with C{host} and C{port} attributes reflecting the
address the port is bound to.
rQ )r� r� Nr ) r r= r` rV r� r. r� r� r r� re rf )rm r� re rd rf rg rV rV rW �test_portGetHostOnIPv6� s z(TCPPortTestsMixin.test_portGetHostOnIPv6c C sF t � }| �� }| �|t� d|�}|�� }| �|t� | �||j� dS )a(
When a link-local IPv6 address including a scope identifier is passed
as the C{interface} argument to L{IReactorTCP.listenTCP}, the resulting
L{IListeningPort} reports its address as an L{IPv6Address} with a host
value that includes the scope identifier.
r N) r\ r r� r. r� r� r r� re )rm � linkLocalr� rf rg rV rV rW � test_portGetHostOnIPv6ScopeID� s z/TCPPortTestsMixin.test_portGetHostOnIPv6ScopeIDc
s� G � fdd�dt �}|� }| �� � | �� |d|�}|�d� zt||�� j|�� jf� W n tyJ } z| � |j
t
j t
j f� W Y d}~nd}~ww | �
� � |jS )a
Connect C{client} to a server listening on C{interface} started with
L{IReactorTCP.listenTCP} and return the address passed to the factory's
C{buildProtocol} method.
@param client: A C{SOCK_STREAM} L{socket.socket} created with an address
family such that it will be able to connect to a server listening on
C{interface}.
@param interface: A C{str} giving an address for a server to listen on.
This should almost certainly be the loopback address for some
address family supported by L{IReactorTCP.listenTCP}.
@return: Whatever object, probably an L{IAddress} provider, is passed to
a server factory's C{buildProtocol} method when C{client}
establishes a connection.
c s e Zd Z� fdd�ZdS )zCTCPPortTestsMixin._buildProtocolAddressTest.<locals>.ObserveAddressc s � � � || _t� S rU )r
�observedAddressr- r� r� rV rW r� s zQTCPPortTestsMixin._buildProtocolAddressTest.<locals>.ObserveAddress.buildProtocolNr� rV r� rV rW �ObserveAddress s r r FN)r. r r� rq rb r� re rf r� r� r� � EINPROGRESS� EWOULDBLOCKr r )rm rc r� r r� rf r� rV r� rW �_buildProtocolAddressTest s
"��
z+TCPPortTestsMixin._buildProtocolAddressTestc C s@ d}t | tjtj�}| �||�}| �tdg|�� �R � |� dS )z�
When a connection is accepted over IPv4, an L{IPv4Address} is passed
to the factory's C{buildProtocol} method giving the peer's address.
r� r/ N)r{ r` rT � SOCK_STREAMr r� r � getsockname)rm r� rc r rV rV rW � test_buildProtocolIPv4Address1 s z/TCPPortTestsMixin.test_buildProtocolIPv4Addressc C sV d}t | tjtj�}| �||�}|�� }t�|tj�d }| �t d||d �|� dS )z�
When a connection is accepted to an IPv6 address, an L{IPv6Address} is
passed to the factory's C{buildProtocol} method giving the peer's
address.
rQ r r/ r� N)
r{ r` rV r r r
� getnameinfo�NI_NUMERICHOSTr� r �rm r� rc r r �hostnamerV rV rW � test_buildProtocolIPv6Address; s z/TCPPortTestsMixin.test_buildProtocolIPv6Addressc C sb t � }t| tjtj�}| �||�}|�� }t�|tj�d }| � t
d|g|dd� �R � |� dS )z�
When a connection is accepted to a link-local IPv6 address, an
L{IPv6Address} is passed to the factory's C{buildProtocol} method
giving the peer's address, including a scope identifier.
r r/ r� N) r\ r{ r` rV r r r
r r
r� r r rV rV rW �$test_buildProtocolIPv6AddressScopeIDK s &z6TCPPortTestsMixin.test_buildProtocolIPv6AddressScopeIDc
s� G � �fdd�dt �}| �� � t� }||_| �� |d|�}|�d� zt||�� j|�� j f� W n t
yN } z| � |j t j
t jf� W Y d}~nd}~ww | �� � |jS )a�
Connect C{client} to a server listening on C{interface} started with
L{IReactorTCP.listenTCP} and return the address returned by one of the
server transport's address lookup methods, C{getHost} or C{getPeer}.
@param client: A C{SOCK_STREAM} L{socket.socket} created with an address
family such that it will be able to connect to a server listening on
C{interface}.
@param interface: A C{str} giving an address for a server to listen on.
This should almost certainly be the loopback address for some
address family supported by L{IReactorTCP.listenTCP}.
@param which: A C{str} equal to either C{"getHost"} or C{"getPeer"}
determining which address will be returned.
@return: Whatever object, probably an L{IAddress} provider, is returned
from the method indicated by C{which}.
c r )zITCPPortTestsMixin._serverGetConnectionAddressTest.<locals>.ObserveAddressc s � � � t|��� | j_d S rU )r
�getattrr� rg r! �r� �whichrV rW r% q s zXTCPPortTestsMixin._serverGetConnectionAddressTest.<locals>.ObserveAddress.makeConnectionNr& rV r rV rW r p r( r r FN)r- r r. r� r� rq rb r� re rf r� r� r� r r r rg )rm rc r� r r r� rf r� rV r rW � _serverGetConnectionAddressTest[ s
"��
z1TCPPortTestsMixin._serverGetConnectionAddressTestc C �B d}t | tjtj�}| �||d�}| �tdg|�� �R � |� dS )z�
When a connection is accepted over IPv4, the server
L{ITransport.getHost} method returns an L{IPv4Address} giving the
address on which the server accepted the connection.
r� r� r/ N)r{ r` rT r r r� r � getpeername)rm r� rc � hostAddressrV rV rW �test_serverGetHostOnIPv4� � z*TCPPortTestsMixin.test_serverGetHostOnIPv4c C �b d}t | tjtj�}| �||d�}|�� }t�|tj�d }| �t d|g|dd� �R � |� dS )z�
When a connection is accepted over IPv6, the server
L{ITransport.getHost} method returns an L{IPv6Address} giving the
address on which the server accepted the connection.
rQ r� r r/ r� N)
r{ r` rV r r r r r
r� r �rm r� rc r r r rV rV rW �test_serverGetHostOnIPv6� � &z*TCPPortTestsMixin.test_serverGetHostOnIPv6c C �d t � }t| tjtj�}| �||d�}|�� }t�|tj�d }| � t
d|g|dd� �R � |� dS )z�
When a connection is accepted over IPv6, the server
L{ITransport.getHost} method returns an L{IPv6Address} giving the
address on which the server accepted the connection, including the scope
identifier.
r� r r/ r� N) r\ r{ r` rV r r r r r
r� r r rV rV rW � test_serverGetHostOnIPv6ScopeID� � &z1TCPPortTestsMixin.test_serverGetHostOnIPv6ScopeIDc C r )z�
When a connection is accepted over IPv4, the server
L{ITransport.getPeer} method returns an L{IPv4Address} giving the
address of the remote end of the connection.
r� r r/ N)r{ r` rT r r r� r r
)rm r� rc � peerAddressrV rV rW �test_serverGetPeerOnIPv4� r z*TCPPortTestsMixin.test_serverGetPeerOnIPv4c C r )z�
When a connection is accepted over IPv6, the server
L{ITransport.getPeer} method returns an L{IPv6Address} giving the
address on the remote end of the connection.
rQ r r r/ r� N)
r{ r` rV r r r
r r
r� r �rm r� rc r" r r rV rV rW �test_serverGetPeerOnIPv6� r z*TCPPortTestsMixin.test_serverGetPeerOnIPv6c C r )z�
When a connection is accepted over IPv6, the server
L{ITransport.getPeer} method returns an L{IPv6Address} giving the
address on the remote end of the connection, including the scope
identifier.
r r r/ r� N) r\ r{ r` rV r r r
r r
r� r r$ rV rV rW � test_serverGetPeerOnIPv6ScopeID� r! z1TCPPortTestsMixin.test_serverGetPeerOnIPv6ScopeIDN) r� r� r� r� r) rQ r r r r r� r� r� r r
�ipv6Skip�ipv6SkipReasonr r r r r r r r r r r# r% r& rV rV rV rW r� � s6
&
&
r� c @ � e Zd ZdS )�TCPPortTestsBuilderN�r� r� r� rV rV rV rW r* � � r* c @ r) )�TCPFDPortTestsBuilderNr+ rV rV rV rW r- � r, r- c @ r� ) �StopStartReadingProtocolzD
Protocol that pauses and resumes the transport a few times
c C s d| _ | �d� d S )Nr� r� )rj �pauseResumeProducingr~ rV rV rW �connectionMade� s z'StopStartReadingProtocol.connectionMadec C sP | j �� | j �� |r| jj�d| j|d � dS | jj�d| jjj| � dS )z?
Toggle transport read state, then count down.
r r� N) r" �pauseProducing�resumeProducingr� r� � callLaterr/ �ready�callback)rm �counterrV rV rW r/ � s
z-StopStartReadingProtocol.pauseResumeProducingc C sD t �dt|�� | j|7 _t| j�dkr | jj�| j� d S d S )Nzgot datai @ )rC r* rw rj r� r
r5 rl rV rV rW r� s
�z%StopStartReadingProtocol.dataReceivedN)r� r� r� r� r0 r/ r� rV rV rV rW r. � s
r. c s t � �� fdd��}|S )aG
Decorate a L{ReactorBuilder} test function which tests one reactor and one
connected transport. Run that test method in the context of
C{connectionMade}, and immediately drop the connection (and end the test)
when that completes.
@param testMethod: A unit test method on a L{ReactorBuilder} test suite;
taking two additional parameters; a C{reactor} as built by the
L{ReactorBuilder}, and an L{ITCPTransport} provider.
@type testMethod: 3-argument C{function}
@return: a no-argument test method.
@rtype: 1-argument C{function}
c s8 t � �G � ��fdd�dt �}|� }t� |�t� � d S )Nc s e Zd Z� ��fdd�ZdS )zBoneTransportTest.<locals>.actualTestMethod.<locals>.ServerProtocolc sp z!�� | j | j� W | jd ur| j�� �jd ur �j�� d S d S | jd ur,| j�� �jd ur7�j�� w w rU )r� r" �loseConnectionr~ ��builder�other�
testMethodrV rW r0 s
�
�
�zQoneTransportTest.<locals>.actualTestMethod.<locals>.ServerProtocol.connectionMadeN�r� r� r� r0 rV r8 rV rW �ServerProtocol s r= )r7 r> r� )r9 r= �serverProtocol�r; )r9 r: rW �actualTestMethod s
z*oneTransportTest.<locals>.actualTestMethodr )r; r@ rV r? rW �oneTransportTest s rA c C s< t �|�r| �||�� � dS | �||j� | �|j� dS )a^
Use the given test to assert that the given transport is actively reading
in the given reactor.
@note: Maintainers; for more information on why this is a function rather
than a method on a test case, see U{this document on how we structure
test tools
<http://twistedmatrix.com/trac/wiki/Design/KeepTestToolsOutOfFixtures>}
@param testCase: a test case to perform the assertion upon.
@type testCase: L{TestCase}
@param reactor: A reactor, possibly one providing L{IReactorFDSet}, or an
IOCP reactor.
@param transport: An L{ITCPTransport}
N)r' r? r� r� �handlesr� �reading�r� r� r" rV rV rW �
assertReading+ s
rE c C s. t �|�r| �||�� � dS | �|j� dS )ae
Use the given test to assert that the given transport is I{not} actively
reading in the given reactor.
@note: Maintainers; for more information on why this is a function rather
than a method on a test case, see U{this document on how we structure
test tools
<http://twistedmatrix.com/trac/wiki/Design/KeepTestToolsOutOfFixtures>}
@param testCase: a test case to perform the assertion upon.
@type testCase: L{TestCase}
@param reactor: A reactor, possibly one providing L{IReactorFDSet}, or an
IOCP reactor.
@param transport: An L{ITCPTransport}
N)r' r? � assertNotInr� r� rC rD rV rV rW �assertNotReadingE s
rG c @ sR e Zd ZdZefZdd� Zedd� �Zedd� �Z edd � �Z
d
d � Z d d
� Z dS )�TCPConnectionTestsBuilderzP
Builder defining tests relating to L{twisted.internet.tcp.Connection}.
c s� t � ���� ��_ddg}�jj}||v r t�� r td��t�_ t
� �_ t
� �_ ��
d��}|�� j}� ���fdd�}�fdd�� t�d |�}t� }t|_ t|�|��j g��||�}|�tj� ���� d
S ) zt
This test verifies transport socket read state after multiple
pause/resumeProducing calls.
� Glib2Reactor� Gtk2Reactorz.This test is broken on gtk/glib under Windows.r c sn �� | d � �� | d � | d d | d d f} | d j�dd dd � �j�� | |���fdd��S )zL
Send several IOCPReactor's buffers' worth of data.
r r� � x� � yc rF rU r rG r� rV rW rX � rI zRTCPConnectionTestsBuilder.test_stopStartReading.<locals>.proceed.<locals>.<lambda>)r� r" r� r
r )�protosrf ��cleanupr� rm �sfrV rW �proceedx s
�z@TCPConnectionTestsBuilder.test_stopStartReading.<locals>.proceedc sF � � | dd dd d� tt|d jj�t|d jj�t|j�g�S )z�
Make sure IOCPReactor didn't start several WSARecv operations
that clobbered each other's results.
rK rL rM zdid not get the right datar r� )r� r r r" r7 r� )rj rN rf r~ rV rW rP � s ���z@TCPConnectionTestsBuilder.test_stopStartReading.<locals>.cleanupr� N)r. r r� r6 r� rE � isWindowsrL r. r� r r4 r
r� r� rf r r, r- r rb r r rC r r ) rm �skippedReactors�reactorClassNamero rf rR �cc�cf�drV rO rW �test_stopStartReadinge s&
z/TCPConnectionTestsBuilder.test_stopStartReadingc C s, |� � t| ||� |�� t| ||� dS )zz
When a L{Server} is connected, its C{resumeProducing} method adds it as
a reader to the reactor.
N)r1 rG r2 rE �rm r� r� rV rV rW �test_resumeProducing� s z.TCPConnectionTestsBuilder.test_resumeProducingc C s |� � |�� t| ||� dS )z�
When a L{Server} has already started disconnecting via
C{loseConnection}, its C{resumeProducing} method does not add it as a
reader to its reactor.
N)r7 r2 rG rZ rV rV rW �&test_resumeProducingWhileDisconnecting� s z@TCPConnectionTestsBuilder.test_resumeProducingWhileDisconnectingc C s6 |� ttd��� t| ||� |�� t| ||� dS )z�
When a L{Server} has already lost its connection, its
C{resumeProducing} method does not add it as a reader to its reactor.
�dummyN)r� rD r� rG r2 rZ rV rV rW �%test_resumeProducingWhileDisconnected� s z?TCPConnectionTestsBuilder.test_resumeProducingWhileDisconnectedc C sN G dd� dt �}G dd� dt �}|� }t| ||� t� � | �|jg d�� dS )a�
Alice connects to Bob. Alice writes some bytes and then shuts down the
connection. Bob receives the bytes from the connection and then pauses
the transport object. Shortly afterwards Bob resumes the transport
object. At that point, Bob is notified that the connection has been
closed.
This is no problem for most reactors. The underlying event notification
API will probably just remind them that the connection has been closed.
It is a little tricky for win32eventreactor (MsgWaitForMultipleObjects).
MsgWaitForMultipleObjects will only deliver the close notification once.
The reactor needs to remember that notification until Bob resumes the
transport.
c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
zQTCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Pauserc S �
g | _ d S rU )�eventsr~ rV rV rW rn � rr zZTCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Pauser.__init__c S s* | j �d� | j�� | j�d| j� d S )N�pausedr )r` rv r" r1 r� r3 �resumerx rV rV rW r� � s
z^TCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Pauser.dataReceivedc S s | j �d� | j�� d S )N�resumed)r` rv r" r2 r~ rV rV rW rb � � zXTCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Pauser.resumec S s | j �d� t�| |� d S )N�lost)r` rv r7 r� �rm r rV rV rW r� � � z`TCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Pauser.connectionLostN)r� r� r� rn r� rb r� rV rV rV rW �Pauser� s
rh c @ r� )zQTCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Clientc S s | j �d� | j �� d S )Ns some bytes for you)r" r� r7 r~ rV rV rW r0 � rd z`TCPConnectionTestsBuilder.test_connectionLostAfterPausedTransport.<locals>.Client.connectionMadeNr< rV rV rV rW �Client� r� ri )ra rc re N)r7 r> r� r� r` )rm rh ri �pauserrV rV rW �'test_connectionLostAfterPausedTransport� s
zATCPConnectionTestsBuilder.test_connectionLostAfterPausedTransportc C s@ t t�G dd� dt��}G dd� dt�}t| |� |� t� � dS )a-
If one side half-closes its connection, and then the other side of the
connection calls C{loseWriteConnection}, and then C{loseConnection} in
{writeConnectionLost}, the connection is closed correctly.
This rather obscure case used to fail (see ticket #3037).
c @ s e Zd Zdd� Zdd� ZdS )zHTCPConnectionTestsBuilder.test_doubleHalfClose.<locals>.ListenerProtocolc S � | j �� d S rU )r" �loseWriteConnectionr~ rV rV rW �readConnectionLost� r z[TCPConnectionTestsBuilder.test_doubleHalfClose.<locals>.ListenerProtocol.readConnectionLostc S rl rU �r" r7 r~ rV rV rW �writeConnectionLost� r z\TCPConnectionTestsBuilder.test_doubleHalfClose.<locals>.ListenerProtocol.writeConnectionLostN)r� r� r� rn rp rV rV rV rW �ListenerProtocol� s rq c @ r� )z>TCPConnectionTestsBuilder.test_doubleHalfClose.<locals>.Clientc S rl rU ro r~ rV rV rW r0 � r zMTCPConnectionTestsBuilder.test_doubleHalfClose.<locals>.Client.connectionMadeNr< rV rV rV rW ri � r� ri N)r r# r7 r> r� )rm rq ri rV rV rW �test_doubleHalfClose� s z.TCPConnectionTestsBuilder.test_doubleHalfCloseN)
r� r� r� r� r) rQ rY rA r[ r\ r^ rk rr rV rV rV rW rH ^ s 8
+rH c @ sT e Zd ZU dZefZeeee e
d< dd� Z dd� Z dd� Z
d d
� Zd d � Zd
S )�WriteSequenceTestsMixinzM
Test for L{twisted.internet.abstract.FileDescriptor.writeSequence}.
rQ c C s$ t |dd�dur
||_dS ||_dS )z�
Set the write buffer size for the given transport, mananing possible
differences (ie, IOCP). Bug #4322 should remove the need of that hack.
�writeBufferSizeN)r rt �
bufferSize)rm r" �valuerV rV rW �setWriteBufferSize s
z*WriteSequenceTestsMixin.setWriteBufferSizec �H � fdd�}� � � }� �|dtj�}|�|� |�tj� � �|� dS )zV
C{writeSequence} sends the data even if C{write} hasn't been called.
c s2 | \� }}� �fdd�}||_ � j�g d�� d S )Nc s( t �d| � ��| d� � j�� d S )Nzdata received: %rs Some sequence splitted)rC r* r� r" r7 rs )rc rm rV rW r� s z^WriteSequenceTestsMixin.test_writeSequeceWithoutWrite.<locals>.connected.<locals>.dataReceived)s Some s sequence s splitted)r� r" r� )� protocolsr� rf r� r~ )rc rW r s
zHWriteSequenceTestsMixin.test_writeSequeceWithoutWrite.<locals>.connectedr� N� r �getConnectedClientAndServerr` rT r r rC r r �rm r r� rX rV r~ rW � test_writeSequeceWithoutWrite s
z5WriteSequenceTestsMixin.test_writeSequeceWithoutWritec rx )zo
C{writeSequence} with an element in the sequence of type unicode raises
C{TypeError}.
c s< | \}}}� � t|jjdg�}� �t|�d� |j�� d S )NzUnicode is not kosherzData must be bytes)r@ � TypeErrorr" r� r� rO r7 )ry rc r� rf �excr~ rV rW r 0 s
�zWWriteSequenceTestsMixin.test_writeSequenceWithUnicodeRaisesException.<locals>.connectedr� Nrz r| rV r~ rW �,test_writeSequenceWithUnicodeRaisesException* s
zDWriteSequenceTestsMixin.test_writeSequenceWithUnicodeRaisesExceptionc sx t t�G dd� d��}|� � � �fdd�}��� }��|dtj�}|�|� |�tj � ��
|� �� � j ddg� dS ) z{
C{writeSequence} pauses its streaming producer if too much data is
buffered, and then resumes it.
c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zdd � ZdS )
zJWriteSequenceTestsMixin.test_streamingProducer.<locals>.SaveActionProducerNc S r_ rU ��actionsr~ rV rV rW rn L rr zSWriteSequenceTestsMixin.test_streamingProducer.<locals>.SaveActionProducer.__init__c S � | j �d� d S )N�pause�r� rv r~ rV rV rW r1 O r� zYWriteSequenceTestsMixin.test_streamingProducer.<locals>.SaveActionProducer.pauseProducingc S s( | j �d� | jj�� | jj�� d S )Nrb )r� rv rc r" �unregisterProducerr� r7 r~ rV rV rW r2 R s zZWriteSequenceTestsMixin.test_streamingProducer.<locals>.SaveActionProducer.resumeProducingc S r� �Nr
r� r~ rV rV rW �
stopProducing\ r� zXWriteSequenceTestsMixin.test_streamingProducer.<locals>.SaveActionProducer.stopProducing) r� r� r� rc r� rn r1 r2 r� rV rV rV rW �SaveActionProducerG s
r� c sl | d d� \}}|� _ |� _|j�� d� ��� jg � ��|jd� |j�dgd � ��� jdg� d S )Nr T�� �2 xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� r� )rc r� r" �registerProducerr� r� rw r� �ry rc r� ��producerrm rV rW r a s zAWriteSequenceTestsMixin.test_streamingProducer.<locals>.connectedr� r� rb N)
r r& r r{ r` rT r r rC r r r� r� �rm r� r r� rX rV r� rW �test_streamingProducerA s
z.WriteSequenceTestsMixin.test_streamingProducerc s� ��t t�G �fdd�d��}|� � � �fdd�}��� }��|dtj�}|�|� |�tj � ��
|� �� � j ddg� dS )z�
C{writeSequence} pauses its producer if too much data is buffered only
if this is a streaming producer.
c s, e Zd ZdZdd� Z� fdd�Zdd� ZdS )zMWriteSequenceTestsMixin.test_nonStreamingProducer.<locals>.SaveActionProducerNc S r_ rU r� r~ rV rV rW rn � rr zVWriteSequenceTestsMixin.test_nonStreamingProducer.<locals>.SaveActionProducer.__init__c sT | j �d� | j �d�dkr| jj�� d S � �| jjd� | jj�dgd � d S )Nrb r r� r� r� )r� rv r� rc r" �
stopConsumingrw r� r~ �rx rV rW r2 � s
z]WriteSequenceTestsMixin.test_nonStreamingProducer.<locals>.SaveActionProducer.resumeProducingc S r� r� r� r~ rV rV rW r� � r� z[WriteSequenceTestsMixin.test_nonStreamingProducer.<locals>.SaveActionProducer.stopProducing)r� r� r� rc rn r2 r� rV r� rV rW r� } s
r� c s0 | d }|� _ |j�� d� ��� jdg� d S )Nr Frb )rc r" r� r� r� )ry rc r� rV rW r � s zDWriteSequenceTestsMixin.test_nonStreamingProducer.<locals>.connectedr� rb N)
r r% r r{ r` rT r r rC r r r� r� r� rV )r� rm rx rW �test_nonStreamingProducerv s
z1WriteSequenceTestsMixin.test_nonStreamingProducerN)r� r� r� r� r) rQ r r r r r� rw r} r� r� r� rV rV rV rW rs s
5rs c @ s: e Zd ZdZdd� Zdd� Zdd� Zeee �dd � �Z
d
S ) �"TCPTransportServerAddressTestMixinzE
Test mixing for TCP server address building and log prefix.
c C s t � �)z�
Helper method returnine a L{Deferred} firing with a tuple of a client
protocol, a server protocol, and a running TCP port.
)�NotImplementedError)rm r� r� ry rV rV rW r{ � s z>TCPTransportServerAddressTestMixin.getConnectedClientAndServerc sJ � ��fdd�}�� � }��|�|�}|�|� |�tj� ��|� dS )zT
Helper method to test TCP server addresses on either IPv4 or IPv6.
c s� | \}}}zA�� d|jj|�� jf t|j�� �� d|jj�f |jj� |jj\}�� |� � �� d|j
� �� �|j � W |j� � d S |j� � w )Nz <AccumulatingProtocol #%s on %s>zAccumulatingProtocol,%s,%sr/ )
r� r" � sessionnor� rf rO �logstrr� �
peerAddressesr� ra re r7 )ry rc r� rf r" �� adressClassr� rm rV rW r � s&
��
��
zHTCPTransportServerAddressTestMixin._testServerAddress.<locals>.connectedN)r r{ r r rC r r )rm r� ry r� r r� rX rV r� rW �_testServerAddress� s
z5TCPTransportServerAddressTestMixin._testServerAddressc C s | � dtjt�S )z�
L{Server} instances have a string representation indicating on which
port they're running, and the connected address is stored on the
C{peerAddresses} attribute of the factory.
r� )r� r` rT r r~ rV rV rW �test_serverAddressTCP4� r� z9TCPTransportServerAddressTestMixin.test_serverAddressTCP4c C s | � t� tjt�S )z�
IPv6 L{Server} instances have a string representation indicating on
which port they're running, and the connected address is stored on the
C{peerAddresses} attribute of the factory.
)r� r\ r` rV r r~ rV rV rW �test_serverAddressTCP6� s
�z9TCPTransportServerAddressTestMixin.test_serverAddressTCP6N) r� r� r� r� r{ r� r� r
r' r( r� rV rV rV rW r� � s "r� c @ r� )�TCPTransportTestsBuilderzT
Test standard L{ITCPTransport}s built with C{listenTCP} and C{connectTCP}.
c
s� t � }t� |_t� |_t� }t� |_t� |_�jd||d��t|j|jg�}�fdd�}|�|� t|j|jg�}t� � � �fdd�} |�| � �� |��
� j |� � S )z�
Return a L{Deferred} firing with a L{MyClientFactory} and
L{MyServerFactory} connected pair, and the listening C{Port}.
r r� c s � � � | S rU r r� r� rV rW r
� s zBTCPTransportTestsBuilder.getConnectedClientAndServer.<locals>.stopc �8 | \}}t �d| � t �d| � � �||�f� d S �Nzclient connected %szserver connected %s�rC r* r5 r� �r� rf rV rW �start � zCTCPTransportTestsBuilder.getConnectedClientAndServer.<locals>.start) rK r r� r rJ r� r rJ r r� r� rf )
rm r� r� ry r� rc � lostDeferredr
�
startDeferredr� rV )r� rf r� rW r{ � s(
�
�
z4TCPTransportTestsBuilder.getConnectedClientAndServerN)r� r� r� r� r{ rV rV rV rW r� � s r� c @ s e Zd ZdZeefZdd� ZdS )�!AdoptStreamConnectionTestsBuilderzF
Test server transports built using C{adoptStreamConnection}.
c s� t � }t� |_t � �t� �_t� �_t� }t� |_t� |_�jd||d��� ��fdd�}|j�|� t|j�jg�}�fdd�}|�|� t� ��� |� t|j�jg�} ��fdd�}
| �|
� ��
|�� � j |� �S ) a0
Return a L{Deferred} firing with a L{MyClientFactory} and
L{MyServerFactory} connected pair, and the listening C{Port}. The
particularity is that the server protocol has been obtained after doing
a C{adoptStreamConnection} against the original server connection.
r r� c s0 �� | j� ��| j� ��| j�� � �� d S rU )r� r" r� �adoptStreamConnectionr� )r� )ry r� r� rV rW �firtServerConnected5 s
�zZAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.firtServerConnectedc s � j r� �� | S rU )r� r
r� r� rV rW r
B s zKAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.stopc r� r� r� r� r� rV rW r� P r� zLAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer.<locals>.start)
rK r r� r rJ r� r r rJ r r� r� rf ) rm r� r� ry � firstServerrc r� r� r
r� r� rV )ry r� rf r� r� rW r{ ! s2
�
�
z=AdoptStreamConnectionTestsBuilder.getConnectedClientAndServerN)r� r� r� r� r' r( rQ r{ rV rV rV rW r� s r� c @ r� )�ServerAbortsTwicez'
Call abortConnection() twice.
c C s | j �� | j �� d S rU )r" �abortConnectionrl rV rV rW r� l �
z ServerAbortsTwice.dataReceivedNr� rV rV rV rW r� g r� r� c @ r� )�ServerAbortsThenLosesz>
Call abortConnection() followed by loseConnection().
c C s | j �� | j �� d S rU )r" r� r7 rl rV rV rW r� v r� z"ServerAbortsThenLoses.dataReceivedNr� rV rV rV rW r� q r� r� c @ r� )�AbortServerWritingProtocolz4
Protocol that writes data upon connection.
c C s | j �d� dS )zW
Tell the client that the connection is set up and it's time to abort.
s readyN�r" r� r~ rV rV rW r0 � r� z)AbortServerWritingProtocol.connectionMadeN�r� r� r� r� r0 rV rV rV rW r� { r� r� c @ r� )�ReadAbortServerProtocolz�
Server that should never receive any data, except 'X's which are written
by the other side of the connection before abortConnection, and so might
possibly arrive.
c C s |� dd�r
td��d S )N� Xr� zUnexpectedly received data.)�replacer� rl rV rV rW r� � s �z$ReadAbortServerProtocol.dataReceivedNr� rV rV rV rW r� � s r� c @ r� )� NoReadServerz�
Stop reading immediately on connection.
This simulates a lost connection that will cause the other side to time
out, and therefore call abortConnection().
c C rl rU )r" � stopReadingr~ rV rV rW r0 � r zNoReadServer.connectionMadeNr� rV rV rV rW r� � s r� c @ s8 e Zd ZdZdZdZdd� Zdd� Zdd� Zd d
� Z d S ) �EventualNoReadServerz�
Like NoReadServer, except we Wait until some bytes have been delivered
before stopping reading. This means TLS handshake has finished, where
applicable.
Fc C s. | j sd| _ | j�| d� | j�d� d S d S )NTF� hello)�gotDatar" r� r� rl rV rV rW r� � s
�z!EventualNoReadServer.dataReceivedc C s | j rd S d| _ | j�� d S �NT)�stoppedReadingr" r� r~ rV rV rW r2 � s z$EventualNoReadServer.resumeProducingc C r� rU rV r~ rV rV rW r1 � r� z#EventualNoReadServer.pauseProducingc C r� rU rV r~ rV rV rW r� � r� z"EventualNoReadServer.stopProducingN)
r� r� r� r� r� r� r� r2 r1 r� rV rV rV rW r� � s r� c @ r� )�BaseAbortingClientz/
Base class for abort-testing clients.
Fc C s | j rtd��t�| |� d S )Nz,BUG: connectionLost was called re-entrantly!)�inReactorMethod� RuntimeErrorr7 r� rf rV rV rW r� � s z!BaseAbortingClient.connectionLostN)r� r� r� r� r� r� rV rV rV rW r� � s r� c @ r� )�WritingButNotAbortingClientz&
Write data, but don't abort.
c C r� )Nr� r� r~ rV rV rW r0 � r� z*WritingButNotAbortingClient.connectionMadeNr� rV rV rV rW r� � r� r� c @ r� )�AbortingClientz9
Call abortConnection() after writing some data.
c C s d| _ | �� d| _ dS )zF
Some data was received, so the connection is set up.
TFN)r� �
writeAndAbortrl rV rV rW r� � s
zAbortingClient.dataReceivedc C s. | j �dd � | j �� | j �dd � d S )Nr� i' � Y)r" r� r� r~ rV rV rW r� � s
z AbortingClient.writeAndAbortN)r� r� r� r� r� r� rV rV rV rW r� � s r� c @ r� )�AbortingTwiceClientz@
Call abortConnection() twice, after writing some data.
c C � t �| � | j�� d S rU )r� r� r" r� r~ rV rV rW r� � r� z!AbortingTwiceClient.writeAndAbortN�r� r� r� r� r� rV rV rV rW r� � r� r� c @ r� )�AbortingThenLosingClientz;
Call abortConnection() and then loseConnection().
c C r� rU )r� r� r" r7 r~ rV rV rW r� � r� z&AbortingThenLosingClient.writeAndAbortNr� rV rV rV rW r� � r� r� c @ s@ e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )�ProducerAbortingClientzA
Call abortConnection from doWrite, via resumeProducing.
TFc C s. | j �dd � d| _| j �| d� d| _d S )Ns lalalai� TF)r" r� �inRegisterProducerr� r~ rV rV rW r�
s
z ProducerAbortingClient.writec C rl rU �r� r~ rV rV rW r0
r� z%ProducerAbortingClient.connectionMadec C s d| _ | js | j�� d| _ d S )NTF)r� r� r" r� r~ rV rV rW r2
s
z&ProducerAbortingClient.resumeProducingc C �
d| _ d S r� )�producerStoppedr~ rV rV rW r�
rr z$ProducerAbortingClient.stopProducingc C s, | j std��| jrtd��t�| |� d S )Nz&BUG: stopProducing() was never called.z(BUG: connectionLost called re-entrantly!)r� r� r� r7 r� rf rV rV rW r�
s
z%ProducerAbortingClient.connectionLostN) r� r� r� r� r� r� r� r0 r2 r� r� rV rV rV rW r� � s r� c @ sT e Zd ZdZdZdZdZdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dS )�StreamingProducerClienta�
Call abortConnection() when the other side has stopped reading.
In particular, we want to call abortConnection() only once our local
socket hits a state where it is no longer writeable. This helps emulate
the most common use case for abortConnection(), closing a connection after
a timeout, with write buffers being full.
Since it's very difficult to know when this actually happens, we just
write a lot of data, and assume at that point no more writes will happen.
Fr c C rl rU r� r~ rV rV rW r0 0
r� z&StreamingProducerClient.connectionMadec C s0 | j �| d� td�D ]
}| j �dd � q dS )zh
Write large amount to transport, then wait for a while for buffers to
fill up.
T�d s
1234567890i } N)r" r� �ranger� )rm �irV rV rW r� 3
s �z StreamingProducerClient.writec C r� )NF)ra r~ rV rV rW r2 <
rr z'StreamingProducerClient.resumeProducingc C r� rU rV r~ rV rV rW r� ?
r� z%StreamingProducerClient.stopProducingc C s$ | j rdS d| _ | j�d| j� dS )a&
Called when local buffer fills up.
The goal is to hit the point where the local file descriptor is not
writeable (or the moral equivalent). The fact that pauseProducing has
been called is not sufficient, since that can happen when Twisted's
buffers fill up but OS hasn't gotten any writes yet. We want to be as
close as possible to every buffer (including OS buffers) being full.
So, we wait a bit more after this for Twisted to write out a few
chunks, then abortConnection.
NTg{�G�z�?)ra r� r3 �doAbortr~ rV rV rW r1 B
s
z&StreamingProducerClient.pauseProducingc C s. | j s
t�td�� d| _| j�� d| _d S )Nz&BUG: We should be paused a this point.TF)ra rC r r� r� r" r� r~ rV rV rW r� W
s
z StreamingProducerClient.doAbortc C s | j j�� t�| |� d S rU )�
otherProtocolr" � startReadingr7 r� rf rV rV rW r� ^
rg z&StreamingProducerClient.connectionLostN)r� r� r� r� ra � extraWritesr� r0 r� r2 r� r1 r� r� rV rV rV rW r�
s r� c @ r� )� StreamingProducerClientLaterzX
Call abortConnection() from dataReceived, after bytes have been
exchanged.
c C s | j �d� d| _d S )Nr� F)r" r� r� r~ rV rV rW r0 j
s
z+StreamingProducerClientLater.connectionMadec C s | j s d| _ | �� d S d S r� )r� r� rl rV rV rW r� n
s �z)StreamingProducerClientLater.dataReceivedN�r� r� r� r� r0 r� rV rV rV rW r� d
s r� c @ r� )�ProducerAbortingClientLaterz�
Call abortConnection from doWrite, via resumeProducing.
Try to do so after some bytes have already been exchanged, so we
don't interrupt SSL handshake.
c C r� rU rV r~ rV rV rW r0 |
s z*ProducerAbortingClientLater.connectionMadec C rl rU r� rl rV rV rW r� �
r� z(ProducerAbortingClientLater.dataReceivedNr� rV rV rV rW r� t
s r� c @ r� )�DataReceivedRaisingClientzN
Call abortConnection(), and then throw exception, from dataReceived.
c C s | j �� td��)N�ONO)r" r� �ZeroDivisionErrorrl rV rV rW r� �
s
z&DataReceivedRaisingClient.dataReceivedNr� rV rV rV rW r� �
r� r� c @ r� )�ResumeThrowsClientzL
Call abortConnection() and throw exception from resumeProducing().
c C s | j s | j�� td��d S )Nzono!)r� r" r� r� r~ rV rV rW r2 �
s
�z"ResumeThrowsClient.resumeProducingc C s t �| |� d S rU )r7 r� rf rV rV rW r� �
r� z!ResumeThrowsClient.connectionLostN)r� r� r� r� r2 r� rV rV rV rW r� �
s r� c @ s� e Zd ZU dZdZee ed< d dd�Zdd� Z dd � Z
d
d � Z d d
� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS ) �AbortConnectionMixinz7
Unit tests for L{ITransport.abortConnection}.
Nr� c C s� t tf}ttf}tr|tjf }|tjf }|� }|� }||_||_t| ||| j�}| � |�
� g � | � |� � g � |durJ| � |j
j|f| � n| � |j
j|� | � |j
j|� dS )z�
A test runner utility function, which hooks up a matched pair of client
and server protocols.
We then run the reactor until both sides have disconnected, and then
verify that the right exception resulted.
N)r r r r� rO �Errorr� r> r� r� r� �getDelayedCallsr� �disconnectReasonrv ) rm � clientClass� serverClass�clientConnectionLostReason�clientExpectedExceptions�serverExpectedExceptionsrc r� r� rV rV rW � runAbortTest�
s* ��z!AbortConnectionMixin.runAbortTestc C � | � tt�S )z�
abortConnection() is called in dataReceived. The protocol should be
disconnected, but connectionLost should not be called re-entrantly.
)r� r� r� r~ rV rV rW �test_dataReceivedAbort�
r z+AbortConnectionMixin.test_dataReceivedAbortc C r� )z�
abortConnection() is called twice by client.
No exception should be thrown, and the connection will be closed.
)r� r� r� r~ rV rV rW � test_clientAbortsConnectionTwice�
� z5AbortConnectionMixin.test_clientAbortsConnectionTwicec C r� )z�
Client calls abortConnection(), followed by loseConnection().
No exception should be thrown, and the connection will be closed.
)r� r� r� r~ rV rV rW �.test_clientAbortsConnectionThenLosesConnection�
r� zCAbortConnectionMixin.test_clientAbortsConnectionThenLosesConnectionc C � | j tttd�S )z�
abortConnection() is called twice by server.
No exception should be thrown, and the connection will be closed.
�r� )r� r� r� r r~ rV rV rW � test_serverAbortsConnectionTwice�
�
�z5AbortConnectionMixin.test_serverAbortsConnectionTwicec C r� )z�
Server calls abortConnection(), followed by loseConnection().
No exception should be thrown, and the connection will be closed.
r� )r� r� r� r r~ rV rV rW �.test_serverAbortsConnectionThenLosesConnection�
r� zCAbortConnectionMixin.test_serverAbortsConnectionThenLosesConnectionc C � | � tt� dS )z�
abortConnection() is called in resumeProducing, before any bytes have
been exchanged. The protocol should be disconnected, but
connectionLost should not be called re-entrantly.
N)r� r� r7 r~ rV rV rW �test_resumeProducingAbort�
r� z.AbortConnectionMixin.test_resumeProducingAbortc C r� )z�
abortConnection() is called in resumeProducing, after some
bytes have been exchanged. The protocol should be disconnected.
)r� r� r� r~ rV rV rW � test_resumeProducingAbortLater s �z3AbortConnectionMixin.test_resumeProducingAbortLaterc C r� )a{
abortConnection() triggered by the write buffer being full.
In particular, the server side stops reading. This is supposed
to simulate a realistic timeout scenario where the client
notices the server is no longer accepting data.
The protocol should be disconnected, but connectionLost should not be
called re-entrantly.
N)r� r� r� r~ rV rV rW �test_fullWriteBuffer s z)AbortConnectionMixin.test_fullWriteBufferc C r� )z�
abortConnection() is triggered by a write buffer being full.
However, this buffer is filled after some bytes have been exchanged,
allowing a TLS handshake if we're testing TLS. The connection will
then be lost.
)r� r� r� r~ rV rV rW �%test_fullWriteBufferAfterByteExchange s z:AbortConnectionMixin.test_fullWriteBufferAfterByteExchangec C �. | j tttd� | �t�}| �t|�d� dS )aJ
dataReceived calls abortConnection(), and then raises an exception.
The connection will be lost, with the thrown exception
(C{ZeroDivisionError}) as the reason on the client. The idea here is
that bugs should not be masked by abortConnection, in particular
unexpected exceptions.
r� r� N)r� r� r� r� �flushLoggedErrorsr� rw �rm �errorsrV rV rW �test_dataReceivedThrows# � �
z,AbortConnectionMixin.test_dataReceivedThrowsc C r� )aM
resumeProducing calls abortConnection(), and then raises an exception.
The connection will be lost, with the thrown exception
(C{ZeroDivisionError}) as the reason on the client. The idea here is
that bugs should not be masked by abortConnection, in particular
unexpected exceptions.
r� r� N)r� r� r7 r� r� r� rw r� rV rV rW �test_resumeProducingThrows4 r� z/AbortConnectionMixin.test_resumeProducingThrowsrU )r� r� r� r� r� r r9 r� r� r� r� r� r� r� r� r� r� r� r� r� rV rV rV rW r� �
s
%
r� c @ s e Zd ZdZefZe� ZdS )�AbortConnectionTestsz5
TCP-specific L{AbortConnectionMixin} tests.
N)r� r� r� r� r) rQ r� r� rV rV rV rW r� F s
r� c @ s6 e Zd ZdZdd� Zee�� d�dd� �Zdd� Z d S )
�SimpleUtilityTestszJ
Simple, direct tests for helpers within L{twisted.internet.tcp}.
c C �* | � tjtdd�}| �|jd tj� dS )a
L{_resolveIPv6} raises a L{socket.gaierror} (L{socket.EAI_NONAME}) when
invoked with a non-numeric host. (In other words, it is passing
L{socket.AI_NUMERICHOST} to L{socket.getaddrinfo} and will not
accidentally block if it receives bad input.)
� localhostr� r N�r@ r` �gaierrorr5 r� r� �
EAI_NONAME�rm r rV rV rW �test_resolveNumericHostY s z*SimpleUtilityTests.test_resolveNumericHostz@The AI_NUMERICSERV flag is not supported by Microsoft providers.c C r� )a
L{_resolveIPv6} raises a L{socket.gaierror} (L{socket.EAI_NONAME}) when
invoked with a non-numeric port. (In other words, it is passing
L{socket.AI_NUMERICSERV} to L{socket.getaddrinfo} and will not
accidentally block if it receives bad input.)
rQ �httpr Nr r rV rV rW �test_resolveNumericServicec s z-SimpleUtilityTests.test_resolveNumericServicec C sR t dd�}| �t|�d� | �|d t� | �|d t� | �|dd� d� dS )zb
L{_resolveIPv6} discovers the flow info and scope ID of an IPv6
address.
rQ r r_ r� N)rQ r )r5 r� rw r� r� r� rV rV rW �test_resolveIPv6r s
z#SimpleUtilityTests.test_resolveIPv6N)
r� r� r� r� r r
rE rS r r rV rV rV rW r� S s
�
r� c @ �0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �BuffersLogsTestsz$
Tests for L{_BuffersLogs}.
c C s" d| _ g | _t| j | jj�| _d S )Nz
name.space)� namespacer` r1 rv � logBufferr~ rV rV rW r� � s zBuffersLogsTests.setUpc C sB | j �}|�d� | �| j� W d � dS 1 sw Y dS )zj
The context manager's logger does not log to provided observer
inside the block.
�An eventN)r r� r� r` )rm �loggerrV rV rW �test_buffersInBlock� s
"�z$BuffersLogsTests.test_buffersInBlockc C s| | j �}|�d� | �| j� W d � n1 sw Y | �dt| j�� | j\}| �|d d� | �|d | j� dS )zw
The context manager flushes its buffered logs when the block
terminates without an exception.
r
Nr� �
log_format�
log_namespace)r r� r� r` r� rw r )rm r �eventrV rV rW �test_flushesOnExit� s
�z#BuffersLogsTests.test_flushesOnExitc C s� G dd� dt �}| �|��" | j�}|�d� | �| j� |� �1 s#w Y W d � n1 s2w Y | �dt| j�� | j\}| �|d d� | �|d | j� dS )zz
The context manager flushes its buffered logs when the block
terminates because of an exception.
c @ r� )zGBuffersLogsTests.test_flushesOnExitWithException.<locals>.TestExceptionz@
An exception only raised by this test.
N�r� r� r� r� rV rV rV rW �
TestException� r� r r
Nr� r r ) r� r@ r r� r� r` r� rw r )rm r r r rV rV rW � test_flushesOnExitWithException� s
� �z0BuffersLogsTests.test_flushesOnExitWithExceptionN)r� r� r� r� r� r r r rV rV rV rW r
� s
r
c @ r� )� FileDescriptorReservationTestsz2
Tests for L{_FileDescriptorReservation}.
c s* g � _ � �� � _� fdd�}t|�� _d S )Nc s � j �t� jd�� � j d S )N�wr. )�reservedFileObjectsrv r� �tempfilerV r~ rV rW �fakeFileFactory� s
z=FileDescriptorReservationTests.setUp.<locals>.fakeFileFactory)r �mktempr r2 �
reservedFD)rm r rV r~ rW r� � s
z$FileDescriptorReservationTests.setUpc C r� )zi
L{_FileDescriptorReservation} instances provide
L{_IFileDescriptorReservation}.
N)r r3 r r~ rV rV rW r� � r� z5FileDescriptorReservationTests.test_providesInterfacec C sR | � t| j�d� td�D ]}| j�� | � t| j�d� | �| jd j� q
dS )zh
Multiple acquisitions without releases open the reservation
file exactly once.
r r� r� N)r� rw r r� r �reserver� rn )rm r] rV rV rW �test_reserveOpensFileOnce� s
�z8FileDescriptorReservationTests.test_reserveOpensFileOncec C sz t � }| �|j� |�� | �| j�� � | j�� | �| j�� � | �t t
�}| � t |�d� | � |d j
jtj� dS )z�
If reserving the file descriptor fails because of C{EMFILE},
the exception is suppressed but logged and the reservation
remains unavailable.
r� r N)r� rw r r} r� r � availabler r� r� r� r� rw rv r� r� )rm r� r� rV rV rW �test_reserveEMFILELogged� s
z7FileDescriptorReservationTests.test_reserveEMFILELoggedc C s6 t ttfD ]}|fdd�}t|�}| �||j� qdS )z�
Any exception raised while opening the reserve file that is
not an L{OSError} or L{IOError} whose errno is C{EMFILE} is
allowed through to the caller.
c S s | t jd d��)Nr� r� )r� r� )�
errorClassrV rV rW � failsWith� r� zWFileDescriptorReservationTests.test_reserveRaisesNonEMFILEExceptions.<locals>.failsWithN)r� r� rA r2 r@ r )rm r" r# � reserveFDrV rV rW �%test_reserveRaisesNonEMFILEExceptions� s
�zDFileDescriptorReservationTests.test_reserveRaisesNonEMFILEExceptionsc C s. | � | j�� � | j�� | �| j�� � dS )z]
The reservation is available after the file descriptor is
reserved.
N)r� r r r r� r~ rV rV rW �test_available� s
z-FileDescriptorReservationTests.test_availablec C sf | � t��$ | j� W d � n1 sw Y W d � dS W d � dS 1 s,w Y dS )zp
A reservation without an open file used as a context manager
raises a L{RuntimeError}.
N)r@ r� r r~ rV rV rW �test_enterFailsWithoutFile s ��"�z9FileDescriptorReservationTests.test_enterFailsWithoutFilec C sd | j �� | �| j �� � | j � | �| j �� � W d � n1 s#w Y | �| j �� � dS )zq
Entering a reservation closes its file for the duration of the
context manager's block.
N)r r r� r r� r~ rV rV rW �!test_enterClosesFileExitOpensFile s
�z@FileDescriptorReservationTests.test_enterClosesFileExitOpensFilec C sh G dd� dt �}| j�� | �|�� | j� |� �1 s w Y W d � dS 1 s-w Y dS )z�
An exception raised within a reservation context manager's
block does not prevent the file from being reopened.
c @ r� )zSFileDescriptorReservationTests.test_exitOpensFileOnException.<locals>.TestExceptionz>
An exception only used by this test.
Nr rV rV rV rW r " r� r N)r� r r r@ )rm r rV rV rW � test_exitOpensFileOnException s
�"�z<FileDescriptorReservationTests.test_exitOpensFileOnExceptionc s� G dd� dt �}G dd� dt �� dg�� �fdd�}t|�}|�� | �|�� � | �|�� |� |� �1 s8w Y W d � n1 sGw Y | �� �}| �t|�d � dS )
z�
An exception raised while re-opening the reserve file exiting
a reservation's context manager block is suppressed but
logged, allowing an exception raised within the block through.
c @ r� )z`FileDescriptorReservationTests.test_exitSuppressesReservationException.<locals>.AllowedExceptionzA
The exception allowed out of the block.
Nr rV rV rV rW �AllowedException3 r� r* c @ r� )zcFileDescriptorReservationTests.test_exitSuppressesReservationException.<locals>.SuppressedExceptionzM
An exception raised by the file descriptor factory.
Nr rV rV rV rW �SuppressedException8 r� r+ Fc s �d r� � �d�d< t �� S )Nr T)�io�BytesIOrV �r+ �calledrV rW �+failsWithSuppressedExceptionAfterSecondOpen? s z{FileDescriptorReservationTests.test_exitSuppressesReservationException.<locals>.failsWithSuppressedExceptionAfterSecondOpenNr� ) r� r2 r r� r r@ r� r� rw )rm r* r0 r r� rV r. rW �'test_exitSuppressesReservationException, s � � �
zFFileDescriptorReservationTests.test_exitSuppressesReservationExceptionN)r� r� r� r� r� r� r r! r% r& r' r( r) r1 rV rV rV rW r � s
r c @ r ) �"NullFileDescriptorReservationTestsz6
Tests for L{_NullFileDescriptorReservation}.
c C r� rU )r4 �nullReservedFDr~ rV rV rW r� Z r� z(NullFileDescriptorReservationTests.setUpc C r� )zd
L{_NullFileDescriptorReservation} provides
L{_IFileDescriptorReservation}.
N)r r3 r3 r~ rV rV rW r� ] r� z9NullFileDescriptorReservationTests.test_providesInterfacec C s | � | j�� � dS )zG
The null reserved file descriptor is never available.
N�r� r3 r r~ rV rV rW r& d s z1NullFileDescriptorReservationTests.test_availablec C sZ | � | j�� � | j� | � | j�� � W d � n1 s w Y | � | j�� � dS )zN
The null reserved file descriptor is a null context manager.
Nr4 r~ rV rV rW �test_contextManagerj s
�z6NullFileDescriptorReservationTests.test_contextManagerN)r� r� r� r� r� r� r& r5 rV rV rV rW r2 U s r2 )�r� r� r� r, r� r` � functoolsr �typingr r r r r r �unittestr
�zope.interfacer r �zope.interface.verifyr
r r� �twisted.internet.addressr r �twisted.internet.deferr r r r r r �twisted.internet.endpointsr r �twisted.internet.errorr r r r r r r r r! �twisted.internet.interfacesr" r# r$ r% r& r' r( r) r* r+ �twisted.internet.protocolr, r- r. �twisted.internet.tcpr/ r0 r1 r2 r3 r4 r5 �&twisted.internet.test.connectionmixinsr6 r7 r8 r9 r: r; r< r= r> �#twisted.internet.test.reactormixinsr? r@ rA �twisted.loggerrB �twisted.pythonrC �twisted.python.failurerD �twisted.python.runtimerE �twisted.test.proto_helpersrF rG �twisted.test.test_tcprH rI rJ rK �twisted.trial.unittestrL rM rN �OpenSSLrO � ImportErrorr� �twisted.internet.sslrP �srV r r� r� r� r' r( r rS �twisted.internet.testrS � win32GetLinkLocalIPv6AddressesrZ r� rT � posixGetLinkLocalIPv6Addressesr\ rb ri r� r� r� r� r� r� r� r� r� rS rU rW ru rv r{ r| r� r� r� r� r� r� r� r� r� r* r- r. rA rE rG rH rs r� r� r� �globals�update�makeTestCaseClassesr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r
r r2 rV rV rV rW �<module> sJ ,0 $, � ��
�<
,40
s
G MLD! %
�
� $ % "
B�
3�D
"E
(
07