Your IP : 216.73.216.213
B
;��]d3 � @ s$ d Z ddlmZmZ ddlZddlZddlZddlmZ ddl m
Z
m Z ddl m
Z
ddlmZ ddlmZmZmZ dd lmZmZmZmZmZ dd
lmZ d Z dZ e� � r�e
d �dkr�d
Z e ej �Z!ej"�#ej$�e!d< n"e� Z!ej"�#ej$��%e�&� �e!d< G dd� dej'�Z(G dd� dej)�Z*dS )z�
Tests for L{twisted.internet.stdio}.
@var properEnv: A copy of L{os.environ} which has L{bytes} keys/values on POSIX
platforms and native L{str} keys/values on Windows.
� )�absolute_import�divisionN)�unittest)�filepath�log)�
requireModule)�platform)�range�
intToBytes� bytesEnviron)�error�defer�protocol�stdio�reactor)� ConnectionLostNotifyingProtocols xyz123abc Twisted is great!Z win32processzIOn windows, spawnProcess is not available in the absence of win32process.�
PYTHONPATHs
PYTHONPATHc @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� ZdS ) � StandardIOTestProcessProtocola�
Test helper for collecting output from a child process and notifying
something when it exits.
@ivar onConnection: A L{defer.Deferred} which will be called back with
L{None} when the connection to the child process is established.
@ivar onCompletion: A L{defer.Deferred} which will be errbacked with the
failure associated with the child process exiting when it exits.
@ivar onDataReceived: A L{defer.Deferred} which will be called back with
this instance whenever C{childDataReceived} is called, or L{None} to
suppress these callbacks.
@ivar data: A C{dict} mapping file descriptors to strings containing all
bytes received from the child process on each file descriptor.
Nc C s t �� | _t �� | _i | _d S )N)r
�Deferred� onConnection� onCompletion�data)�self� r �J/opt/alt/python37/lib64/python3.7/site-packages/twisted/test/test_stdio.py�__init__B s
z&StandardIOTestProcessProtocol.__init__c C s | j �d � d S )N)r �callback)r r r r �connectionMadeH s z,StandardIOTestProcessProtocol.connectionMadec C s> | j �|d�| | j |< | jdk r:| jd }| _|�| � dS )z�
Record all bytes received from the child process in the C{data}
dictionary. Fire C{onDataReceived} if it is not L{None}.
� N)r �get�onDataReceivedr )r �name�bytes�dr r r �childDataReceivedL s
z/StandardIOTestProcessProtocol.childDataReceivedc C s | j �|� d S )N)r r )r �reasonr r r � processEndedW s z*StandardIOTestProcessProtocol.processEnded) �__name__�
__module__� __qualname__�__doc__r r r r$ r& r r r r r . s r c @ s~ e Zd ZeZdd� Zdd� Zdd� Zdd� Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Ze�� rzde_dS )�StandardInputOutputTestsc O s: t jdd| tjjgt|� }tj|t j|fdti|��S )a_
Launch a child Python process and communicate with it using the
given ProcessProtocol.
@param proto: A L{ProcessProtocol} instance which will be connected
to the child process.
@param sibling: The basename of a file containing the Python program
to run in the child process.
@param *args: strings which will be passed to the child process on
the command line as C{argv[2:]}.
@param **kw: additional arguments to pass to L{reactor.spawnProcess}.
@return: The L{IProcessTransport} provider for the spawned process.
s -ms
twisted.test.�env)�sys�
executabler � __class__r( �listZ spawnProcess� properEnv)r �protoZsibling�args�kwr r r �
_spawnProcess` s z&StandardInputOutputTests._spawnProcessc s$ �fdd�}� fdd�}|� ||�S )Nc s � � d| f � d S )Nz'Process terminated with non-Failure: %r)Zfail)�result)r r r �cb~ s z4StandardInputOutputTests._requireFailure.<locals>.cbc s � | �S )Nr )�err)r r r �eb� s z4StandardInputOutputTests._requireFailure.<locals>.eb)Z addCallbacks)r r# r r7 r9 r )r r r �_requireFailure} s z(StandardInputOutputTests._requireFailurec sL �� � � t�d� � t� ��j}���d� � � ��fdd�}��||�S )z�
Verify that a protocol connected to L{StandardIO} can disconnect
itself using C{transport.loseConnection}.
zChild process logging to s stdio_test_loseconnc sT t � d��&}x |D ]}t�d|�� � qW W d Q R X ��d�j� | �tj� d S )N�rzChild logged: � ) �openr �msg�rstripZfailIfInr �trapr � ProcessDone)r% �f�line)� errorLogFile�pr r r r& � s
zBStandardInputOutputTests.test_loseConnection.<locals>.processEnded)�mktempr r> r r r5 r: )r r# r& r )rD rE r r �test_loseConnection� s z,StandardInputOutputTests.test_loseConnectionc sf | � � }t�d| � t� � t�� � _� fdd�}� j�|� dd� }| �� j|�}| � � d|� |S )z�
When stdin is closed and the protocol connected to it implements
L{IHalfCloseableProtocol}, the protocol's C{readConnectionLost} method
is called.
zChild process logging to c s � j }� j�� |S )N)r Z transportZ
closeStdin)Zignoredr# )rE r r �cbBytes� s
zAStandardInputOutputTests.test_readConnectionLost.<locals>.cbBytesc S s | � tj� d S )N)r@ r rA )r% r r r r& � s zFStandardInputOutputTests.test_readConnectionLost.<locals>.processEndeds stdio_test_halfclose)
rF r r> r r
r r � addCallbackr: r5 )r rD rH r& r# r )rE r �test_readConnectionLost� s
z0StandardInputOutputTests.test_readConnectionLostc
sj t � � y�j� dtdd� W n0 tk
rL } zt�t|���W dd}~X Y nX � �fdd�}��� j|�S )z�
Verify that a write made directly to stdout using L{os.write}
after StandardIO has finished is reliably received by the
process reading that stdout.
s stdio_test_lastwriteT)ZusePTYNc s0 �� � jd �t�d� jf � | �tj� dS )z�
Asserts that the parent received the bytes written by the child
immediately after the child starts.
r< z4Received %r from child, did not find expected bytes.N)�
assertTruer �endswith�UNIQUE_LAST_WRITE_STRINGr@ r rA )r% )rE r r r r& � s
zEStandardInputOutputTests.test_lastWriteReceived.<locals>.processEnded) r r5 rM �
ValueErrorr ZSkipTest�strr: r )r �er& r )rE r r �test_lastWriteReceived� s
z/StandardInputOutputTests.test_lastWriteReceivedc s2 t � � � j}��� d� � �fdd�}��||�S )z�
Verify that the transport of a protocol connected to L{StandardIO}
has C{getHost} and C{getPeer} methods.
s stdio_test_hostpeerc s6 � j d �� \}}��|� ��|� | �tj� d S )Nr< )r �
splitlinesrK r@ r rA )r% �hostZpeer)rE r r r r& � s
z?StandardInputOutputTests.test_hostAndPeer.<locals>.processEnded)r r r5 r: )r r# r& r )rE r r �test_hostAndPeer� s
z)StandardInputOutputTests.test_hostAndPeerc s2 t � � � j}��� d� � �fdd�}��||�S )z�
Verify that the C{write} method of the transport of a protocol
connected to L{StandardIO} sends bytes to standard out.
s stdio_test_writec s" �� � jd d� | �tj� d S )Nr< s ok!)� assertEqualr r@ r rA )r% )rE r r r r& � s z9StandardInputOutputTests.test_write.<locals>.processEnded)r r r5 r: )r r# r& r )rE r r �
test_write� s
z#StandardInputOutputTests.test_writec s2 t � � � j}��� d� � �fdd�}��||�S )z�
Verify that the C{writeSequence} method of the transport of a
protocol connected to L{StandardIO} sends bytes to standard out.
s stdio_test_writeseqc s" �� � jd d� | �tj� d S )Nr< s ok!)rU r r@ r rA )r% )rE r r r r& s zAStandardInputOutputTests.test_writeSequence.<locals>.processEnded)r r r5 r: )r r# r& r )rE r r �test_writeSequence� s
z+StandardInputOutputTests.test_writeSequencec C sF | � � }t|d��*}x"td�D ]}|�t|�d � q W W d Q R X |S )N�wbi �
)rF r= r �writer
)r �junkPathZjunkFile�ir r r � _junkPath
s
z"StandardInputOutputTests._junkPathc sd t � ��j}g �ttd���� ���fdd�� ���d���j�� � ����fdd�}��||�S )z�
Verify that the transport of a protocol connected to L{StandardIO}
is a working L{IProducer} provider.
�d c s: �r6�� t��� �d � ���d � t�d� d � d S )NrY ���g{�G�z�?)�appendr
�poprZ r � callLater)Zign)r �proc�toWrite�writtenr r r s z>StandardInputOutputTests.test_producer.<locals>.connectionMades stdio_test_producerc s> �� � jd d���� ���dt��f � | �tj� d S )Nr< r z*Connection lost with %d writes left to go.)rU r �joinZ assertFalse�lenr@ r rA )r% )rE r rd re r r r& * s
z<StandardInputOutputTests.test_producer.<locals>.processEnded)r r r0 r r5 r rI r: )r r# r& r )r rE rc r rd re r �
test_producer s z&StandardInputOutputTests.test_producerc s> t � ��j}��� � ���d� � � ��fdd�}��||�S )z�
Verify that the transport of a protocol connected to L{StandardIO}
is a working L{IConsumer} provider.
s stdio_test_consumerc s< t � d�� }���jd |�� � W d Q R X | �tj� d S )N�rbr< )r= rU r �readr@ r rA )r% rB )r[ rE r r r r& ? s z<StandardInputOutputTests.test_consumer.<locals>.processEnded)r r r] r5 r: )r r# r& r )r[ rE r r �
test_consumer3 s z&StandardInputOutputTests.test_consumerc s� t �� }t|�}t���� ����d� �_}��|j � t
|� � d�}t �
� s�t�� \}}��tj |� ��tj |� ||d< tj|f|�� d�t�� �� ���fdd��t�d�� ����fdd �}|�|� |S )
aE
If L{StandardIO} is created with a file descriptor which refers to a
normal file (ie, a file from the filesystem), L{StandardIO.write}
writes bytes to that file. In particular, it does not immediately
consider the file closed or call its protocol's C{connectionLost}
method.
rX )�stdout�stdin� c sB x0�D ](} | �kr � � � d S � �t| �� P qW t�d�� d S )Nr )ZloseConnectionrZ r
r rb )�value)�
connection�count�howMany�spinr r rs e s
zAStandardInputOutputTests.test_normalFileStandardOut.<locals>.spinr c s8 �� t� ��d � �� ��� d�ttt����� d S )Nr< r )rU �nextZ
getContentrf �mapr
r )r% )rq rr �pathr r r �cbLostq s zCStandardInputOutputTests.test_normalFileStandardOut.<locals>.cbLost)r
r r r ZFilePathrF r= �normalZ
addCleanup�close�dict�filenor � isWindows�os�piper Z
StandardIO� itertoolsrq r rb rI )r Z
onConnLostr2 rx �kwargsr; �wrw r )rp rq rr rv r rs r �test_normalFileStandardOutF s&
z3StandardInputOutputTests.test_normalFileStandardOutzpStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.N)r' r( r) �skipWindowsNopywin32�skipr5 r: rG rJ rQ rT rV rW r] rh rk r� r r| r r r r r+ \ s ' 3r+ )+r* �
__future__r r r} r- r Z
twisted.trialr Ztwisted.pythonr r Ztwisted.python.reflectr Ztwisted.python.runtimer Ztwisted.python.compatr r
r Ztwisted.internetr r
r r r Ztwisted.test.test_tcpr rM r� r| rz �environr1 �pathseprf rv �encode�getfilesystemencodingZProcessProtocolr ZTestCaser+ r r r r �<module> s.
.