�
Ϫ�f~t � �f � d Z ddlmZ ddlmZ ddlmZmZ ddlm Z m
Z
m Z m Z ddl
mZmZ ddlmZ ddlmZmZ dd lmZmZmZmZ dd
lmZmZ dd l m Z m Z m Z dd l m!Z! dd
l"m#Z# ddl$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5m6Z6 ddl7m8Z8 ddl9m:Z: ddl;m<Z< g d�Z= ed� Z> G d� dej~ � Z@ G d� de8j� � ZB G d� d� ZC ee0e&e)� G d � d � � ZD G d � d eD� ZE G d � d!e� ZF ee'� G d"� d#� � ZG ee%� G d$� d%� � ZH ee*e.e-e/e,e+� G d&� d'� � ZI eeI� D ] ZJ eeJeI� �
G d(� d)eIe2� ZK ee.e-e/e,� G d*� d+� � ZL G d,� d-� ZMd.� ZN ee4� G d/� d0e e5 � � ZOy1)2zJ
Assorted functionality which is commonly useful when writing unit tests.
� )� annotations)�BytesIO)�AF_INET�AF_INET6)�Callable�Iterator�Sequence�overload)�
implementedBy� implementer)� verifyClass)� ParamSpec�Self)�address�error�protocol�task)�_dataMustBeBytes�
isIPv6Address)� IPv4Address� IPv6Address� UNIXAddress)�Deferred)�UnsupportedAddressFamily) �
IConnector� IConsumer�IListeningPort� IProtocol�
IPushProducer� IReactorCore�
IReactorFDSet�IReactorSocket� IReactorSSL� IReactorTCP� IReactorUNIX�
ITransport)�Clock)� ILogObserver�LogEvent� LogPublisher)�basic)�failure)�TestCase)�AccumulatingProtocol�LineSendingProtocol�FakeDatagramTransport�StringTransport� StringTransportWithDisconnection�StringIOWithoutClosing�_FakeConnector� _FakePort�
MemoryReactor�MemoryReactorClock�RaisingMemoryReactor�NonStreamingProducer�waitUntilAllDisconnected�EventLoggingObserver�_Pc �: � e Zd ZdZdxZZdZdZdZdZ d� Z
d� Z d� Z y)r. ad
L{AccumulatingProtocol} is an L{IProtocol} implementation which collects
the data delivered to it and can fire a Deferred when it is connected or
disconnected.
@ivar made: A flag indicating whether C{connectionMade} has been called.
@ivar data: Bytes giving all the data passed to C{dataReceived}.
@ivar closed: A flag indicated whether C{connectionLost} has been called.
@ivar closedReason: The value of the I{reason} parameter passed to
C{connectionLost}.
@ivar closedDeferred: If set to a L{Deferred}, this will be fired when
C{connectionLost} is called.
r N� c �� � d| _ | j �P| j j �9| j j }d | j _ |j | � y y y �N� )�made�factory�protocolConnectionMade�callback��self�ds �:/usr/lib/python3/dist-packages/twisted/internet/testing.py�connectionMadez#AccumulatingProtocol.connectionMadeX sR � ��� � �<�<� #�� � �(K�(K�(W�� � �3�3�A�26�D�L�L� /�
�J�J�t� � )X� #r> c �. � | xj |z
c_ y �N)�data�rG rM s rI � dataReceivedz!AccumulatingProtocol.dataReceived_ s � � � � �T�� r> c � � d| _ || _ | j �&| j d c}| _ |j d � y y r@ )�closed� closedReason�closedDeferredrE )rG �reasonrH s rI �connectionLostz#AccumulatingProtocol.connectionLostb sC � ��� � "��� � � � *�%)�%8�%8�$� "�A�t�"�
�J�J�t� � +r> )
�__name__�
__module__� __qualname__�__doc__rB rQ rR rS rM rC rJ rO rU � r> rI r. r. @ s6 � � � ��D�6��L��N� �D��G� �� r> r. c �* � e Zd ZdZdd�Zd� Zd� Zd� Zy)r/ Fc �4 � |d d | _ g | _ || _ y rL )�lines�response�start)rG r] r_ s rI �__init__z LineSendingProtocol.__init__m s � ��1�X��
���
���
r> c �b � | j r#| j D ] }| j |� � y y rL )r_ r] �sendLine�rG �lines rI rJ z"LineSendingProtocol.connectionMader s- � � �:�:� �
�
�
$���
�
�d�#�
$� r> c � � | j s)| j D ] }| j |� � g | _ | j j |� y rL )r_ r] rb r^ �appendrc s rI � lineReceivedz LineSendingProtocol.lineReceivedw sA � ��z�z� �
�
�
$���
�
�d�#�
$��D�J� �
�
� � �T�"r> c � � d| _ y �NT)�lostConn)rG rT s rI rU z"LineSendingProtocol.connectionLost~ s � � ��
r> N)T)rV rW rX rj r` rJ rg rU rZ r> rI r/ r/ j s � ��H��
$�
#� r> r/ c �* � e Zd Z e� Zd� Zefd�Zy)r0 c � � g | _ y rL )�written�rG s rI r` z FakeDatagramTransport.__init__� s � ��� r> c �> � | j j ||f� y rL )rm rf )rG �packet�addrs rI �writezFakeDatagramTransport.write� s � � � � ���V�T�N�+r> N)rV rW rX �object�noAddrr` rr rZ r> rI r0 r0 � s � �
�X�F�� "(� ,r> r0 c � � e Zd ZdZdZdZdZdZdZdZ dZ
dd�Z d� Z d� Z
d� Zd � Zd
� Zd � Zd � Zd
� Zd� Zd� Zd� Zd� Zd� Zd� Zy)r1 a�
A transport implementation which buffers data in memory and keeps track of
its other state without providing any behavior.
L{StringTransport} has a number of attributes which are not part of any of
the interfaces it claims to implement. These attributes are provided for
testing purposes. Implementation code should not use any of these
attributes; they are not provided by other transports.
@ivar disconnecting: A C{bool} which is C{False} until L{loseConnection} is
called, then C{True}.
@ivar disconnected: A C{bool} which is C{False} until L{abortConnection} is
called, then C{True}.
@ivar producer: If a producer is currently registered, C{producer} is a
reference to it. Otherwise, L{None}.
@ivar streaming: If a producer is currently registered, C{streaming} refers
to the value of the second parameter passed to C{registerProducer}.
@ivar hostAddr: L{None} or an object which will be returned as the host
address of this transport. If L{None}, a nasty tuple will be returned
instead.
@ivar peerAddr: L{None} or an object which will be returned as the peer
address of this transport. If L{None}, a nasty tuple will be returned
instead.
@ivar producerState: The state of this L{StringTransport} in its capacity
as an L{IPushProducer}. One of C{'producing'}, C{'paused'}, or
C{'stopped'}.
@ivar io: A L{io.BytesIO} which holds the data which has been written to
this transport since the last call to L{clear}. Use L{value} instead
of accessing this directly.
@ivar _lenient: By default L{StringTransport} enforces that
L{resumeProducing} is not called after the connection is lost. This is
to ensure that any code that does call L{resumeProducing} after the
connection is lost is not blindly expecting L{resumeProducing} to have
any impact.
However, if your test case is calling L{resumeProducing} after
connection close on purpose, and you know it won't block expecting
further data to show up, this flag may safely be set to L{True}.
Defaults to L{False}.
@type lenient: L{bool}
FN� producingc �d � | j � |�|| _ |�|| _ d| _ || _ y ri )�clear�hostAddr�peerAddr� connected�_lenient)rG � hostAddress� peerAddress�lenients rI r` zStringTransport.__init__� s3 � � �
�
� � � "� '�D�M� � "� '�D�M� ��� ��
r> c �" � t � | _ y)z�
Discard all data written to this transport so far.
This is not a transport method. It is intended for tests. Do not use
it in implementation code.
N)r �iorn s rI rx zStringTransport.clear� s
� � �)��r> c �6 � | j j � S )aN
Retrieve all data which has been buffered by this transport.
This is not a transport method. It is intended for tests. Do not use
it in implementation code.
@return: A C{bytes} giving all data written to this transport since the
last call to L{clear}.
@rtype: C{bytes}
)r� �getvaluern s rI �valuezStringTransport.value� s � � �w�w� � �!�!r> c �P � t |� | j j |� y rL )r r� rr rN s rI rr zStringTransport.write� s � ��� � ���
�
�d�r> c �X � | j j dj |� � y )Nr> )r� rr �joinrN s rI �
writeSequencez StringTransport.writeSequence� s � � ���
�
�c�h�h�t�n�%r> c � � d| _ y)z~
Close the connection. Does nothing besides toggle the C{disconnecting}
instance variable to C{True}.
TN)�
disconnectingrn s rI �loseConnectionz StringTransport.loseConnection� s � �
"��r> c �2 � d| _ | j � y)z�
Abort the connection. Same as C{loseConnection}, but also toggles the
C{aborted} instance variable to C{True}.
TN)� disconnectedr� rn s rI �abortConnectionz StringTransport.abortConnection� s � �
!��� ��� r> c �` � | j �t j ddd� S | j S )N�TCPz 192.168.1.1i1� )rz r r rn s rI �getPeerzStringTransport.getPeer s+ � � �=�=� ��&�&�u�m�U�C� C��}�}� r> c �` � | j �t j ddd� S | j S )Nr� z10.0.0.1i90 )ry r r rn s rI �getHostzStringTransport.getHost s+ � � �=�=� ��&�&�u�j�%�@� @��}�}� r> c �N � | j � t d� �|| _ || _ y )Nz Cannot register two producers��producer� RuntimeError� streaming)rG r� r� s rI �registerProducerz StringTransport.registerProducer
s& � � �=�=� $� � >�?� ?� ��
�"��r> c �N � | j � t d� �d | _ d | _ y )Nz5Cannot unregister a producer unless one is registeredr� rn s rI �unregisterProducerz"StringTransport.unregisterProducer s& � � �=�=� � � V�W� W� ��
� ��r> c �~ � | j r| j s t d� �| j dk( r t d� �y )Nz,Cannot resume producing after loseConnection�stoppedz Cannot resume a stopped producer)r� r| r� �
producerStatern s rI � _checkStatezStringTransport._checkState s< � � � � �d�m�m� � M�N� N� � � �� *� � A�B� B� +r> c �2 � | j � d| _ y )N�paused�r� r� rn s rI �pauseProducingz StringTransport.pauseProducing s � � ���� %��r> c � � d| _ y )Nr� )r� rn s rI �
stopProducingz StringTransport.stopProducing$ s
� � &��r> c �2 � | j � d| _ y )Nrv r� rn s rI �resumeProducingz StringTransport.resumeProducing' s � � ���� (��r> )NNF)rV rW rX rY r� r� r� r� ry rz r� r` rx r� rr r� r� r� r� r� r� r� r� r� r� r� rZ r> rI r1 r1 � sy � �1�f �M��L��H��I��H��H� �M� � � "� �&�"� � �
� #� �C� &�'�)r> r1 c �" � e Zd ZU dZded< d� Zy)r2 zt
A L{StringTransport} which on disconnection will trigger the connection
lost on the attached protocol.
r r c � � | j rId| _ | j j t j t j d� � � y y )NFzBye.)r{ r rU r, �Failurer �ConnectionDonern s rI r� z/StringTransportWithDisconnection.loseConnection4 s; � � �>�>� "�D�N� �M�M� (� (�� � ��9M�9M�f�9U�)V� W� r> N)rV rW rX rY �__annotations__r� rZ r> rI r2 r2 , s � ��
��Xr> r2 c � � e Zd ZdZd� Zy)r3 z)
A BytesIO that can't be closed.
c � � y)z
Do nothing.
NrZ rn s rI �closez StringIOWithoutClosing.close? � � r> N)rV rW rX rY r� rZ r> rI r3 r3 : s � �� r> r3 c �( � e Zd ZdZd� Zd� Zd� Zd� Zy)r5 z�
A fake L{IListeningPort} to be used in tests.
@ivar _hostAddress: The L{IAddress} this L{IListeningPort} is pretending
to be listening on.
c � � || _ y)zz
@param hostAddress: An L{IAddress} this L{IListeningPort} should
pretend to be listening on.
N�� _hostAddress)rG r} s rI r` z_FakePort.__init__N s � �
(��r> c � � y)zQ
Fake L{IListeningPort.startListening} that doesn't do anything.
NrZ rn s rI �startListeningz_FakePort.startListeningU r� r> c � � y)zP
Fake L{IListeningPort.stopListening} that doesn't do anything.
NrZ rn s rI �
stopListeningz_FakePort.stopListeningZ r� r> c � � | j S )zN
Fake L{IListeningPort.getHost} that returns our L{IAddress}.
r� rn s rI r� z_FakePort.getHost_ s � � � � � r> N)rV rW rX rY r` r� r� r� rZ r> rI r5 r5 E s � ��(� �
�
!r> r5 c �6 � e Zd ZdZdZdZd� Zd� Zd� Zd� Z d� Z
y) r4 a'
A fake L{IConnector} that allows us to inspect if it has been told to stop
connecting.
@ivar stoppedConnecting: has this connector's
L{_FakeConnector.stopConnecting} method been invoked yet?
@ivar _address: An L{IAddress} provider that represents our destination.
Fc � � || _ y)zs
@param address: An L{IAddress} provider that represents this
connector's destination.
N��_address)rG r s rI r` z_FakeConnector.__init__u s � �
��
r> c � � d| _ y)zw
Implement L{IConnector.stopConnecting} and set
L{_FakeConnector.stoppedConnecting} to C{True}
TN)�stoppedConnectingrn s rI �stopConnectingz _FakeConnector.stopConnecting| s � �
"&�� r> c � � d| _ y)z@
Implement L{IConnector.disconnect} as a no-op.
TN)�
_disconnectedrn s rI �
disconnectz_FakeConnector.disconnect� s � � "��r> c � � y)z=
Implement L{IConnector.connect} as a no-op.
NrZ rn s rI �connectz_FakeConnector.connect� r� r> c � � | j S )zp
Implement L{IConnector.getDestination} to return the C{address} passed
to C{__init__}.
r� rn s rI �getDestinationz _FakeConnector.getDestination� s � �
�}�}� r> N) rV rW rX rY r� r� r` r� r� r� r� rZ r> rI r4 r4 f s- � �� �M� �� �&�"� �
r> r4 c �� � e Zd ZdZd� Zd� Zd d�Zd� Zd� Zd� Z d d�Z
d � Z d d
�Z d � Z
d!d �Zd
� Zd� Zd"d�Zd#d�Zd$d�Zd#d�Z d$d�Zd%d�Zd&d�Zd� Zd� Zd� Zd� Zd� Z d � Z d � Z y)'r6 a:
A fake reactor to be used in tests. This reactor doesn't actually do
much that's useful yet. It accepts TCP connection setup attempts, but
they will never succeed.
@ivar hasInstalled: Keeps track of whether this reactor has been installed.
@type hasInstalled: L{bool}
@ivar running: Keeps track of whether this reactor is running.
@type running: L{bool}
@ivar hasStopped: Keeps track of whether this reactor has been stopped.
@type hasStopped: L{bool}
@ivar hasCrashed: Keeps track of whether this reactor has crashed.
@type hasCrashed: L{bool}
@ivar whenRunningHooks: Keeps track of hooks registered with
C{callWhenRunning}.
@type whenRunningHooks: L{list}
@ivar triggers: Keeps track of hooks registered with
C{addSystemEventTrigger}.
@type triggers: L{dict}
@ivar tcpClients: Keeps track of connection attempts (ie, calls to
C{connectTCP}).
@type tcpClients: L{list}
@ivar tcpServers: Keeps track of server listen attempts (ie, calls to
C{listenTCP}).
@type tcpServers: L{list}
@ivar sslClients: Keeps track of connection attempts (ie, calls to
C{connectSSL}).
@type sslClients: L{list}
@ivar sslServers: Keeps track of server listen attempts (ie, calls to
C{listenSSL}).
@type sslServers: L{list}
@ivar unixClients: Keeps track of connection attempts (ie, calls to
C{connectUNIX}).
@type unixClients: L{list}
@ivar unixServers: Keeps track of server listen attempts (ie, calls to
C{listenUNIX}).
@type unixServers: L{list}
@ivar adoptedPorts: Keeps track of server listen attempts (ie, calls to
C{adoptStreamPort}).
@ivar adoptedStreamConnections: Keeps track of stream-oriented
connections added using C{adoptStreamConnection}.
c � � d| _ d| _ d| _ d| _ d| _ g | _ i | _ g | _ g | _ g | _ g | _
g | _ g | _ g | _
g | _ g | _ t! � | _ t! � | _ y)z0
Initialize the tracking lists.
FTN)� hasInstalled�running�hasRun�
hasStopped�
hasCrashed�whenRunningHooks�triggers�
tcpClients�
tcpServers�
sslClients�
sslServers� unixClients� unixServers� adoptedPorts�adoptedStreamConnections�
connectors�set�readers�writersrn s rI r` zMemoryReactor.__init__� s� � � "��� �� ��� � ��� ��� "�� ���
� ��� ��� ��� ��� ��� ��� ���(*��%� ����u�� ��u�� r> c � � d| _ y)zO
Fake install callable to emulate reactor module installation.
TN)r� rn s rI �installzMemoryReactor.install� s � � !��r> c � � t � ��zA
Not implemented; raises L{NotImplementedError}.
��NotImplementedError)rG �name�timeouts rI �resolvezMemoryReactor.resolve� �
� � "�#�#r> c � � | j du sJ �d| _ d| _ | j D ] \ }}} ||i |�� � | j � y)a
Fake L{IReactorCore.run}.
Sets C{self.running} to L{True}, runs all of the hooks passed to
C{self.callWhenRunning}, then calls C{self.stop} to simulate a request
to stop the reactor.
Sets C{self.hasRun} to L{True}.
FTN)r� r� r� �stop)rG �f�args�kwargss rI �runzMemoryReactor.run� sW � � �|�|�u�$�$�$��� ��� � #� 4� 4� �O�A�t�V�
�t� �v� � �
� � � r> c � � d| _ d| _ y)z�
Fake L{IReactorCore.run}.
Sets C{self.running} to L{False}.
Sets C{self.hasStopped} to L{True}.
FTN)r� r� rn s rI r� zMemoryReactor.stop s � � �� � ��r> c � � d| _ d| _ y)z�
Fake L{IReactorCore.crash}.
Sets C{self.running} to L{None}, because that feels crashy.
Sets C{self.hasCrashed} to L{True}.
NT)r� r� rn s rI �crashzMemoryReactor.crash s � � �� � ��r> c � � t � �r� r� )rG �delays rI �iteratezMemoryReactor.iterate r� r> c � � t � �r� r� )rG � eventTypes rI �fireSystemEventz MemoryReactor.fireSystemEvent# r� r> c � � | j j |i � }|j |g � }|j |||f� y)z�
Fake L{IReactorCore.run}.
Keep track of trigger by appending it to
self.triggers[phase][eventType].
N)r� �
setdefaultrf )rG �phaser� �callabler� �kw�
phaseTriggers�eventTypeTriggerss rI �addSystemEventTriggerz#MemoryReactor.addSystemEventTrigger) sB � � �
�
�0�0���;�
� )� 4� 4�Y�� C��� � �(�D�"�!5�6r> c � � t � �r� r� )rG � triggerIDs rI �removeSystemEventTriggerz&MemoryReactor.removeSystemEventTrigger: r� r> c �@ � | j j |||f� y)z�
Fake L{IReactorCore.callWhenRunning}.
Keeps a list of invocations to make in C{self.whenRunningHooks}.
N)r� rf )rG r� r� r� s rI �callWhenRunningz MemoryReactor.callWhenRunning@ s � �
� � �$�$�h��b�%9�:r> c �� � |t k( rt ddd� }n!|t k( rt ddd� }n
t � �| j
j
|||f� t |� S )zv
Fake L{IReactorSocket.adoptStreamPort}, that logs the call and returns
an L{IListeningPort}.
r� �0.0.0.0�� �::�r r r r r r� rf r5 )rG �fileno�
addressFamilyrC rq s rI �adoptStreamPortz MemoryReactor.adoptStreamPortI s^ � �
�G� #� �u�i��6�D�
�h�
&� �u�d�D�1�D�*�,� ,� ��� � �&�-��!A�B���� r> c �@ � | j j |||f� y)z�
Record the given stream connection in C{adoptedStreamConnections}.
@see:
L{twisted.internet.interfaces.IReactorSocket.adoptStreamConnection}
N)r� rf �rG �fileDescriptorr rC s rI �adoptStreamConnectionz#MemoryReactor.adoptStreamConnectionX s � �
�%�%�,�,�n�m�W�-U�Vr> c �� � |t k( rt ddd� }n!|t k( rt ddd� }n
t � �| j
j
||||f� t |� S )z�
Fake L{IReactorSocket.adoptDatagramPort}, that logs the call and
returns a fake L{IListeningPort}.
@see: L{twisted.internet.interfaces.IReactorSocket.adoptDatagramPort}
�UDPr r r r )rG r r r �
maxPacketSizerq s rI �adoptDatagramPortz MemoryReactor.adoptDatagramPorta s` � � �G� #� �u�i��6�D�
�h�
&� �u�d�D�1�D�*�,� ,� ��� � �&�-��=�!Q�R���� r> c � � | j j ||||f� t |� rt d||� }t |� S t dd|� }t |� S )zm
Fake L{IReactorTCP.listenTCP}, that logs the call and
returns an L{IListeningPort}.
r� r )r� rf r r r r5 )rG �portrC �backlog� interfacer s rI � listenTCPzMemoryReactor.listenTCPr s] � �
��� � ��g�w� � B�C� �� #�!�%��D�9�G� ��!�!� "�%��D�9�G���!�!r> Nc � � | j j |||||f� t |� rt t d||� � }nt t d||� � }|j
|� | j j |� |S )zj
Fake L{IReactorTCP.connectTCP}, that logs the call and
returns an L{IConnector}.
r� )r� rf r r4 r r �startedConnectingr� )rG �hostr rC r� � bindAddress�conns rI �
connectTCPzMemoryReactor.connectTCP~ st � �
��� � ��d�G�W�k� J�K� �� �!�+�e�T�4�"@�A�D�!�+�e�T�4�"@�A�D��!�!�$�'� ��� � �t�$�� r> c �n � | j j |||||f� t t dd|� � S )zm
Fake L{IReactorSSL.listenSSL}, that logs the call and
returns an L{IListeningPort}.
r� r )r� rf r5 r �rG r rC �contextFactoryr r s rI � listenSSLzMemoryReactor.listenSSL� s5 � �
��� � ��g�~�w� � R�S���U�I�t�<�=�=r> c �� � | j j ||||||f� t t d||� � }|j |� | j
j |� |S )zj
Fake L{IReactorSSL.connectSSL}, that logs the call and returns an
L{IConnector}.
r� )r� rf r4 r r r� )rG r r rC r r� r r s rI �
connectSSLzMemoryReactor.connectSSL� s` � �
��� � �
�4��.�'�;� G�
� �k�%��t� <�=���!�!�$�'� ��� � �t�$�� r> c �j � | j j |||||f� t t |� � S )zo
Fake L{IReactorUNIX.listenUNIX}, that logs the call and returns an
L{IListeningPort}.
)r� rf r5 r �rG r rC r �mode�wantPIDs rI �
listenUNIXzMemoryReactor.listenUNIX� s3 � �
��� � ��'�7�D�'� J�K���W�-�.�.r> c �� � | j j ||||f� t t |� � }|j |� | j
j |� |S )zl
Fake L{IReactorUNIX.connectUNIX}, that logs the call and returns an
L{IConnector}.
)r� rf r4 r r r� )rG r rC r� �checkPIDr s rI � connectUNIXzMemoryReactor.connectUNIX� sU � �
��� � ��'�7�H� E�F� �k�'� 2�3���!�!�$�'� ��� � �t�$�� r> c �: � | j j |� y)zW
Fake L{IReactorFDSet.addReader} which adds the reader to a local set.
N)r� �add�rG �readers rI � addReaderzMemoryReactor.addReader� � � �
� � ���� r> c �: � | j j |� y)zg
Fake L{IReactorFDSet.removeReader} which removes the reader from a
local set.
N)r� �discardr. s rI � removeReaderzMemoryReactor.removeReader� � � �
� � � � �V�$r> c �: � | j j |� y)zW
Fake L{IReactorFDSet.addWriter} which adds the writer to a local set.
N)r� r- �rG �writers rI � addWriterzMemoryReactor.addWriter� r1 r> c �: � | j j |� y)zg
Fake L{IReactorFDSet.removeWriter} which removes the writer from a
local set.
N)r� r3 r7 s rI � removeWriterzMemoryReactor.removeWriter� r5 r> c �, � t | j � S )zn
Fake L{IReactorFDSet.getReaders} which returns a list of readers from
the local set.
)�listr� rn s rI �
getReaderszMemoryReactor.getReaders� � � �
�D�L�L�!�!r> c �, � t | j � S )zn
Fake L{IReactorFDSet.getWriters} which returns a list of writers from
the local set.
)r= r� rn s rI �
getWriterszMemoryReactor.getWriters� r? r> c �l � | j j � | j j � y)zt
Fake L{IReactorFDSet.removeAll} which removed all readers and writers
from the local sets.
N)r� rx r� rn s rI � removeAllzMemoryReactor.removeAll� s$ � �
� � ��� � � � ��� r> )�
)r ) r� �strr� rE r� �Callable[_P, object]r� �_P.argsr� � _P.kwargs�return�None)r� rF r� rG r� rH rI rJ )i ��2 � �� N�rL i� r �rO r ) rV rW rX rY r` r� r� r� r� r� r� r� r� r r r
r r r r r! r# r( r+ r0 r4 r9 r; r> rA rC rZ r> rI r6 r6 � s� � �6�p �6!� $� �$ � �$� $� 7��7� �7� '� 7�
� 7� �
7�
�7�"$� ;�,�;�5<�;�DM�;�
�;�
� W� �"
"� � >� LP�
� /� �!� %�!� %�"�"� r> r6 c � � e Zd Zd� Zy)r7 c �X � t j | � t j | � y rL )r6 r` r' rn s rI r` zMemoryReactorClock.__init__� s � �� � �t�$�
���t� r> N)rV rW rX r` rZ r> rI r7 r7 � s � � r> r7 c �\ � e Zd ZdZd
d�Zd� Zdd�Zdd�Zdd�Z dd�Z dd �Z
dd
�Z d � Z d � Z
y)r8 z�
A fake reactor to be used in tests. It accepts TCP connection setup
attempts, but they will fail.
@ivar _listenException: An instance of an L{Exception}
@ivar _connectException: An instance of an L{Exception}
Nc � � || _ || _ y)z�
@param listenException: An instance of an L{Exception} to raise
when any C{listen} method is called.
@param connectException: An instance of an L{Exception} to raise
when any C{connect} method is called.
N)�_listenException�_connectException)rG �listenException�connectExceptions rI r` z RaisingMemoryReactor.__init__� s � � !0�� �!1�� r> c � � | j �)zb
Fake L{IReactorSocket.adoptStreamPort}, that raises
L{_listenException}.
�rV )rG r r rC s rI r
z$RaisingMemoryReactor.adoptStreamPort s � �
�#�#�#r> c � � | j �)zQ
Fake L{IReactorTCP.listenTCP}, that raises L{_listenException}.
r[ )rG r rC r r s rI r z RaisingMemoryReactor.listenTCP
� � � �#�#�#r> c � � | j �)zS
Fake L{IReactorTCP.connectTCP}, that raises L{_connectException}.
�rW )rG r r rC r� r s rI r z RaisingMemoryReactor.connectTCP � � � �$�$�$r> c � � | j �)zQ
Fake L{IReactorSSL.listenSSL}, that raises L{_listenException}.
r[ r s rI r! z RaisingMemoryReactor.listenSSL r] r> c � � | j �)zS
Fake L{IReactorSSL.connectSSL}, that raises L{_connectException}.
r_ )rG r r rC r r� r s rI r# z RaisingMemoryReactor.connectSSL s � � �$�$�$r> c � � | j �)zS
Fake L{IReactorUNIX.listenUNIX}, that raises L{_listenException}.
r[ r% s rI r( z RaisingMemoryReactor.listenUNIX' r] r> c � � | j �)zU
Fake L{IReactorUNIX.connectUNIX}, that raises L{_connectException}.
r_ )rG r rC r� r* s rI r+ z RaisingMemoryReactor.connectUNIX- r` r> c � � | j �)ze
Fake L{IReactorSocket.adoptDatagramPort}, that raises
L{_connectException}.
r_ )rG r
r r r s rI r z&RaisingMemoryReactor.adoptDatagramPort3 � � �
�$�$�$r> c � � | j �)zi
Fake L{IReactorSocket.adoptStreamConnection}, that raises
L{_connectException}.
r_ r s rI r z*RaisingMemoryReactor.adoptStreamConnection: rf r> )NNrK rN rP rQ )rV rW rX rY r` r
r r r! r# r( r+ r r rZ r> rI r8 r8 � s@ � �� 2�$�$� %� $� LP�%�$� %� %�%r> r8 c �6 � e Zd ZdZdZdZd� Zd� Zd� Zd� Z d� Z
y )
r9 z5
A pull producer which writes 10 times only.
r Fc �0 � || _ t � | _ y rL )�consumerr �result)rG rj s rI r` z NonStreamingProducer.__init__J s � � ��
� �j�� r> c �@ � | j �| j dk\ r t d� �| j j d| j fz � | xj dz
c_ | j dk( r+| j j � | j � yy)z/
Write the counter value once.
NrD z"BUG: resume after unregister/stop.s %drA )rj �counterr� rr r� �_donern s rI r� z$NonStreamingProducer.resumeProducingN sz � � �=�=� �D�L�L�B�$6� � C�D� D� �M�M� � ��� � �� 7� 8� �L�L�A� �L��|�|�r�!��
�
�0�0�2��
�
� � "r> c � � t d� �)z�
An implementation of C{IPushProducer.pauseProducing}. This should never
be called on a pull producer, so this just raises an error.
z"BUG: pause should never be called.)r� rn s rI r� z#NonStreamingProducer.pauseProducing[ s � �
�?�@�@r> c �P � d| _ | j }| `|j d� y)zQ
Fire a L{Deferred} so that users can wait for this to complete.
N)rj rk rE rF s rI rn zNonStreamingProducer._doneb s% � � ��
� �K�K�� �K� �
�
�4�r> c �2 � d| _ | j � y)z&
Stop all production.
TN)r� rn rn s rI r� z"NonStreamingProducer.stopProducingk s � � �� � �
�
� r> N) rV rW rX rY rm r� r` r� r� rn r� rZ r> rI r9 r9 B s- � �� �G��G�!� �A��r> r9 c �t ��� d���fd�}t j |� �| �_ �j dd�� S )a�
Take a list of disconnecting protocols, callback a L{Deferred} when they're
all done.
This is a hack to make some older tests less flaky, as
L{ITransport.loseConnection} is not atomic on all reactors (for example,
the CoreFoundation, which sometimes takes a reactor turn for CFSocket to
realise). New tests should either not use real sockets in testing, or take
the advice in
I{https://jml.io/pages/how-to-disconnect-in-twisted-really.html} to heart.
@param reactor: The reactor to schedule the checks on.
@type reactor: L{IReactorTime}
@param protocols: The protocols to wait for disconnecting.
@type protocols: A L{list} of L{IProtocol}s.
Nc �~ �� d�D � cg c] } | j j �� c} vr�j � y y c c} w ri )� transportr{ r� )�x�lc� protocolss ��rI �_checkz(waitUntilAllDisconnected.<locals>._check� s2 �� � �y�A�!�� � �-�-�A� A� �G�G�I� B��As � :g{�G�z�?T)�now)r � LoopingCall�clockr_ )�reactorrw rx rv s ` @rI r: r: s s; �� �$
�B��
� � �&� !�B��B�H�
�8�8�D�d�8� #�#r> c �n � e Zd ZdZd d�Zd d�Zed
d�� Zedd�� Zdd�Zdd�Zdd�Z e
dd �� Z y
)r; a�
L{ILogObserver} That stores its events in a list for later inspection.
This class is similar to L{LimitedHistoryLogObserver} save that the
internal buffer is public and intended for external inspection. The
observer implements the sequence protocol to ease iteration of the events.
@ivar _events: The events captured by this observer
@type _events: L{list}
c � � g | _ y rL ��_eventsrn s rI r` z EventLoggingObserver.__init__� s � �')�� r> c �, � t | j � S rL )�lenr� rn s rI �__len__z EventLoggingObserver.__len__� s � ��4�<�<� � r> c � � y rL rZ �rG �indexs rI � __getitem__z EventLoggingObserver.__getitem__� � � � r> c � � y rL rZ r� s rI r� z EventLoggingObserver.__getitem__� r� r> c � � | j | S rL r r� s rI r� z EventLoggingObserver.__getitem__� s � ��|�|�E�"�"r> c �, � t | j � S rL )�iterr� rn s rI �__iter__z EventLoggingObserver.__iter__� s � ��D�L�L�!�!r> c �: � | j j |� y)z'
@see: L{ILogObserver}
N)r� rf )rG �events rI �__call__z EventLoggingObserver.__call__� s � �
� � ���E�"r> c �d ��� | � ��j �� |j ��fd�� �S )a-
Create an L{EventLoggingObserver} instance that observes the provided
publisher and will be cleaned up with addCleanup().
@param testInstance: Test instance in which this logger is used.
@type testInstance: L{twisted.trial.unittest.TestCase}
@param publisher: Log publisher to observe.
@type publisher: twisted.logger.LogPublisher
@return: An EventLoggingObserver configured to observe the provided
publisher.
@rtype: L{twisted.test.proto_helpers.EventLoggingObserver}
c �& �� �j � � S rL )�removeObserver)�obs� publishers ��rI �<lambda>z8EventLoggingObserver.createWithCleanup.<locals>.<lambda>� s �� � �(@�(@��(E� r> )� addObserver�
addCleanup)�cls� testInstancer� r� s `@rI �createWithCleanupz&EventLoggingObserver.createWithCleanup� s. �� � �e��� � �c�"�� � � E�F��
r> N)rI rJ )rI �int)r� r� rI r) )r� �slicerI zSequence[LogEvent])r� z int | slicerI z LogEvent | Sequence[LogEvent])rI zIterator[LogEvent])r� r) rI rJ )r� r- r� r* rI r ) rV rW rX rY r` r� r
r� r� r� � classmethodr� rZ r> rI r; r; � s\ � ��*�!� � � � � � � � �#�"�#� �� �r> r; N)PrY �
__future__r r� r �socketr r �typingr r r r
�zope.interfacer r �zope.interface.verifyr
r r �twisted.internetr r r r �twisted.internet.abstractr r �twisted.internet.addressr r r �twisted.internet.deferr �twisted.internet.errorr �twisted.internet.interfacesr r r r r r r! r" r# r$ r% r& �twisted.internet.taskr' �twisted.loggerr( r) r* �twisted.protocolsr+ �twisted.pythonr, �twisted.trial.unittestr- �__all__r< �Protocolr. � LineReceiverr/ r0 r1 r2 r3 r5 r4 r6 �ifacer7 r8 r9 r: r; rZ r> rI �<module>r� s� ��
� #� � $� 9� 9� 5� -� "� ;� ;� E� J� J� +� ;�
�
�
�
� (� ?� ?� #� "� +� ��" �t�_��' �8�,�,� ' �T �%�,�,� �0,� ,�
�Z��M�2�\)� \)� 3�\)�~ X�� X� �W� �
�^� � !� !� � !�@
�Z��, � , � �, �^
��+�{�L�.�-��K � K ��K �\
�=�
)� &�E���}�%�&� � �� �
�[�+�|�^�D�M%� M%� E�M%�`.� .�b$�:
�\��8�8�H�-� 8� �8r> |