File: //usr/local/lib/python3.7/test/__pycache__/test_file_eintr.cpython-37.pyc
B
��gf* � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZe� e j
dkd�G dd� d��Z G dd� de ej �Z
G dd � d e ej �ZG d
d � d e �ZG d d
� d
eej �ZG dd� deej �ZG dd� de �ZG dd� deej �ZG dd� deej �Zedk�re�� dS )� N�posixz tests requires a posix system.c @ sR e Zd Zdd� Zdd� Zdd� Zdd d
�Zd d � Zd
Zdd� Z dd� Z
dd� Z dS )�TestFileIOSignalInterruptc C s
d | _ d S )N)�_process)�self� r �0/usr/local/lib/python3.7/test/test_file_eintr.py�setUp s z TestFileIOSignalInterrupt.setUpc C s< | j r8| j �� d kr8y| j �� W n tk
r6 Y nX d S )N)r �poll�kill�OSError)r r r r �tearDown s
z"TestFileIOSignalInterrupt.tearDownc C s
d| j S )z�Returns the infile = ... line of code for the reader process.
subclasseses should override this to test different IO objects.
z=import %s as io ;infile = io.FileIO(sys.stdin.fileno(), "rb"))�modname)r r r r �_generate_infile_setup_code$ s z5TestFileIOSignalInterrupt._generate_infile_setup_code� Tc C s~ | j �� dkr<t�d� y| j �� W n tk
r: Y nX |r^| j �� \}}||7 }||7 }| �d||�� |�� f � dS )a; A common way to cleanup and fail with useful debug output.
Kills the process if it is still running, collects remaining output
and fails the test with an error message including the output.
Args:
why: Text to go after "Error from IO process" in the message.
stdout, stderr: standard output and error from the process so
far to include in the error message.
communicate: bool, when True we call communicate() on the process
after killing it to gather additional output.
Ng�������?z/Error from IO process %s:
STDOUT:
%sSTDERR:
%s
) r r �timeZsleepZ terminater � communicate�fail�decode)r Zwhy�stdout�stderrr Z
stdout_endZ
stderr_endr r r �fail_with_process_info- s
z0TestFileIOSignalInterrupt.fail_with_process_infoc C sL | � � }t|�dk s td��tjtjddd| d d | d d gtjtjtjd �| _| jj �
td
��}|d
kr�| j d |d � | jj �
|� d
}g }xV|s�t�| jj gddd�\}}}| j�tj� |d7 }|dkr�| j�� | �d� q�W | jj �� }|dk�r| j d|d � | jjdd�\} }
| jj�rH| j d| jj | |
dd� dS )an Generic buffered read method test harness to validate EINTR behavior.
Also validates that Python signal handlers are run during the read.
Args:
data_to_write: String to write to the child process for reading
before sending it a signal, confirming the signal was handled,
writing a final newline and closing the infile pipe.
read_and_verify_code: Single "line" of code to read from a file
object named 'infile' and validate the result. This will be
executed as part of a python subprocess fed data_to_write.
i z#data_to_write must fit in pipe buf.z-uz-czXimport signal, sys ;signal.signal(signal.SIGINT, lambda s, f: sys.stderr.write("$\n")) ;z ;z"sys.stderr.write("Worm Sign!\n") ;zinfile.close())�stdinr r s Worm Sign!
zwhile awaiting a sign)r r r g�������?� �� z,reader process failed to handle our signals.s $
zwhile awaiting signal�
)�inputz exited rc=%dF)r N)r �len�AssertionError�
subprocess�Popen�sys�
executable�PIPEr r �readr r �write�selectZ send_signal�signal�SIGINTr
r �readliner �
returncode) r �
data_to_write�read_and_verify_codeZinfile_setup_codeZ worm_signZ signals_sentZrlist�_Z signal_liner r r r r �
_test_readingH s>
z'TestFileIOSignalInterrupt._test_readingz�got = infile.{read_method_name}() ;expected = {expected!r} ;assert got == expected, ("{read_method_name} returned wrong data.\n""got data %r\nexpected %r" % (got, expected))c C s | j d| jjddd�d� dS )z1readline() must handle signals and not lose data.s
hello, world!r( s hello, world!
)�read_method_name�expected)r* r+ N)r- �_READING_CODE_TEMPLATE�format)r r r r �
test_readline� s
z'TestFileIOSignalInterrupt.test_readlinec C s" | j d| jjdddgd�d� dS )z2readlines() must handle signals and not lose data.s hello
world!� readliness hello
s world!
)r. r/ )r* r+ N)r- r0 r1 )r r r r �test_readlines� s
z(TestFileIOSignalInterrupt.test_readlinesc C s8 | j d| jjddd�d� | j d| jjddd�d� dS )z0readall() must handle signals and not lose data.s hello
world!�readalls
hello
world!
)r. r/ )r* r+ r# N)r- r0 r1 )r r r r � test_readall� s z&TestFileIOSignalInterrupt.test_readallN)r r T) �__name__�
__module__� __qualname__r r r r r- r0 r2 r4 r6 r r r r r s
Ir c @ s e Zd ZdZdS )�CTestFileIOSignalInterrupt�_ioN)r7 r8 r9 r
r r r r r: � s r: c @ s e Zd ZdZdS )�PyTestFileIOSignalInterrupt�_pyioN)r7 r8 r9 r
r r r r r< � s r< c @ s e Zd Zdd� Zdd� ZdS )� TestBufferedIOSignalInterruptc C s
d| j S )z?Returns the infile = ... line of code to make a BufferedReader.ziimport %s as io ;infile = io.open(sys.stdin.fileno(), "rb") ;assert isinstance(infile, io.BufferedReader))r
)r r r r r � s z9TestBufferedIOSignalInterrupt._generate_infile_setup_codec C s | j d| jjddd�d� dS )z<BufferedReader.read() must handle signals and not lose data.s hello
world!r# s
hello
world!
)r. r/ )r* r+ N)r- r0 r1 )r r r r r6 � s
z*TestBufferedIOSignalInterrupt.test_readallN)r7 r8 r9 r r6 r r r r r> � s r> c @ s e Zd ZdZdS )� CTestBufferedIOSignalInterruptr; N)r7 r8 r9 r
r r r r r? � s r? c @ s e Zd ZdZdS )� PyTestBufferedIOSignalInterruptr= N)r7 r8 r9 r
r r r r r@ � s r@ c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestTextIOSignalInterruptc C s
d| j S )z>Returns the infile = ... line of code to make a TextIOWrapper.zvimport %s as io ;infile = io.open(sys.stdin.fileno(), "rt", newline=None) ;assert isinstance(infile, io.TextIOWrapper))r
)r r r r r � s z5TestTextIOSignalInterrupt._generate_infile_setup_codec C s | j d| jjddd�d� dS )z1readline() must handle signals and not lose data.s
hello, world!r( zhello, world!
)r. r/ )r* r+ N)r- r0 r1 )r r r r r2 � s
z'TestTextIOSignalInterrupt.test_readlinec C s" | j d| jjdddgd�d� dS )z2readlines() must handle signals and not lose data.s
hello
world!r3 zhello
zworld!
)r. r/ )r* r+ N)r- r0 r1 )r r r r r4 � s
z(TestTextIOSignalInterrupt.test_readlinesc C s | j d| jjddd�d� dS )z-read() must handle signals and not lose data.s hello
world!r# z
hello
world!
)r. r/ )r* r+ N)r- r0 r1 )r r r r r6 � s
z&TestTextIOSignalInterrupt.test_readallN)r7 r8 r9 r r2 r4 r6 r r r r rA � s rA c @ s e Zd ZdZdS )�CTestTextIOSignalInterruptr; N)r7 r8 r9 r
r r r r rB � s rB c @ s e Zd ZdZdS )�PyTestTextIOSignalInterruptr= N)r7 r8 r9 r
r r r r rC � s rC �__main__)�osr% r& r r r Zunittestr; r= Z
skipUnless�namer ZTestCaser: r< r> r? r@ rA rB rC r7 �mainr r r r �<module> s(