File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_fdset.cpython-310.pyc
o
�b�4 � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZmZ ddlmZ dd lmZ dd
lmZ d d � ZG d
d� de�Zee �G dd� d��Ze� �e�� � dS )z0
Tests for implementations of L{IReactorFDSet}.
� N)�skipIf)� implementer)�FileDescriptor)�
IReactorFDSet�IReadDescriptor)� EINPROGRESS� EWOULDBLOCK)�ReactorBuilder)�platform)�SkipTestc
C s� t � � } | �d� | �d� zQt � � }z7|�d� z
|�d| �� d f� W n tyB } z|jd tt fvr8� W Y d }~nd }~ww | �
� \}}W n t yV |� � � w W | � � ||fS | � � w )N)� 127.0.0.1r � Fr r )
�socket�bind�listen� setblocking�connect� getsockname�OSError�argsr r �accept�
BaseException�close)� serverSocket�client�e�server�addr� r �B/usr/lib/python3/dist-packages/twisted/internet/test/test_fdset.py�
socketpair s0
��� ��
�r c @ s� e Zd ZdZegZdd� Zdd� Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zee�� d�dd � �Zd 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 �
addCleanupr )�selfr r r r r �_connectedPair8 s
z'ReactorFDSetTestsBuilder._connectedPairc C s. | � � }| �� \}}t|�}|j|_|||fS �N)� buildReactorr$ r �fileno)r# �reactorr r �fdr r r � _simpleSetupA 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 � �� � � ��� d S r% )� removeReader�stopr �r) r( r r �
removeAndStopR �
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 � � � d� d S �NzdoRead should not be called��failr �r# r r r; f � z8ReactorFDSetTestsBuilder.test_removeReader.<locals>.failr1 r N)r* r2 r3 r, r4 � callLaterr- r5 �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 r+ r% )� removeWriterr- r r. r r r/ ~ r0 z>ReactorFDSetTestsBuilder.test_addWriter.<locals>.removeAndStopN)r* �doWrite� addWriterr5 r6 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$ � assertNotIn�assertIn)r# �kindr( rE rG rH r r r r r �
_getFDTest� s z#ReactorFDSetTestsBuilder._getFDTestc C � | � d� dS )z�
L{IReactorFDSet.getReaders} reflects the additions and removals made
with L{IReactorFDSet.addReader} and L{IReactorFDSet.removeReader}.
�ReaderN�rM r< r r r �test_getReaders� � 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 r8 �Nz doWrite should not be calledr: r r< r r r; � r= z8ReactorFDSetTestsBuilder.test_removeWriter.<locals>.failr N)r* rB rC rA r> r- r5 r? r r< r �test_removeWriter� s
z*ReactorFDSetTestsBuilder.test_removeWriterc C rN )z�
L{IReactorFDSet.getWriters} reflects the additions and removals made
with L{IReactorFDSet.addWriter} and L{IReactorFDSet.removeWriter}.
�WriterNrP r< r r r �test_getWriters� rR 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 �
� � d�S r9 r: r r< r r �<lambda>� �
z9ReactorFDSetTestsBuilder.test_removeAll.<locals>.<lambda>c rW rS r: r r< r r rX � rY r1 r N) r& r* r2 rB r4 r3 rC � removeAllr> r- r5 � assertEqual)r# r( r) r �removedr 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�startr5 r[ �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>.DisappearingDescriptor� 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)re � _received�recv�sendr< )r r r r r2 � 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' re rh r2 ro r �r r( r r r �DisappearingDescriptor� s rt r1 N)r& r$ r r3 rj r5 r[ rh )r# rt rb r rs r � test_negativeOneFileDescriptor� s
z7ReactorFDSetTestsBuilder.test_negativeOneFileDescriptorz*Cannot duplicate socket filenos on Windowsc s� �� � � � jj}|dv rt|�d������ \}�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.
)� EPollReactor�
KQueueReactor� CFReactor�AsyncioSelectorReactorz$ 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 � zGReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victim.filenoc S s t d��)Nz$Victim.doRead should never be called)� Exceptionr< r r r r2 6 rz 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.
Nrk rl rn r r ro 9 s zOReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.Victim.connectionLostN)rp rq rr �__doc__r' r2 ro r )r( r r r �Victim* s
r} c s: � � � \} }��� }��� t�|�� |� | �d� d S )Nr1 )r$ r' r �os�dup2rj )�newC�newSr' )r# r r r �messItUpH s
zBReactorFDSetTestsBuilder.test_lostFileDescriptor.<locals>.messItUpr N) r& � __class__rp r r$ r r3 r> r5 �flushLoggedErrorsr �error)r# �namer r} r� r )r( r# r r �test_lostFileDescriptor
s
z0ReactorFDSetTestsBuilder.test_lostFileDescriptorc 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 � d S r% r r< r r r r2 _ � zZReactorFDSetTestsBuilder.test_connectionLostOnShutdown.<locals>.DoNothingDescriptor.doReadc S r� r% r r< r r r rB b r� z[ReactorFDSetTestsBuilder.test_connectionLostOnShutdown.<locals>.DoNothingDescriptor.doWriteN)rp rq rr r2 rB r r r r �DoNothingDescriptor^ s r� N) r& r r$ r' r3 rC r_ r- r5 �
assertTrue� disconnected)r# r( r� r r �fd1�fd2r r r � test_connectionLostOnShutdownW s
z6ReactorFDSetTestsBuilder.test_connectionLostOnShutdownN)rp rq rr r| r �requiredInterfacesr$ r* r7 r@ rD rM rQ rT rV r] rc ru r r
� isWindowsr� r� r r r r r! 1 s$
Lr! c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )r^ 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( �
insideReactorra r �read�write)r# r( r r r �__init__� s zRemovingDescriptor.__init__c C s" d| _ | j�| � | j�d� d S )NT� a)r� r( r3 r� rj r< r r r r` � s zRemovingDescriptor.startc C s dS )N�foor r< r r r � logPrefix� r� z RemovingDescriptor.logPrefixc C s4 | j �| � d| _| j �� | j�� | j�� d S r� )r( r, r� r- r� r r� r< r r r r2 � s
zRemovingDescriptor.doReadc C s, | j s| j�tjdd�d d� � | j�� S )N� )�limitrf )r� ra �append� traceback�
extract_stackr� r' r< r r r r' � s
zRemovingDescriptor.filenoc C r� r% r rl r r r ro � s z!RemovingDescriptor.connectionLostN)
rp rq rr r| r� r` r� r2 r' ro r r r r r^ t s r^ ) r| r~ r r� �unittestr �zope.interfacer �twisted.internet.abstractr �twisted.internet.interfacesr r �twisted.internet.tcpr r �#twisted.internet.test.reactormixinsr �twisted.python.runtimer
�twisted.trial.unittestr r r! r^ �globals�update�makeTestCaseClassesr r r r �<module> s& E.