File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_fdset.cpython-38.pyc
U
��W[�4 � @ s� d Z eZddlZddlZddlZddlmZ ddlm Z ddl
m Z ddl m
Z
mZ ddlmZ ddlmZ dd lmZmZ d
d � ZG d d
� d
e�Zee�G dd� de��Ze� �e� � � dS )z0
Tests for implementations of L{IReactorFDSet}.
� N)� implementer)�platform)�SkipTest)�
IReactorFDSet�IReadDescriptor)�FileDescriptor)�ReactorBuilder)� EINPROGRESS� EWOULDBLOCKc
C s� t � � } | �d� | �d� z�t � � }zn|�d� z|�d| �� d f� W n8 t jk
r� } z|jd t t
fkrt� W 5 d }~X Y nX | � � \}}W n |�� � Y nX W 5 | �� X ||fS )N)� 127.0.0.1r � Fr r ) �socketZbindZlisten�closeZ setblockingZconnectZ getsockname�error�argsr r
Zaccept)Z serverSocket�client�e�serverZaddr� r �B/usr/lib/python3/dist-packages/twisted/internet/test/test_fdset.py�
socketpair s$
r c @ s� e Zd ZdZegZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Ze�� r�d e_d d � Zd S ) �ReactorFDSetTestsBuilderz>
Builder defining tests relating to L{IReactorFDSet}.
c C s* t � \}}| �|j� | �|j� ||fS )zL
Return the two sockets which make up a new TCP connection.
)r Z
addCleanupr )�selfr r r r r �_connectedPair6 s
z'ReactorFDSetTestsBuilder._connectedPairc C s. | � � }| �� \}}t|�}|j|_|||fS �N)� buildReactorr r �fileno)r �reactorr r �fdr r r � _simpleSetup@ s
z%ReactorFDSetTestsBuilder._simpleSetupc sD | � � \�� }� �fdd�}|� _��� � |�d� | ��� dS )z�
C{reactor.addReader()} accepts an L{IReadDescriptor} provider and calls
its C{doRead} method when there may be data available on its C{fileno}.
c s �� � � ��� d S r )� removeReader�stopr �r r r r �
removeAndStopR s
z>ReactorFDSetTestsBuilder.test_addReader.<locals>.removeAndStop� xN)r �doRead� addReader�sendall�
runReactor�r r r# r r" r �test_addReaderK s
z'ReactorFDSetTestsBuilder.test_addReaderc s` � � � \}}}� fdd�}||_|�|� |�|� |�d� |�d|jd|j� � �|� dS )z�
L{reactor.removeReader()} accepts an L{IReadDescriptor} provider
previously passed to C{reactor.addReader()} and causes it to no longer
be monitored for input events.
c s � � d� d S �NzdoRead should not be called��failr �r r r r- f s z8ReactorFDSetTestsBuilder.test_removeReader.<locals>.failr$ r N)r r% r& r r' � callLaterr! r( �r r r r r- r r. r �test_removeReader^ s
z*ReactorFDSetTestsBuilder.test_removeReaderc s: | � � \�� }� �fdd�}|� _��� � | ��� dS )z�
C{reactor.addWriter()} accepts an L{IWriteDescriptor} provider and
calls its C{doWrite} method when it may be possible to write to its
C{fileno}.
c s �� � � ��� d S r )� removeWriterr! r r" r r r# ~ s
z>ReactorFDSetTestsBuilder.test_addWriter.<locals>.removeAndStopN)r �doWrite� addWriterr( r) r r" r �test_addWriterv s
z'ReactorFDSetTestsBuilder.test_addWriterc C s� | � � }t|d| d �}t|d| �}t|d| �}| �� \}}| �||� � | �||� � ||� | �||� � | �||� � ||� | �||� � | �||� � dS )z=
Helper for getReaders and getWriters tests.
�get�s�add�removeN)r �getattrr Z assertNotInZassertIn)r Zkindr r6 r8 r9 r r r r r �
_getFDTest� s z#ReactorFDSetTestsBuilder._getFDTestc C s | � d� dS )z�
L{IReactorFDSet.getReaders} reflects the additions and removals made
with L{IReactorFDSet.addReader} and L{IReactorFDSet.removeReader}.
�ReaderN�r; r. r r r �test_getReaders� s z(ReactorFDSetTestsBuilder.test_getReadersc sV � � � \}}}� fdd�}||_|�|� |�|� |�d|jd|j� � �|� dS )z�
L{reactor.removeWriter()} accepts an L{IWriteDescriptor} provider
previously passed to C{reactor.addWriter()} and causes it to no longer
be monitored for outputability.
c s � � d� d S �Nz doWrite should not be calledr, r r. r r r- � s z8ReactorFDSetTestsBuilder.test_removeWriter.<locals>.failr N)r r3 r4 r2 r/ r! r( r0 r r. r �test_removeWriter� s
z*ReactorFDSetTestsBuilder.test_removeWriterc C s | � d� dS )z�
L{IReactorFDSet.getWriters} reflects the additions and removals made
with L{IReactorFDSet.addWriter} and L{IReactorFDSet.removeWriter}.
�WriterNr= r. r r r �test_getWriters� s z(ReactorFDSetTestsBuilder.test_getWritersc s� � � � }� �� \}}}� fdd�|_� fdd�|_|�d� |�|� |�|� |�� }|�d|jd|j � � �
|� � � ||g� dS )z�
C{reactor.removeAll()} removes all registered L{IReadDescriptor}
providers and all registered L{IWriteDescriptor} providers and returns
them.
c s
� � d�S r+ r, r r. r r �<lambda>� � z9ReactorFDSetTestsBuilder.test_removeAll.<locals>.<lambda>c s
� � d�S r? r, r r. r r rC � rD r$ r N) r r r% r3 r' r& r4 Z removeAllr/ r! r( � assertEqual)r r r r Zremovedr r. r �test_removeAll� s
z'ReactorFDSetTestsBuilder.test_removeAllc C s8 | � � }t|�}|�|j� | �|� | �|jg � dS )z�
A descriptor's C{fileno} method should not be called after the
descriptor has been removed from the reactor.
N)r �RemovingDescriptor�callWhenRunning�startr( rE �calls)r r �
descriptorr r r �test_removedFromReactor� s
z0ReactorFDSetTestsBuilder.test_removedFromReactorc sd | � � �| �� \� �G � ��fdd�dt�}|��}��|� � �d� | ��� | �|jd� dS )zp
If L{FileDescriptor.fileno} returns C{-1}, the descriptor is removed
from the reactor.
c s: e Zd Z��� ZdZdd� Z� �fdd�Z�fdd�ZdS ) zWReactorFDSetTestsBuilder.test_negativeOneFileDescriptor.<locals>.DisappearingDescriptorrD c S s | j S r )�_filenor. r r r r � s z^ReactorFDSetTestsBuilder.test_negativeOneFileDescriptor.<locals>.DisappearingDescriptor.filenoc s( d| _ | j��d�7 _� �d� d S )N���r � y)rM � _receivedZrecv�sendr. )r r r r r% s z^ReactorFDSetTestsBuilder.test_negativeOneFileDescriptor.<locals>.DisappearingDescriptor.doReadc s � � � d S r �r! �r �reason�r r r �connectionLost s zfReactorFDSetTestsBuilder.test_negativeOneFileDescriptor.<locals>.DisappearingDescriptor.connectionLostN)�__name__�
__module__� __qualname__r rM rP r% rV r �r r r r r �DisappearingDescriptor� s
r[ r$ N)r r r r& rQ r( rE rP )r r[ rK r rZ r � test_negativeOneFileDescriptor� s
z7ReactorFDSetTestsBuilder.test_negativeOneFileDescriptorc s� �� � � � jj}|dkr&td|f ����� \}�G � �fdd�dt�}� �|� � ��fdd�}� �d|� ��� � �� t
j � dS ) a
The file descriptor underlying a FileDescriptor may be closed and
replaced by another at some point. Bytes which arrive on the new
descriptor must not be delivered to the FileDescriptor which was
originally registered with the original descriptor of the same number.
Practically speaking, this is difficult or impossible to detect. The
implementation relies on C{fileno} raising an exception if the original
descriptor has gone away. If C{fileno} continues to return the original
file descriptor value, the reactor may deliver events from that
descriptor. This is a best effort attempt to ease certain debugging
situations. Applications should not rely on it intentionally.
)Z EPollReactorZ
KQueueReactorZ CFReactorZAsyncioSelectorReactorz&%r cannot detect lost file descriptorsc s0 e Zd ZdZ�fdd�Zdd� Z� fdd�ZdS ) z@ReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victima�
This L{FileDescriptor} will have its socket closed out from under it
and another socket will take its place. It will raise a
socket.error from C{fileno} after this happens (because socket
objects remember whether they have been closed), so as long as the
reactor calls the C{fileno} method the problem will be detected.
c s � � � S r )r r. )r r r r 3 s zGReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victim.filenoc S s t d��d S )Nz$Victim.doRead should never be called)� Exceptionr. r r r r% 6 s zGReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victim.doReadc s � � � dS )z�
When the problem is detected, the reactor should disconnect this
file descriptor. When that happens, stop the reactor so the
test ends.
NrR rS rU r r rV 9 s zOReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victim.connectionLostN)rW rX rY �__doc__r r% rV r )r r r r �Victim+ s r_ c s: � � � \} }��� }��� t�|�� |� | �d� d S )Nr$ )r r r �os�dup2rQ )ZnewCZnewSr )r r r r �messItUpH s
zBReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.messItUpr N) r � __class__rW r r r r& r/ r( ZflushLoggedErrorsr
r )r �namer r_ rb r )r r r r �test_lostFileDescriptor s
z0ReactorFDSetTestsBuilder.test_lostFileDescriptorz*Cannot duplicate socket filenos on Windowsc C s� | � � }G dd� dt�}| �� \}}||�}|j|_||�}|j|_|�|� |�|� |�|j� | �|� | � |j
� | � |j
� dS )z�
Any file descriptors added to the reactor have their C{connectionLost}
called when C{reactor.stop} is called.
c @ s e Zd Zdd� Zdd� ZdS )zSReactorFDSetTestsBuilder.test_connectionLostOnShutdown.<locals>.DoNothingDescriptorc S s d S r r r. r r r r% b s zZReactorFDSetTestsBuilder.test_connectionLostOnShutdown.<locals>.DoNothingDescriptor.doReadc S s d S r r r. r r r r3 d s z[ReactorFDSetTestsBuilder.test_connectionLostOnShutdown.<locals>.DoNothingDescriptor.doWriteN)rW rX rY r% r3 r r r r �DoNothingDescriptora s rf N) r r r r r& r4 rH r! r( Z
assertTrueZ disconnected)r r rf r r Zfd1Zfd2r r r � test_connectionLostOnShutdownZ s
z6ReactorFDSetTestsBuilder.test_connectionLostOnShutdownN)rW rX rY r^ r ZrequiredInterfacesr r r* r1 r5 r; r> r@ rB rF rL r\ re r Z isWindows�skiprg r r r r r 0 s&
E�r c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )rG a�
A read descriptor which removes itself from the reactor as soon as it
gets a chance to do a read and keeps track of when its own C{fileno}
method is called.
@ivar insideReactor: A flag which is true as long as the reactor has
this descriptor as a reader.
@ivar calls: A list of the bottom of the call stack for any call to
C{fileno} when C{insideReactor} is false.
c C s$ || _ d| _g | _t� \| _| _d S �NF)r �
insideReactorrJ r �read�write)r r r r r �__init__� s zRemovingDescriptor.__init__c C s" d| _ | j�| � | j�d� d S )NT� a)rj r r& rl rQ r. r r r rI � s zRemovingDescriptor.startc C s dS )NZfoor r. r r r � logPrefix� s z RemovingDescriptor.logPrefixc C s4 | j �| � d| _| j �� | j�� | j�� d S ri )r r rj r! rk r rl r. r r r r% � s
zRemovingDescriptor.doReadc C s, | j s"| j�tjdd�d d� � | j�� S )N� )�limitrN )rj rJ �append� traceback�
extract_stackrk r r. r r r r � s zRemovingDescriptor.filenoc C s d S r r rS r r r rV � s z!RemovingDescriptor.connectionLostN)
rW rX rY r^ rm rI ro r% r rV r r r r rG w s
rG ) r^ �typeZ
__metaclass__r` r
rs Zzope.interfacer Ztwisted.python.runtimer Ztwisted.trial.unittestr Ztwisted.internet.interfacesr r Ztwisted.internet.abstractr Z#twisted.internet.test.reactormixinsr Ztwisted.internet.tcpr r
r r �objectrG �globals�updateZmakeTestCaseClassesr r r r �<module> s I3