/usr/lib/python3/dist-packages/twisted/internet/test/__pycache__
Edit: /usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/connectionmixins.cpython-310.pyc (24452B)
o
bN @ s d Z ddlZddlmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZmZ dd lmZmZmZ dd
lmZ dd lmZmZ dd lmZ dd
lm Z m Z m Z ddl m Z ddl!m"Z" ddl#m$Z$ dej%ej&fddZ'G dd deZ(G dd dZ)G dd deZ*dd Z+d d Z,G d d d eZ-G d d! d!Z.G d"d# d#eZ/G d$d% d%e(Z0G d&d' d'Z1G d(d) d)Z2G d*d+ d+Z3G d,d- d-Z4dS ).z?
Various helpers for tests for connection-oriented transports.
N)collect)Optional)ref) verifyObject)Deferred
gatherResults)
IConnector
IReactorFDSet)
ClientFactoryProtocol
ServerFactory)needsRunningReactor)contextlog)Failure) ILogContexterrmsg)platform)ClosingProtocol)SkipTestz 127.0.0.1c C s t | dd d }t ||}z.|| |t jkr6| }t |t jt jB d }||d fW | S | W | S | w )a
Ask the platform to allocate a free port on the specified interface, then
release the socket and return the address which was allocated.
@param interface: The local address to try to bind the port on.
@type interface: C{str}
@param type: The socket type which will use the resulting port.
@return: A two-tuple of address and port, like that returned by
L{socket.getsockname}.
r ) socket getaddrinfobindAF_INET6 getsockname getnameinfoNI_NUMERICHOSTNI_NUMERICSERVclose) interfacefamilytypeaddrprobesocknamehostname r) H/usr/lib/python3/dist-packages/twisted/internet/test/connectionmixins.py findFreePort s
r+ c @ $ e Zd ZdZdZdd Zdd ZdS )ConnectableProtocolab
A protocol to be used with L{runProtocolsWithReactor}.
The protocol and its pair should eventually disconnect from each other.
@ivar reactor: The reactor used in this test.
@ivar disconnectReason: The L{Failure} passed to C{connectionLost}.
@ivar _done: A L{Deferred} which will be fired when the connection is
lost.
Nc C s || _ || _dS )aN
Set attributes on the protocol that are known only externally; this
will be called by L{runProtocolsWithReactor} when this protocol is
instantiated.
@param reactor: The reactor used in this test.
@param done: A L{Deferred} which will be fired when the connection is
lost.
N)reactor_done)selfr. doner) r) r* _setAttributesJ s
z"ConnectableProtocol._setAttributesc C s || _ | jd | `d S N)disconnectReasonr/ callbackr0 reasonr) r) r* connectionLostX s z"ConnectableProtocol.connectionLost)__name__
__module__ __qualname____doc__r4 r2 r8 r) r) r) r* r- : s
r- c @ e Zd ZdZdd Zdd ZdS )EndpointCreatorzT
Create client and server endpoints that know how to connect to each other.
c C t )z
Return an object providing C{IStreamServerEndpoint} for use in creating
a server to use to establish the connection type to be tested.
NotImplementedErrorr0 r. r) r) r* serverc zEndpointCreator.serverc C r? )z
Return an object providing C{IStreamClientEndpoint} for use in creating
a client to use to establish the connection type to be tested.
r@ )r0 r.
serverAddressr) r) r* clientj rD zEndpointCreator.clientN)r9 r: r; r< rC rF r) r) r) r* r> ^ s r> c @ r= )_SingleProtocolFactoryz
Factory to be used by L{runProtocolsWithReactor}.
It always returns the same protocol (i.e. is intended for only a single
connection).
c C
|| _ d S r3 _protocol)r0 protocolr) r) r* __init__z
z _SingleProtocolFactory.__init__c C | j S r3 rI )r0 r% r) r) r*
buildProtocol} z$_SingleProtocolFactory.buildProtocolNr9 r: r; r< rL rO r) r) r) r* rG r s rG c s | |t |t t|}t| }||} fdd}|| dd }t|j|jg|fdd | | |fdd |
S )a
Connect two protocols using endpoints and a new reactor instance.
A new reactor will be created and run, with the client and server protocol
instances connected to each other using the given endpoint creator. The
protocols should run through some set of tests, then disconnect; when both
have disconnected the reactor will be stopped and the function will
return.
@param reactorBuilder: A L{ReactorBuilder} instance.
@param serverProtocol: A L{ConnectableProtocol} that will be the server.
@param clientProtocol: A L{ConnectableProtocol} that will be the client.
@param endpointCreator: An instance of L{EndpointCreator}.
@return: The reactor run by this test.
c s | }| S r3 )rF getHostconnect)pclientEndpoint)
clientFactoryendpointCreatorr. r) r* gotPort s
z(runProtocolsWithReactor.
.gotPortc S s t | d d S )NzConnection setup failed.)r r )resultr) r) r* failed z'runProtocolsWithReactor..failedc S r3 r) _) disconnectedr) r* z)runProtocolsWithReactor..c s t jS r3 )r
stopr] r. r) r* r` ) buildReactorr2 r rG rC listen addCallbackr r/
addErrback
runReactor) reactorBuilderserverProtocolclientProtocolrW
serverFactoryserverEndpointdrX rZ r) )rV r_ rW r. r* runProtocolsWithReactor s
rp c C s2 t | r | S d| jjv r| jS td| )zZ
Like L{IReactorFDSet.getWriters}, but with support for IOCP reactor as
well.
IOCPzCannot find writers on )r
providedBy
getWriters __class__r9 handles Exceptionrc r) r) r* _getWriters s
rw c @ r= )_AcceptOneClientaN
This factory fires a L{Deferred} with a protocol instance shortly after it
is constructed (hopefully long enough afterwards so that it has been
connected to a transport).
@ivar reactor: The reactor used to schedule the I{shortly}.
@ivar result: A L{Deferred} which will be fired with the protocol instance.
c C s || _ || _d S r3 )r. rY )r0 r. rY r) r) r* rL
z_AcceptOneClient.__init__c C s$ t | |}| jd| jj| |S )Nr )r rO r. callLaterrY r5 )r0 r% rK r) r) r* rO s z _AcceptOneClient.buildProtocolNrQ r) r) r) r* rx s
rx c @ s( e Zd ZdZdd Zdd Zdd ZdS ) _SimplePullProducerz
A pull producer which writes one byte whenever it is resumed. For use by
C{test_unregisterProducerAfterDisconnect}.
c C rH r3 )consumer)r0 r| r) r) r* rL rM z _SimplePullProducer.__init__c C s d S r3 r) r0 r) r) r*
stopProducing s z!_SimplePullProducer.stopProducingc C s t d | jd d S )NProducer.resumeProducing x)r r r| writer} r) r) r* resumeProducing s
z#_SimplePullProducer.resumeProducingN)r9 r: r; r< rL r~ r r) r) r) r* r{ s
r{ c @ r, )StopzQ
A client factory which stops a reactor when a connection attempt fails.
Nc C rH r3 rc rB r) r) r* rL rM z
Stop.__init__c C s& || _ td| | j d S )NzStop(CF) cCFailed: )
failReasonr getErrorMessager. rb )r0 connectorr7 r) r) r* clientConnectionFailed s zStop.clientConnectionFailed)r9 r: r; r< r rL r r) r) r) r* r s
r c @ s0 e Zd ZdZdd Zdd Zdd Zdd Zd
S ) ClosingLaterProtocolz
ClosingLaterProtocol exchanges one byte with its peer and then disconnects
itself. This is mostly a work-around for the fact that connectionMade is
called before the SSL handshake has completed.
c C s d | _ || _d S r3 )lostConnectionReasononConnectionLost)r0 r r) r) r* rL ry z ClosingLaterProtocol.__init__c C s t d d S )N#ClosingLaterProtocol.connectionMade)r r} r) r) r* connectionMade s r c C s t d| | j d S )Nz"ClosingLaterProtocol.dataReceived r transportloseConnectionr0 bytesr) r) r* dataReceived z!ClosingLaterProtocol.dataReceivedc C s t d || _| j| d S )N#ClosingLaterProtocol.connectionLost)r r r r5 r6 r) r) r* r8 s r N)r9 r: r; r< rL r r r8 r) r) r) r* r s r c @ s: e Zd ZU dZdZee ed< dd Zdd Z dd Z
dS )
ConnectionTestsMixinzg
This mixin defines test methods which should apply to most L{ITransport}
implementations.
N endpointsc C sP G dd dt }|d}|d}t| ||| j | d|j | d|j dS )z
Client and server transports implement L{ILoggingContext.logPrefix} to
return a message reflecting the protocol they are running.
c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
zDConnectionTestsMixin.test_logPrefix..CustomLogPrefixProtocolc S s || _ d | _d S r3 )_prefixsystem)r0 prefixr) r) r* rL ry zMConnectionTestsMixin.test_logPrefix..CustomLogPrefixProtocol.__init__c S s | j d d S )N a)r r r} r) r) r* r $ r[ zSConnectionTestsMixin.test_logPrefix..CustomLogPrefixProtocol.connectionMadec S rN r3 )r r} r) r) r* logPrefix' rP zNConnectionTestsMixin.test_logPrefix..CustomLogPrefixProtocol.logPrefixc S s6 t td | _| jd d|v r| j d S d S )Nr b)r getr r r r r r r) r) r* r * s
zQConnectionTestsMixin.test_logPrefix..CustomLogPrefixProtocol.dataReceivedN)r9 r: r; rL r r r r) r) r) r* CustomLogPrefixProtocol s
r z
Custom Clientz
Custom ServerN)r- rp r assertInr )r0 r rF rC r) r) r* test_logPrefix s z#ConnectionTestsMixin.test_logPrefixc s g t fdd}jt| fddfdd}t| ddg dS ) z}
After a connection is disconnected, L{ITransport.write} and
L{ITransport.writeSequence} are no-ops.
c t S r3 r r) ) serverConnectionLostDeferredr) r* r` B z@ConnectionTestsMixin.test_writeAfterDisconnect..c s t d| j| }t fdd}|t|}dd }|| j fdd}
|
| t gS )N
Listening on c r r3 r r) lostConnectionDeferredr) r* r` L r zSConnectionTestsMixin.test_writeAfterDisconnect..listening..c S s t d| | jd d S )NzAbout to write to r )r r r protor) r) r* r O s zPConnectionTestsMixin.test_writeAfterDisconnect..listening..writec s8 t | d | jd | jddg d d S )Nz
disconnecteds some bytes to get losts somes moreT)r r r
writeSequenceappendr finishedr) r* r_ U s zWConnectionTestsMixin.test_writeAfterDisconnect..listening..disconnected) r rR r rF r rS r
forProtocol addCallbackserrbackrg r )portendpointrK rF r r_ )r r. r0 r r r* listeningG s
zAConnectionTestsMixin.test_writeAfterDisconnect..listeningc s* t fdd d S )Nc S r3 rb ignoredrc r) r* r` b r zRConnectionTestsMixin.test_writeAfterDisconnect..onListen..)rg rh r r) r portDeferredr. r) r* onListen_
z@ConnectionTestsMixin.test_writeAfterDisconnect..onListenTN)
re r r rC rf r r r
ri assertEqual)r0 rK r r) )r r r r. r0 r r* test_writeAfterDisconnect8 s
z.ConnectionTestsMixin.test_writeAfterDisconnectc s t t t } jtt fddfdd}t
| d t
| dS )z
After the connection a protocol is being used for is closed, the
reactor discards all of its references to the protocol.
c s^ t d| j| }|t fdd}dd }|| |j S )Nr c r\ r3 r) r) )rl r) r* r` { ra zaConnectionTestsMixin.test_protocolGarbageAfterLostConnection..listening..c S s t d| | j d S )NzAbout to disconnect r r r) r) r*
disconnect} r zcConnectionTestsMixin.test_protocolGarbageAfterLostConnection..listening..disconnect)
r rR r rF rS r
r rg rh r )r r rF r )rl r r. r0 r) r* r w s
zOConnectionTestsMixin.test_protocolGarbageAfterLostConnection..listeningc s* t fdd d S )Nc r r3 r r rc r) r* r` r zcConnectionTestsMixin.test_protocolGarbageAfterLostConnection..onListening..)rg rh r addBothr) r r) r* onListening r zQConnectionTestsMixin.test_protocolGarbageAfterLostConnection..onListeningN)r r r re r rC rf r r r r
ri r assertIsNone)r0 clientRefr r) )rl r r r r. r0 r* 'test_protocolGarbageAfterLostConnectioni s
z __annotations__r r r r) r) r) r* r s
1r c @ s e Zd ZdZdd ZdS )LogObserverMixinzL
Mixin for L{TestCase} subclasses which want to observe log events.
c C s$ g }t |j | t j|j |S r3 )r addObserverr
addCleanupremoveObserver)r0 loggedMessagesr) r) r* observe s zLogObserverMixin.observeN)r9 r: r; r< r r) r) r) r* r s r c @ s e Zd ZdZdZdd ZdS )BrokenContextFactoryzv
A context factory with a broken C{getContext} method, for exercising the
error handling for such a case.
zSome path was wrong maybec C s
t | jr3 )
ValueErrormessager} r) r) r*
getContext rM z BrokenContextFactory.getContextN)r9 r: r; r< r r r) r) r) r* r s r c @ s8 e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd S )
StreamClientTestsMixinau
This mixin defines tests applicable to SOCK_STREAM client implementations.
This must be mixed in to a L{ReactorBuilder
} subclass, as it
depends on several of its methods.
Then the methods C{connect} and C{listen} must defined, defining a client
and a server communicating with each other.
c C s* | }| |t }| tt| dS )zR
The C{connect} method returns an object providing L{IConnector}.
N)re rS r
assertTruer r )r0 r. r r) r) r* test_interface s z%StreamClientTestsMixin.test_interfacec s* t fdd dS )zl
The reactor can be stopped by a client factory's
C{clientConnectionFailed} method.
c s t S r3 )rS r r) r. r0 r) r* r` s zPStreamClientTestsMixin.test_clientConnectionFailedStopsReactor..N)re r
ri r} r) r r* 'test_clientConnectionFailedStopsReactor s z>StreamClientTestsMixin.test_clientConnectionFailedStopsReactorc sn tt g G fdddt}t | _t fdd dS )z
This test checks that we correctly get notifications event for a
client. This ought to prevent a regression under Windows using the
GTK2 reactor. See #3925.
c s e Zd Z fddZdS )zAStreamClientTestsMixin.test_connectEvent..CheckConnectionc s | d S r3 )r rb r} connectedr. r) r* r s
zPStreamClientTestsMixin.test_connectEvent..CheckConnection.connectionMadeN)r9 r: r; r r) r r) r* CheckConnection s r c s S r3 )rS r) )rV r. r0 r) r* r` rd z:StreamClientTestsMixin.test_connectEvent..N)
re rf r r r r rK r
runr )r0 r r) )rV r r. r0 r* test_connectEvent s z(StreamClientTestsMixin.test_connectEventc s | | tt t tj fdd g G fdddt
}t }||_ |
| | | d d dS )a3
If a producer is unregistered from a transport after the transport has
been disconnected (by the peer) and after C{loseConnection} has been
called, the transport is not re-added to the reactor as a writer as
would be necessary if the transport were still connected.
c r r3 r ignrc r) r* r` r zOStreamClientTestsMixin.test_unregisterProducerAfterDisconnect..c s0 e Zd ZdZdd Z fddZdd ZdS ) zUStreamClientTestsMixin.test_unregisterProducerAfterDisconnect..ClientProtocolz
Protocol to connect, register a producer, try to lose the
connection, wait for the server to disconnect from us, and then
unregister the producer.
c S s, t d | jt| jd | j d S )N ClientProtocol.connectionMadeF)r r r registerProducerr{ r r} r) r) r* r s
zdStreamClientTestsMixin.test_unregisterProducerAfterDisconnect..ClientProtocol.connectionMadec s4 t d | | jtv d d S Nz ClientProtocol.connectionLost)r r
unregisterr r rw r5 r6 r r. writingr) r* r8 s
zdStreamClientTestsMixin.test_unregisterProducerAfterDisconnect..ClientProtocol.connectionLostc S s t d | j d S )NClientProtocol unregister)r r r unregisterProducerr} r) r) r* r
z`StreamClientTestsMixin.test_unregisterProducerAfterDisconnect..ClientProtocol.unregisterN)r9 r: r; r< r r8 r r) r r) r* ClientProtocol s
r r z/Transport was writing after unregisterProducer.N)re rf r r r r rh r r rg r r
rK rS ri assertFalse)r0 r rV r) r r* &test_unregisterProducerAfterDisconnect s
z=StreamClientTestsMixin.test_unregisterProducerAfterDisconnectc s | ddg}jj}||v rt rtdG dd d | tt t
t j
fdd G fdd d t }t }||_| | | d
S ) z
If C{loseConnection} is called while a producer is registered with the
transport, the connection is closed after the producer is unregistered.
Glib2Reactor Gtk2Reactorz=A pygobject/pygtk bug disables this functionality on Windows.c @ s e Zd Zdd ZdS )zFStreamClientTestsMixin.test_disconnectWhileProducing..Producerc S s t d d S )Nr )r r r} r) r) r* r % s zVStreamClientTestsMixin.test_disconnectWhileProducing..Producer.resumeProducingN)r9 r: r; r r) r) r) r* Producer$ s r c r r3 r r rc r) r* r` , r zFStreamClientTestsMixin.test_disconnectWhileProducing..c s8 e Zd ZdZ fddZfddZfddZdS ) zLStreamClientTestsMixin.test_disconnectWhileProducing..ClientProtocolz
Protocol to connect, register a producer, try to lose the
connection, unregister the producer, and wait for the connection to
actually be lost.
c s< t d | j d | j djd| j d S )Nr Fr )r r r r r rz r r} )r r. r) r* r 5 s
z[StreamClientTestsMixin.test_disconnectWhileProducing..ClientProtocol.connectionMadec s0 t d | j d jttd d S )Nr g ?zConnection was not lost)r r r r rz r r rv r} )r r. r) r* r ? s
zWStreamClientTestsMixin.test_disconnectWhileProducing..ClientProtocol.unregisterc s t d d d S r )r r r5 r6 r r) r* r8 I r z[StreamClientTestsMixin.test_disconnectWhileProducing..ClientProtocol.connectionLostN)r9 r: r; r< r r r8 r) r r r. r) r* r . s
r N)re rt r9 r isWindowsr rf r r r r rh r r rg r
rK rS ri )r0 skippedReactorsreactorClassNamer rV r) r r* test_disconnectWhileProducing s" z4StreamClientTestsMixin.test_disconnectWhileProducingN) r9 r: r; r< r r r r r r) r) r) r* r s .r )5r< r gcr typingr weakrefr zope.interface.verifyr twisted.internet.deferr r twisted.internet.interfacesr r twisted.internet.protocolr
r r #twisted.internet.test.reactormixinsr
twisted.pythonr r twisted.python.failurer twisted.python.logr r r twisted.python.runtimer twisted.test.test_tcpr twisted.trial.unittestr AF_INET SOCK_STREAMr+ r- r> rG rp rw rx r{ r r r r r r r) r) r) r* s> $4