File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_subprocess.cpython-38.pyc
U
/D�gSe � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlZd dlmZ d dlmZ d dlm Z
d dl m Z ej
dkrzd dlmZ ejd d
gZejd d �d �gZd
d� ZG dd� dej�ZG dd� de
j�ZG dd� d�Zej
dk�rJG dd� de�ZG dd� dee
j�Ze�d�G dd� dee
j��Z G d d � d ee
j�Z G d d � d ee
j�Z nG d d!� d!ee
j�Z G d"d#� d#�Z e!d$k�r~e�"� dS )%� N)�mock)�base_subprocess)�
subprocess)�utils)�support�win32)� unix_events�-cz import time; time.sleep(3600)�;)�
import sysz data = sys.stdin.buffer.read()z sys.stdout.buffer.write(data)c C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �@/opt/cppython/lib/python3.8/test/test_asyncio/test_subprocess.py�tearDownModule s r c @ s e Zd Zdd� ZdS )�TestSubprocessTransportc O s. t �� | _d | j_d | j_d | j_d| j_d S )N���)r �Mock�_proc�stdin�stdout�stderr�pid)�self�args�kwargsr r r �_start s
z TestSubprocessTransport._startN)�__name__�
__module__� __qualname__r r r r r r s r c s6 e Zd Z� fdd�Zd
dd�Zdd� Zdd � Z� ZS ) �SubprocessTransportTestsc s$ t � �� | �� | _| �| j� d S r )�super�setUpZ
new_test_loop�loop�set_event_loop�r �� __class__r r r" ) s
z SubprocessTransportTests.setUpNc C s> t �� }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� | j �� }| �|�\}}|�d� | j �|� | �|�� d� | �|jj � | �|j
j � | �|j j � | �|j j d d� | �
|�� � | �|j� | �|j� | �|j� | �t|jtj� | �t|j� | �t|j� |�� d S )N� r r )r# �
create_futurer. �_process_exited�run_until_complete� assertEqual�get_returncode�
assertTruer* �calledr+ Zconnection_lostZ 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_exited7 s(
�z)SubprocessTransportTests.test_proc_exitedc C s| | j �� }| �|�\}}|�d� | j �|� | �t|�d� d |_| �t|�d� d |_d |_| �t|�d� |� � d S )Nr/ z-<TestSubprocessTransport pid=-1 returncode=6>z(<TestSubprocessTransport pid=-1 running>z%<TestSubprocessTransport not started>)
r# r0 r. r1 r2 r3 �reprZ _returncode�_pidr@ rA r r r �test_subprocess_reprQ s(
���z-SubprocessTransportTests.test_subprocess_repr)N)r r r r" r. rB rE �
__classcell__r r r& r r ( s
r c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze �
e j d
kd�dd� �Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z d7d8� Z!d9d:� Z"d;d<� Z#d=d>� Z$d?S )@�SubprocessMixinc sP t � � fdd�}|d�}t�|d�}| j�|�\}}| �|d� | �|d� d S )Nc � sh t j� tjtjd��I d H }|j�| � |j�� I d H |j�� |j� � I d H } |�
� I d H }|| fS �N)r r ) r
�create_subprocess_execr �PIPEr �write�drainr@ r �read�wait)�data�proc�exitcode�r r r �runn s �
z.SubprocessMixin.test_stdin_stdout.<locals>.run� some data� N@r �� PROGRAM_CATr
�wait_forr# r2 r3 �r rS �taskrQ r r rR r �test_stdin_stdoutk s z!SubprocessMixin.test_stdin_stdoutc sP t � � fdd�}|d�}t�|d�}| j�|�\}}| �|d� | �|d� d S )Nc � s: t j� tjtjd��I d H }|�| �I d H \}}|j|fS rH )r
rI r rJ � communicate�
returncode)rO rP r r rR r r rS � s �z-SubprocessMixin.test_communicate.<locals>.runrT rU r rV rY r rR r �test_communicate� s z SubprocessMixin.test_communicatec C s2 | j �t�d��}| j �|�� �}| �|d� d S )Nzexit 7� �r# r2 r
�create_subprocess_shellrN r3 �r rP rQ r r r �
test_shell� s
�zSubprocessMixin.test_shellc C s6 | j �tjddd��}| j �|�� �}| �|d� d S )Nzexit 8T)�start_new_session� r` rb r r r �test_start_new_session� s ��z&SubprocessMixin.test_start_new_sessionc C sZ t }| j�tj|� �}|�� | j�|�� �}tjdkrF| � |t
� n| � t j
|� d S �Nr )�PROGRAM_BLOCKEDr# r2 r
rI r? rN �sys�platform�assertIsInstance�intr3 r< �SIGKILL�r r rP r] r r r � test_kill� s �
zSubprocessMixin.test_killc C sZ t }| j�tj|� �}|�� | j�|�� �}tjdkrF| � |t
� n| � t j
|� d S rg )rh r# r2 r
rI r> rN ri rj rk rl r3 r< r= rn r r r �test_terminate� s �
z SubprocessMixin.test_terminater zDon't have SIGHUPc
s� t � t jt j�}zZd}tjd|g}� j�tj|dt j
i��}� fdd�}� j�||��}� � t j |� W 5 t � t j|� X d S )Nz<import time; print("sleeping", flush=True); time.sleep(3600)r r c � s: | j �� I d H }� �|d� | �tj� | �� I d H }|S )Ns sleeping
)r �readliner3 r; r< �SIGHUPrN )rP �liner] r% r r r; � s
z5SubprocessMixin.test_send_signal.<locals>.send_signal) r< rr �SIG_DFLri �
executabler# r2 r
rI r rJ r3 )r Z old_handler�coder rP r; r] r r% r �test_send_signal� s ��� z SubprocessMixin.test_send_signalc C s0 dt j }| j�tjtjddtj d��}||fS )N� xr �pass�r )
r Z
PIPE_MAX_SIZEr# r2 r
rI ri ru r rJ )r �
large_datarP r r r �prepare_broken_pipe_test� s
��z(SubprocessMixin.prepare_broken_pipe_testc C s\ | � � \}}dd� }|||�}t�� � | �ttf| jj|� W 5 Q R X | j�|�� � d S )Nc � s0 t �d�I d H | j�|� | j�� I d H d S )Ng �?)r
�sleepr rK rL )rP rO r r r � write_stdin� s z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) r| �
test_utils�disable_loggerr9 �BrokenPipeError�ConnectionResetErrorr# r2 rN )r rP r{ r~ �coror r r �test_stdin_broken_pipe� s
�z&SubprocessMixin.test_stdin_broken_pipec C sB | � � \}}| j�dd� � | j�|�|�� | j�|�� � d S )Nc S s d S r r )r# �msgr r r �<lambda> � zESubprocessMixin.test_communicate_ignore_broken_pipe.<locals>.<lambda>)r| r# Zset_exception_handlerr2 r\ rN )r rP r{ r r r �#test_communicate_ignore_broken_pipe� s z3SubprocessMixin.test_communicate_ignore_broken_pipec sb d� � d d �� ��fdd�}�j �|� �\}}��|d� � ��|jj� ��|jj� d S )N�
� � c � s| d� dd� df�} �jj� � fdd�}|�j_tjtjd| tjjtjj�d�I d H }|j �
d �}|� � I d H \}}||fS )
N�
r zsys.stdout.write("x" * %s)�sys.stdout.flush()c � s0 � | |�I d H \}}t �� |_t �� |_||fS r )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_mockr )r r �limitr� ) �joinr# r� r
rI ri ru r rJ Z
_transportZget_pipe_transportr\ )rv r� rP Zstdout_transportr r �r� r �sizer� r �test_pause_reading
s& � � z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingrx )r# r2 r3 r5 r� r6 r� )r r� r r- r r� r r� s "z"SubprocessMixin.test_pause_readingc C s< dd� }| j �|d��\}}| �|�� d� | �|d� d S )Nc � sZ d}t jtjd|t jjt jjt jjdd�I d H }|�| �I d H \}}|�� I d H }||fS �N�5import sys; data = sys.stdin.read(); print(len(data))r F�r r r � close_fds�r
rI ri ru r rJ r\ rN )�messagerv rP r r rQ r r r � len_message5 s � z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_message� abc� 3r �r# r2 r3 �rstrip)r r� �outputrQ r r r �test_stdin_not_inheritable2 s
z*SubprocessMixin.test_stdin_not_inheritablec C s: dd� }| j �|� �\}}| �|�� d� | �|d� d S )Nc � sZ d} t jtjd| t jjt jjt jjdd�I d H }|�d�I d H \}}|�� I d H }||fS )Nr� r Fr� r� r� �rv rP r r rQ r r r � empty_inputH s � z5SubprocessMixin.test_empty_input.<locals>.empty_input� 0r r� �r r� r� rQ r r r �test_empty_inputF s
z SubprocessMixin.test_empty_inputc C s: dd� }| j �|� �\}}| �|�� d� | �|d� d S )Nc � sX d} t jtjd| t jjt jjt jjdd�I d H }|�� I d H \}}|�� I d H }||fS r� ) r
rI ri ru r �DEVNULLrJ r\ rN r� r r r r� [ s � z7SubprocessMixin.test_devnull_input.<locals>.empty_inputr� r r� r� r r r �test_devnull_inputY s
z"SubprocessMixin.test_devnull_inputc C s6 dd� }| j �|� �\}}| �|d � | �|d� d S )Nc � sZ d} t jtjd| t jjt jjt jjdd�I d H }|�d�I d H \}}|�� I d H }||fS �Nr� r Fr� r� � r
rI ri ru r rJ r� r\ rN r� r r r � empty_outputn s � z9SubprocessMixin.test_devnull_output.<locals>.empty_outputr �r# r2 r3 )r r� r� rQ r r r �test_devnull_outputl s
z#SubprocessMixin.test_devnull_outputc C s6 dd� }| j �|� �\}}| �|d � | �|d� d S )Nc � sZ d} t jtjd| t jjt jjt jjdd�I d H }|�d�I d H \}}|�� I d H }||fS r� r� r� r r r � empty_error� s � z7SubprocessMixin.test_devnull_error.<locals>.empty_errorr r� )r r� r� rQ r r r �test_devnull_error s
z"SubprocessMixin.test_devnull_errorc s � fdd�}� j �|� � d S )Nc � sv t jt� I d H } � j�| �� �}� j�|j� z|I d H W n t jk
rR Y nX |�� | � � | �� I d H d S r )
r
rI rh r# � create_taskrN � call_soon�cancel�CancelledErrorr? )rP rZ r% r r � cancel_wait� s z=SubprocessMixin.test_cancel_process_wait.<locals>.cancel_wait�r# r2 )r r� r r% r �test_cancel_process_wait� s z(SubprocessMixin.test_cancel_process_waitc s2 � fdd�}t �� � � j�|� � W 5 Q R X d S )Nc � sN t jt� } � j�| �}� j�|j� z|I d H W n t jk
rH Y nX d S r )r
rI rh r# r� r� r� r� �r� rZ r% r r �cancel_make_transport� s
zYSubprocessMixin.test_cancel_make_subprocess_transport_exec.<locals>.cancel_make_transport)r r� r# r2 �r r� r r% r �*test_cancel_make_subprocess_transport_exec� s
z:SubprocessMixin.test_cancel_make_subprocess_transport_execc s> � fdd�}t �� � � j�|� � t �� j� W 5 Q R X d S )Nc � sX � j jtjft�� } � j �| �}� j �|j� z|I d H W n tjk
rR Y nX d S r ) r# �subprocess_execr
�SubprocessProtocolrh r� r� r� r� r� r% r r r� � s � zDSubprocessMixin.test_cancel_post_init.<locals>.cancel_make_transport)r r� r# r2 � run_brieflyr� r r% r �test_cancel_post_init� s
z%SubprocessMixin.test_cancel_post_initc sV � fdd�}t �� � � j�|� �\}}W 5 Q R X � �|� � �|� t �� j� d S )Nc � sp �j jtjft�� } | I d H \}}d� � �fdd�}|�d�}|j�||_|�� }|�� |� � I d H |� fS )NFc s d� �� d S )NTr r �Z kill_calledZ orig_killr r r? � s zKSubprocessMixin.test_close_kill_running.<locals>.kill_running.<locals>.killr )
r# r� r
r� rh �get_extra_infor? r4 r@ �_wait)�creater- r, r? rP r] r% r� r � kill_running� s �
z=SubprocessMixin.test_close_kill_running.<locals>.kill_running)r r� r# r2 r8 r5 r� )r r� r] �killedr r% r �test_close_kill_running� s
z'SubprocessMixin.test_close_kill_runningc s� � fdd�}t �� � � j�|� �}t �� j� W 5 Q R X |\}}}� �|� � �|� � �|� tj dkr�t
� t �r�t �
� j�� d S )Nc � sl � j jtjft�� } | I d H \}}|�d�}|�� |�� t� � |_|�
� }|� � }|� � |||jj
fS )Nr )r# r� r
r� rh r� r? rN r r �pollr4 r@ r6 )r� r- r, rP �proc_returncodeZtransport_returncoder% r r r� � s �
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_runningr )r r� r# r2 r� ZassertIsNotNoner8 r7 ri rj �
isinstance�SubprocessFastWatcherTestsr
�get_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 � s� t jdkrd}nd}t�|��d}t}||_tjdd��B}| �|�� t j
t j dd|d�I d H W 5 Q R X | � |g � W 5 Q R X W 5 Q R X d S ) Nr zasyncio.windows_utils.Popenzsubprocess.PopenT)�recordr ry rz )
ri rj r Zpatch�ZeroDivisionErrorZ side_effect�warnings�catch_warningsr9 r
rI ru r3 )r r �target�popen�excZwarnsr r r �_test_popen_error s
�z!SubprocessMixin._test_popen_errorc C s | j �| jd d�� d S �Nrz )r# r2 r� r% r r r �test_popen_error' s z SubprocessMixin.test_popen_errorc C s | j �| jtjd�� d S r� )r# r2 r� r rJ r% r r r � test_popen_error_with_stdin_pipe, s �z0SubprocessMixin.test_popen_error_with_stdin_pipec C s dd� }| j �|� � d S )Nc � sb d� dddddg�} tjtjd| tjjd�I d H }|j�d �I d H }|r^t� d
�I d H q2q^q2d S ) Nr� r zfor _ in range(64):z sys.stdout.write("x" * 4096)r� z sys.exit(1)r )r i g333333�?)
r� r
rI ri ru r rJ r rM r} )rv �processrO r r r �execute4 s � � zDSubprocessMixin.test_read_stdout_after_process_exit.<locals>.executer� �r r� r r r �#test_read_stdout_after_process_exit2 s z3SubprocessMixin.test_read_stdout_after_process_exitc s � fdd�}� j �|� � d S )Nc � s� � � t�� tjtjdd�I d H W 5 Q R X � � t�� tjtjdd�I d H W 5 Q R X � � t�� tjtjdd�I d H W 5 Q R X d S �NT)�textzutf-8)�encoding�strict)�errors)r9 �
ValueErrorr rI ri ru r r% r r r� J s � � �zLSubprocessMixin.test_create_subprocess_exec_text_mode_fails.<locals>.executer� r� r r% r �+test_create_subprocess_exec_text_mode_failsI s
z;SubprocessMixin.test_create_subprocess_exec_text_mode_failsc s � fdd�}� j �|� � d S )Nc � s� � � t�� tjtjdd�I d H W 5 Q R X � � t�� tjtjdd�I d H W 5 Q R X � � t�� tjtjdd�I d H W 5 Q R X d S r� )r9 r� r ra ri ru r r% r r r� [ s � � �zMSubprocessMixin.test_create_subprocess_shell_text_mode_fails.<locals>.executer� r� r r% r �,test_create_subprocess_shell_text_mode_failsY s
z<SubprocessMixin.test_create_subprocess_shell_text_mode_failsc C s dd� }| � | j�|� �� d S )Nc � sZ t �t�tj�dd�I d H } | �� I d H t �tjddt�d��I d H } | �� I d H d S )Nr ry �.)r rI r �FakePathri ru rN )�pr r r r� k s
�
�
zFSubprocessMixin.test_create_subprocess_exec_with_path.<locals>.execute�r8 r# r2 r� r r r �%test_create_subprocess_exec_with_pathj s z5SubprocessMixin.test_create_subprocess_exec_with_pathc s � fdd�}� j �|� � d S )Nc � sD � � t��" tjtjdd� jd�I d H } W 5 Q R X | �� I d H d S )Nr ry �r# )� assertWarns�DeprecationWarningr
rI ri ru r# rN �rP r% r r �gov s �z5SubprocessMixin.test_exec_loop_deprecated.<locals>.gor� �r r� r r% r �test_exec_loop_deprecatedu s z)SubprocessMixin.test_exec_loop_deprecatedc s � fdd�}� j �|� � d S )Nc � s> � � t�� tjd� jd�I d H } W 5 Q R X | �� I d H d S )Nzexit 0r� )r� r� r
ra r# rN r� r% r r r� � s �z6SubprocessMixin.test_shell_loop_deprecated.<locals>.gor� r� r r% r �test_shell_loop_deprecated s z*SubprocessMixin.test_shell_loop_deprecatedN)%r r r r[ r^ rc rf ro rp �unittestZskipIfri rj rw r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rG i s>
, )
rG c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )�SubprocessWatcherMixinNc sJ t � �� t�� }|�� | _| �| j� | �� }|�| j� |� |� d S r )
r! r" r
�get_event_loop_policyZnew_event_loopr# r$ �Watcher� attach_loop�set_child_watcher�r Zpolicy�watcherr& r r r" � s
z SubprocessWatcherMixin.setUpc s: t � �� t�� }|�� }|�d � |�d � |�� d S r )r! �tearDownr
r� r� r� r� r@ r� r& r r r� � s
z SubprocessWatcherMixin.tearDown)r r r r� r" r� rF r r r& r r� � s
r� c @ s e Zd ZejZdS )� SubprocessThreadedWatcherTestsN)r r r r ZThreadedChildWatcherr� r r r r r� � s r� zqbpo-38323: MultiLoopChildWatcher has a race condition and these tests can hang the test suitec @ s e Zd ZejZdS )� SubprocessMultiLoopWatcherTestsN)r r r r ZMultiLoopChildWatcherr� 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 � �� t�� | _| �| j� d S r )r! r" r
ZProactorEventLoopr# r$ r% r& r r r" � s
z SubprocessProactorTests.setUp)r r r r" rF r r r& r r � s r c @ s e Zd Zdd� ZdS )�GenericWatcherTestsc s$ � fdd�}� � � j�|� �� d S )Nc � s^ t �tj�} d| j_t�| � � �t��" t �
t � t
j�dd�I d H W 5 Q R X | j�� d S )NFr ry )r Zcreate_authspecr
ZAbstractChildWatcherZ is_activeZ return_valuer� r9 � RuntimeErrorr rI r r� ri ru Zadd_child_handlerZassert_not_called)r� r% r r r� � s
�zWGenericWatcherTests.test_create_subprocess_fails_with_inactive_watcher.<locals>.executer� r� r r% r �2test_create_subprocess_fails_with_inactive_watcher� s zFGenericWatcherTests.test_create_subprocess_fails_with_inactive_watcherN)r r r r r r r r r � s r �__main__)#r< ri r� r� r r
r r Ztest.test_asyncior r r( r rj r ru rh r� rW r ZBaseSubprocessTransportr ZTestCaser rG r� r� �skipr r r� r r r �mainr r r r �<module> sV
� A % � � � �