Your IP : 216.73.216.213
3
� f�O � @ s` d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZ d dlmZ d dlmZ yd dl m
Z
W n e k
r� d dlm Z
Y nX ej
dkr�d d lmZ ejd
d gZejd
d jd �gZG dd� dej�ZG dd� dej�ZG dd� d�Zej
dk�r6G dd� de�ZG dd� deej�ZG dd� deej�ZnG d d � d eej�Ze d k�r\ej � dS ) � N)�mock)�base_subprocess)�
subprocess)�
test_utils)�support)� test_support�win32)� unix_eventsz-cz import time; time.sleep(3600)�;�
import sys� data = sys.stdin.buffer.read()� sys.stdout.buffer.write(data)c @ s e Zd Zdd� ZdS )�TestSubprocessTransportc O s& t j� | _d | j_d | j_d | j_d S )N)r �Mock�_proc�stdin�stdout�stderr)�self�args�kwargs� r �F/opt/alt/python36/lib64/python3.6/test/test_asyncio/test_subprocess.py�_start s
z TestSubprocessTransport._startN)�__name__�
__module__� __qualname__r r r r r r s r c s. e Zd Z� fdd�Zddd�Zdd� Z� ZS ) �SubprocessTransportTestsc s$ t � j� | j� | _| j| j� d S )N)�super�setUpZ
new_test_loop�loop�set_event_loop)r )� __class__r r r % s
z SubprocessTransportTests.setUpNc C s> t j� }d|j_d|j_t| j|dgdd d d d|d� }||fS )NF�testr )�waiter)r r �connection_madeZ
_is_coroutine�process_exitedr r )r r$ �protocol� transportr r r �create_transport+ s z)SubprocessTransportTests.create_transportc C s� t j| jd�}| j|�\}}|jd� | jj|� | j|j� d� | j|j j
� | j|j j
� | j|j j
� | j|j j
d d� | j|j� � | j|j� | j|j� | j|j� | jt|jtj� | jt|j� | jt|j� |j� d S )N)r � r )N) �asyncioZFuturer r) Z_process_exited�run_until_complete� assertEqual�get_returncode�
assertTruer% �calledr& Zconnection_lost� call_args� assertFalseZ
is_closing� assertIsNoneZ_loopr Z _protocol� assertRaises�ProcessLookupError� send_signal�signal�SIGTERM� terminate�kill�close)r r$ r( r' r r r �test_proc_exited4 s$
z)SubprocessTransportTests.test_proc_exited)N)r r r r r) r< �
__classcell__r r )r" r r $ s
r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
d
d�dd� �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+S ),�SubprocessMixinc s^ t � tj� �fdd��}|d�}tj|d�jd�}�jj|�\}}�j|d� �j|d� d S )Nc 3 sl t j� tjtj�jd��E d H }|jj| � |jj� E d H |jj� |j j
� E d H } |j � E d H }|| fS )N)r r r ) r+ �create_subprocess_execr �PIPEr r �write�drainr; r �read�wait)�data�proc�exitcode)r r r r �runT s
z.SubprocessMixin.test_stdin_stdout.<locals>.runs some datag N@)r r )� PROGRAM_CATr+ � coroutine�wait_forr r, r- )r rH �taskrG r r )r r r �test_stdin_stdoutQ s z!SubprocessMixin.test_stdin_stdoutc s^ t � tj� �fdd��}|d�}tj|d�jd�}�jj|�\}}�j|d� �j|d� d S )Nc 3 s> t j� tjtj�jd��E d H }|j| �E d H \}}|j|fS )N)r r r )r+ r? r r@ r � communicate�
returncode)rE rF r r )r r r r rH o s z-SubprocessMixin.test_communicate.<locals>.runs some datag N@)r r )rI r+ rJ rK r r, r- )r rH rL rG r r )r r r �test_communicatel s
z SubprocessMixin.test_communicatec C s< t jd| jd�}| jj|�}| jj|j� �}| j|d� d S )Nzexit 7)r � )r+ �create_subprocess_shellr r, rD r- )r �createrF rG r r r �
test_shell s
zSubprocessMixin.test_shellc C s> t jdd| jd�}| jj|�}| jj|j� �}| j|d� d S )Nzexit 8T)Zstart_new_sessionr � )r+ rR r r, rD r- )r rS rF rG r r r �test_start_new_session� s
z&SubprocessMixin.test_start_new_sessionc C sf t }tj|d| ji�}| jj|�}|j� | jj|j� �}tjdkrR| j |t
� n| j t j
|� d S )Nr r )�PROGRAM_BLOCKEDr+ r? r r, r: rD �sys�platform�assertIsInstance�intr- r7 �SIGKILL)r r rS rF rO r r r � test_kill� s
zSubprocessMixin.test_killc C sf t }tj|d| ji�}| jj|�}|j� | jj|j� �}tjdkrR| j |t
� n| j t j
|� d S )Nr r )rW r+ r? r r, r9 rD rX rY rZ r[ r- r7 r8 )r r rS rF rO r r r �test_terminate� s
z SubprocessMixin.test_terminateTz"Temporarily skipped (rhbz#1088233)c s� t j t jt j�}zhd}tjd|g}tj|tj� j d��}� j j
|�}tj � fdd��}� j j
||��}� j t j |� W d t j t j|� X d S )Nz<import time; print("sleeping", flush=True); time.sleep(3600)z-c)r r c 3 s: | j j� E d H }� j|d� | jtj� | j� E d H }|S )Ns sleeping
)r �readliner- r6 r7 �SIGHUPrD )rF �linerO )r r r r6 � s
z5SubprocessMixin.test_send_signal.<locals>.send_signal)
r7 r` �SIG_DFLrX �
executabler+ r? r r@ r r, rJ r- )r Z old_handler�coder rS rF r6 rO r )r r �test_send_signal� s
z SubprocessMixin.test_send_signalc C s8 dt j }tjtjddtj| jd�}| jj |�}||fS )N� xz-c�pass)r r )
r Z
PIPE_MAX_SIZEr+ r? rX rc r r@ r r, )r �
large_datarS rF r r r �prepare_broken_pipe_test� s
z(SubprocessMixin.prepare_broken_pipe_testc sf � j � \}}tj� fdd��}|||�}tj� � � jttf� jj |� W d Q R X � jj |j
� � d S )Nc 3 s6 t jd� jd�E d H | jj|� | jj� E d H d S )Ng �?)r )r+ �sleepr r rA rB )rF rE )r r r � write_stdin� s z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) ri r+ rJ r �disable_loggerr4 �BrokenPipeError�ConnectionResetErrorr r, rD )r rF rh rk �coror )r r �test_stdin_broken_pipe� s
z&SubprocessMixin.test_stdin_broken_pipec
C sF | j � \}}tj� � | jj|j|�� W d Q R X | jj|j� � d S )N)ri r rl r r, rN rD )r rF rh r r r �#test_communicate_ignore_broken_pipe� s
z3SubprocessMixin.test_communicate_ignore_broken_pipec sh d� � d d �t j� ��fdd��}�jj|� �\}}�j|d� � �j|jj� �j|jj� d S )N�
� � c 3 s� dj dd� df�} �jj� tj� fdd��}|�j_tjtjd| tjj tjj ��jd�E d H }|j
j d �}|j � E d H \}}||fS )
N�
z
import syszsys.stdout.write("x" * %s)zsys.stdout.flush()c ? s0 � | |�E d H \}}t j� |_t j� |_||fS )N)r r �
pause_reading�resume_reading)r �kwr( r' )�connect_read_piper r �connect_read_pipe_mock� s
z^SubprocessMixin.test_pause_reading.<locals>.test_pause_reading.<locals>.connect_read_pipe_mockz-c)r r �limitr rt )
�joinr ry r+ rJ r? rX rc r r@ Z
_transportZget_pipe_transportrN )rd rz rF Zstdout_transportr r )r{ r �size)ry r �test_pause_reading� s z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingrf ) r+ rJ r r, r- r/ rv r0 rw )r r~ r r( r )r{ r r} r r~ � s $z"SubprocessMixin.test_pause_readingc sF t j� fdd��}� jj|d��\}}� j|j� d� � j|d� d S )Nc
3 s^ d}t jtjd|t jjt jjt jjd� jd�E d H }|j| �E d H \}}|j� E d H }||fS )Nz5import sys; data = sys.stdin.read(); print(len(data))z-cF)r r r � close_fdsr ) r+ r? rX rc r r@ r rN rD )�messagerd rF r r rG )r r r � len_message s z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_messages abc� 3r )r+ rJ r r, r- �rstrip)r r� �outputrG r )r r �test_stdin_not_inheritable s z*SubprocessMixin.test_stdin_not_inheritablec sD t j� fdd��}� jj|� �\}}� j|j� d� � j|d� d S )Nc
3 s^ d} t jtjd| t jjt jjt jjd� jd�E d H }|jd�E d H \}}|j� E d H }||fS )Nz5import sys; data = sys.stdin.read(); print(len(data))z-cF)r r r r r � ) r+ r? rX rc r r@ r rN rD )rd rF r r rG )r r r � empty_input, s z5SubprocessMixin.test_empty_input.<locals>.empty_input� 0r )r+ rJ r r, r- r� )r r� r� rG r )r r �test_empty_input+ s z SubprocessMixin.test_empty_inputc s$ t j� fdd��}� jj|� � d S )Nc 3 s~ t jtd� ji�E d H } � jj| j� �}� jj|j� y|E d H W n t jk
rZ Y nX |j� | j � | j� E d H d S )Nr )
r+ r? rW r � create_taskrD � call_soon�cancel�CancelledErrorr: )rF rL )r r r � cancel_waitA s z=SubprocessMixin.test_cancel_process_wait.<locals>.cancel_wait)r+ rJ r r, )r r� r )r r �test_cancel_process_wait> s z(SubprocessMixin.test_cancel_process_waitc s8 t j� fdd��}tj� � � jj|� � W d Q R X d S )Nc 3 sV t jtd� ji�} � jj| �}� jj|j� y|E d H W n t jk
rP Y nX d S )Nr )r+ r? rW r r� r� r� r� )ro rL )r r r �cancel_make_transportY s
zYSubprocessMixin.test_cancel_make_subprocess_transport_exec.<locals>.cancel_make_transport)r+ rJ r rl r r, )r r� r )r r �*test_cancel_make_subprocess_transport_execX s
z:SubprocessMixin.test_cancel_make_subprocess_transport_execc sD t j� fdd��}tj� � � jj|� � tj� j� W d Q R X d S )Nc 3 sX � j jtjft�� } � j j| �}� j j|j� y|E d H W n tjk
rR Y nX d S )N) r �subprocess_execr+ �SubprocessProtocolrW r� r� r� r� )ro rL )r r r r� k s zDSubprocessMixin.test_cancel_post_init.<locals>.cancel_make_transport)r+ rJ r rl r r, � run_briefly)r r� r )r r �test_cancel_post_initj s
z%SubprocessMixin.test_cancel_post_initc s\ t j� fdd��}tj� � � jj|� �\}}W d Q R X � j|� � j|� tj� j� d S )Nc 3 sp �j jtjft�� } | E d H \}}d� � �fdd�}|jd�}|j�||_|j� }|j� |j � E d H |� fS )NFc s d� �� d S )NTr r )� kill_called� orig_killr r r: � s zKSubprocessMixin.test_close_kill_running.<locals>.kill_running.<locals>.killr )
r r� r+ r� rW �get_extra_infor: r. r; Z_wait)rS r( r' r: rF rO )r )r� r� r � kill_running~ s
z=SubprocessMixin.test_close_kill_running.<locals>.kill_running) r+ rJ r rl r r, r3 r/ r� )r r� rO �killedr )r r �test_close_kill_running} s
z'SubprocessMixin.test_close_kill_runningc
s� t j� fdd��}tj� � � jj|� �}tj� j� W d Q R X |\}}}� j|� � j|� � j |� t
j dkr�t � t
�r�t j� jj� d S )Nc 3 sl � j jtjft�� } | E d H \}}|jd�}|j� |j� tj � |_|j
� }|j � }|j � |||jj
fS )Nr )r r� r+ r� rW r� r: rD r r Zpollr. r; r0 )rS r( r' rF �proc_returncodeZtransport_returncode)r r r r� � s
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_runningr )r+ rJ r rl r r, r� ZassertIsNotNoner3 r2 rX rY �
isinstance�SubprocessFastWatcherTestsZget_child_watcherZ
_callbacks�clear)r r� �resultr� Ztransport_return_coder� r )r r � test_close_dont_kill_finished� s
z-SubprocessMixin.test_close_dont_kill_finishedc C s� t jdkrd}nd}tj|��l}t}||_tjt jdd| j d�}t
j dd��4}| j |�� | j j
|� W d Q R X | j|g � W d Q R X W d Q R X d S ) Nr zasyncio.windows_utils.Popenzsubprocess.Popenz-crg )r T)�record)rX rY r Zpatch�ZeroDivisionErrorZ side_effectr+ r? rc r �warnings�catch_warningsr4 r, r- )r �target�popen�excrS Zwarnsr r r �test_popen_error� s
z SubprocessMixin.test_popen_errorc s$ t j� fdd��}� jj|� � d S )Nc 3 st dj dddddg�} tjtjd| tjj� jd�}|E d H }x4|jj d �E d H }|rjtj
d
� jd �E d H q<P q<W d S ) Nru z
import syszfor _ in range(64):z sys.stdout.write("x" * 4096)zsys.stdout.flush()z sys.exit(1)z-c)r r i g333333�?)r ) r| r+ r? rX rc r r@ r r rC rj )rd ZfutZprocessrE )r r r �execute� s
zDSubprocessMixin.test_read_stdout_after_process_exit.<locals>.execute)r+ rJ r r, )r r� r )r r �#test_read_stdout_after_process_exit� s z3SubprocessMixin.test_read_stdout_after_process_exitN)r r r rM rP rT rV r] r^ �unittestZskipIfre ri rp rq r~ r� r� r� r� r� r� r� r� r� r r r r r> O s( . )r> c s e Zd ZdZ� fdd�Z� ZS )�SubprocessWatcherMixinNc sX t � j� tj� }|j� | _| j| j� | j� }|j| j� |j |� | j
|j d � d S )N) r r r+ Zget_event_loop_policyZnew_event_loopr r! �WatcherZ attach_loopZset_child_watcherZ
addCleanup)r ZpolicyZwatcher)r" r r r � s
z SubprocessWatcherMixin.setUp)r r r r� r r= r r )r" r r� � s r� c @ s e Zd ZejZdS )�SubprocessSafeWatcherTestsN)r r r r ZSafeChildWatcherr� r r r r r� s r� c @ s e Zd ZejZdS )r� N)r r r r ZFastChildWatcherr� r r r r r� s r� c s e Zd Z� fdd�Z� ZS )�SubprocessProactorTestsc s$ t � j� tj� | _| j| j� d S )N)r r r+ ZProactorEventLoopr r! )r )r" r r r s
z SubprocessProactorTests.setUp)r r r r r= r r )r" r r� s r� �__main__)r r r
) r7 rX r� r� r r+ r r r r# r � ImportErrorr rY r rc rW r| rI ZBaseSubprocessTransportr ZTestCaser r> r� r� r� r� r �mainr r r r �<module> sD
+ #