File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_stdio.cpython-38.pyc
U
W[1 � @ sp d Z ddlmZmZ ddlmZ ddlmZ ddlm Z e�
� sLddl m Z G dd� de�Z
e� �e
�� � d S )
z&
Tests for L{twisted.internet.stdio}.
� )�absolute_import�division)�platform)�ReactorBuilder)�Protocol)�
StandardIOc @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z e �
� rXdZdS )�StdioFilesTestszI
L{StandardIO} supports reading and writing to filesystem files.
c C s4 | � � }t|d��� t|d�| _| �| jj� d S )N�wbzrb+)�mktemp�open�close� extraFile�
addCleanup)�self�path� r �B/usr/lib/python3/dist-packages/twisted/internet/test/test_stdio.py�setUp s zStdioFilesTests.setUpc s� | � � � G � fdd�dt�}| �� }t|d��}|�d� W 5 Q R X t|d��0}|� }t||�� | j�� � d� | �� � W 5 Q R X | � |j
d� dS )zc
Adding a filesystem file reader to a reactor will make sure it is
polled.
c s e Zd ZdZ� fdd�ZdS )z4StdioFilesTests.test_addReader.<locals>.DataProtocol� c s$ | j |7 _ | j dkr � �� d S )N� hello!)�data�stop)r r ��reactorr r � dataReceived( s
zAStdioFilesTests.test_addReader.<locals>.DataProtocol.dataReceivedN)�__name__�
__module__� __qualname__r r r r r r � DataProtocol& s r r r �rb��stdin�stdoutr N) � buildReactorr r
r �writer �filenor
�
runReactor� assertEqualr )r r r �f�protocolr r r �test_addReader s
�z StdioFilesTests.test_addReaderc s� | � � � G � fdd�dt�}| �� }t|d��R}|� }t||�� | j�� � d� |j�d� |j�d� |j� � | �
� � W 5 Q R X t|d��}| � |� � d� W 5 Q R X d S )
zc
Adding a filesystem file writer to a reactor will make sure it is
polled.
c s e Zd Z� fdd�ZdS )z:StdioFilesTests.test_addWriter.<locals>.DisconnectProtocolc s � � � d S )N)r )r �reasonr r r �connectionLostH s zIStdioFilesTests.test_addWriter.<locals>.DisconnectProtocol.connectionLostN)r r r r, r r r r �DisconnectProtocolG s r- r �r" r! r � hellos , worldr s hello, worldN)
r# r r
r r r% r
� transportr$ ZloseConnectionr& r' �read)r r- r r( r) r r r �test_addWriter@ s
�
z StdioFilesTests.test_addWriterc C s� | � � }| �| j|� | �� }t|d��� t|d��P}tt� |�� | j �� |d�}| �
|j |� � � |j �
� | �|j |� � � W 5 Q R X dS )zq
Removing a filesystem file reader from a reactor will make sure it is
no longer polled.
r r r N)r# r �unbuildReactorr
r r r r r% r
�assertIn�_reader�
getReadersZ stopReading� assertNotIn�r r r r( �stdior r r �test_removeReader] s �
z!StdioFilesTests.test_removeReaderc C s� | � � }| �| j|� t| �� d� | _}t� }t||�� | j �� |d�}|j
� d� | � |j
|�� � |j
�� | �|j
|�� � dS )zq
Removing a filesystem file writer from a reactor will make sure it is
no longer polled.
r r. r/ N)r# r r3 r r
r( r r r% r
r0 r$ r4 �_writer�
getWritersZ stopWritingr7 )r r r( r) r9 r r r �test_removeWriterr s
�
z!StdioFilesTests.test_removeWriterc C s� | � � }| �| j|� | �� }t|d��� t|d� | _}tt� |� � | j
� � |d�}|� � }| � |j
|� | �|j
|�� � dS )zk
Calling C{removeAll} on a reactor includes descriptors that are
filesystem files.
r r r N)r# r r3 r
r r r( r r r% r
Z removeAllr4 r5 r7 r6 )r r r r( r9 Zremovedr r r �test_removeAll� s �z StdioFilesTests.test_removeAllc C st | � � }| �| j|� | �� }t|d��� t|d��4}tt� |�� | j �� |d�}| �
|j |� � � W 5 Q R X dS )zW
C{reactor.getReaders} includes descriptors that are filesystem files.
r r r N)
r# r r3 r
r r r r r% r
r4 r5 r6 r8 r r r �test_getReaders� s �z StdioFilesTests.test_getReadersc C sx | � � }| �| j|� t| �� d� | _}tt� |�� | j �� |d�}| �
|j |� � � |j �
� | �|j |� � � dS )zW
C{reactor.getWriters} includes descriptors that are filesystem files.
r r. N)r# r r3 r r
r( r r r% r
r7 r; r< Z startWritingr4 )r r r( r9 r r r �test_getWriters� s �
z StdioFilesTests.test_getWriterszoStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.N)r r r �__doc__r r* r2 r: r= r> r? r@ r � isWindows�skipr r r r r s ! r N)rA Z
__future__r r Ztwisted.python.runtimer Z#twisted.internet.test.reactormixinsr Ztwisted.internet.protocolr rB Ztwisted.internet.stdior r �globals�updateZmakeTestCaseClassesr r r r �<module> s 5