Your IP : 216.73.216.213
a
XC?h]h � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZd dlmZ d dlmZ d dl m
Z d dl m
Z
ejdkr�d dlmZ ejd d
gZejd d �d �gZd
d� ZG dd� dej�ZG dd� de j�ZG dd� d�Zejdk�r|G 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 d d!� Z e�!e � d"�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 �F/opt/alt/python39/lib64/python3.9/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_exited8 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 _returncodeZ_pidr@ rA r r r �test_subprocess_reprR s(
���z-SubprocessTransportTests.test_subprocess_repr)N)r r r r" r. rB rD �
__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 �runo s �
z.SubprocessMixin.test_stdin_stdout.<locals>.run� some datag N@r )� PROGRAM_CATr
�wait_forr# r2 r3 �r rR �taskrP r r rQ r �test_stdin_stdoutl s z!SubprocessMixin.test_stdin_stdoutc sR t � � fdd�}|d�}t�|tj�}| j�|�\}}| �|d� | �|d� d S )Nc � s: t j� tjtjd��I d H }|�| �I d H \}}|j|fS rG )r
rH r rI � communicate�
returncode)rN rO r r rQ r r rR � s �z-SubprocessMixin.test_communicate.<locals>.runrS r )rT r
rU r Z LONG_TIMEOUTr# r2 r3 rV r rQ r �test_communicate� s z SubprocessMixin.test_communicatec C s2 | j �t�d��}| j �|�� �}| �|d� d S )Nzexit 7� �r# r2 r
�create_subprocess_shellrM r3 �r rO rP r r r �
test_shell� s
�zSubprocessMixin.test_shellc C s6 | j �tjddd��}| j �|�� �}| �|d� d S )Nzexit 8T)Zstart_new_session� r] r_ 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
rH r? rM �sys�platform�assertIsInstance�intr3 r< �SIGKILL�r r rO rZ r r r � test_kill� s �
zSubprocessMixin.test_killc C sZ t }| j�tj|� �}|�� | j�|�� �}tjdkrF| � |t
� n| � t j
|� d S rc )rd r# r2 r
rH r> rM re rf rg rh r3 r< r= rj r r r �test_terminate� s �
z SubprocessMixin.test_terminater zDon't have SIGHUPc
s� t � t jt j�}zhd}tjd|g}� j�tj|dt j
i��}� fdd�}� j�||��}� � t j |� W t � t j|� nt � t j|� 0 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< �SIGHUPrM )rO �linerZ r% r r r; � s
z5SubprocessMixin.test_send_signal.<locals>.send_signal) r< rn �SIG_DFLre �
executabler# r2 r
rH r rI r3 )r Z old_handler�coder rO r; rZ 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
rH re rq r rI )r �
large_datarO r r r �prepare_broken_pipe_test� s
��z(SubprocessMixin.prepare_broken_pipe_testc C sp | � � \}}dd� }|||�}t�� �& | �ttf| jj|� W d � n1 sR0 Y | j�|�� � d S )Nc � s0 t �d�I d H | j�|� | j�� I d H d S )Ng �?)r
�sleepr rJ rK )rO rN r r r � write_stdin� s z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) rx �
test_utils�disable_loggerr9 �BrokenPipeError�ConnectionResetErrorr# r2 rM )r rO rw rz �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>)rx r# Zset_exception_handlerr2 rY rM )r rO rw 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 � s4 � | i |��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
rH re rq r rI Z
_transportZget_pipe_transportrY )rr r� rO Zstdout_transportr r �r� r �sizer� r �test_pause_reading s" � � z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingrt )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 Z close_fds�r
rH re rq r rI rY rM )�messagerr rO r r rP r r r � len_message6 s � z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_message� abc� 3r �r# r2 r3 �rstrip)r r� �outputrP r r r �test_stdin_not_inheritable3 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� �rr rO r r rP r r r � empty_inputI s � z5SubprocessMixin.test_empty_input.<locals>.empty_input� 0r r� �r r� r� rP r r r �test_empty_inputG 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
rH re rq r �DEVNULLrI rY rM r� r r r r� \ s � z7SubprocessMixin.test_devnull_input.<locals>.empty_inputr� r r� r� r r r �test_devnull_inputZ 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
rH re rq r rI r� rY rM r� r r r � empty_outputo s � z9SubprocessMixin.test_devnull_output.<locals>.empty_outputr �r# r2 r3 )r r� r� rP r r r �test_devnull_outputm 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� rP r r r �test_devnull_error� s
z"SubprocessMixin.test_devnull_errorc s � fdd�}� j �|� � d S )Nc � st t jt� I d H } � j�| �� �}� j�|j� z|I d H W n t jyP Y n0 |�� | � � | �� I d H d S r )
r
rH rd r# � create_taskrM � call_soon�cancel�CancelledErrorr? )rO rW 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 sF � fdd�}t �� � � j�|� � W d � n1 s80 Y d S )Nc � sL t jt� } � j�| �}� j�|j� z|I d H W n t jyF Y n0 d S r )r
rH rd r# r� r� r� r� �r rW 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 sR � fdd�}t �� �* � j�|� � t �� j� W d � n1 sD0 Y d S )Nc � sX � j jtjgt�R � } � j �| �}� j �|j� z|I d H W n tjyR Y n0 d S r ) r# �subprocess_execr
�SubprocessProtocolrd 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 s� � fdd�}t �� �F z� j�|� �\}}W n tjyJ � �d� Y n0 W d � n1 s`0 Y � �|� � �|� t � � j� d S )Nc � sz �j jtjgt�R � } | I d H \}}d� � �fdd�}|�d�}|j�||_|�� }|�� t� |�
� d�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� rd �get_extra_infor? r4 r@ rU Z_wait)�creater- r, r? rO rZ r% r� r � kill_running� s �
z=SubprocessMixin.test_close_kill_running.<locals>.kill_runningz2Timeout failure on waiting for subprocess stopping)
r{ r| r# r2 r
� TimeoutErrorZskipTestr8 r5 r� )r r� rZ �killedr r% r �test_close_kill_running� s
� �(
z'SubprocessMixin.test_close_kill_runningc s� � fdd�}t �� �* � j�|� �}t �� j� W d � n1 sD0 Y |\}}}� �|� � �|� � �|� tj dkr�t
� t �r�t �
� j�� d S )Nc � sn � j jtjgt�R � } | I d H \}}|�d�}|�� |�� t� � |_|�
� }|� � }|� � |||jj
fS )Nr )r# r� r
r� rd r� r? rM r r Zpollr4 r@ r6 )r� r- r, rO �proc_returncodeZtransport_returncoder% r r r� � s �
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_runningr )r{ r| r# r2 r� ZassertIsNotNoner8 r7 re rf �
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�|���}t}||_tjdd��`}| �|��* t j
t j dd|d�I d H W d � n1 sr0 Y | � |g � W d � n1 s�0 Y W d � n1 s�0 Y d S ) Nr zasyncio.windows_utils.Popenzsubprocess.PopenT)�recordr ru rv )
re rf r Zpatch�ZeroDivisionErrorZ side_effect�warnings�catch_warningsr9 r
rH rq r3 )r r �target�popen�excZwarnsr r r �_test_popen_error s
�*z!SubprocessMixin._test_popen_errorc C s | j �| jd d�� d S �Nrv )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 rI r% r r r � test_popen_error_with_stdin_pipe4 s �z0SubprocessMixin.test_popen_error_with_stdin_pipec C s dd� }| j �|� � d S )Nc � s\ d� g d��} tjtjd| tjjd�I d H }|j�d�I d H }|rXt� d�I d H q,qXq,d S )Nr� )r zfor _ in range(64):z sys.stdout.write("x" * 4096)r� z sys.exit(1)r )r i g333333�?)
r� r
rH re rq r rI r rL ry )rr ZprocessrN r r r �execute< s � zDSubprocessMixin.test_read_stdout_after_process_exit.<locals>.executer� �r r� r r r �#test_read_stdout_after_process_exit: s z3SubprocessMixin.test_read_stdout_after_process_exitc s � fdd�}� j �|� � d S )Nc � s� � � t��& tjtjdd�I d H W d � n1 s60 Y � � t��& tjtjdd�I d H W d � n1 sv0 Y � � t��& tjtjdd�I d H W d � n1 s�0 Y d S �NT)�textzutf-8)�encoding�strict)�errors)r9 �
ValueErrorr rH re rq r r% r r r� R s �* �* �zLSubprocessMixin.test_create_subprocess_exec_text_mode_fails.<locals>.executer� r� r r% r �+test_create_subprocess_exec_text_mode_failsQ s
z;SubprocessMixin.test_create_subprocess_exec_text_mode_failsc s � fdd�}� j �|� � d S )Nc � s� � � t��& tjtjdd�I d H W d � n1 s60 Y � � t��& tjtjdd�I d H W d � n1 sv0 Y � � t��& tjtjdd�I d H W d � n1 s�0 Y d S r� )r9 r� r r^ re rq r r% r r r� c s �* �* �zMSubprocessMixin.test_create_subprocess_shell_text_mode_fails.<locals>.executer� r� r r% r �,test_create_subprocess_shell_text_mode_failsa 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 ru �.)r rH r �FakePathre rq rM )�pr r r r� s s �
�
zFSubprocessMixin.test_create_subprocess_exec_with_path.<locals>.execute�r8 r# r2 r� r r r �%test_create_subprocess_exec_with_pathr s z5SubprocessMixin.test_create_subprocess_exec_with_pathc s � fdd�}� j �|� � d S )Nc � sX � � t��, tjtjdd� jd�I d H } W d � n1 s<0 Y | �� I d H d S )Nr ru �r# )� assertWarns�DeprecationWarningr
rH re rq r# rM �rO r% r r �go~ s �*z5SubprocessMixin.test_exec_loop_deprecated.<locals>.gor� �r r� r r% r �test_exec_loop_deprecated} s z)SubprocessMixin.test_exec_loop_deprecatedc s � fdd�}� j �|� � d S )Nc � sR � � t��& tjd� jd�I d H } W d � n1 s60 Y | �� I d H d S )Nzexit 0r� )r� r� r
r^ r# rM 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 rX r[ r` rb rk rl �unittestZskipIfre rf rs rx 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 rF j s>
,%)
rF 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� rE 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 C s@ t td�sdS zt�t�t�� �� W n ty: Y dS 0 dS )N�
pidfd_openFT)�hasattr�osr@ r� �getpid�OSErrorr r r r �has_pidfd_support� s
r z(operating system does not support pidfdsc @ s e Zd ZejZdS )�SubprocessPidfdWatcherTestsN)r r r r ZPidfdChildWatcherr� 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" rE r r r& r r � s r c @ s e Zd Zdd� ZdS )�GenericWatcherTestsc s$ � fdd�}� � � j�|� �� d S )Nc � sr t �tj�} d| j_t�| � � �t��, t �
t � t
j�dd�I d H W d � n1 sZ0 Y | j�� d S )NFr ru )r Zcreate_authspecr
ZAbstractChildWatcherZ is_activeZ return_valuer� r9 � RuntimeErrorr rH r r� re rq 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� r< re r� r� r r
r r Ztest.test_asyncior r{ r( r rf r rq rd r� rT r ZBaseSubprocessTransportr ZTestCaser rF r� r� �skipr� r� r� r Z
skipUnlessr r r r �mainr r r r �<module> sf
� A , � � � � � �