← Back
Editing: 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
Save File
Cancel