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