Your IP : 216.73.217.129
3
��2aE � @ sd d dl Z d dl mZ d dlmZ d dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ y d dlZW n ek
r� dZY n
X d dlZy d dlZW n ek
r� dZY nX y d dlZW n ek
�r& dZY nX ej�r:e j d��ejdkZ e �rPdZ ndZ d&Z e e!e"fZ#G d
d � d e j$�Z%G d d
� d
e&�Z'G dd� dej(�Z)G dd� de%�Z*G dd� de%�Z+e j,e d�G dd� de%��Z-e j.e d�G dd� de%��Z/G dd� de j$�Z0e j.e1ed �d �G d d � d e*��Z2e j.e d �G d!d"� d"e%��Z3G d#d$� d$e%�Z4e5d%k�r`e j6� dS )'� N)�mock)�support)�FakePathztest is not helpful for PGO�win32z@import msvcrt; msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY);� �nonexisting_i_hopec @ s& e Zd Zdd� Zdd� Zddd�ZdS ) � BaseTestCasec C s t j� d S )N)r �
reap_children)�self� r �4/opt/alt/python36/lib64/python3.6/test_subprocess.py�setUp< s zBaseTestCase.setUpc C sB xt jD ] }|j� qW t j� | jt jd� | j� tj� d S )Nz subprocess._active not empty)�
subprocess�_active�waitZ_cleanup� assertFalseZ
doCleanupsr r )r
Zinstr r r �tearDownA s zBaseTestCase.tearDownNc C s$ t j|�}|j� }| j|||� d S )N)r �strip_python_stderr�strip� assertEqual)r
�stderr�expected�msg�actualr r r �assertStderrEqualI s
z BaseTestCase.assertStderrEqual)N)�__name__�
__module__� __qualname__r
r r r r r r r ; s r c @ s e Zd ZdS )�PopenTestExceptionN)r r r r r r r r S s r c @ s e Zd Zdd� ZdS )�PopenExecuteChildRaisesc O s t d��d S )NzForced Exception for Test)r )r
�args�kwargsr r r �_execute_child[ s z&PopenExecuteChildRaises._execute_childN)r r r r" r r r r r W s r c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zejed-�d.d/� �Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!ejed:�d;d<� �Z"ejed:�d=d>� �Z#d?d@� Z$eje%j&e%j'kdA�dBdC� �Z(eje%j&e%j'kdA�eje)j*� dD�dEdF� ��Z+dGdH� Z,dIdJ� Z-dKdL� Z.dMdN� Z/dOdP� Z0dQdR� Z1dSdT� Z2dUdV� Z3dWdX� Z4dYdZ� Z5d[d\� Z6d]d^� Z7d_d`� Z8dadb� Z9dcdd� Z:dedf� Z;dgdh� Z<eje%j=dikdj�eje)j>dk�dlk dm�dndo� ��Z?dpdq� Z@drds� ZAdtdu� ZBdvdw� ZCdxdy� ZDdzd{� ZEd|d}� ZFd~d� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�� Z_eje`dlkd��d�d�� �Zaeje`dlkd��d�d�� �Zbd�d�� Zcd�d�� Zdd�d�� Zed�d�� Zfejgeheid��d��ejgehejd��d��ejgehejd��dd�dĄ ���Zkejgejjljmd�ejjn� �dƃd�dȄ �ZoejedɃd�d˄ �ZpejedɃd�d̈́ �ZqdlS )��ProcessTestCasec
C s� t jtjddgt jt jt jd�}z4| j|jtj� | j|j tj� | j|j
tj� W d |jj � |j j � |j
j � |j � X d S )Nz-czimport sys; sys.exit(0))�stdin�stdoutr )
r �Popen�sys�
executable�PIPE�assertIsInstancer$ �io�BufferedIOBaser% r �closer )r
�pr r r �test_io_buffered_by_defaulta s
z+ProcessTestCase.test_io_buffered_by_defaultc
C s� t jtjddgt jt jt jdd�}z4| j|jtj� | j|j tj� | j|j
tj� W d |jj � |j j � |j
j � |j � X d S )Nz-czimport sys; sys.exit(0)r )r$ r% r �bufsize)
r r& r' r( r) r* r$ r+ � RawIOBaser% r r- r )r
r. r r r �test_io_unbuffered_workso s
z(ProcessTestCase.test_io_unbuffered_worksc C s" t jtjddg�}| j|d� d S )Nz-czimport sys; sys.exit(47)�/ )r �callr' r( r )r
�rcr r r �
test_call_seq} s
z ProcessTestCase.test_call_seqc C s" | j tjtjtjddgdd� d S )Nz-czwhile True: passg�������?)�timeout)� assertRaisesr �TimeoutExpiredr4 r' r( )r
r r r �test_call_timeout� s
z!ProcessTestCase.test_call_timeoutc C s" t jtjddg�}| j|d� d S )Nz-czimport sys; sys.exit(0)r )r �
check_callr' r( r )r
r5 r r r �test_check_call_zero� s
z$ProcessTestCase.test_check_call_zeroc C s> | j tj��}tjtjddg� W d Q R X | j|jjd� d S )Nz-czimport sys; sys.exit(47)r3 ) r8 r �CalledProcessErrorr; r' r( r � exception�
returncode)r
�cr r r �test_check_call_nonzero� s
z'ProcessTestCase.test_check_call_nonzeroc C s" t jtjddg�}| jd|� d S )Nz-cz
print('BDFL')s BDFL)r � check_outputr' r( �assertIn)r
�outputr r r �test_check_output� s z!ProcessTestCase.test_check_outputc C s> | j tj��}tjtjddg� W d Q R X | j|jjd� d S )Nz-czimport sys; sys.exit(5)� ) r8 r r= rB r' r( r r> r? )r
r@ r r r �test_check_output_nonzero� s z)ProcessTestCase.test_check_output_nonzeroc C s( t jtjddgt jd�}| jd|� d S )Nz-cz$import sys; sys.stderr.write('BDFL'))r s BDFL)r rB r' r( �STDOUTrC )r
rD r r r �test_check_output_stderr� s
z(ProcessTestCase.test_check_output_stderrc C sN t j� }| j|j� |jd� |jd� tjtj ddg|d�}| j
d|� d S )Ns pearr z-cz6import sys; sys.stdout.write(sys.stdin.read().upper()))r$ s PEAR) �tempfile�
TemporaryFile�
addCleanupr- �write�seekr rB r' r( rC )r
�tfrD r r r �test_check_output_stdin_arg� s
z+ProcessTestCase.test_check_output_stdin_argc C s& t jtjddgdd�}| jd|� d S )Nz-cz6import sys; sys.stdout.write(sys.stdin.read().upper())s pear)�inputs PEAR)r rB r' r( rC )r
rD r r r �test_check_output_input_arg� s
z+ProcessTestCase.test_check_output_input_argc C sP | j t��(}tjtjddgtjd�}| jd� W d Q R X | jd|j j
d � d S )Nz-czprint('will not be run'))r% z-Expected ValueError when stdout arg supplied.r% r ) r8 �
ValueErrorr rB r' r( r% �failrC r> r )r
r@ rD r r r � test_check_output_stdout_arg� s
z,ProcessTestCase.test_check_output_stdout_argc C s� t j� }| j|j� |jd� |jd� | jt��(}tj t
j ddg|dd�}| j d� W d Q R X | j
d|jjd � | j
d |jjd � d S )
Ns pearr z-czprint('will not be run')s hare)r$ rQ z7Expected ValueError when stdin and input args supplied.r$ rQ )rJ rK rL r- rM rN r8 rS r rB r' r( rT rC r> r )r
rO r@ rD r r r �&test_check_output_stdin_with_input_arg� s
z6ProcessTestCase.test_check_output_stdin_with_input_argc C sL | j tj��&}tjtjddgdd�}| jd� W d Q R X | j|jj d� d S )Nz-czMimport sys, time
sys.stdout.write('BDFL')
sys.stdout.flush()
time.sleep(3600)� )r7 zExpected TimeoutExpired.s BDFL)
r8 r r9 rB r' r( rT r r> rD )r
r@ rD r r r �test_check_output_timeout� s z)ProcessTestCase.test_check_output_timeoutc C s8 t jj� }d|d< tjtjddg|d�}| j|d� d S )N�banana�FRUITz-cz5import sys, os;sys.exit(os.getenv("FRUIT")=="banana"))�env� )�os�environ�copyr r4 r' r( r )r
�newenvr5 r r r �test_call_kwargs� s
z ProcessTestCase.test_call_kwargsc C sh t j� �F}| jttjdd� tjjjj}dg|d }| jttjf|�� W d Q R X | j |j
� d� d S )Nr\ )Zinvalid_arg_namer r ) r Zcaptured_stderrr8 � TypeErrorr r& �__init__�__code__� co_argcountr �getvalue)r
�sZargcountZ
too_many_argsr r r �test_invalid_args� s
z!ProcessTestCase.test_invalid_argsc C sR t jtjddgt jt jd�}| j|jj� | j|jj� |j � | j
|j d � d S )Nz-czprint("banana"))r% r ) r r& r' r( r) rL r% r- r r r r$ )r
r. r r r �test_stdin_none� s z ProcessTestCase.test_stdin_nonec C sl d}t jtjd|gt jt jd�}| j|jj� | j|jj� |j � \}}| j
|j d|� | j
|j � d� d S )Nz�import sys; from subprocess import Popen, PIPE;p = Popen([sys.executable, "-c", "print('test_stdout_none')"], stdin=PIPE, stderr=PIPE);p.wait(); assert p.stdout is None;z-c)r% r r s test_stdout_none)
r r& r' r( r) rL r% r- r � communicater r? �rstrip)r
�coder. �out�errr r r �test_stdout_none s z ProcessTestCase.test_stdout_nonec C sR t jtjddgt jt jd�}| j|jj� | j|jj� |j � | j
|j d � d S )Nz-czprint("banana"))r$ r% ) r r& r' r( r) rL r% r- r$ r r r )r
r. r r r �test_stderr_none s z ProcessTestCase.test_stderr_nonec K s2 |dg }t j|f|�}|j� | jd|j� d S )Nzimport sys; sys.exit(47)r3 )r r& r r r? )r
�pre_argsr! r r. r r r �_assert_python$ s
z ProcessTestCase._assert_pythonc C s0 t jjt jjtj�d�}| j|dgtjd� d S )N� doesnotexistz-c)r( )r] �path�join�dirnamer' r( rr )r
rs r r r �test_executable, s z ProcessTestCase.test_executablec C s0 t jdg}| j|� | jt| j|td d� d S )Nz-cr )r( )r' r( rr r8 �NONEXISTING_ERRORS�NONEXISTING_CMD)r
rq r r r � test_executable_takes_precedence7 s
z0ProcessTestCase.test_executable_takes_precedencez"executable argument replaces shellc C s | j g tjdd� d S )NT)r( �shell)rr r' r( )r
r r r � test_executable_replaces_shellA s z.ProcessTestCase.test_executable_replaces_shellc C s t j|��
tj� S Q R X d S )N)r Z
change_cwdr] �getcwd)r
�cwdr r r �_normalize_cwdH s z ProcessTestCase._normalize_cwdc C s t jjtj�}t jj|�S )N)r] rt �realpathr' r( �split)r
Z python_pathr r r �_split_python_pathP s z"ProcessTestCase._split_python_pathc K sn t j|ddgfdt ji|��}| j|jj� |j� | jd|j� t j
j }| j||�||jj � j
d��� d S )Nz-cz;import os, sys; sys.stdout.write(os.getcwd()); sys.exit(47)r% r3 zutf-8)r r& r) rL r% r- r r r? r] rt �normcase�read�decode)r
Z expected_cwdZ
python_argr! r. r� r r r � _assert_cwdV s
zProcessTestCase._assert_cwdc C s( t j� }| j|�}| j|tj|d� d S )N)r~ )rJ �
gettempdirr r� r' r( )r
�temp_dirr r r �test_cwdg s
zProcessTestCase.test_cwdc C s, t j� }| j|�}| j|tjt|�d� d S )N)r~ )rJ r� r r� r' r( r )r
r� r r r �test_cwd_with_pathlikem s
z&ProcessTestCase.test_cwd_with_pathlikez"pending resolution of issue #15533c
C sv | j � \}}tjjtj|�}tj� �H}| jtt j
|g� | jtt j
|g|d� | j |�}| j |||d� W d Q R X d S )N)r~ )
r� r] rt ru �curdirr �temp_cwdr8 �FileNotFoundErrorr r& r r� )r
�
python_dir� python_base�
rel_python� wrong_dirr r r �test_cwd_with_relative_argr s
z*ProcessTestCase.test_cwd_with_relative_argc C s� | j � \}}tjjtj|�}d}tj� �P}| jtt j
|g|d� | jtt j
|g||d� | j |�}| j ||||d� W d Q R X d S )N�somethingyoudonthave)r( )r( r~ )
r� r] rt ru r� r r� r8 r� r r& r r� )r
r� r� r� Z doesntexistr� r r r �!test_cwd_with_relative_executable� s
z1ProcessTestCase.test_cwd_with_relative_executablec
C sr | j � \}}tjj||�}tjjtj|�}tj� �6}| jtt j
|g|d� | j |�}| j |||d� W d Q R X d S )N)r~ )
r� r] rt ru r� r r� r8 r� r r& r r� )r
r� r� Z
abs_pythonr� r� r r r �test_cwd_with_absolute_arg� s
z*ProcessTestCase.test_cwd_with_absolute_argzTest is not venv-compatiblec C s. | j � \}}| j|�}| j|dtj|d� d S )Nr� )r( r~ )r� r r� r' r( )r
r� r� r r r �test_executable_with_cwd� s
z(ProcessTestCase.test_executable_with_cwdz#need an installed Python. See #7774c C s | j tj� dtjd� d S )Nr� )r( )r� r] r} r' r( )r
r r r �test_executable_without_cwd� s z+ProcessTestCase.test_executable_without_cwdc C sH t jtjddgt jd�}|jjd� |jj� |j� | j |j
d� d S )Nz-cz0import sys; sys.exit(sys.stdin.read() == "pear"))r$ s pearr\ ) r r& r' r( r) r$ rM r- r r r? )r
r. r r r �test_stdin_pipe� s
z ProcessTestCase.test_stdin_pipec C sf t j� }| j|j� |j� }tj|d� tj|dd� tj t
j ddg|d�}|j � | j
|jd� d S )Ns pearr z-cz0import sys; sys.exit(sys.stdin.read() == "pear"))r$ r\ )rJ rK rL r- �filenor] rM �lseekr r& r' r( r r r? )r
rO �dr. r r r �test_stdin_filedes� s
z"ProcessTestCase.test_stdin_filedesc C sX t j� }| j|j� |jd� |jd� tjtj ddg|d�}|j
� | j |j d� d S )Ns pearr z-cz0import sys; sys.exit(sys.stdin.read() == "pear"))r$ r\ )
rJ rK rL r- rM rN r r& r' r( r r r? )r
rO r. r r r �test_stdin_fileobj� s
z"ProcessTestCase.test_stdin_fileobjc
C s> t jtjddgt jd�}|� | j|jj� d� W d Q R X d S )Nz-cz&import sys; sys.stdout.write("orange"))r% s orange)r r& r' r( r) r r% r� )r
r. r r r �test_stdout_pipe� s
z ProcessTestCase.test_stdout_pipec C s` t j� }| j|j� |j� }tjtjddg|d�}|j � t
j |dd� | j t
j
|d�d� d S )Nz-cz&import sys; sys.stdout.write("orange"))r% r i s orange)rJ rK rL r- r� r r& r' r( r r] r� r r� )r
rO r� r. r r r �test_stdout_filedes� s
z#ProcessTestCase.test_stdout_filedesc C sP t j� }| j|j� tjtjddg|d�}|j� |j d� | j
|j � d� d S )Nz-cz&import sys; sys.stdout.write("orange"))r% r s orange) rJ rK rL r- r r& r' r( r rN r r� )r
rO r. r r r �test_stdout_fileobj� s
z#ProcessTestCase.test_stdout_fileobjc
C s> t jtjddgt jd�}|� | j|jj� d� W d Q R X d S )Nz-cz*import sys; sys.stderr.write("strawberry"))r s
strawberry)r r& r' r( r) r r r� )r
r. r r r �test_stderr_pipe� s
z ProcessTestCase.test_stderr_pipec C s` t j� }| j|j� |j� }tjtjddg|d�}|j � t
j |dd� | j t
j
|d�d� d S )Nz-cz*import sys; sys.stderr.write("strawberry"))r r i s
strawberry)rJ rK rL r- r� r r& r' r( r r] r� r r� )r
rO r� r. r r r �test_stderr_filedes� s
z#ProcessTestCase.test_stderr_filedesc C sP t j� }| j|j� tjtjddg|d�}|j� |j d� | j
|j � d� d S )Nz-cz*import sys; sys.stderr.write("strawberry"))r r s
strawberry) rJ rK rL r- r r& r' r( r rN r r� )r
rO r. r r r �test_stderr_fileobj s
z#ProcessTestCase.test_stderr_fileobjc C sR t jtjddgt jt jd�}|j� \}}| j|d� | j|d� | j|jd� d S )Nz-cz�import sys, subprocess;rc = subprocess.call([sys.executable, "-c", "import sys;" "sys.stderr.write('42')"], stderr=subprocess.STDOUT);sys.exit(rc))r% r s 42� r ) r r& r' r( r) rj r r r? )r
r. r% r r r r �,test_stderr_redirect_with_no_stdout_redirect s
z<ProcessTestCase.test_stderr_redirect_with_no_stdout_redirectc
C sB t jtjddgt jt jd�}|� | j|jj� d� W d Q R X d S )Nz-czRimport sys;sys.stdout.write("apple");sys.stdout.flush();sys.stderr.write("orange"))r% r s appleorange) r r& r' r( r) rH r r% r� )r
r. r r r �test_stdout_stderr_pipe+ s
z'ProcessTestCase.test_stdout_stderr_pipec C sR t j� }| j|j� tjtjddg||d�}|j� |j d� | j
|j � d� d S )Nz-czRimport sys;sys.stdout.write("apple");sys.stdout.flush();sys.stderr.write("orange"))r% r r s appleorange) rJ rK rL r- r r& r' r( r rN r r� )r
rO r. r r r �test_stdout_stderr_file7 s
z'ProcessTestCase.test_stdout_stderr_filec C sl d}t jtjd|gt jt jd�}| j|jj� | j|jj� |j � \}}| j
|j d|� | j
|j � d� d S )Nz�import sys, subprocess; rc = subprocess.call([sys.executable, "-c", "import os, sys; sys.exit(os.write(sys.stdout.fileno(), b'test with stdout=1'))"], stdout=1); assert rc == 18z-c)r% r r s test with stdout=1)
r r& r' r( r) rL r% r- r rj r r? rk )r
rl r. rm rn r r r � test_stdout_filedes_of_stdoutF s z-ProcessTestCase.test_stdout_filedes_of_stdoutc C s2 t jtjddgt jd�}|j� | j|jd � d S )Nz-cz'for i in range(10240):print("x" * 1024))r% )r r& r' r( �DEVNULLr r r% )r
r. r r r �test_stdout_devnullY s
z#ProcessTestCase.test_stdout_devnullc C s2 t jtjddgt jd�}|j� | j|jd � d S )Nz-cz=import sys
for i in range(10240):sys.stderr.write("x" * 1024))r )r r& r' r( r� r r r )r
r. r r r �test_stderr_devnulla s
z#ProcessTestCase.test_stderr_devnullc C s2 t jtjddgt jd�}|j� | j|jd � d S )Nz-cz import sys;sys.stdin.read(1))r$ )r r& r' r( r� r r r$ )r
r. r r r �test_stdin_devnullj s
z"ProcessTestCase.test_stdin_devnullc
C sT t jj� }d|d< tjtjddgtj|d�� }|j� \}}| j |d� W d Q R X d S )N�orangerZ z-cz2import sys,os;sys.stdout.write(os.getenv("FRUIT")))r% r[ s orange)
r] r^ r_ r r& r' r( r) rj r )r
r` r. r% r r r r �test_envr s
zProcessTestCase.test_envr z#cannot test an empty env on Windows�Py_ENABLE_SHAREDNz=the python library cannot be loaded with an empty environmentc C sF t jtjddgt ji d��"}|j� \}}| j|j� d� W d Q R X d S )Nz-cz)import os; print(list(os.environ.keys())))r% r[ � []� ['__CF_USER_TEXT_ENCODING'])r� r� )r r& r' r( r) rj rC r )r
r. r% r r r r �test_empty_env s
z ProcessTestCase.test_empty_envc C s\ t jd }| jt�� tj|ddg� W d Q R X | jt�� tjt jddg� W d Q R X d S )N� z-c�passzpass# )r' r( r8 rS r r& )r
�cmdr r r �test_invalid_cmd� s
z ProcessTestCase.test_invalid_cmdc C s t jj� }d|d< | jt�� tjtjddg|d� W d Q R X t jj� }d|d< | jt�� tjtjddg|d� W d Q R X t jj� }d|d < | jt�� tjtjddg|d� W d Q R X t jj� }d
|d< tjtjdd gtj |d �� }|j
� \}}| j |d
� W d Q R X d S )NZcabbagezFRUIT VEGETABLEz-cr� )r[ zorange VEGETABLE=cabbagerZ Zlemonz FRUIT=ORANGEz orange=lemonz3import sys, os;sys.stdout.write(os.getenv("FRUIT")))r% r[ s orange=lemon) r] r^ r_ r8 rS r r& r' r( r) rj r )r
r` r. r% r r r r �test_invalid_env� s(
z ProcessTestCase.test_invalid_envc C s4 t jtjddgt jd�}|jd� | j|jd� d S )Nz-cz/import sys;sys.exit(sys.stdin.read() == "pear"))r$ s pearr\ )r r& r' r( r) rj r r? )r
r. r r r �test_communicate_stdin� s
z&ProcessTestCase.test_communicate_stdinc C s@ t jtjddgt jd�}|j� \}}| j|d� | j|d � d S )Nz-cz)import sys; sys.stdout.write("pineapple"))r% s pineapple)r r& r' r( r) rj r )r
r. r% r r r r �test_communicate_stdout� s
z'ProcessTestCase.test_communicate_stdoutc C s@ t jtjddgt jd�}|j� \}}| j|d � | j|d� d S )Nz-cz)import sys; sys.stderr.write("pineapple"))r s pineapple)r r& r' r( r) rj r r )r
r. r% r r r r �test_communicate_stderr� s
z'ProcessTestCase.test_communicate_stderrc C st t jtjddgt jt jt jd�}| j|jj� | j|jj� | j|j j� |j
d�\}}| j |d� | j |d� d S )Nz-czNimport sys,os;sys.stderr.write("pineapple");sys.stdout.write(sys.stdin.read()))r$ r% r s bananas pineapple)
r r& r' r( r) rL r% r- r r$ rj r r )r
r. r% r r r r �test_communicate� s
z ProcessTestCase.test_communicatec C sd t jtjddgdt jt jt jd�}| jt j|jddd� |j� \}}| j|d� | j |j
� d� d S ) Nz-cz~import sys,os,time;sys.stderr.write("pineapple\n");time.sleep(1);sys.stderr.write("pear\n");sys.stdout.write(sys.stdin.read())T)�universal_newlinesr$ r% r rY g333333�?)r7 s pineapple
pear
) r r& r' r( r) r8 r9 rj r r �encode)r
r. r% r r r r �test_communicate_timeout� s
z(ProcessTestCase.test_communicate_timeoutc C sL t jtjddgt jd�}| jt j|jdd� |j� \}}| jt |�d
� d S ) Nz-cz�import sys,os,time;sys.stdout.write("a" * (64 * 1024));time.sleep(0.2);sys.stdout.write("a" * (64 * 1024));time.sleep(0.2);sys.stdout.write("a" * (64 * 1024));time.sleep(0.2);sys.stdout.write("a" * (64 * 1024));)r% g�������?)r7 � �@ i � i )
r r& r' r( r) r8 r9 rj r �len)r
r. r% �_r r r �%test_communicate_timeout_large_output� s
z5ProcessTestCase.test_communicate_timeout_large_outputc C s� x�dD ]�}x�d D ]�}x�d
D ]�}i }|r0t j|d< |r>t j|d< |rLt j|d< |sRqt jtjddff|�}|j� |jd k r�| j|jj� |j d k r�| j|j j� |j
d k r| j|j
j� qW qW qW d S ) NFTr$ r% r z-cr� )FT)FT)FT) r r) r& r' r( rj r$ �
assertTrue�closedr% r )r
Z
stdin_pipeZ stdout_pipeZ stderr_pipe�optionsr. r r r � test_communicate_pipe_fd_leak s(
z-ProcessTestCase.test_communicate_pipe_fd_leakc C s: t jtjddg�}|j� \}}| j|d � | j|d � d S )Nz-czimport sys; sys.exit(47))r r& r' r( rj r )r
r. r% r r r r �test_communicate_returns s
z(ProcessTestCase.test_communicate_returnsc C s� t j� \}}t j|� t j|� tjtjddtj gtj tj tj d�}| j
|j j� | j
|j j� | j
|j
j� dtj }|j|�\}}| j||� d S )Nz-czpimport sys,os;sys.stdout.write(sys.stdin.read(47));sys.stderr.write("x" * %d);sys.stdout.write(sys.stdin.read()))r$ r% r � a)r] �piper- r r& r' r( r �
PIPE_MAX_SIZEr) rL r% r r$ rj r )r
�x�yr. Zstring_to_writer% r r r r �test_communicate_pipe_buf" s
z)ProcessTestCase.test_communicate_pipe_bufc C s� t jtjddgt jt jt jd�}| j|jj� | j|jj� | j|j j� |j j
d� |j d�\}}| j |d� | j
|d� d S )Nz-cz0import sys,os;sys.stdout.write(sys.stdin.read()))r$ r% r s bananas splits bananasplitr� )r r& r' r( r) rL r% r- r r$ rM rj r r )r
r. r% r r r r � test_writes_before_communicate9 s
z.ProcessTestCase.test_writes_before_communicatec
C s� t jtjddt d gt jt jdd�}|�� |jjd� |jj� | j |j
j � d� |jjd� |jj � | j
|j
j � | j |j
j � d� | j |j
jd �d� | j |j
j� d
� W d Q R X d S ) Nz-czimport sys,os;a@ buf = sys.stdout.buffer;buf.write(sys.stdin.readline().encode());buf.flush();buf.write(b"line2\n");buf.flush();buf.write(sys.stdin.read().encode());buf.flush();buf.write(b"line4\n");buf.flush();buf.write(b"line5\r\n");buf.flush();buf.write(b"line6\r");buf.flush();buf.write(b"\nline7");buf.flush();buf.write(b"\nline8");r\ )r$ r% r� zline1
zline3
zline2
� z line4
line5
line6
line7
line8)r r& r' r( � SETBINARYr) r$ rM �flushr r% �readliner- rL r� )r
r. r r r �test_universal_newlinesI s&
z'ProcessTestCase.test_universal_newlinesc C s^ t jtjddt d gt jt jdd�}| j|jj� | j|j j� |j
� \}}| j |d� d S )Nz-czimport sys,os;z�buf = sys.stdout.buffer;buf.write(b"line2\n");buf.flush();buf.write(b"line4\n");buf.flush();buf.write(b"line5\r\n");buf.flush();buf.write(b"line6\r");buf.flush();buf.write(b"\nline7");buf.flush();buf.write(b"\nline8");r\ )r r% r� z#line2
line4
line5
line6
line7
line8) r r& r' r( r� r) rL r% r- r rj r )r
r. r% r r r r �#test_universal_newlines_communicatem s
z3ProcessTestCase.test_universal_newlines_communicatec C sH t jtjddt tjd� gt jdd�}|jd�\}}| j |j
d� d S )Nz-czimport sys,os;a
s = sys.stdin.readline()
assert s == "line1\n", repr(s)
s = sys.stdin.read()
assert s == "line3\n", repr(s)
r\ )r$ r� z line1
line3
r ) r r& r' r( r� �textwrap�dedentr) rj r r? )r
r. r% r r r r �)test_universal_newlines_communicate_stdin� s
z9ProcessTestCase.test_universal_newlines_communicate_stdinc C s8 t jtjddgt jt jdd�}|j� | j|jd� d S )Nz-cr� T)r$ r% r� r )r r& r' r( r) rj r r? )r
r. r r r �.test_universal_newlines_communicate_input_none� s z>ProcessTestCase.test_universal_newlines_communicate_input_nonec C s� t jtjddt tjd� gt jt jt jdd�}| j|j j
� | j|j j
� |j d�\}}| j
|jd� | j
d|� | j|jd �� d S )
Nz-czimport sys,os;a�
s = sys.stdin.buffer.readline()
sys.stdout.buffer.write(s)
sys.stdout.buffer.write(b"line2\r")
sys.stderr.buffer.write(b"eline2\n")
s = sys.stdin.buffer.read()
sys.stdout.buffer.write(s)
sys.stdout.buffer.write(b"line4\n")
sys.stdout.buffer.write(b"line5\r\n")
sys.stderr.buffer.write(b"eline6\r")
sys.stderr.buffer.write(b"eline7\r\nz")
T)r$ r r% r� z line1
line3
r z line1
line2
line3
line4
line5
zeline2
eline6
eline7
)r r& r' r( r� r� r� r) rL r% r- r rj r r? r� �
startswith)r
r. r% r r r r �7test_universal_newlines_communicate_stdin_stdout_stderr� s
zGProcessTestCase.test_universal_newlines_communicate_stdin_stdout_stderrc C sX xRd D ]J}d| }t jd|g}tj|tjtj|d�}|jdd�\}}| j|d� qW d S )
N�utf-16� utf-32-bez@import sys; sys.stdout.buffer.write('1\r\n2\r3\n4'.encode('%s'))z-c)r$ r% �encodingr )rQ z1
2
3
4)r� r� )r' r( r r& r) rj r )r
r� rl r �popenr% r r r r �-test_universal_newlines_communicate_encodings� s
z=ProcessTestCase.test_universal_newlines_communicate_encodingsc C s` xZdD ]R\}}d }t jd
|g}tj|tjtjd |d �}|jdd
�\}}| j|dj|�� qW d S )N�ignorer �replace� ���surrogateescape� ���backslashreplace�\x80\x80z2import sys; sys.stdout.buffer.write(b'[\x80\x80]')z-czutf-8)r$ r% r� �errors)rQ z[{}]�r� r �r� r� �r� r� �r� r� )r� r� r� r� )r' r( r r& r) rj r �format)r
r� r rl r r� r% r r r r �test_communicate_errors� s z'ProcessTestCase.test_communicate_errorsc
C s8 t s
d}nd}g }tj� }z�x�t|�D ]f}y.tjj|tj�}|j tj
|tj tj B �� W q& t
k
r� } z|jtjkrx� P W Y d d }~X q&X q&W | jd| � x td�D ]}tj|j� � q�W xJtd�D ]>}tjtjddgtjtjtjd�}|jd �d
}| j|d � q�W W d x|D ]} tj| � �qW tj|� X d S ) Ni i z4failed to reach the file descriptor limit (tried %d)�
� z-cz-import sys;sys.stdout.write(sys.stdin.read()))r$ r% r s limer ) � mswindowsrJ �mkdtemp�ranger] rt ru r ZTESTFN�append�open�O_WRONLY�O_CREAT�OSError�errnoZEMFILE�skipTestr- �popr r& r' r( r) rj r �shutilZrmtree)
r
Z max_handlesZhandlesZtmpdir�iZtmpfile�er. �data�hr r r �test_no_leaking� s:
z ProcessTestCase.test_no_leakingc C s� | j tjdddg�d� | j tjdddg�d� | j tjdddg�d � | j tjd
d d g�d
� | j tjdddg�d� | j tjdddg�d� | j tjdddg�d� | j tjddg�d� d S )Nza b cr� r� z "a b c" d ezab"c�\z ab\"c \ dz \z
ab\"c " \\" dza\\\bzde fgr za\\\b "de fg" hza\"br@ z
a\\\"b c dza\\b cz "a\\b c" d eza\\b\ cz
"a\\b\ c" d eZabr zab "")r r � list2cmdline)r
r r r �test_list2cmdline s z!ProcessTestCase.test_list2cmdlinec C sb t jtjddgt jd�}| j|jj� | j|j � � t
j |jj � d� |j
� | j|j � d� d S )Nz-czimport os; os.read(0, 1))r$ � Ar )r r& r' r( r) rL r$ r- � assertIsNone�pollr] rM r� r r )r
r. r r r � test_poll! s
zProcessTestCase.test_pollc C s6 t jtjddg�}| j|j� d� | j|j� d� d S )Nz-cr� r )r r& r' r( r r )r
r. r r r � test_wait, s zProcessTestCase.test_waitc C s` t jtjddg�}| jt j��}|jdd� W d Q R X | jdt|j �� | j
|jdd�d� d S )Nz-cz import time; time.sleep(0.3)g-C ��6?)r7 z0.0001rW r ) r r& r' r( r8 r9 r rC �strr> r )r
r. r@ r r r �test_wait_timeout2 s
z!ProcessTestCase.test_wait_timeoutc C s� t jtjddg�}zJy.| jt��}|jtj� d d� W d Q R X W n t jk
rX Y nX W d |j � X | j
d|j � | j
dt |j
�� d S )Nz-cr� g{�G�z�?)�endtimeztest_subprocess.pyr
)r r& r' r( � assertWarns�DeprecationWarningr �timer9 �killrC �filenamer �warning)r
r. Zwarn_cmr r r �test_wait_endtime< s "
z!ProcessTestCase.test_wait_endtimec C s. | j t�� tjtjddgd� W d Q R X d S )Nz-cr� r� )r8 rb r r& r' r( )r
r r r �test_invalid_bufsizeI s z$ProcessTestCase.test_invalid_bufsizec C sN t jtjddgd �}| j|j� d� t jtjddgd d�}| j|j� d� d S )Nz-cr� r )r0 )r r& r' r( r r )r
r. r r r �test_bufsize_is_noneO s z$ProcessTestCase.test_bufsize_is_nonec C s� t jtjddgt jt jt jd|d��j}|jj|� tj |jj
� � |j j � }t
j� �* y|jj � W n tk
rx Y nX W d Q R X d |_W d Q R X | j|jd� | j||� d S )Nz-czDimport sys;sys.stdout.write(sys.stdin.readline());sys.stdout.flush()r\ )r$ r% r r0 r� r )r r& r' r( r) r� r$ rM r] r- r� r% r� r �SuppressCrashReportr� r r? )r
�liner r� r. Z read_liner r r �_test_bufsize_equal_oneW s"
z'ProcessTestCase._test_bufsize_equal_onec C s d}| j ||dd� d S )Nzline
T)r� )r )r
r r r r � test_bufsize_equal_one_text_modem s z0ProcessTestCase.test_bufsize_equal_one_text_modec C s" dt jj� }| j|ddd� d S )Ns liner� F)r� )r] �linesepr� r )r
r r r r �"test_bufsize_equal_one_binary_modes s z2ProcessTestCase.test_bufsize_equal_one_binary_modec
C s@ x:t d�D ].}| jt�� tjttjtjd� W d Q R X q
W d S )Ni )r% r )r� r8 rx r r& ry r) )r
r� r r r �test_leaking_fds_on_errory s
z)ProcessTestCase.test_leaking_fds_on_errorc C s� ydd l }|j W n" ttfk
r4 | jd� Y nX tjdt�d��}tj d|g}t
j |t
j dd�}|� |j
� d }W d Q R X | j|d � | j|jd� d S )
Nr z need msvcrt.CrtSetReportModezL
import msvcrt
import subprocess
cmd = a
for report_type in [msvcrt.CRT_WARN,
msvcrt.CRT_ERROR,
msvcrt.CRT_ASSERT]:
msvcrt.CrtSetReportMode(report_type, msvcrt.CRTDBG_MODE_FILE)
msvcrt.CrtSetReportFile(report_type, msvcrt.CRTDBG_FILE_STDERR)
try:
subprocess.Popen(cmd,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
except OSError:
pass
z-cT)r r� r\ r )�msvcrtZCrtSetReportMode�AttributeError� ImportErrorr� r� r� ry r' r( r r& r) rj r r? )r
r rl r� �procr r r r �test_nonexisting_with_pipes� s
z+ProcessTestCase.test_nonexisting_with_pipeszthreading requiredc s� g � � fdd�}t j|d�}|j� z2| jt�� tjttjtjtjd� W d Q R X W d |j � d }xB� D ]:}yt
j |� W qj t k
r� } z
|}W Y d d }~X qjX qjW |d k r�|�X d S )Nc s. x(t d�D ] } � jtj� � tjd� q
W d S )N� g����MbP?)r� �extendr] r� r �sleep)r� )�fdsr r �open_fds� s z<ProcessTestCase.test_double_close_on_error.<locals>.open_fds)�target)r$ r% r )
� threadingZThread�startr8 �EnvironmentErrorr r& ry r) ru r] r- r� )r
r' �t�exc�fdr� r )r& r �test_double_close_on_error� s&
z*ProcessTestCase.test_double_close_on_errorc s� t jtjddg�� | j� jd � g �� �fdd�}tjd|�}|j� t rPd}nd}� j
dd � | j� j|d
d � � j
� | j� j|d d � |j � | jdd|fd|fg�� d S )Nz-czimport time; time.sleep(12)c sN �j d� j� f� � j� � j� �j d� jf� � j� �j d� jf� d S )Nzthread-start-poll-resultzthread-after-kill-and-waitzthread-after-second-wait)r� r r r r? r )r! �resultsr r �kill_proc_timer_thread� s zDProcessTestCase.test_threadsafe_wait.<locals>.kill_proc_timer_threadg�������?r\ � r# )r7 z*unexpected result in wait from main thread)r z&unexpected result in second main wait.�thread-start-poll-resultzthread-after-kill-and-waitzthread-after-second-waiti����)r3 N) r r& r' r( r r? r) ZTimerr* r� r ru )r
r1 r, Zexpected_errorcoder )r! r0 r �test_threadsafe_wait� s*
z$ProcessTestCase.test_threadsafe_waitc C s6 dj d�}tjtjd|g�}| j|jd�t|�� d S )N�;�import subprocess, sys�Jretcode = subprocess.call([sys.executable, '-c', 'print("Hello World!")'])�assert retcode == 0z-cs Hello World!)r6 r7 r8 )ru r rB r' r( r� r� �ascii)r
rl rD r r r �test_issue8780 s z ProcessTestCase.test_issue8780c C s� t j� \}}t j� \}}t j� \}}ytjdg|||d� W nP tk
r� tj|� tj|� tj|� tj|� tj|� tj|� Y nX | jtj j
|�� | jtj j
|�� | jtj j
|�� d S )N�*)r$ r% r ) rJ �mkstempr r& r� r] r- �remover rt �exists)r
ZifhandleZifnameZofhandleZofnameZefhandleZefnamer r r � test_handles_closed_on_exception s
z0ProcessTestCase.test_handles_closed_on_exceptionc C s\ t jtjddgt jt jt jd�}| j|jj� | j|jj� | j|j j� |j
dd � d S )Nz-cr� )r$ r% r � x� r# i ) r r& r' r( r) rL r% r- r r$ rj )r
r. r r r �test_communicate_epipe s
z&ProcessTestCase.test_communicate_epipec C s@ t jtjddgt jd�}| j|jj� |j� |j dd � d S )Nz-cr� )r$ r@ rA r# i )
r r& r' r( r) rL r$ r- r rj )r
r. r r r �!test_communicate_epipe_only_stdin+ s
z1ProcessTestCase.test_communicate_epipe_only_stdin�SIGUSR1zRequires signal.SIGUSR1r zRequires os.kill�getppidzRequires os.getppidc
C sr dd� }t j t j|�}| jt j t j|� tjddg}x8dD ]0}|tji}tj|f|��}|j� W d Q R X q:W d S )Nc S s d S )Nr )Zsignum�framer r r �handler; s z7ProcessTestCase.test_communicate_eintr.<locals>.handlerz-cz7import os, signal;os.kill(os.getppid(), signal.SIGUSR1)r% r )r% r ) �signalrD rL r' r( r r) r& rj )r
rG � old_handlerr �stream�kwZprocessr r r �test_communicate_eintr3 s
z&ProcessTestCase.test_communicate_eintrz /proc/%d/fdzLinux specificc
C sd dt j� }t j|�}| jt��$ ttjddgtj tj tj d� W d Q R X t j|�}| j
||� d S )Nz /proc/%d/fdz-cr� )r$ r% r ) r] �getpid�listdirr8 r r r' r( r r) r )r
Z fd_directoryZfds_before_popenZfds_after_exceptionr r r �!test_failed_child_execute_fd_leakL s
z1ProcessTestCase.test_failed_child_execute_fd_leakz+behavior currently not supported on Windowsc C s< | j t��}tjddddg� W d Q R X | j|jjd� d S )Nz/opt/nonexistent_binary�withZsomer )r8 r� r r4 r r> r )r
r@ r r r �%test_file_not_found_includes_filename^ s z5ProcessTestCase.test_file_not_found_includes_filenamec C s< | j t��}tjddgdd� W d Q R X | j|jjd� d S )N�exit�0z/some/nonexistent/directory)r~ )r8 r� r r& r r> r )r
r@ r r r � test_file_not_found_with_bad_cwdd s z0ProcessTestCase.test_file_not_found_with_bad_cwd)rr r r r/ r2 r6 r: r< rA rE rG rI rP rR rU rV rX ra rh ri ro rp rr rw rz �unittest�skipIfr� r| r r� r� r� r� r� r� r� r' � base_prefix�prefixr� � sysconfig�is_python_buildr� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �platform�get_config_varr� 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 r r r r r" r) r/ r4 r: r? rB rC �
skipUnless�hasattrrH r] rL rt �isdirrM rO rQ rT r r r r r# _ s�
$ '
) 5 r# c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�RunFuncTestCasec K s t jd|g}tj|f|�S )Nz-c)r' r( r Zrun)r
rl r! �argvr r r �
run_pythonl s zRunFuncTestCase.run_pythonc C s< | j d�}| j|jd� | jtj�� |j� W d Q R X d S )Nzimport sys; sys.exit(47)r3 )rb r r? r8 r r= Zcheck_returncode)r
�cpr r r �test_returncodeq s
z RunFuncTestCase.test_returncodec C s: | j tj��}| jddd� W d Q R X | j|jjd� d S )Nzimport sys; sys.exit(47)T)�checkr3 )r8 r r= rb r r> r? )r
r@ r r r �
test_checkx s zRunFuncTestCase.test_checkc C s | j ddd�}| j|jd� d S )Nzimport sys; sys.exit(0)T)re r )rb r r? )r
rc r r r �test_check_zero} s z RunFuncTestCase.test_check_zeroc C s* | j tj�� | jddd� W d Q R X d S )Nzwhile True: passg-C ��6?)r7 )r8 r r9 rb )r
r r r � test_timeout� s z RunFuncTestCase.test_timeoutc C s" | j dtjd�}| jd|j� d S )Nz
print('BDFL'))r% s BDFL)rb r r) rC r% )r
rc r r r �test_capture_stdout� s z#RunFuncTestCase.test_capture_stdoutc C s" | j dtjd�}| jd|j� d S )Nz$import sys; sys.stderr.write('BDFL'))r s BDFL)rb r r) rC r )r
rc r r r �test_capture_stderr� s
z#RunFuncTestCase.test_capture_stderrc C sL t j� }| j|j� |jd� |jd� | jd|tjd�}| j d|j
� d S )Ns pearr z6import sys; sys.stdout.write(sys.stdin.read().upper()))r$ r% s PEAR) rJ rK rL r- rM rN rb r r) rC r% )r
rO rc r r r rP � s
z+RunFuncTestCase.test_check_output_stdin_argc C s$ | j ddtjd�}| jd|j� d S )Nz6import sys; sys.stdout.write(sys.stdin.read().upper())s pear)rQ r% s PEAR)rb r r) rC r% )r
rc r r r rR � s z+RunFuncTestCase.test_check_output_input_argc C s~ t j� }| j|j� |jd� |jd� | jtdd��}| jd|dd�}W d Q R X | j d|j
j d � | j d |j
j d � d S )
Ns pearr z7Expected ValueError when stdin and input args supplied.)r zprint('will not be run')s hare)r$ rQ r$ rQ ) rJ rK rL r- rM rN r8 rS rb rC r> r )r
rO r@ rD r r r rV � s
z6RunFuncTestCase.test_check_output_stdin_with_input_argc C sN | j tj��}| jddtjd�}W d Q R X | j|jjd� | j|jjd� d S )NzMimport sys, time
sys.stdout.write('BDFL')
sys.stdout.flush()
time.sleep(3600)rW )r7 r% s BDFL) r8 r r9 rb r) r r> rD r% )r
r@ rc r r r rX � s z)RunFuncTestCase.test_check_output_timeoutc C s2 t jj� }d|d< | jd|d�}| j|jd� d S )NrY rZ zCimport sys, os;sys.exit(33 if os.getenv("FRUIT")=="banana" else 31))r[ �! )r] r^ r_ rb r r? )r
r` rc r r r �test_run_kwargs� s
z RunFuncTestCase.test_run_kwargsN)r r r rb rd rf rg rh ri rj rP rR rV rX rl r r r r r` k s
r` zPOSIX specific testsc s2 e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� ZG d d � d ej �Z
e j d
�dd� �Z
e j d
�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� ZG d"d#� d#ej �Zejejj d$� d%�d&d'� �Z d(d)� Z eje j d*kd+�d,d-� �Z!d.d/� Z"d0d1� Z#d2d3� Z$d4d5� Z%d6d7� Z&d8d9� Z'd:d;� Z(eje j j)d��d>�d?d@� �Z*dAdB� Z+dCdD� Z,dEdF� Z-dGdH� Z.dIdJ� Z/dKdL� Z0dMdN� Z1dOdP� Z2dQdR� Z3dSdT� Z4dUdV� Z5dWdX� Z6dYdZ� Z7d[d\� Z8d]d^� Z9d_d`� Z:dadb� Z;dcdd� Z<dedf� Z=dgdh� Z>didj� Z?dkdl� Z@dmdn� ZAdodp� ZBdqdr� ZCdsdt� ZDdudv� ZEdwdx� ZFeje j j)dy��oXejGdz�jHejGd{�jHkd|�d}d~� �ZIeJjKdd��d�d�� �ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVeJjWd�d�� �ZXeJjWd�d�� �ZYd�d�� ZZd�d�� Z[d�d�� Z\d�d�� Z]ej^e_d�k �o e`e_d��d��d�d�� �Za� ZbS )��POSIXProcessTestCasec s t � j� d| _d S )Nz/_this/pa.th/does/not/exist)�superr
�_nonexistent_dir)r
)� __class__r r r
� s
zPOSIXProcessTestCase.setUpc C sd yt j| j� W n> tk
rN } z"|}| jdt| j� 7 _W Y d d }~X nX | jd| j � |S )Nz: z,chdir to nonexistent directory %s succeeded.)r] �chdirro r� �strerror�reprrT )r
r� �desired_exceptionr r r �_get_chdir_exception� s *
z)POSIXProcessTestCase._get_chdir_exceptionc C sz | j � }y tjtjddg| jd�}W nB tk
rf } z&| j|j|j� | j|j |j � W Y d d }~X nX | j
d| � d S )Nz-cr )r~ zExpected OSError: %s) ru r r& r' r( ro r� r r� rr rT )r
rt r. r� r r r �test_exception_cwd� s "z'POSIXProcessTestCase.test_exception_cwdc C sz | j � }y tjtjddg| jd�}W nB tk
rf } z&| j|j|j� | j|j |j � W Y d d }~X nX | j
d| � d S )Nz-cr )r( zExpected OSError: %s) ru r r& r' r( ro r� r r� rr rT )r
rt r. r� r r r � test_exception_bad_executable� s "z2POSIXProcessTestCase.test_exception_bad_executablec C st | j � }ytj| jddg�}W nB tk
r` } z&| j|j|j� | j|j|j� W Y d d }~X nX | jd| � d S )Nz-cr zExpected OSError: %s) ru r r& ro r� r r� rr rT )r
rt r. r� r r r �test_exception_bad_args_0� s "z.POSIXProcessTestCase.test_exception_bad_args_0c @ s e Zd Zdd� ZdS )z&POSIXProcessTestCase.PopenNoDestructorc C s d S )Nr )r
r r r �__del__ s z.POSIXProcessTestCase.PopenNoDestructor.__del__N)r r r ry r r r r �PopenNoDestructor s rz z%subprocess._posixsubprocess.fork_execc C sN dd� }||_ tjdtd��( | jt�� | jdg� W d Q R X W d Q R X d S )Nc W s0 | d }dj tj�j� }tj|d| d � dS )N�
z{:x}s OSError:� :r )r� r� ZEISDIRr� r] rM )r �
errpipe_writeZerr_coder r r � proper_error s zHPOSIXProcessTestCase.test_exception_errpipe_normal.<locals>.proper_errorzsubprocess.os.waitpid)� side_effect�non_existent_command)r r �patch�ChildProcessErrorr8 �IsADirectoryErrorrz )r
� fork_execr~ r r r � test_exception_errpipe_normal s
z2POSIXProcessTestCase.test_exception_errpipe_normalc sn d� � fdd�}||_ tjdtd��* | jtj��}| jdg� W d Q R X W d Q R X | jt � �t
|j �� d S )Ns � ޭc s | d }t j|� � dS )Nr{ r )r] rM )r r} )�
error_datar r � bad_error) s zGPOSIXProcessTestCase.test_exception_errpipe_bad_data.<locals>.bad_errorzsubprocess.os.waitpid)r r� ) r r r� r� r8 r �SubprocessErrorrz rC rs r r> )r
r� r� r� r )r� r � test_exception_errpipe_bad_data$ s
z4POSIXProcessTestCase.test_exception_errpipe_bad_datac C s0 t jtjddgdd� t jtjddgdd� d S )Nz-cr T)Zrestore_signalsF)r r4 r' r( )r
r r r �test_restore_signals; s z)POSIXProcessTestCase.test_restore_signalsc C sr yt jtjddgdd�}W n0 tk
rJ } z|jtjkr:� W Y d d }~X n$X tjtj � �}t
|�}| j ||� d S )Nz-cz)import os; print(os.getpgid(os.getpid()))T)Zstart_new_session) r rB r' r( r� r� ZEPERMr] �getpgidrM �int�assertNotEqual)r
rD r� Z parent_pgidZ
child_pgidr r r �test_start_new_sessionB s z+POSIXProcessTestCase.test_start_new_sessionc C sD t j� � tjtjddg�}|j� W d Q R X | j|j t j
� d S )Nz-czimport os; os.abort()) r r r r& r' r( r r r? rH �SIGABRT)r
r. r r r �test_run_abortS s
z#POSIXProcessTestCase.test_run_abortc C sN t jttj� d�}t|�}| jd|j� � | jd|� | jttj�|� d S )Nzfake cmdrH ZSIG)r r= r� rH r� r rC �lower)r
rn � error_stringr r r �"test_CalledProcessError_str_signal[ s
z7POSIXProcessTestCase.test_CalledProcessError_str_signalc C s$ t jdd�}t|�}| jd|� d S )Ni?�� zfake cmdzunknown signal 9876543.i�Ki�)r r= r rC )r
rn r� r r r �*test_CalledProcessError_str_unknown_signalf s z?POSIXProcessTestCase.test_CalledProcessError_str_unknown_signalc C s$ t jdd�}t|�}| jd|� d S )NrA zfake cmdznon-zero exit status 2.)r r= r rC )r
rn r� r r r �$test_CalledProcessError_str_non_zerok s z9POSIXProcessTestCase.test_CalledProcessError_str_non_zeroc
C sD t jtjddgt jdd� d�}|� | j|jj� d� W d Q R X d S )Nz-cz2import sys,os;sys.stdout.write(os.getenv("FRUIT"))c S s t jdd�S )NrZ �apple)r] �putenvr r r r �<lambda>w s z3POSIXProcessTestCase.test_preexec.<locals>.<lambda>)r% �
preexec_fns apple)r r& r' r( r) r r% r� )r
r. r r r � test_preexecp s
z!POSIXProcessTestCase.test_preexecc C s� dd� }yt jtjddg|d�}W nd t jk
rT } z| jt jd� W Y d d }~X n> tk
r� } z| jd|j d � W Y d d }~X n X | j
d � d S )
Nc S s t d��d S )Nz'What if two swallows carried a coconut?)rS r r r r �raise_it| s z=POSIXProcessTestCase.test_preexec_exception.<locals>.raise_itz-cr )r� z)Expected a ValueError from the preexec_fnZcoconutr zEException raised by preexec_fn did not make it to the parent process.) r r& r' r( r� r� �_posixsubprocessrS rC r rT )r
r� r. r� r r r �test_preexec_exception{ s $z+POSIXProcessTestCase.test_preexec_exceptionc @ s e Zd Zdd� Zdd� ZdS )z+POSIXProcessTestCase._TestExecuteChildPopenc O s || _ tjj| f|�|� d S )N)� _testcaser r& rc )r
Ztestcaser r! r r r rc � s z4POSIXProcessTestCase._TestExecuteChildPopen.__init__c O s� zt jj| f|�|� W d dd� td�D �}z<x6|D ].}| jj|| jj� | jj� | j j� fdd� q4W W d x|D ]}t
j |� qpW X X d S )Nc S s g | ]}t jd t j��qS )z /dev/zero)r] r� �O_RDONLY)�.0r� r r r �
<listcomp>� s zNPOSIXProcessTestCase._TestExecuteChildPopen._execute_child.<locals>.<listcomp>� z!At least one fd was closed early.)r ) r r& r" r� r� � assertNotInr$ r� r% r r] r- )r
r r! Z devzero_fdsr. r r r r" � s
z:POSIXProcessTestCase._TestExecuteChildPopen._execute_childN)r r r rc r" r r r r �_TestExecuteChildPopen� s r� z /dev/zeroz/dev/zero required.c C sH dd� }| j tj��* | j| tjddgtjtjtj|d� W d Q R X d S )Nc S s t jd��d S )Nz-force the _execute_child() errpipe_data path.)r r� r r r r r� � s zWPOSIXProcessTestCase.test_preexec_errpipe_does_not_double_close_pipes.<locals>.raise_itz-cr� )r$ r% r r� )r8 r r� r� r' r( r) )r
r� r r r �0test_preexec_errpipe_does_not_double_close_pipes� s zEPOSIXProcessTestCase.test_preexec_errpipe_does_not_double_close_pipesc C s dd� }t j� }t j}t j}z�t j� | jt j� � tjtjddgdd� d� | jt j� d� t j� | j t j� � tjtjddgd d� d� | j t j� d
� |t _| j
t tj tjddgd d� d� t `| j
t
tj tjddgd d� d� W d |t _|t _|�st j� X d S )
Nc S s t d��d S )Nzthis shouldn't escape)� RuntimeErrorr r r r �raise_runtime_error� s zPPOSIXProcessTestCase.test_preexec_gc_module_failure.<locals>.raise_runtime_errorz-cr c S s d S )Nr r r r r r� � s zEPOSIXProcessTestCase.test_preexec_gc_module_failure.<locals>.<lambda>)r� z#Popen enabled gc when it shouldn't.c S s d S )Nr r r r r r� � s zPopen left gc disabled.c S s d S )Nr r r r r r� � s c S s d S )Nr r r r r r� � s )�gc� isenabled�disabler r r4 r' r( �enabler� r8 r� r& r )r
r� ZenabledZorig_gc_disableZorig_gc_isenabledr r r � test_preexec_gc_module_failure� s8
z3POSIXProcessTestCase.test_preexec_gc_module_failure�darwinz!setrlimit() seems to fail on OS Xc C s� yddl m}m}m} W n, tk
rD } z| j|� W Y d d }~X nX ||�}|\}}||d|f� | j|||� y tjt j
ddgdd� d� W n t k
r� Y n X | jd� d S ) Nr )� getrlimit� setrlimit� RLIMIT_NPROCz-cr c S s d S )Nr r r r r r� � s z@POSIXProcessTestCase.test_preexec_fork_failure.<locals>.<lambda>)r� z.RLIMIT_NPROC had no effect; probably superuser) Zresourcer� r� r� r r� rL r r4 r' r( �BlockingIOError)r
r� r� r� rn Zlimitsr� Zhardr r r �test_preexec_fork_failure� s z.POSIXProcessTestCase.test_preexec_fork_failurec
C s� t j� \}}t|ddd��&}|jdtj � |jdtj � W d Q R X tj |d� t
j |�}|j � tj
|� | j|jd� d S )N�wr� )r� z#!%s
z(exec '%s' -c 'import sys; sys.exit(47)'
i� r3 )rJ r< r� rM r �
unix_shellr' r( r] �chmodr r& r r= r r? )r
r. �fname�fobjr. r r r �test_args_string� s
z%POSIXProcessTestCase.test_args_stringc C s< | j ttjtjddgdd� | j ttjtjddgdd� d S )Nz-czimport sys; sys.exit(47)r3 )� startupinfo)�
creationflags)r8 rS r r4 r' r( )r
r r r rh � s
z&POSIXProcessTestCase.test_invalid_argsc
C sT t jj� }d|d< tjdgdtj|d�}|� | j|jj� j d�d� W d Q R X d S )Nr� rZ z echo $FRUITr\ )r{ r% r[ s
s apple)
r] r^ r_ r r& r) r r% r� r )r
r` r. r r r �test_shell_sequence s
z(POSIXProcessTestCase.test_shell_sequencec
C sR t jj� }d|d< tjddtj|d�}|� | j|jj� j d�d� W d Q R X d S )Nr� rZ z echo $FRUITr\ )r{ r% r[ s
s apple)
r] r^ r_ r r& r) r r% r� r )r
r` r. r r r �test_shell_string s
z&POSIXProcessTestCase.test_shell_stringc
C sv t j� \}}t|ddd��&}|jdtj � |jdtj � W d Q R X tj |d� t
j |�}tj |� | j
|d� d S )Nr� r� )r� z#!%s
z(exec '%s' -c 'import sys; sys.exit(47)'
i� r3 )rJ r< r� rM r r� r' r( r] r� r r4 r= r )r
r. r� r� r5 r r r �test_call_string s
z%POSIXProcessTestCase.test_call_stringc
C s� g }x>d D ]6}x0d
D ](}t jj||�}t jj|�r|j|� qW q
W |sR| jd� d}t jj|�rzt jj|� rz|j|� xL|D ]D}tjd|d tj d
�}|�" | j
|j j � j
� t|d �� W d Q R X q�W d S )N�/bin� /usr/bin/�/usr/local/bin�bash�kshz"bash or ksh required for this testz/bin/shzecho $0T)r( r{ r% r9 )r� r� r� )r� r� )r] rt ru �isfiler� r� �islinkr r& r) r r% r� r �bytes)r
ZshellsrX �name�shr. r r r �test_specific_shell' s
z(POSIXProcessTestCase.test_specific_shellc
G sf t j t jt j�}z&tjtjddgdtjtjtjd�}W d t j t j|� X |jj d� t
||�|� |S )Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
time.sleep(30)
T)� close_fdsr$ r% r r\ ) rH �SIGINT�default_int_handlerr r& r' r( r) r% r� �getattr)r
�methodr rI r. r r r �
_kill_process; s
z"POSIXProcessTestCase._kill_process�netbsd�openbsdz"Due to known OS bug (issue #16762)c G sR t jtjddgdt jt jt jd�}|jjd� tjd� t ||�|� |j
� d S )Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
T)r� r$ r% r r\ ) r r& r' r( r) r% r� r r% r� rj )r
r� r r. r r r �_kill_dead_processT s
z'POSIXProcessTestCase._kill_dead_processc C s: | j dtj�}|j� \}}| jd|� | j|j� d� d S )N� send_signals KeyboardInterruptr )r� rH r� rj rC r� r )r
r. r� r r r r �test_send_signalk s z%POSIXProcessTestCase.test_send_signalc C s: | j d�}|j� \}}| j|d� | j|j� tj � d S )Nr r� )r� rj r r r rH �SIGKILL)r
r. r� r r r r � test_killq s
z POSIXProcessTestCase.test_killc C s: | j d�}|j� \}}| j|d� | j|j� tj � d S )N� terminater� )r� rj r r r rH �SIGTERM)r
r. r� r r r r �test_terminatew s
z#POSIXProcessTestCase.test_terminatec C s | j dtj� d S )Nr� )r� rH r� )r
r r r �test_send_signal_dead} s z*POSIXProcessTestCase.test_send_signal_deadc C s | j d� d S )Nr )r� )r
r r r �test_kill_dead� s z#POSIXProcessTestCase.test_kill_deadc C s | j d� d S )Nr� )r� )r
r r r �test_terminate_dead� s z(POSIXProcessTestCase.test_terminate_deadc C s: g }x0|D ](}t j|�}t j|�}|j|||f� q
W |S )N)r] �get_inheritable�dupr� )r
Zsave_fdsr& r. � inheritable�savedr r r � _save_fds� s
z POSIXProcessTestCase._save_fdsc C s2 x,|D ]$\}}}t j|||d� t j|� qW d S )N)r� )r] �dup2r- )r
r& r. r� r� r r r � _restore_fds� s z!POSIXProcessTestCase._restore_fdsc C s� d}| j |�}x |D ]\}}}|dkr|}P qW z\x|D ]}tj|� q8W tjtjddg|tjtjd�j� \}}t j
|�}| j ||fd� W d | j |� X d S )Nr z-czRimport sys;sys.stdout.write("apple");sys.stdout.flush();sys.stderr.write("orange"))r$ r% r � apple� orange)r� r� )
r� r] r- r r& r' r( r) rj r r r r� ) r
r& r$ � saved_fdsr. r� r� rm rn r r r �check_close_std_fds� s"
z(POSIXProcessTestCase.check_close_std_fdsc C s | j dg� d S )Nr )r� )r
r r r �test_close_fd_0� s z$POSIXProcessTestCase.test_close_fd_0c C s | j dg� d S )Nr\ )r� )r
r r r �test_close_fd_1� s z$POSIXProcessTestCase.test_close_fd_1c C s | j dg� d S )NrA )r� )r
r r r �test_close_fd_2� s z$POSIXProcessTestCase.test_close_fd_2c C s | j ddg� d S )Nr r\ )r� )r
r r r �test_close_fds_0_1� s z'POSIXProcessTestCase.test_close_fds_0_1c C s | j ddg� d S )Nr rA )r� )r
r r r �test_close_fds_0_2� s z'POSIXProcessTestCase.test_close_fds_0_2c C s | j ddg� d S )Nr\ rA )r� )r
r r r �test_close_fds_1_2� s z'POSIXProcessTestCase.test_close_fds_1_2c C s | j dddg� d S )Nr r\ rA )r� )r
r r r �test_close_fds_0_1_2� s z)POSIXProcessTestCase.test_close_fds_0_1_2c C sv t jd�}t jd�}z.t jd� t jd� tjtjddg�j� W d t j|d� t j|d� t j|� t j|� X d S )Nr r\ z-cz*print('AssertionError:0:CLOEXEC failure.')) r] r� r- r r& r' r( r r� )r
Z new_stdinZ
new_stdoutr r r �test_small_errpipe_write_fd� s
z0POSIXProcessTestCase.test_small_errpipe_write_fdc
C sP dd� t d�D �}�zdd� |D �}x|D ]\}}tj|� q*W tj|d d� tj|d dd� | jt d��}zTx t|�D ]\}}tj||� q|W tj t
j dd g|d |d
|d d �}|j � W d | j
|� X x|D ]}tj|dd� q�W tj|d
d �}tjtj|d d ��} | j|d
� | j| d� W d x|D ]}tj|� �q6W X d S )Nc S s g | ] }t j� �qS r )rJ r< )r� r� r r r r� � s z?POSIXProcessTestCase.test_remapping_std_fds.<locals>.<listcomp>rW c S s g | ] \}}|�qS r r )r� r. r� r r r r� � s r\ s STDINr z-czZimport sys; got = sys.stdin.read();sys.stdout.write("got %s"%got); sys.stderr.write("err")rA )r$ r% r i s got STDINs err)r� r] �unlinkrM r� r� � enumerater� r r& r' r( r r� r� r r r r- )
r
�temps�temp_fdsr. r� r� �temp_fdr. rm rn r r r �test_remapping_std_fds� s4
z+POSIXProcessTestCase.test_remapping_std_fdsc
C s2 dd� t d�D �}dd� |D �}z�x|D ]\}}tj|� q(W | jt d��}z�x t|�D ]\}} tj| |� qXW tj|d� tj|dd� tj t
j ddg|||d �}
|
j � x|D ]}tj|dd� q�W tj
|d
�} tjtj
|d
��} W d | j|� X | j| d � | j| d � W d x|D ]}tj|� �qW X d S )
Nc S s g | ] }t j� �qS r )rJ r< )r� r� r r r r�
s z7POSIXProcessTestCase.check_swap_fds.<locals>.<listcomp>rW c S s g | ] \}}|�qS r r )r� r. r� r r r r� s s STDINr z-czZimport sys; got = sys.stdin.read();sys.stdout.write("got %s"%got); sys.stderr.write("err"))r$ r% r i s got STDINs err)r� r] r� r� r� r� rM r� r r& r' r( r r� r r r� r r- )
r
Zstdin_noZ stdout_noZ stderr_nor� r� r. r� r� r� r. rm rn r r r �check_swap_fds s4
z#POSIXProcessTestCase.check_swap_fdsc C sX | j ddd� | j ddd� | j ddd� | j ddd� | j ddd� | j ddd� d S )Nr r\ rA )r� )r
r r r �
test_swap_fds8 s z"POSIXProcessTestCase.test_swap_fdsc C sx | j td��}�zVx0|D ](}tj� �}tj|j� |� W d Q R X qW ttd��t|� j� }tj |� dddg}i }x"t
||�D ]\}} |||| < q�W t j d�}
ttd��t|� j� } t
jtjd|
t| �gf|�} | j| d� x�t
||�D ]r\}} tj|dtj� tj|d�}
ttt|
jd ���}t j d
|� d |� d |� d
|� d| � d� �}| j| g||� q�W W d | j|� X d S )NrW r$ r% r z�
import os, sys
skipped_fd = int(sys.argv[1])
for fd in range(3):
if fd != skipped_fd:
os.write(fd, str(fd).encode('ascii'))
z-cr i r9 z"
When testing z to z4 redirection,
parent descriptor z5 got redirected
to descriptor(s) z instead of descriptor z.
)r� r� rJ rK r] r� r� �setr� r- �zipr� r� r r4 r' r( r r r� �SEEK_SETr� �list�mapr� r� r� )r
�from_fds�to_fdsr� Zfrom_fd�fZ fd_to_closeZ arg_namesr! Zto_fdrl Z
skipped_fdr5 Z
read_bytesZread_fdsr r r r �#_check_swap_std_fds_with_one_closed@ s2
&z8POSIXProcessTestCase._check_swap_std_fds_with_one_closedc C sD x>t jtd�d�D ]*}x$t jtd�d�D ]}| j||� q(W qW d S )NrW rA )� itertools� combinationsr� � permutationsr� )r
r� r� r r r �!test_swap_std_fds_with_one_closedl s z6POSIXProcessTestCase.test_swap_std_fds_with_one_closedc C s� dd� }yt jtjddg|d� W n| tk
r` } z"| jt j� | jt|�d� W Y d d }~X nJ t j k
r� } z"| j
t j� | jt|�d� W Y d d }~X n X | j d� d S ) Nc S s t d��d S )Nu
surrogate:�)rS r r r r �preparer s zCPOSIXProcessTestCase.test_surrogates_error_message.<locals>.preparez-cr� )r� u
surrogate:�z!Exception occurred in preexec_fn.z1Expected ValueError or subprocess.SubprocessError) r r4 r' r( rS r r� r r r� ZassertIsNotNonerT )r
r rn r r r � test_surrogates_error_messageq s
"z2POSIXProcessTestCase.test_surrogates_error_messagec C s� � x�dD ]�\}}|j dd�}dt|� }tjj� }|||< d|d < tjjd
�r\|jd d�}n|}t j
tj d |g|d
�}|j d�}| j
|jd�t|�� |j dd�}dt|� }tjj� }|||< t j
tj d |g|d
�}|j d�}| j
|jd�t|�� qW d S )N�test� abc�� test��42r9 r� z&import os; print(ascii(os.getenv(%s)))�C�LC_ALL�aix�latin1z-c)r[ s
z'import os; print(ascii(os.getenvb(%s)))�r r �r r )r r )r� rs r] r^ r_ r' r[ r� r� r rB r( rk r r9 )r
�key�valueZ
encoded_value�scriptr[ Z
decoded_valuer% r r r �test_undecodable_env� s.
z)POSIXProcessTestCase.test_undecodable_envc C s� t jtj�}t jjtj�\}}t j|�}tj|ddg�}| j|d� d| d }tj|dd�}| j|d� t j j
� }||d< tj|ddg|d �}| j|d� t j j
� }t j|�|d
< tj|ddg|d �}| j|d� d S ) Nz-cr� r � 's ' -c passT)r{ �PATH)r[ s PATH) r] �fsencoder' r( rt r� r r4 r r^ r_ �environb)r
Z abs_programrt ZprogramZexitcoder� r[ Zenvbr r r �test_bytes_program� s
z'POSIXProcessTestCase.test_bytes_programc C s� t jddd�}t jddd�}tjtj|gtjtjtjdd�}| j|jd� tjtj|gtjdd�}|j� \}}t t
t |j d ���}t |j
j� |jj� |jj� g�}| j||@ d
|||@ f � d S ) Nzinput_reader.py�subprocessdata)�subdirz fd_status.pyF)r$ r% r r� r� )r% r� � ,z5Expected no fds from %r to be open in child, found %r)r �findfiler r& r' r( r) rL rj r� r� r� r� r$ r� r% r r ) r
Zsleeper� fd_status�p1�p2rD �errorZ
result_fdsZ unwanted_fdsr r r �test_pipe_cloexec� s
z&POSIXProcessTestCase.test_pipe_cloexecc
s t jddd�}t jddd�}d}|d d }tjtj|gtjtjdd �� tjtj||g� jtjdd ��| j� j � | j�j � � fd
d �}�fd d
�}| j|� | j|� � j
j |� � j
j � t
j
�jgg g d�\}}} | j|d� | j�jj� |� � jj � �jj � d S )Nzqcat.pyr )r zqgrep.pys zxcvbnr� �
F)r$ r% r� c s& y � j � W n tk
r Y nX d S )N)r� �ProcessLookupErrorr )r r r �kill_p1� s zBPOSIXProcessTestCase.test_pipe_cloexec_real_tools.<locals>.kill_p1c s& y � j � W n tk
r Y nX d S )N)r� r# r )r r r �kill_p2� s zBPOSIXProcessTestCase.test_pipe_cloexec_real_tools.<locals>.kill_p2r� zThe child hung)r r r r& r' r( r) r% rL r r$ rM r- �selectr� r r� )
r
ZqcatZqgrepZsubdatar r$ r% Z readfilesZignored1Zignored2r )r r r � test_pipe_cloexec_real_tools� s.
z1POSIXProcessTestCase.test_pipe_cloexec_real_toolsc
s� t jddd�}tj� }| jtj|d � | jtj|d � t|�� x8td�D ],}tjtj tj
�}| jtj|� � j |� qLW x� D ]}tj |d� q�W t
jtj|gt
jdd �}|j� \}}ttt|jd
���}| j|� @ � d � t
jtj|gt
jdd �}|j� \}}ttt|jd
���}| j|� @ d � | jd|d
� t� fdd�td�D ��} t
jtj|gt
jdf d�}|j� \}}ttt|jd
���}| j|| @ � @ d� | jd|d
� d S )Nz fd_status.pyr )r r r\ r2 TF)r% r� r zSome fds were closedzSome fds were left openzSubprocess failedc 3 s | ]}� j � V qd S )N)r� )r� r� )r' r r � <genexpr>% s z6POSIXProcessTestCase.test_close_fds.<locals>.<genexpr>r� )r% r� �pass_fdsz'Some fds not in pass_fds were left open)r r r] r� rL r- r� r� r� �devnullr� �add�set_inheritabler r& r' r( r) rj r� r� r� r r rC )
r
r r& r� r. r. rD �ignored�
remaining_fds� fds_to_keepr )r' r �test_close_fds sB
z#POSIXProcessTestCase.test_close_fdsZfreebsdz/devz/dev/fdz/Requires fdescfs mounted on /dev/fd on FreeBSD.c C s� t jddd�}tjtjdtjd| �gtjd�}|j � \}}|j
� }| j t |�dd| d � t
tt|d
j� jd ���}t
tt|d j� jd ���}| j||@ d
d � d S )Nz fd_status.pyr )r z-caK
import os, resource, subprocess, sys, textwrap
open_fds = set()
# Add a bunch more fds to pass down.
for _ in range(40):
fd = os.open(os.devnull, os.O_RDONLY)
open_fds.add(fd)
# Leave a two pairs of low ones available for use by the
# internal child error pipe and the stdout pipe.
# We also leave 10 more open as some Python buildbots run into
# "too many open files" errors during the test if we do not.
for fd in sorted(open_fds)[:14]:
os.close(fd)
open_fds.remove(fd)
for fd in open_fds:
#self.addCleanup(os.close, fd)
os.set_inheritable(fd, True)
max_fd_open = max(open_fds)
# Communicate the open_fds to the parent unittest.TestCase process.
print(','.join(map(str, sorted(open_fds))))
sys.stdout.flush()
rlim_cur, rlim_max = resource.getrlimit(resource.RLIMIT_NOFILE)
try:
# 29 is lower than the highest fds we are leaving open.
resource.setrlimit(resource.RLIMIT_NOFILE, (29, rlim_max))
# Launch a new Python interpreter with our low fd rlim_cur that
# inherits open fds above that limit. It then uses subprocess
# with close_fds=True to get a report of open fds in the child.
# An explicit list of fds to check is passed to fd_status.py as
# letting fd_status rely on its default logic would miss the
# fds above rlim_cur as it normally only checks up to that limit.
subprocess.Popen(
[sys.executable, '-c',
textwrap.dedent("""
import subprocess, sys
subprocess.Popen([sys.executable, %r] +
[str(x) for x in range({max_fd})],
close_fds=True).wait()
""".format(max_fd=max_fd_open+1))],
close_fds=False).wait()
finally:
resource.setrlimit(resource.RLIMIT_NOFILE, (rlim_cur, rlim_max))
)r% rA z(expected exactly two lines of output:
%r)r r r r\ zSome fds were left open.)r r r r& r' r( r� r� r) rj �
splitlinesr r� r� r� r� r r� r )r
r r. rD Z
unused_stderrZ output_linesZ
opened_fdsr. r r r �%test_close_fds_when_max_fd_is_lowered1 s
0
z:POSIXProcessTestCase.test_close_fds_when_max_fd_is_loweredr� rF c C s: t jddd�}t� }xftd�D ]Z}tj� }| jtj|d � | jtj|d � tj|d d� tj|d d� |j |� q W x�|D ]�}t
j t j
|gt
jd|fd�}|j� \}}ttt|jd ���} ||h }
| j|| d
� | j| |
@ d � | jt��&} | jt
jt j
d d
gd|fd�� W d Q R X | jdt| j�� q�W d S )Nz fd_status.pyr )r rF r r\ T)r% r� r) r zfd to be passed not passedzfd to be closed passedz-czimport sys; sys.exit(0)F)r� r) zoverriding close_fds)r r r� r� r] r� rL r- r, �updater r& r' r( r) rj r� r� r� rC r r �RuntimeWarningr4 r r ) r
r r' r� r& r. r. rD r- r. Z to_be_closed�contextr r r �
test_pass_fds� s0
z"POSIXProcessTestCase.test_pass_fdsc
C s� t jddd�}tj� \}}| jtj|� | jtj|� tj|d� tj|d� ||f}tj|g}|t t
t |��7 }t j
|t jd|d�}|j� \}}tt
t|jd���} | j| t|�d| � | jtj|�d� | jtj|�d� d S ) Nz fd_status.pyr )r TF)r% r� r) r z output=%a)r r r] r� rL r- r, r' r( r� r� r r r& r) rj r� r� r� r r� )
r
r r� Znon_inheritabler) r r. rD r- r& r r r �test_pass_fds_inheritable� s"
z.POSIXProcessTestCase.test_pass_fds_inheritablec C s� t jddd�}g }x8td�D ],}tjtjtj�}| jtj|� |j |� q W tj
� \}}| jtj|� | jtj|� |j d|� t j
tj|g|d |d |d d|d�� tj|d �}W d Q R X d
d � |jd �D �}| j|dddht|�B d
|��� d S )Nz fd_status.pyr )r rA r\ r T)r$ r% r r� r) i c S s h | ] }t |��qS r )r� )r� Znumr r r � <setcomp>� s z@POSIXProcessTestCase.test_pass_fds_redirected.<locals>.<setcomp>r zoutput=)r r r� r] r� r* �O_RDWRrL r- r� r� �insertr r& r' r( r� r� r � frozenset) r
r r) r� r. Zstdout_rZstdout_wrD r& r r r �test_pass_fds_redirected� s&
z-POSIXProcessTestCase.test_pass_fds_redirectedc C s> t jtjd��&}tjtjddg||d�}|j� W d Q R X d S )Nzr+z-czimport sys; sys.exit(0))r% r$ ) r+ r� r] r* r r& r' r( r )r
�inoutr. r r r �%test_stdout_stdin_are_single_inout_fd� s
z:POSIXProcessTestCase.test_stdout_stdin_are_single_inout_fdc C s> t jtjd��&}tjtjddg||d�}|j� W d Q R X d S )Nzr+z-czimport sys; sys.exit(0))r% r ) r+ r� r] r* r r& r' r( r )r
r= r. r r r �&test_stdout_stderr_are_single_inout_fd� s
z;POSIXProcessTestCase.test_stdout_stderr_are_single_inout_fdc C s> t jtjd��&}tjtjddg||d�}|j� W d Q R X d S )Nzr+z-czimport sys; sys.exit(0))r r$ ) r+ r� r] r* r r& r' r( r )r
r= r. r r r �%test_stderr_stdin_are_single_inout_fd� s
z:POSIXProcessTestCase.test_stderr_stdin_are_single_inout_fdc C sR t jddd�}tjtj|gtjtjd�}|j� \}}| jd|j d|j
d� � d S )Nzsigchild_ignore.pyr )r )r% r r z6sigchild_ignore.py exited non-zero with this error:
%szutf-8) r r r r& r' r( r) rj r r? r� )r
Zsigchild_ignorer. r% r r r r � test_wait_when_sigchild_ignored� s z4POSIXProcessTestCase.test_wait_when_sigchild_ignoredc
C sz t jd�}tjtjddgtjdd�}|j}| j|j � z4| j
|j d�d� | j ||j
|gg g d�d � W d |j� X d S ) Nr& z-cz$import sys;sys.stdout.write("apple")r )r% r0 r� s applg )r �
import_moduler r& r' r( r) r% rL r- r r� rC r& r )r
r& r. r� r r r �test_select_unbuffered� s
"z+POSIXProcessTestCase.test_select_unbufferedc C s� t jtjddgt jt jd�}| j|jj� | j|jj� t |�}|j
}t j dt
f��
d }W d Q R X | j|dd� t jD �� d S )Nz-cz import sys, time;time.sleep(0.2))r% r r c S s g | ] }t |��qS r )�id)r� �or r r r�
s zEPOSIXProcessTestCase.test_zombie_fast_process_del.<locals>.<listcomp>)r r& r' r( r) rL r% r- r rD �pidr �check_warnings�ResourceWarningrC r )r
r. �identrF r r r � test_zombie_fast_process_del
s
z1POSIXProcessTestCase.test_zombie_fast_process_delc C s� t jtjddgt jt jd�}| j|jj� | j|jj� t |�}|j
}t j dt
f��
d }W d Q R X tj|tj� | j|dd� t jD �� tjd� | jt��& t jtt jt jd��}W d Q R X W d Q R X | jttj|d� | j|d d� t jD �� d S )
Nz-czimport time;time.sleep(3))r% r r c S s g | ] }t |��qS r )rD )r� rE r r r r� 4
s zJPOSIXProcessTestCase.test_leak_fast_process_del_killed.<locals>.<listcomp>g�������?r c S s g | ] }t |��qS r )rD )r� rE r r r r� @
s )r r& r' r( r) rL r% r- r rD rF r rG rH r] r rH r� rC r r r% r8 r� ry �waitpidr� )r
r. rI rF r! r r r �!test_leak_fast_process_del_killed
s(
z6POSIXProcessTestCase.test_leak_fast_process_del_killedc sx t jddd�}tjd�� | jtj� � tjtj |gtj
d� fdd�d�}|j � \}}t t
t|jd ���}| j� |� d S )
Nz fd_status.pyr )r r\ Tc s t jd� �S )Nr\ )r] r� r )r. r r r� L
s zCPOSIXProcessTestCase.test_close_fds_after_preexec.<locals>.<lambda>)r% r� r� r )r r r] r� rL r- r r& r' r( r) rj r� r� r� r� r� )r
r r. rD r- r. r )r. r � test_close_fds_after_preexecB
s
z1POSIXProcessTestCase.test_close_fds_after_preexecc C s� dd l }tj� }z�dd� }tj� x�ddgd dgfdgdd dgfdgdgddgfdgdgd dffD ]L\}}}}| jt��0 |j||df ||d
dddd d
d d dd|� W d Q R X qbW W d |s�tj� X d S )Nr c S s d S )Nr r r r r r� ]
s z5POSIXProcessTestCase.test_fork_exec.<locals>.<lambda>�{ s exes envs argTr\ rA rW r� ���rO rO rO )r� r� r� r� r8 rb r� r� )r
r� �
gc_enabled�funcr Zexe_listr~ Zenv_listr r r �test_fork_execS
s& z#POSIXProcessTestCase.test_fork_execc C s� dd l }G dd� d�}tj� }z�tj� x�dddd d |d�|d�ffD ]f}| jtdj|�d��6}|jdgdgd|d dgd d d d!ddddddd � W d Q R X | jdt |j
�� qFW W d |s�tj � X d S )"Nr c @ s e Zd ZdZdd� Zdd� ZdS )zJPOSIXProcessTestCase.test_fork_exec_sorted_fd_sanity_check.<locals>.BadIntTc S s
|| _ d S )N)r )r
r r r r rc w
s zSPOSIXProcessTestCase.test_fork_exec_sorted_fd_sanity_check.<locals>.BadInt.__init__c S s | j rd| _ | jS t�d S )NF)�firstr rS )r
r r r �__int__y
s zRPOSIXProcessTestCase.test_fork_exec_sorted_fd_sanity_check.<locals>.BadInt.__int__N)r r r rS rc rT r r r r �BadIntu
s rU r\ rA rW r� rF r � � �* �? r� � r� zfds_to_keep={})r s falseTs envr/ rO )rO rA rW r� rF )r r� � )rV rW rX r[ )rF r� )r� rZ rZ r� rO rO rO rO ) r� r� r� r� r8 rS r� r� rC r r> r� )r
r� rU rP r/ r@ r r r �%test_fork_exec_sorted_fd_sanity_checkq
s0
z:POSIXProcessTestCase.test_fork_exec_sorted_fd_sanity_checkc C sZ t jtjddg�}|�: tjj|d�� }t|j_ |j
� |jj � W d Q R X W d Q R X d S )Nz-cr� r$ ) r r& r' r( r r� �object�BrokenPipeErrorr- r rj Zassert_called_with)r
r! �mock_proc_stdinr r r �,test_communicate_BrokenPipeError_stdin_close�
s
zAPOSIXProcessTestCase.test_communicate_BrokenPipeError_stdin_closec C sh t jtjddg�}|�H tjj|d��.}t|j_ |j
d� |jj d� |j j � W d Q R X W d Q R X d S )Nz-cr� r$ s stuff)
r r& r' r( r r� r] r^ rM r rj �assert_called_once_withr- )r
r! r_ r r r �,test_communicate_BrokenPipeError_stdin_write�
s
zAPOSIXProcessTestCase.test_communicate_BrokenPipeError_stdin_writec C s� t jtjdgt jt jd�}|�` tjj|d��F}tt j
d��.}t |j _
|j� |j_|jd� |j j� W d Q R X W d Q R X W d Q R X d S )Nz-h)r$ r% r$ �wbs stuff)r r& r' r( r) r r� r] r� r] r* r^ r� r r� � return_valuerj ra )r
r! r_ Zdev_nullr r r �,test_communicate_BrokenPipeError_stdin_flush�
s
zAPOSIXProcessTestCase.test_communicate_BrokenPipeError_stdin_flushc C sf t jtjdgt jt jd�}|�> tjj|d��$}t|j _
|j dd� |j j � W d Q R X W d Q R X d S )Nz-h)r$ r% r$ i� )r7 )
r r& r' r( r) r r� r] r^ r- r rj ra )r
r! r_ r r r �9test_communicate_BrokenPipeError_stdin_close_with_timeout�
s
zNPOSIXProcessTestCase.test_communicate_BrokenPipeError_stdin_close_with_timeoutN�
W_STOPCODEzneed _testcapi.W_STOPCODEc
C sx t jddg}tj|�}|j}tj|d�\}}| j|d� tj d�}t
j d||fd�� |j � }W d Q R X | j|d� d S )Nz-cr� r rW zsubprocess.os.waitpid)rd ���)
r' r( r r& rF r] rK r � _testcapirg r r� r )r
r r! rF �statusr? r r r � test_stopped�
s
z!POSIXProcessTestCase.test_stopped)r� r� )cr r r r
ru rv rw rx r r& rz r r� r� r� r� r� r� r� r� r� r� r� r� rU rV r] rt r> r� r� r' r[ r� r� rh 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� r� r r r r r! r' r0 �stat�st_devr2 r Zrequires_mac_verr6 r7 r< r> r? r@ rA rC rJ rL rM � cpython_onlyrR r\ r` rb re rf r] ri r^ rk �
__classcell__r r )rp r rm � s� %
-0,#)0Q" " (
rm zWindows specific testsc @ s� e Zd Zdd� Zdd� Zdd� Zejdd� �Zd d
� Z d d � Z
d
d� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�Win32ProcessTestCasec C s6 d}d}t j� }||_||_t jtjddg|d� d S )Nr\ rW z-czimport sys; sys.exit(0))r� )r Z STARTUPINFOZdwFlagsZ wShowWindowr4 r' r( )r
ZSTARTF_USESHOWWINDOWZ SW_MAXIMIZEr� r r r �test_startupinfo�
s z%Win32ProcessTestCase.test_startupinfoc C s( d}t jjd� tjt jd |d� d S )N� z' a DOS box should flash briefly ...
z# -c "import time; time.sleep(0.25)")r� )r' r rM r r4 r( )r
ZCREATE_NEW_CONSOLEr r r �test_creationflags�
s
z'Win32ProcessTestCase.test_creationflagsc C sD | j ttjtjddgdd� d� | j ttjtjddgtjdd� d S )Nz-czimport sys; sys.exit(47)c S s dS )Nr\ r r r r r r� �
s z8Win32ProcessTestCase.test_invalid_args.<locals>.<lambda>)r� T)r% r� )r8 rS r r4 r' r( r) )r
r r r rh �
s
z&Win32ProcessTestCase.test_invalid_argsc C sB G dd� dt �}| jt�� tjtjddg|� d� W d Q R X d S )Nc @ s e Zd ZdZdS )z4Win32ProcessTestCase.test_issue31471.<locals>.BadEnvN)r r r �keysr r r r �BadEnv s ru z-cr� )r[ )�dictr8 rb r r& r' r( )r
ru r r r �test_issue31471 s z$Win32ProcessTestCase.test_issue31471c C s& t jtjddgdd�}| j|d� d S )Nz-czimport sys; sys.exit(47)T)r� r3 )r r4 r' r( r )r
r5 r r r r0
s
z#Win32ProcessTestCase.test_close_fdsc
C sN t jj� }d|d< tjdgdtj|d�}|� | jd|jj� � W d Q R X d S )N�physalisrZ r� r\ )r{ r% r[ s physalis) r] r^ r_ r r& r) rC r% r� )r
r` r. r r r r� s
z(Win32ProcessTestCase.test_shell_sequencec
C sL t jj� }d|d< tjddtj|d�}|� | jd|jj� � W d Q R X d S )Nrx rZ r� r\ )r{ r% r[ s physalis) r] r^ r_ r r& r) rC r% r� )r
r` r. r r r r� s
z&Win32ProcessTestCase.test_shell_stringc C s^ xXdD ]P}t jj� }d|d< tjddtj||d�}|� | jd|jj� |� W d Q R X qW d S ) N�ansi�oemrx rZ r� r\ )r{ r% r[ r� )ry rz ) r] r^ r_ r r& r) rC r% r� )r
�encr` r. r r r �test_shell_encodings% s
z)Win32ProcessTestCase.test_shell_encodingsc C s t jtjd �}| j|d� d S )Nz -c "import sys; sys.exit(47)"r3 )r r4 r' r( r )r
r5 r r r r� 1 s z%Win32ProcessTestCase.test_call_stringc
G sz t jtjddgt jt jt jd�}|�@ |jjd� t||�|� |j� \}}| j |d� |j
� }W d Q R X | j |d� d S )Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
time.sleep(30)
)r$ r% r r\ r� r ) r r& r' r( r) r% r� r� rj r r r� )r
r� r r. r� r r? r r r r� 7 s
z"Win32ProcessTestCase._kill_processc
G s� t jtjddgt jt jt jd�}|�J |jjd� tjd� t ||�|� |j
� \}}| j |d� |j � }W d Q R X | j
|d� d S )Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
sys.exit(42)
)r$ r% r r\ r� rX )r r& r' r( r) r% r� r r% r� rj r r r )r
r� r r. r� r r5 r r r r� L s
z'Win32ProcessTestCase._kill_dead_processc C s | j dtj� d S )Nr� )r� rH r� )r
r r r r� c s z%Win32ProcessTestCase.test_send_signalc C s | j d� d S )Nr )r� )r
r r r r� f s z Win32ProcessTestCase.test_killc C s | j d� d S )Nr� )r� )r
r r r r� i s z#Win32ProcessTestCase.test_terminatec C s | j dtj� d S )Nr� )r� rH r� )r
r r r r� l s z*Win32ProcessTestCase.test_send_signal_deadc C s | j d� d S )Nr )r� )r
r r r r� o s z#Win32ProcessTestCase.test_kill_deadc C s | j d� d S )Nr� )r� )r
r r r r� r s z(Win32ProcessTestCase.test_terminate_deadN)r r r rq rs rh r rn rw r0 r� r� r| r� r� r� r� r� r� r� r� r� r r r r rp �
s"
rp c @ s e Zd Zdd� Zdd� ZdS )� MiscTestsc C s� | j tjd�d� | j tjd�d� d }z@tj� }tjj|d�}tjt rLdnd| �\}}| j
|d� W d |d k r|tj |� X d S )Nz
echo xyzzy�xyzzyr Zfooztype zcat )r r~ ) r r Z getoutputZgetstatusoutputrJ r� r] rt ru r� r� �rmdir)r
�dirr� rj rD r r r �test_getoutputv s zMiscTests.test_getoutputc C sr ddh}t tj�}t � }dd l}x<tjj� D ].\}}|jd�r@q,t||jf�rPq,|j |� q,W | j
||| � d S )Nr ZHandler r� ) r� r �__all__�types�__dict__�itemsr� �
isinstance�
ModuleTyper+ r )r
Zintentionally_excludedZexportedZpossible_exportsr� r� r r r r � test__all__� s
zMiscTests.test__all__N)r r r r� r� r r r r r} u s r} Z PollSelectorz!Test needs selectors.PollSelectorc @ s e Zd Zdd� Zdd� ZdS )�ProcessTestCaseNoPollc C s t j| _tjt _tj| � d S )N)r �_PopenSelector�
orig_selector� selectorsZSelectSelectorr# r
)r
r r r r
� s zProcessTestCaseNoPoll.setUpc C s | j t_tj| � d S )N)r� r r� r# r )r
r r r r � s z ProcessTestCaseNoPoll.tearDownN)r r r r
r r r r r r� � s r� zWindows-specific testsc sP e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z � Z
S )�CommandsWithSpacesc s> t � j� tjdd�\}}|j� | _tj|d� tj|� d S )Nz.pyzte stsV import sys;sys.stdout.write('%d %s' % (len(sys.argv), [a.lower () for a in sys.argv]))) rn r
rJ r< r� r� r] rM r- )r
r� r� )rp r r r
� s
zCommandsWithSpaces.setUpc s t j| j� t� j� d S )N)r] r= r� rn r )r
)rp r r r � s zCommandsWithSpaces.tearDownc O sH t j|d< t j||�}|�$ | j|jj� jd�d| j � W d Q R X d S )Nr% �mbcsz2 [%r, 'ab cd'])r r) r& r r% r� r� r� )r
r r! r. r r r � with_spaces� s
z CommandsWithSpaces.with_spacesc C s | j dtj| jdf dd� d S )Nz"%s" "%s" "%s"zab cdr\ )r{ )r� r' r( r� )r
r r r � test_shell_string_with_spaces� s z0CommandsWithSpaces.test_shell_string_with_spacesc C s | j tj| jdgdd� d S )Nzab cdr\ )r{ )r� r' r( r� )r
r r r � test_shell_sequence_with_spaces� s z2CommandsWithSpaces.test_shell_sequence_with_spacesc C s | j dtj| jdf � d S )Nz"%s" "%s" "%s"zab cd)r� r' r( r� )r
r r r � test_noshell_string_with_spaces� s z2CommandsWithSpaces.test_noshell_string_with_spacesc C s | j tj| jdg� d S )Nzab cd)r� r' r( r� )r
r r r �!test_noshell_sequence_with_spaces� s z4CommandsWithSpaces.test_noshell_sequence_with_spaces) r r r r
r r� r� r� r� r� ro r r )rp r r� � s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ContextManagerTestsc
C sl t jtjddgt jt jd��*}| j|jj� d� | j|j j� d� W d Q R X | j
|jj � | j
|j j � d S )Nz-czAimport sys;sys.stdout.write('stdout');sys.stderr.write('stderr');)r% r s stdouts stderr) r r& r' r( r) r r% r� r r r� r� )r
r! r r r � test_pipe� s
z ContextManagerTests.test_pipec
C s0 t jtjddg��}W d Q R X | j|jd� d S )Nz-czimport sys; sys.exit(100)�d )r r& r' r( r r? )r
r! r r r rd � s
z#ContextManagerTests.test_returncodec
C s@ t jtjddgt jd�� }|jd� | j|jd� W d Q R X d S )Nz-cz2import sys;sys.exit(sys.stdin.read() == 'context'))r$ s contextr\ )r r& r' r( r) rj r r? )r
r! r r r r� � s
z*ContextManagerTests.test_communicate_stdinc C s: | j t��& tjttjtjd��}W d Q R X W d Q R X d S )N)r% r )r8 rx r r& ry r) )r
r! r r r rh � s
z%ContextManagerTests.test_invalid_argsc C sz t jtjddgt jtjd d�}|j� }|jj dtj � | j
|j � | j t
|jd d d � | j|j d� | j|jj� d S )Nz-cr� rA )r$ r0 r@ r )r r& r' r( r) r r� � __enter__r$ rM r r? r8 r� �__exit__r r� r� )r
r! r r r �test_broken_pipe_cleanup� s z,ContextManagerTests.test_broken_pipe_cleanupN)r r r r� rd r� rh r� r r r r r� � s
r� �__main__)r )7rU r r r r r' r[ rH r+ r r] r� rJ r r� rY r& r� r� r� Z test.supportr Zctypesr Z ctypes.utilr) ri ZPGOZSkipTestr� r� ry r� �NotADirectoryError�PermissionErrorrx ZTestCaser � Exceptionr r& r r# r` rV rm r] rp r} r^ r� r� r� r �mainr r r r �<module> s�
_ # +6