File: //usr/lib/python3/dist-packages/twisted/internet/test/__pycache__/test_stdio.cpython-310.pyc
o
�b� � @ s` d Z ddlmZ ddlmZ ddlmZ e�� s ddlm Z G dd� de�Z
e � � e
�
� � dS ) z&
Tests for L{twisted.internet.stdio}.
� )�Protocol)�ReactorBuilder)�platform)�
StandardIOc @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z e �
� r.dZdS dS )�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 d � n1 s'w Y t|d�� }|� }t||�� | j�� � d� | �� � W d � n1 sQw Y | � |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 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 N�rb��stdin�stdoutr ) � 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��.}|� }t||�� | j�� � d� |j�d� |j�d� |j� � | �
� � W d � n1 sHw Y t|d��}| � |� � d � W d � dS 1 sfw Y dS )
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 , worldNr s hello, world)
r! r r r r r# r � transportr" �loseConnectionr$ r% �read)r
r+ r r& r' r r r �test_addWriter@ s& �
� "�z StdioFilesTests.test_addWriterc C s� | � � }| �| j|� | �� }t|d��� t|d��.}tt� |�� | j �� |d�}| �
|j |� � � |j �
� | �|j |� � � W d � dS 1 sLw Y 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�
getReaders� 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 r2 r r r& r r r# r r. r" r3 �_writer�
getWriters� stopWritingr7 )r
r r& r' r9 r r r �test_removeWriterw 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 r2 r r r
r& r r r# r � removeAllr3 r4 r7 r5 )r
r r r& r9 �removedr r r �test_removeAll� s �z StdioFilesTests.test_removeAllc C s� | � � }| �| j|� | �� }t|d��� t|d�� }tt� |�� | j �� |d�}| �
|j |� � � W d � dS 1 s>w Y dS )zW
C{reactor.getReaders} includes descriptors that are filesystem files.
r r r N)
r! r r2 r r r
r r r# r r3 r4 r5 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 r2 r r r& r r r# r r7 r; r< � startWritingr3 )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( r1 r: r> rA rB rD r � isWindows�skipr r r r r s $ ��r N)rE �twisted.internet.protocolr �#twisted.internet.test.reactormixinsr �twisted.python.runtimer rF �twisted.internet.stdior r �globals�update�makeTestCaseClassesr r r r �<module> s G