File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_subprocess.cpython-310.pyc
o
�i i � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZd dlmZ d dlmZ d dl m
Z d dl m
Z
d dlmZ ejdkrGd d lmZ ejd
d gZejd
d �d
�gZdd� ZG dd� dej�ZG dd� de j�ZG dd� d�Zejdkr�G dd� de�Z G dd� de e j�Z e� d�G dd � d e e j��Z G d d � d e e j�Z G d d � d e e j�Z!d!d"� Z"e�#e"� d#�G d$d%� d%e e j��Z$n
G d&d'� d'ee j�Z%G d(d)� d)�Z&e'd*kr�e�(� dS dS )+� N)�mock)�base_subprocess)�
subprocess)�utils)�support)� os_helper�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 �>/usr/local/lib/python3.10/test/test_asyncio/test_subprocess.py�tearDownModule s r c @ � 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$ + �
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_transport0 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_futurer1 �_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, r0 r/ r r r �test_proc_exited9 s&
� z)SubprocessTransportTests.test_proc_exitedc C s| | j �� }| �|�\}}|�d� | j �|� | �t|�d� d |_| �t|�d� d |_d |_| �t|�d� |� � d S )Nr2 z-<TestSubprocessTransport pid=-1 returncode=6>z(<TestSubprocessTransport pid=-1 running>z%<TestSubprocessTransport not started>)
r% r3 r1 r4 r5 r6 �reprZ _returncode�_pidrC rD r r r �test_subprocess_reprS s(
��� z-SubprocessTransportTests.test_subprocess_reprr
)r r r! r$ r1 rE rH �
__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"� Ze �
e j d#kd$�d%d&� �Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/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 � sj �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�drainrC r �read�wait)�data�proc�exitcode�r r r �runp s ��
z.SubprocessMixin.test_stdin_stdout.<locals>.run� some datag N@r )� PROGRAM_CATr �wait_forr% r5 r6 �r rV �taskrT r r rU r �test_stdin_stdoutm 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 rK )r rL r rM � communicate�
returncode)rR rS r r rU r r rV � s ��
z-SubprocessMixin.test_communicate.<locals>.runrW r )rX r rY r Z LONG_TIMEOUTr% r5 r6 rZ r rU r �test_communicate� s z SubprocessMixin.test_communicatec C s2 | j �t�d��}| j �|�� �}| �|d� d S )Nzexit 7� �r% r5 r �create_subprocess_shellrQ r6 �r rS rT r r r �
test_shell� s
�zSubprocessMixin.test_shellc C s6 | j �tjddd��}| j �|�� �}| �|d� d S )Nzexit 8T)�start_new_session� ra rc r r r �test_start_new_session� s ��z&SubprocessMixin.test_start_new_sessionc C �\ t }| j�tj|� �}|�� | j�|�� �}tjdkr$| � |t
� d S | � t j
|� d S �Nr )�PROGRAM_BLOCKEDr% r5 r rL rB rQ �sys�platform�assertIsInstance�intr6 r? �SIGKILL�r r rS r^ r r r � test_kill� � �
zSubprocessMixin.test_killc C rh ri )rj r% r5 r rL rA rQ rk rl rm rn r6 r? r@ rp r r r �test_terminate� rr z SubprocessMixin.test_terminater zDon't have SIGHUPc
s� t � t jt j�}z5d}tjd|g}� j�tj|dt j
i��}� fdd�}� j�||��}� � t j |� W t � t j|� d S t � t j|� w )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 �readliner6 r> r? �SIGHUPrQ )rS �liner^ r' r r r> � s � z5SubprocessMixin.test_send_signal.<locals>.send_signal) r? ru �SIG_DFLrk �
executabler% r5 r rL r rM r6 )r Z old_handler�coder rS 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% r5 r rL rk rx r rM )r �
large_datarS r r r �prepare_broken_pipe_test� s
��z(SubprocessMixin.prepare_broken_pipe_testc C sp | � � \}}dd� }|||�}t�� � | �ttf| jj|� W d � n1 s)w Y | j�|�� � d S )Nc � s2 �t �d�I d H | j�|� | j�� I d H d S )Ng �?)r �sleepr rN rO )rS rR r r r � write_stdin� s � z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) r �
test_utils�disable_loggerr< �BrokenPipeError�ConnectionResetErrorr% r5 rQ )r rS 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> s zESubprocessMixin.test_communicate_ignore_broken_pipe.<locals>.<lambda>)r r% Zset_exception_handlerr5 r] rQ )r rS 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 � s6 �� | i |��I d H \}}t �� |_t �� |_||fS r
)r r �
pause_reading�resume_reading)r �kwr0 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 rL rk rx r rM Z
_transportZget_pipe_transportr] )ry r� rS Zstdout_transportr r �r� r �sizer� r �test_pause_reading s$ �� � z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingr{ )r% r5 r6 r8 r� r9 r� )r r� r r0 r r� r r� s "z"SubprocessMixin.test_pause_readingc C �< dd� }| j �|d��\}}| �|�� d� | �|d� d S )Nc � �\ �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 rL rk rx r rM r] rQ ��messagery rS r r rT r r r � len_message7 � � �z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_message� abc� 3r �r% r5 r6 �rstrip)r r� �outputrT r r r �test_stdin_not_inheritable4 �
z*SubprocessMixin.test_stdin_not_inheritablec C �: dd� }| j �|� �\}}| �|�� d� | �|d� d S )Nc � s\ �d} t jtjd| t jjt jjt jjdd�I d H }|�d�I d H \}}|�� I d H }||fS )Nr� r
Fr� � r� �ry rS r r rT r r r � empty_inputJ r� z5SubprocessMixin.test_empty_input.<locals>.empty_input� 0r r� �r r� r� rT r r r �test_empty_inputH �
z SubprocessMixin.test_empty_inputc C r� )Nc � sZ �d} t jtjd| t jjt jjt jjdd�I d H }|�� I d H \}}|�� I d H }||fS r� ) r rL rk rx r �DEVNULLrM r] rQ r� r r r r� ] s � �z7SubprocessMixin.test_devnull_input.<locals>.empty_inputr� r r� r� r r r �test_devnull_input[ r� z"SubprocessMixin.test_devnull_inputc C �6 dd� }| j �|� �\}}| �|d � | �|d� d S )Nc � s\ �d} t jtjd| t jjt jjt jjdd�I d H }|�d�I d H \}}|�� I d H }||fS �Nr� r
Fr� r� � r rL rk rx r rM r� r] rQ r� r r r � empty_outputp r� z9SubprocessMixin.test_devnull_output.<locals>.empty_outputr �r% r5 r6 )r r� r� rT r r r �test_devnull_outputn �
z#SubprocessMixin.test_devnull_outputc C r� )Nc � s\ �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� r� z7SubprocessMixin.test_devnull_error.<locals>.empty_errorr r� )r r� r� rT r r r �test_devnull_error� r� z"SubprocessMixin.test_devnull_error�linuxzDon't have /dev/stdinc C r� )Nc � r� )Nz?file = open("/dev/stdin"); data = file.read(); print(len(data))r
Fr� r� r� r r r �devstdin_input� r� z;SubprocessMixin.test_devstdin_input.<locals>.devstdin_inputr� r� r r� )r r� r� rT r r r �test_devstdin_input� r� z#SubprocessMixin.test_devstdin_inputc � � fdd�}� j �|� � d S )Nc � sv �t jt� I d H } � j�| �� �}� j�|j� z|I d H W n
t jy) Y nw |�� | � � | �� I d H d S r
)
r rL rj r% � create_taskrQ � call_soon�cancel�CancelledErrorrB )rS r[ r' r r � cancel_wait� s ��z=SubprocessMixin.test_cancel_process_wait.<locals>.cancel_wait�r% r5 )r r� r r' r �test_cancel_process_wait� s z(SubprocessMixin.test_cancel_process_waitc sH � fdd�}t �� � � j�|� � W d � d S 1 s w Y d S )Nc � sN �t jt� } � j�| �}� j�|j� z|I d H W d S t jy& Y d S w r
)r rL rj r% r� r� r� r� �r� r[ r' r r �cancel_make_transport� s �
�zYSubprocessMixin.test_cancel_make_subprocess_transport_exec.<locals>.cancel_make_transport)r� r� r% r5 �r r� r r' r �*test_cancel_make_subprocess_transport_exec� s
"�z:SubprocessMixin.test_cancel_make_subprocess_transport_execc sT � fdd�}t �� � � j�|� � t �� j� W d � d S 1 s#w Y d S )Nc � sZ �� j jtjgt�R � } � j �| �}� j �|j� z|I d H W d S tjy, Y d S w r
) r% �subprocess_execr �SubprocessProtocolrj r� r� r� r� r� r' r r r� � s � � �zDSubprocessMixin.test_cancel_post_init.<locals>.cancel_make_transport)r� r� r% r5 � run_brieflyr� r r' r �test_cancel_post_init� s
"�z%SubprocessMixin.test_cancel_post_initc s� � fdd�}t �� �# z � j�|� �\}}W n tjy% � �d� Y nw W d � n1 s0w Y � �|� � �|� t � � j� d S )Nc � s| ��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 rB � s
zKSubprocessMixin.test_close_kill_running.<locals>.kill_running.<locals>.killr � ) r% r� r r� rj �get_extra_inforB r7 rC rY �_wait)�creater0 r/ rB rS r^ r' r� r � kill_running� s � �
z=SubprocessMixin.test_close_kill_running.<locals>.kill_runningz2Timeout failure on waiting for subprocess stopping)
r� r� r% r5 r � TimeoutErrorZskipTestr; r8 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 d � n1 s"w Y |\}}}� �|� � �|� � �|� tj dkrNt
� t �rPt �
� j�� d S d S d S )Nc � sp �� j jtjgt�R � } | I d H \}}|�d�}|�� |�� t� � |_|�
� }|� � }|� � |||jj
fS )Nr )r% r� r r� rj r� rB rQ r r �pollr7 rC r9 )r� r0 r/ rS �proc_returncodeZtransport_returncoder' r r r�
s � �
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_runningr )r� r� r% r5 r� ZassertIsNotNoner; r: rk rl �
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 jdkr d}nd}t�|��T}t}||_tjdd��0}| �|�� t j
t j dd|d�I d H W d � n1 s:w Y | � |g � W d � n1 sOw Y W d � d S W d � d S 1 sgw Y d S ) Nr zasyncio.windows_utils.Popenzsubprocess.PopenT)�recordr
r| r} )
rk rl r Zpatch�ZeroDivisionErrorZ side_effect�warnings�catch_warningsr< r rL rx r6 )r r �target�popen�excZwarnsr r r �_test_popen_error1 s* �
� � ��"�z!SubprocessMixin._test_popen_errorc C s | j �| jd d�� d S �Nr} )r% r5 r� r' r r r �test_popen_errorD s z SubprocessMixin.test_popen_errorc C s | j �| jtjd�� d S r� )r% r5 r� r rM r' r r r � test_popen_error_with_stdin_pipeI 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 }|r,t� d�I d H nd S q)Nr� )r zfor _ in range(64):z sys.stdout.write("x" * 4096)r� z sys.exit(1)r
)r Ti g333333�?)
r� r rL rk rx r rM r rP r� )ry �processrR r r r �executeQ s � ��zDSubprocessMixin.test_read_stdout_after_process_exit.<locals>.executer� �r r� r r r �#test_read_stdout_after_process_exitO s z3SubprocessMixin.test_read_stdout_after_process_exitc r� )Nc � �� �� � t�� tjtjdd�I d H W d � n1 s w Y � � t�� tjtjdd�I d H W d � n1 s<w Y � � t�� tjtjdd�I d H W d � d S 1 s]w Y d S �NT)�textzutf-8)�encoding�strict)�errors)r< �
ValueErrorr rL rk rx r r' r r r� g � � � � � � �"�zLSubprocessMixin.test_create_subprocess_exec_text_mode_fails.<locals>.executer� r� r r' r �+test_create_subprocess_exec_text_mode_failsf s
z;SubprocessMixin.test_create_subprocess_exec_text_mode_failsc r� )Nc � r� r� )r< r� r rb rk rx r r' r r r� x r� zMSubprocessMixin.test_create_subprocess_shell_text_mode_fails.<locals>.executer� r� r r' r �,test_create_subprocess_shell_text_mode_failsv s
z<SubprocessMixin.test_create_subprocess_shell_text_mode_failsc C s dd� }| � | j�|� �� d S )Nc � s\ �t �t�tj�dd�I d H } | �� I d H t �tjddt�d��I d H } | �� I d H d S )Nr
r| �.)r rL r �FakePathrk rx rQ )�pr r r r� � s �
�
�zFSubprocessMixin.test_create_subprocess_exec_with_path.<locals>.execute�r; r% r5 r� r r r �%test_create_subprocess_exec_with_path� s z5SubprocessMixin.test_create_subprocess_exec_with_pathN)$r r r! r\ r_ rd rg rq rs �unittestZskipIfrk rl rz 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 rJ k s@
,
%) rJ 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 rC r r( r r r
� s
z SubprocessWatcherMixin.tearDown)r r r! r r$ r
rI r r r( r r � s
r c @ � e Zd ZejZdS )� SubprocessThreadedWatcherTestsN)r r r! r ZThreadedChildWatcherr r r r r r � �
r zqbpo-38323: MultiLoopChildWatcher has a race condition and these tests can hang the test suitec @ r )� SubprocessMultiLoopWatcherTestsN)r r r! r ZMultiLoopChildWatcherr r r r r r � s
r c @ r )�SubprocessSafeWatcherTestsN)r r r! r ZSafeChildWatcherr r r r r r � r
r c @ r )r� N)r r r! r ZFastChildWatcherr r r r r r� � r
r� c C s> t td�sdS z
t�t�t�� �� W dS ty Y dS w )N�
pidfd_openFT)�hasattr�osrC r �getpid�OSErrorr r r r �has_pidfd_support� s
��r z(operating system does not support pidfdsc @ r )�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$ � r* z SubprocessProactorTests.setUp)r r r! r$ rI r r r( r r � s r c @ r )�GenericWatcherTestsc s$ � fdd�}� � � j�|� �� d S )Nc � st �t �tj�} d| j_t�| � � �t�� t �
t � t
j�dd�I d H W d � n1 s.w Y | j�� d S )NFr
r| )r Zcreate_authspecr ZAbstractChildWatcherZ is_activeZ return_valuer r< � RuntimeErrorr rL r r� rk rx 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? rk r r� r r r r Ztest.test_asyncior r� r+ r Z test.supportr rl r rx rj r� rX r ZBaseSubprocessTransportr ZTestCaser" rJ r r �skipr r r� r Z
skipUnlessr r r r �mainr r r r �<module> sl
� A
, � � � � � � �