Your IP : 216.73.216.213
�
f ff� �
@ s� d d l Z d d l m Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l Z Wn e k
r"d Z Yn Xe j d k Z e rAd Z n d Z Gd d � d e j � Z Gd d
� d
e � Z Gd d � d e j � Z Gd
d � d e � Z e j! e d � Gd d � d e � � Z" e j# e d � Gd d � d e � � Z$ Gd d � d e j � Z% e j# e& e d � d � Gd d � d e � � Z' Gd d � d e j � Z( e j# e d � Gd d � d e � � Z) Gd d! � d! e � Z* d" d# � Z+ e, d$ k r�e j- � n d S)%� N)�
script_helper)�support�win32z@import msvcrt; msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY);� c @ s7 e Z d Z d d � Z d d � Z d d d � Z d S)� BaseTestCasec C s t j � d S)N)r �
reap_children)�self� r �9/opt/alt/python34/lib64/python3.4/test/test_subprocess.py�setUp) s zBaseTestCase.setUpc C s? x t j D] } | j � q
Wt j � | j t j d � d S)Nz subprocess._active not empty)�
subprocess�_active�waitZ_cleanup� assertFalse)r Zinstr r r
�tearDown. s
zBaseTestCase.tearDownNc C s2 t j | � } | j � } | j | | | � d S)N)r �strip_python_stderr�strip� assertEqual)r �stderr�expected�msgZactualr r r
�assertStderrEqual4 s z BaseTestCase.assertStderrEqual)�__name__�
__module__� __qualname__r r r r r r r
r ( s r c @ s e Z d Z d S)�PopenTestExceptionN)r r r r r r r
r > s r c @ s" e Z d Z d Z d d � Z d S)�PopenExecuteChildRaiseszePopen subclass for testing cleanup of subprocess.PIPE filehandles when
_execute_child fails.
c O s t d � � d S)NzForced Exception for Test)r )r �args�kwargsr r r
�_execute_childF s z&PopenExecuteChildRaises._execute_childN)r r r �__doc__r r r r r
r B s r c @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z e j e d- � d. d/ � � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z e j e d8 � d9 d: � � Z! e j e d8 � d; d< � � Z" d= d> � Z# e j e$ j% e$ j& k d? � d@ dA � � Z' e j e$ j% e$ j& k d? � e j e( j) � dB � dC dD � � � Z* dE dF � Z+ dG dH � Z, dI dJ � Z- dK dL � Z. dM dN � Z/ dO dP � Z0 dQ dR � Z1 dS dT � Z2 dU dV � Z3 dW dX � Z4 dY dZ � Z5 d[ d\ � Z6 d] d^ � Z7 d_ d` � Z8 da db � Z9 dc dd � Z: e j e$ j; de k df � e j e( j< dg � dh k di � dj dk � � � Z= dl dm � Z> dn do � Z? dp dq � Z@ dr ds � ZA dt du � ZB dv dw � ZC dx dy � ZD dz d{ � ZE d| d} � ZF d~ d � ZG d� d� � ZH d� d� � ZI d� d� � ZJ d� d� � ZK d� d� � ZL d� d� � ZM d� d� � ZN d� d� � ZO d� d� � ZP d� d� � ZQ d� d� � ZR d� d� � ZS d� d� � ZT d� d� � ZU d� d� � ZV d� d� � ZW d� d� � ZX d� d� � ZY d� d� � ZZ e j e[ dh k d� � d� d� � � Z\ e j e[ dh k d� � d� d� � � Z] d� d� � Z^ d� d� � Z_ d� d� � Z` d� d� � Za e jb ec ed d� � d� � e jb ec ee d� � d� � e jb ec ee d� � d� � d� d� � � � � Zf e jb ee jg jh d� ee ji � � d� � d� d� � � Zj dh S)��ProcessTestCasec
C s� t j t j d d g d t j d t j d t j �} zF | j | j t j � | j | j t j � | j | j
t j � Wd | j j � | j j � | j
j � | j � Xd 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_defaultL s
z+ProcessTestCase.test_io_buffered_by_defaultc
C s� t j t j d d g d t j d t j d t j d d �} zF | j | j t j � | j | j t j � | j | j
t j � Wd | j j � | j j � | j
j � | j � Xd S)Nz-czimport sys; sys.exit(0)r" r# r �bufsizer )
r r$ r% r& r' r( r" r) � RawIOBaser# r r+ r )r r, r r r
�test_io_unbuffered_worksZ s
z(ProcessTestCase.test_io_unbuffered_worksc C s/ t j t j d d g � } | j | d � d S)Nz-czimport sys; sys.exit(47)�/ )r �callr% r& r )r �rcr r r
�
test_call_seqh s z ProcessTestCase.test_call_seqc C s/ | j t j t j t j d d g d d �d S)Nz-czwhile True: pass�timeoutg�������?)� assertRaisesr �TimeoutExpiredr2 r% r& )r r r r
�test_call_timeoutn s z!ProcessTestCase.test_call_timeoutc C s/ t j t j d d g � } | j | d � d S)Nz-czimport sys; sys.exit(0)r )r �
check_callr% r& r )r r3 r r r
�test_check_call_zerow s z$ProcessTestCase.test_check_call_zeroc C sN | j t j � � } t j t j d d g � Wd QX| j | j j d � d S)Nz-czimport sys; sys.exit(47)r1 ) r6 r �CalledProcessErrorr9 r% r& r � exception�
returncode)r �cr r r
�test_check_call_nonzero} s z'ProcessTestCase.test_check_call_nonzeroc C s/ t j t j d d g � } | j d | � 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 sN | j t j � � } t j t j d d g � Wd QX| j | j j d � d S)Nz-czimport sys; sys.exit(5)� ) r6 r r; r@ r% r& r r< r= )r r> r r r
�test_check_output_nonzero� s z)ProcessTestCase.test_check_output_nonzeroc C s8 t j t j d d g d t j �} | j d | � d S)Nz-cz$import sys; sys.stderr.write('BDFL')r s BDFL)r r@ r% r& �STDOUTrA )r rB r r r
�test_check_output_stderr� s z(ProcessTestCase.test_check_output_stderrc C sk t j � } | j | j � | j d � | j d � t j t j d d g 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 r@ r% r& rA )r �tfrB r r r
�test_check_output_stdin_arg� s
z+ProcessTestCase.test_check_output_stdin_argc C s5 t j t j d d g d d �} | j d | � d S)Nz-cz6import sys; sys.stdout.write(sys.stdin.read().upper())�inputs pears PEAR)r r@ r% r& rA )r rB r r r
�test_check_output_input_arg� s
z+ProcessTestCase.test_check_output_input_argc C sg | j t � �8 } t j t j d d g d t j �} | j d � Wd QX| j d | j j
d � d S)Nz-czprint('will not be run')r# z-Expected ValueError when stdout arg supplied.r ) r6 �
ValueErrorr r@ r% r& r# �failrA r< r )r r> rB r r r
� test_check_output_stdout_arg� s z,ProcessTestCase.test_check_output_stdout_argc
C s� t j � } | j | j � | j d � | j d � | j t � �; } t j t
j d d g d | d d �} | j d � Wd QX| j
d | j j d � | j
d | j j d � d S) Ns pearr z-czprint('will not be run')r" rO s harez7Expected ValueError when stdin and input args supplied.)rH rI rJ r+ rK rL r6 rQ r r@ r% r& rR rA r< r )r rM r> rB r r r
�&test_check_output_stdin_with_input_arg� s
z6ProcessTestCase.test_check_output_stdin_with_input_argc C sc | j t j � �5 } t j t j d d g d d �} | j d � Wd QX| j | j j d � d S)Nz-czMimport sys, time
sys.stdout.write('BDFL')
sys.stdout.flush()
time.sleep(3600)r5 � zExpected TimeoutExpired.s BDFL)
r6 r r7 r@ r% r& rR r r< rB )r r> rB r r r
�test_check_output_timeout� s z)ProcessTestCase.test_check_output_timeoutc C sN t j j � } d | d <t j t j d d g d | �} | j | d � d S)N�banana�FRUITz-cz5import sys, os;sys.exit(os.getenv("FRUIT")=="banana")�env� )�os�environ�copyr r2 r% r& r )r �newenvr3 r r r
�test_call_kwargs� s
z ProcessTestCase.test_call_kwargsc C s� t j � �Y } | j t t j d d �t j j j j } d g | d } | j t t j | � Wd QX| j | j
� d � d S)NZinvalid_arg_namerZ r r ) r Zcaptured_stderrr6 � TypeErrorr r$ �__init__�__code__� co_argcountr �getvalue)r �sZargcountZ
too_many_argsr r r
�test_invalid_args� s z!ProcessTestCase.test_invalid_argsc C st t j t j d d g d t j d t j �} | j | j j � | j | j j � | j � | j
| j d � d S)Nz-czprint("banana")r# r ) r r$ r% r& r' rJ r# r+ r r r r" )r r, r r r
�test_stdin_none� s
z ProcessTestCase.test_stdin_nonec C s� d } t j t j d | g d t j d t j �} | j | j j � | j | j j � | 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-cr# r r s test_stdout_none)
r r$ r% r& r' rJ r# r+ r � communicater r= �rstrip)r �coder, �out�errr r r
�test_stdout_none� s z ProcessTestCase.test_stdout_nonec C st t j t j d d g d t j d t j �} | j | j j � | j | j j � | j � | j
| j d � d S)Nz-czprint("banana")r" r# ) r r$ r% r& r' rJ r# r+ r" r r r )r r, r r r
�test_stderr_none s
z ProcessTestCase.test_stderr_nonec K s@ | d g } t j | | � } | j � | j d | j � d S)Nzimport sys; sys.exit(47)r1 )r r$ r r r= )r �pre_argsr r r, r r r
�_assert_python s
z ProcessTestCase._assert_pythonc C sD t j j t j j t j � d � } | j | d g d t j �d S)N� doesnotexistz-cr& )r[ �path�join�dirnamer% r& rp )r rq r r r
�test_executable s z ProcessTestCase.test_executablec C s< t j d g } | j | � | j t | j | d d �d S)Nz-cr& rq )r% r& rp r6 �FileNotFoundError)r ro r r r
� test_executable_takes_precedence" s
z0ProcessTestCase.test_executable_takes_precedencez"executable argument replaces shellc C s | j g d t j d d �d S)Nr& �shellT)rp r% r& )r r r r
� test_executable_replaces_shell+ s z.ProcessTestCase.test_executable_replaces_shellc C s$ t j | � � t j � SWd QXd S)N)r Z
change_cwdr[ �getcwd)r �cwdr r r
�_normalize_cwd2 s z ProcessTestCase._normalize_cwdc C s% t j j t j � } t j j | � S)N)r[ rr �realpathr% r& �split)r Z python_pathr r r
�_split_python_path: s z"ProcessTestCase._split_python_pathc K s� t j | d d g d t j | �} | j | j j � | j � | j d | j � t j
j } | j | | � | | j j � j
d � � � d S)Nz-cz;import os, sys; sys.stdout.write(os.getcwd()); sys.exit(47)r# r1 zutf-8)r r$ r' rJ r# r+ r r r= r[ rr �normcase�read�decode)r Z expected_cwdZ
python_argr r, r� r r r
� _assert_cwd@ s
zProcessTestCase._assert_cwdc C s8 t j � } | j | � } | j | t j d | �d S)Nr{ )rH Z
gettempdirr| r� r% r& )r �temp_dirr r r
�test_cwdQ s zProcessTestCase.test_cwdz"pending resolution of issue #15533c
C s� | j � \ } } t j j t j | � } t j � �d } | j t t j
| g � | j t t j
| g d | �| j | � } | j | | d | �Wd QXd S)Nr{ )
r r[ rr rs �curdirr �temp_cwdr6 rv r r$ r| r� )r �
python_dir� python_base�
rel_python� wrong_dirr r r
�test_cwd_with_relative_argW s
z*ProcessTestCase.test_cwd_with_relative_argc C s� | j � \ } } t j j t j | � } d } t j � �v } | j t t j
| g d | �| j t t j
| g d | d | �| j | � } | j | | d | d | �Wd QXd S)N�somethingyoudonthaver& r{ )
r r[ rr rs r� r r� r6 rv r r$ r| r� )r r� r� r� Z doesntexistr� r r r
�!test_cwd_with_relative_executableg s z1ProcessTestCase.test_cwd_with_relative_executablec
C s� | j � \ } } t j j | | � } t j j t j | � } t j � �K } | j t t j
| g d | �| j | � } | j | | d | �Wd QXd S)Nr{ )
r r[ rr rs r� r r� r6 rv r r$ r| r� )r r� r� Z
abs_pythonr� r� r r r
�test_cwd_with_absolute_argz s z*ProcessTestCase.test_cwd_with_absolute_argzTest is not venv-compatiblec C sD | j � \ } } | j | � } | j | d d t j 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 t j � d d t j �d S)Nr� r& )r� r[ rz r% r& )r r r r
�test_executable_without_cwd� s z+ProcessTestCase.test_executable_without_cwdc C sb t j t j d d g d t j �} | j j d � | j j � | j � | j | j
d � d S)Nz-cz0import sys; sys.exit(sys.stdin.read() == "pear")r" s pearrZ ) r r$ r% r& r' r" rK r+ r r r= )r r, r r r
�test_stdin_pipe� s
z ProcessTestCase.test_stdin_pipec C s� t j � } | j | j � | j � } t j | d � t j | d d � t j t
j d d g d | �} | j � | j
| j d � d S)Ns pearr z-cz0import sys; sys.exit(sys.stdin.read() == "pear")r" rZ )rH rI rJ r+ �filenor[ rK �lseekr r$ r% r& r r r= )r rM �dr, r r r
�test_stdin_filedes� s
z"ProcessTestCase.test_stdin_filedesc C sx t j � } | j | j � | j d � | j d � t j t j d d g d | �} | j
� | j | j d � d S)Ns pearr z-cz0import sys; sys.exit(sys.stdin.read() == "pear")r" rZ )
rH rI rJ r+ rK rL r r$ r% r& r r r= )r rM r, r r r
�test_stdin_fileobj� s
z"ProcessTestCase.test_stdin_fileobjc C sT t j t j d d g d t j �} | j | j j � | j | j j � d � d S)Nz-cz&import sys; sys.stdout.write("orange")r# s orange)
r r$ r% r& r' rJ 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 � } t j t j d d g d | �} | j � t
j | d d � | j t
j
| d � d � d S)Nz-cz&import sys; sys.stdout.write("orange")r# r i s orange)rH rI rJ r+ r� r r$ r% r& r r[ r� r r� )r rM r� r, r r r
�test_stdout_filedes� s
z#ProcessTestCase.test_stdout_filedesc C sn t j � } | j | j � t j t j d d g d | �} | j � | j d � | j
| j � d � d S)Nz-cz&import sys; sys.stdout.write("orange")r# r s orange) rH rI rJ r+ r r$ r% r& r rL r r� )r rM r, r r r
�test_stdout_fileobj� s
z#ProcessTestCase.test_stdout_fileobjc C sT t j t j d d g d t j �} | j | j j � | j | j j � d � d S)Nz-cz*import sys; sys.stderr.write("strawberry")r s
strawberry)
r r$ r% r& r' rJ 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 � } t j t j d d g d | �} | j � t
j | d d � | j t
j
| d � d � d S)Nz-cz*import sys; sys.stderr.write("strawberry")r r i s
strawberry)rH rI rJ r+ r� r r$ r% r& r r[ r� r r� )r rM r� r, r r r
�test_stderr_filedes� s
z#ProcessTestCase.test_stderr_filedesc C sn t j � } | j | j � t j t j d d g d | �} | j � | j d � | j
| j � d � d S)Nz-cz*import sys; sys.stderr.write("strawberry")r r s
strawberry) rH rI rJ r+ r r$ r% r& r rL r r� )r rM r, r r r
�test_stderr_fileobj� s
z#ProcessTestCase.test_stderr_fileobjc C s] t j t j d d g d t j d t j �} | j | j j � | j | j j
� d � 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' rF rJ r# r+ r r� )r r, r r r
�test_stdout_stderr_pipe� s z'ProcessTestCase.test_stdout_stderr_pipec C st t j � } | j | j � t j t j d d g d | 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) rH rI rJ r+ r r$ r% r& r rL r r� )r rM r, r r r
�test_stdout_stderr_file s
z'ProcessTestCase.test_stdout_stderr_filec C s� d } t j t j d | g d t j d t j �} | j | j j � | j | j j � | 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-cr# r r s test with stdout=1)
r r$ r% r& r' rJ r# r+ r rh r r= ri )r rj r, rk rl r r r
� test_stdout_filedes_of_stdout s z-ProcessTestCase.test_stdout_filedes_of_stdoutc C sE t j t j d d g d t j �} | j � | j | j d � 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_devnull) s
z#ProcessTestCase.test_stdout_devnullc C sE t j t j d d g d t j �} | j � | j | j d � 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_devnull1 s
z#ProcessTestCase.test_stderr_devnullc C sE t j t j d d g d t j �} | j � | j | j d � 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_devnull: s
z"ProcessTestCase.test_stdin_devnullc
C sr t j j � } d | d <t j t j d d g d t j d | ��) } | j � \ } } | j | d � Wd QXd S)N�orangerX z-cz2import sys,os;sys.stdout.write(os.getenv("FRUIT"))r# rY s orange)
r[ r\ r] r r$ r% r& r' rh r )r r^ r, r# r r r r
�test_envB 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 s_ t j t j d d g d t j d i ��/ } | j � \ } } | j | j � d � Wd QXd S)Nz-cz)import os; print(list(os.environ.keys()))r# rY � []� ['__CF_USER_TEXT_ENCODING'])r� r� )r r$ r% r& r' rh rA r )r r, r# r r r r
�test_empty_envO s z ProcessTestCase.test_empty_envc C sx t j d } | j t t f � � t j | d d g � Wd QX| j t t f � � t j t j d d g � Wd QXd S)N� z-c�passzpass# )r% r& r6 rQ r` r r$ )r �cmdr r r
�test_invalid_cmda s
z ProcessTestCase.test_invalid_cmdc C sh t j j � } d | d <| j t t f � �$ t j t j d d g d | �Wd QXt j j � } d | d <| j t t f � �$ t j t j d d g d | �Wd QXt j j � } d | d <| j t � �$ t j t j d d g d | �Wd QXt j j � } d
| d <t j t j d d g d t j
d | ��) } | j � \ } } | j | d
� Wd QXd S)NZcabbageZFRUIT VEGETABLEz-cr� rY Zorange VEGETABLE=cabbagerX Zlemonz FRUIT=ORANGEz orange=lemonz3import sys, os;sys.stdout.write(os.getenv("FRUIT"))r# s orange=lemon)
r[ r\ r] r6 rQ r` r r$ r% r& r' rh r )r r^ r, r# r r r r
�test_invalid_envk s(
%
%
%
z ProcessTestCase.test_invalid_envc C sH t j t j d d g d t j �} | j d � | j | j d � d S)Nz-cz/import sys;sys.exit(sys.stdin.read() == "pear")r" s pearrZ )r r$ r% r& r' rh r r= )r r, r r r
�test_communicate_stdin� s
z&ProcessTestCase.test_communicate_stdinc C sZ t j t j d d g d t j �} | j � \ } } | j | d � | j | d � d S)Nz-cz)import sys; sys.stdout.write("pineapple")r# s pineapple)r r$ r% r& r' rh r )r r, r# r r r r
�test_communicate_stdout� s z'ProcessTestCase.test_communicate_stdoutc C sZ t j t j d d g d t j �} | j � \ } } | j | d � | j | d � d S)Nz-cz)import sys; sys.stderr.write("pineapple")r s pineapple)r r$ r% r& r' rh r r )r r, r# r r r r
�test_communicate_stderr� s z'ProcessTestCase.test_communicate_stderrc C s� t j t j d d g d t j d t j d t j �} | j | j j � | j | j j � | 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' rJ r# r+ r r" rh r r )r r, r# r r r r
�test_communicate� s z ProcessTestCase.test_communicatec
C s� t j t j d d g d d d t j d t j d t j �} | j t j | j d d d
�| 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())�universal_newlinesTr" r# r rW r5 g333333�?s pineapple
pear
) r r$ r% r& r' r6 r7 rh r r �encode)r r, r# r r r r
�test_communicate_timeout� s z(ProcessTestCase.test_communicate_timeoutc C sl t j t j d d g d t j �} | j t j | j d d �| j � \ } } | j t | � 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# r5 g�������?� �@ i � i )
r r$ r% r& r' r6 r7 rh r �len)r r, r# �_r r r
�$test_communicate_timeout_large_ouput� s z4ProcessTestCase.test_communicate_timeout_large_ouputc C s" xd D]} x
d D]} x� d
D]� } i } | rC t j | d <n | rY t j | d <n | ro t j | d <n | s{ q! n t j t j d d f | � } | j � | j d k r� | j | j j � n | j d k r� | j | j j � n | j
d k r! | j | j
j � q! q! Wq Wq Wd S) NFTr" r# r z-cr� )FT)FT)FT) r r' r$ r% r& rh r" �
assertTrue�closedr# r )r Z
stdin_pipeZ stdout_pipeZ stderr_pipeZoptionsr, r r r
� test_communicate_pipe_fd_leak� s(
z-ProcessTestCase.test_communicate_pipe_fd_leakc C sQ t j t j d d g � } | j � \ } } | j | d � | j | d � d S)Nz-czimport sys; sys.exit(47))r r$ r% r& rh 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 | � t j t j d d t j g d t j d t j d t j �} | j
| j j � | j
| j j � | j
| j
j � d t j } | 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' rJ r# r r" rh 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 j t j d d g d t j d t j d t j �} | j | j j � | j | j j � | 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 bananasplit� )r r$ r% r& r' rJ r# r+ r r" rK rh r r )r r, r# r r r r
� test_writes_before_communicate s z.ProcessTestCase.test_writes_before_communicatec C s� t j t j d d t d g d t j d t j d d �} | j j d � | j j � | j | j
j � d � | j j d � | j j � | j
| j
j � | j | j
j � d
� | j | j
j d � d � | j | j
j � d � 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� rZ zline1
zline3
zline2
� z line4
line5
line6
line7
line8)r r$ r% r& � SETBINARYr' r" rK �flushr r# �readliner+ rJ r� )r r, r r r
�test_universal_newlines s$
z'ProcessTestCase.test_universal_newlinesc C s� t j t j d d t d g d t j d t j d d �} | j | j j � | 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� rZ z#line2
line4
line5
line6
line7
line8) r r$ r% r& r� r' rJ r# r+ r rh r )r r, r# r r r r
�#test_universal_newlines_communicate< s
z3ProcessTestCase.test_universal_newlines_communicatec C sg t j t j d d t t j d � g d t j d d �} | j d � \ } } | 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� rZ z line1
line3
r ) r r$ r% r& r� �textwrap�dedentr' rh r r= )r r, r# r r r r
�)test_universal_newlines_communicate_stdinU s
z9ProcessTestCase.test_universal_newlines_communicate_stdinc C sT t j t j d d g d t j d t j d d �} | j � | j | j d � d S)Nz-cr� r" r# r� Tr )r r$ r% r& r' rh r r= )r r, r r r
�.test_universal_newlines_communicate_input_nonec s
z>ProcessTestCase.test_universal_newlines_communicate_input_nonec
C s� t j t j d d t t j d � g d t j d t j d t j d d �} | j | j j
� | j | j j
� | j d � \ } } | j
| j d
� | j
d | � | j | j d � � 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")
r" r r# r� Tz line1
line3
r z line1
line2
line3
line4
line5
zeline2
eline6
eline7
)r r$ r% r& r� r� r� r' rJ r# r+ r rh r r= r� �
startswith)r r, r# r r r r
�7test_universal_newlines_communicate_stdin_stdout_stderro s
zGProcessTestCase.test_universal_newlines_communicate_stdin_stdout_stderrc s� d d l } x� d d g D]� � | j } d � f d d � } d � } t j d | g } zL | | _ t j | d d d
t j d t j �} | j d d
� \ } } Wd | | _ X| j | d � q Wd S)Nr zutf-16z utf-32-beTc s � S)Nr )� do_setlocale)�encodingr r
�getpreferredencoding� s z[ProcessTestCase.test_universal_newlines_communicate_encodings.<locals>.getpreferredencodingz@import sys; sys.stdout.buffer.write('1\r\n2\r3\n4'.encode('%s'))z-cr� r" r# rO r z1
2
3
4) � _bootlocaler� r% r& r r$ r' rh r ) r r� Zold_getpreferredencodingr� rj r �popenr# r r )r� r
�-test_universal_newlines_communicate_encodings� s
z=ProcessTestCase.test_universal_newlines_communicate_encodingsc
C s� t s d } n d } g } t j � } zFx� t | � D]� } y? t j j | t j � } | j t j
| t j t j B� � Wq7 t
k
r� } z | j t j k r� � n PWYd d } ~ Xq7 Xq7 W| j d | � x' t d � D] } t j | j � � q� Wxm t d � D]_ } t j t j d d g d t j d t j d
t j �} | j d � d } | j | d � q WWd x | D] } t j | � qwWt j | � Xd 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 ) � mswindowsrH �mkdtemp�ranger[ rr rs r ZTESTFN�append�open�O_WRONLY�O_CREAT�OSError�errnoZEMFILE�skipTestr+ �popr r$ r% r& r' rh r �shutilZrmtree)
r Z max_handlesZhandlesZtmpdir�iZtmpfile�er, �data�hr r r
�test_no_leaking� s: '
z ProcessTestCase.test_no_leakingc C s | j t j d d d g � d � | j t j d d d g � d � | j t j d d d g � d � | j t j d
d d g � d
� | j t j d d d g � d � | j t j d d d g � d � | j t j d d d g � d � | j t j d d g � 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 Z list2cmdline)r r r r
�test_list2cmdline� s z!ProcessTestCase.test_list2cmdlinec C s� t j t j d d g d t j �} | j | j j � | j | j � � t
j | j j � d � | j
� | j | j � d � d S)Nz-czimport os; os.read(0, 1)r" � Ar )r r$ r% r& r' rJ r" r+ � assertIsNone�pollr[ rK r� r r )r r, r r r
� test_poll� s
zProcessTestCase.test_pollc C sK t j t j d d g � } | 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 j t j d d g � } | j t j � � } | j d d � Wd QX| j d t | j � � | j
| j d d � d � d S)Nz-cz import time; time.sleep(0.3)r5 g-C ��6?z0.0001rU r ) r r$ r% r& r6 r7 r rA �strr< r )r r, r> r r r
�test_wait_timeout� s z!ProcessTestCase.test_wait_timeoutc C s6 | j t � �! t j t j d d g d � Wd QXd S)Nz-cr� r� )r6 r` r r$ r% r& )r r r r
�test_invalid_bufsize s z$ProcessTestCase.test_invalid_bufsizec C so t j t j d d g d � } | j | j � d � t j t j d d g d d �} | j | j � d � d S)Nz-cr� r r. )r r$ r% r& r r )r r, r r r
�test_bufsize_is_none s !z$ProcessTestCase.test_bufsize_is_nonec C s� t j t j d d g d t j d t j d t j d d d | ��k } | j j | � t j | j j
� � | j j � } y | j j � Wn t
k
r� Yn Xd | _ Wd QX| j | j d � | j | | � d S)
Nz-czDimport sys;sys.stdout.write(sys.stdin.readline());sys.stdout.flush()r" r# r r. rZ r� r )r r$ r% r& r' r� r" rK r[ r+ r� r# r� r� r r= )r �liner r� r, Z read_liner r r
�_test_bufsize_equal_one s
z'ProcessTestCase._test_bufsize_equal_onec C s d } | j | | d d �d S)Nzline
r� T)r� )r r� r r r
� test_bufsize_equal_one_text_mode# s z0ProcessTestCase.test_bufsize_equal_one_text_modec C s- d t j j � } | j | d d d �d S)Ns liner� r� F)r[ �linesepr� r� )r r� r r r
�"test_bufsize_equal_one_binary_mode) s z2ProcessTestCase.test_bufsize_equal_one_binary_modec C s xx t d � D]j } | j t � �) } t j d g d t j d t j �Wd QX| j j t j t j f k r
| j � q
q
Wd S)Ni �nonexisting_i_hoper# r )
r� r6 r� r r$ r' r< r� �ENOENTZEACCES)r r� r> r r r
�test_leaking_fds_on_error/ s z)ProcessTestCase.test_leaking_fds_on_errorzthreading requiredc s� g � � f d d � } t j d | � } | j � zE | j t � �0 t j d g d t j d t j d t j �Wd QXWd | j � d } xM � D]E } y t j
| � Wq� t k
r� } z | } WYd d } ~ Xq� Xq� W| d k r� | � n Xd S)Nc s; x4 t d � D]& } � j t j � � t j d � q
Wd S)N� g����MbP?)r� �extendr[ r� �time�sleep)r� )�fdsr r
�open_fdsB s z<ProcessTestCase.test_double_close_on_error.<locals>.open_fds�targetr� r" r# r ) � threadingZThread�startr6 �EnvironmentErrorr r$ r' rs 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 j t j d d g � � | j � j d � g � � � f d d � } t j d | � } | j � t rt d } n d } � j
d d
� | j � j | d d �� j
� | j � j | d d
�| j � | j d d | f d | f g � � d S)z?Issue21291: Popen.wait() needs to be threadsafe for returncode.z-czimport time; time.sleep(12)Nc sg � j d � j � f � � j � � j � � j d � j f � � j � � j d � j f � d S)Nzthread-start-poll-resultzthread-after-kill-and-waitzthread-after-second-wait)r� r� �killr r= r )�proc�resultsr r
�kill_proc_timer_threadd s
zDProcessTestCase.test_threadsafe_wait.<locals>.kill_proc_timer_threadg�������?rZ � r5 r r z*unexpected result in wait from main threadz&unexpected result in second main wait.�thread-start-poll-resultzthread-after-kill-and-waitzthread-after-second-waiti����)r N) r r$ r% r& r r= r ZTimerr r� r rs )r r r Zexpected_errorcoder )r r r
�test_threadsafe_wait\ s*
z$ProcessTestCase.test_threadsafe_waitc C sM d j d � } t j t j d | g � } | j | j d � t | � � d S)N�;�import subprocess, sys�Jretcode = subprocess.call([sys.executable, '-c', 'print("Hello World!")'])�assert retcode == 0z-cs Hello World!)r r r )rs r r@ r% r& r� r� �ascii)r rj rB r r r
�test_issue8780� s z ProcessTestCase.test_issue8780c C s t j � \ } } t j � \ } } t j � \ } } y& t j d g d | d | d | �Wn` t k
r� t j | � t j | � t j | � t j | � t j | � t j | � Yn X| j t j j
| � � | j t j j
| � � | j t j j
| � � d S)N�*r" r# r ) rH �mkstempr r$ r� r[ r+ �remover rr �exists)r ZifhandleZifnameZofhandleZofnameZefhandleZefnamer r r
� test_handles_closed_on_exception� s
z0ProcessTestCase.test_handles_closed_on_exceptionc C s� t j t j d d g d t j d t j d t j �} | j | j j � | j | j j � | j | j j � | j
d d � d S)
Nz-cr� r" r# r � x� r i ) r r$ r% r& r' rJ r# r+ r r" rh )r r, r r r
�test_communicate_epipe� s z&ProcessTestCase.test_communicate_epipec C sV t j t j d d g d t j �} | j | j j � | j � | j d d � d S)Nz-cr� r" r! r" r i )
r r$ r% r& r' rJ r" r+ r rh )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 s� d d � } t j t j | � } | j t j t j | � t j d d g } xC d D]; } i t j | 6} t j | | � � } | j � Wd QXqS Wd S)Nc S s d S)Nr )ZsignumZframer r r
�handler� s z7ProcessTestCase.test_communicate_eintr.<locals>.handlerz-cz7import os, signal;os.kill(os.getppid(), signal.SIGUSR1)r# r )zstdoutzstderr) �signalr% rJ r% r& r r' r$ rh )r r' � old_handlerr �stream�kwZprocessr r r
�test_communicate_eintr� s
z&ProcessTestCase.test_communicate_eintrz /proc/%d/fdzLinux specificc C s� d t j � } t j | � } | j t � �6 t t j d d g d t j d t j d t j �Wd QXt j | � } | j
| | � d S)z;Test for the fork() failure fd leak reported in issue16327.z /proc/%d/fdz-cr� r" r# r N) r[ �getpid�listdirr6 r r r% r& r r' r )r Z fd_directoryZfds_before_popenZfds_after_exceptionr r r
�!test_failed_child_execute_fd_leak� s z1ProcessTestCase.test_failed_child_execute_fd_leak)kr r r r- r0 r4 r8 r: r? rC rE rG rN rP rS rT rV r_ rf rg rm rn rp ru rw �unittest�skipIfr� ry 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� �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 r# r$ �
skipUnless�hasattrr( r[ r, rr �isdirr- r/ r r r r
r! J s�
# '
$ $5 r! zPOSIX specific testsc s� e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z Gd d � d e
j � Z e j e j j d � d � d d � � Z d d � Z e j e j d k d � d d � � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z e j e j j! d� � d1 � d2 d3 � � Z" d4 d5 � Z# d6 d7 � Z$ d8 d9 � Z% d: d; � Z& d< d= � Z' d> d? � Z( d@ dA � Z) dB dC � Z* dD dE � Z+ dF dG � Z, dH dI � Z- dJ dK � Z. dL dM � Z/ dN dO � Z0 dP dQ � Z1 dR dS � Z2 dT dU � Z3 dV dW � Z4 dX dY � Z5 dZ d[ � Z6 d\ d] � Z7 d^ d_ � Z8 d` da � Z9 db dc � Z: dd de � Z; df dg � Z< e j e j j! dh � o�e j= di � j> e j= dj � j> k dk � dl dm � � Z? e@ jA dn do � dp dq � � ZB dr ds � ZC dt du � ZD dv dw � ZE dx dy � ZF dz d{ � ZG d| d} � ZH d~ d � ZI d� d� � ZJ d� d� � ZK e@ jL d� d� � � ZM � S)��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 sv y t j | j � WnG t k
r] } z' | } | j d t | j � 7_ WYd d } ~ Xn X| j d | j � | S)Nz: z,chdir to nonexistant directory %s succeeded.)r[ �chdirr= r� �strerror�reprrR )r r� �desired_exceptionr r r
�_get_chdir_exception� s / z)POSIXProcessTestCase._get_chdir_exceptionc C s� | j � } y( t j t j d d g d | j �} WnQ t k
r� } z1 | j | j | j � | j | j | j � WYd d } ~ Xn X| j
d | � d S)z;Test error in the child raised in the parent for a bad cwd.z-cr r{ NzExpected OSError: %s) rC r r$ r% r& r= r� r r� r@ rR )r rB r, r� r r r
�test_exception_cwd s )z'POSIXProcessTestCase.test_exception_cwdc C s� | j � } y( t j t j d d g d | j �} WnQ t k
r� } z1 | j | j | j � | j | j | j � WYd d } ~ Xn X| j
d | � d S)zBTest error in the child raised in the parent for a bad executable.z-cr r& NzExpected OSError: %s) rC r r$ r% r& r= r� r r� r@ rR )r rB r, r� r r r
� test_exception_bad_executable s )z2POSIXProcessTestCase.test_exception_bad_executablec C s� | j � } y t j | j d d g � } WnQ t k
r~ } z1 | j | j | j � | j | j | j � WYd d } ~ Xn X| j d | � d S)z?Test error in the child raised in the parent for a bad args[0].z-cr NzExpected OSError: %s) rC r r$ r= r� r r� r@ rR )r rB r, r� r r r
�test_exception_bad_args_0 s )z.POSIXProcessTestCase.test_exception_bad_args_0c C sB t j t j d d g d d �t j t j d d g d d �d S)Nz-cr Zrestore_signalsTF)r r2 r% r& )r r r r
�test_restore_signals, s z)POSIXProcessTestCase.test_restore_signalsc C s� y% t j t j d d g d d �} Wn= t k
rd } z | j t j k rR � n WYd d } ~ Xn2 Xt j t j � � } t
| � } | j | | � d S)Nz-cz)import os; print(os.getpgid(os.getpid()))Zstart_new_sessionT) r r@ r% r& r� r� ZEPERMr[ �getpgidr- �int�assertNotEqual)r rB r� Z parent_pgidZ
child_pgidr r r
�test_start_new_session3 s
z+POSIXProcessTestCase.test_start_new_sessionc C sS t j � �* t j t j d d g � } | j � Wd QX| j | j t j
� d S)Nz-czimport os; os.abort()) r ZSuppressCrashReportr r$ r% r& r r r= r( �SIGABRT)r r, r r r
�test_run_abortD s
z#POSIXProcessTestCase.test_run_abortc C s` t j t j d d g d t j d d d � �} | j | j j � | j | j j � d � d S)Nz-cz2import sys,os;sys.stdout.write(os.getenv("FRUIT"))r# �
preexec_fnc S s t j d d � S)NrX �apple)r[ �putenvr r r r
�<lambda>S s z3POSIXProcessTestCase.test_preexec.<locals>.<lambda>s apple)
r r$ r% r& r' rJ r# r+ r r� )r r, r r r
� test_preexecL s z!POSIXProcessTestCase.test_preexecc C s� d d � } y% t j t j d d g d | �} Wnv t j k
rn } z | j t j d � WYd d } ~ XnI t k
r� } z | j d | j d � WYd d } ~ Xn X| j
d � d S)
Nc S s t d � � d S)Nz'What if two swallows carried a coconut?)rQ r r r r
�raise_itX s z=POSIXProcessTestCase.test_preexec_exception.<locals>.raise_itz-cr rN 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& �SubprocessErrorr� �_posixsubprocessrQ rA r rR )r rS r, r� r r r
�test_preexec_exceptionW s
*z+POSIXProcessTestCase.test_preexec_exceptionc @ s. e Z d Z d Z d d � Z d d � Z d S)z+POSIXProcessTestCase._TestExecuteChildPopenz3Used to test behavior at the end of _execute_child.c O s# | | _ t j j | | | � d S)N)� _testcaser r$ ra )r Ztestcaser r r r r
ra i s z4POSIXProcessTestCase._TestExecuteChildPopen.__init__c O s� z t j j | | | � Wd d d � t d � D� } zR xK | D]C } | j j | | j j � | j j � | j j � f d d �q@ WWd x | D] } t
j | � q� WXXd S)Nc S s% g | ] } t j d t j � � q S)z /dev/zero)r[ r� �O_RDONLY)�.0r� r r r
�
<listcomp>t s zNPOSIXProcessTestCase._TestExecuteChildPopen._execute_child.<locals>.<listcomp>� r z!At least one fd was closed early.) r r$ r r� rW � assertNotInr" r� r# r r[ r+ )r r r Z devzero_fdsr
r r r
r m s
z:POSIXProcessTestCase._TestExecuteChildPopen._execute_childN)r r r r ra r r r r r
�_TestExecuteChildPopeng s r] z /dev/zeroz/dev/zero required.c C sf d d � } | j t j � �B | j | t j d d g d t j d t j d t j d | �Wd QXd S)
z6Issue16140: Don't double close pipes on preexec error.c S s t j d � � d S)Nz-force the _execute_child() errpipe_data path.)r rT r r r r
rS � s zWPOSIXProcessTestCase.test_preexec_errpipe_does_not_double_close_pipes.<locals>.raise_itz-cr� r" r# r rN N)r6 r rT r] r% r& r' )r rS r r r
�0test_preexec_errpipe_does_not_double_close_pipes� s zEPOSIXProcessTestCase.test_preexec_errpipe_does_not_double_close_pipesc
C sv d d � } t j � } t j } t j } z t j � | j t j � � t j t j d d g d d d � �| j t j � d � t j � | j t j � � t j t j d d g d d d � �| j t j � d
� | t _ | j
t t j t j d d g d d d � �t ` | j
t
t j t j d d g d d d � �Wd | t _ | t _ | sqt j � n Xd 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 rN c S s d S)Nr r r r r
rQ � s zEPOSIXProcessTestCase.test_preexec_gc_module_failure.<locals>.<lambda>z#Popen enabled gc when it shouldn't.c S s d S)Nr r r r r
rQ � s zPopen left gc disabled.c S s d S)Nr r r r r
rQ � s c S s d S)Nr r r r r
rQ � s )�gc� isenabled�disabler r r2 r% r& �enabler� r6 r_ r$ �AttributeError)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� y d d l m } m } m } Wn2 t k
rT } z | j | � WYd d } ~ Xn X| | � } | \ } } | | d | f � | j | | | � | j t � �* t j
t j d d g d d d � �Wd QXd S)Nr )� getrlimit� setrlimit� RLIMIT_NPROCz-cr rN c S s d S)Nr r r r r
rQ � s z@POSIXProcessTestCase.test_preexec_fork_failure.<locals>.<lambda>)
Zresourcerh ri rj � ImportErrorr� rJ r6 �BlockingIOErrorr r2 r% r& )r rh ri rj rl Zlimitsr� Zhardr r r
�test_preexec_fork_failure� s z.POSIXProcessTestCase.test_preexec_fork_failurec
C s� t j � \ } } t | d d d ��( } | j d � | j d t j � Wd QXt j | d � t j | � } | j
� t j | � | j | j
d � d S)N�w�errors�surrogateescapez
#!/bin/sh
z(exec '%s' -c 'import sys; sys.exit(47)'
i� r1 )rH r r� rK r% 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 sT | j t t j t j d d g d d �| j t t j t j d d g d d �d S)Nz-czimport sys; sys.exit(47)� startupinfor1 �
creationflags)r6 rQ r r2 r% r& )r r r r
rf � s z&POSIXProcessTestCase.test_invalid_argsc C sy t j j � } d | d <t j d g d d d t j d | �} | j | j j � | j | j j
� j d � d � d S)
NrO rX z echo $FRUITrx rZ r# rY s
s apple) r[ r\ r] r r$ r' rJ r# r+ r r� r )r r^ r, r r r
�test_shell_sequence� s
z(POSIXProcessTestCase.test_shell_sequencec C sv t j j � } d | d <t j d d d d t j d | �} | j | j j � | j | j j
� j d � d � d S)
NrO rX z echo $FRUITrx rZ r# rY s
s apple) r[ r\ r] r r$ r' rJ r# r+ r r� r )r r^ r, r r r
�test_shell_string� s
z&POSIXProcessTestCase.test_shell_stringc
C s� t j � \ } } t | d d d ��( } | j d � | j d t j � Wd QXt j | d � t j | � } t j
| � | j | d � d S)Nrn ro rp z
#!/bin/sh
z(exec '%s' -c 'import sys; sys.exit(47)'
i� r1 ) rH r r� rK r% r& r[ rq r r2 r r )r r
rr rs r3 r r r
�test_call_string� s
z%POSIXProcessTestCase.test_call_stringc C s3 g } xe d d d g D]T } xK d d g D]= } t j j | | � } t j j | � r) | j | � q) q) Wq W| s� | j d � n d } t j j | � r� t j j | � r� | j | � n xm | D]e } t j d d | d
d d t j �} | j
| j j � | j
| j j � j � t | d
� � q� Wd S)Nz/binz /usr/bin/z/usr/local/binZbashZkshz"bash or ksh required for this testz/bin/shzecho $0r& rx Tr# r )r[ rr rs �isfiler� r� �islinkr r$ r' rJ r# r+ r r� r �bytes)r Zshellsr3 �nameZshr, r r r
�test_specific_shell� s %
z(POSIXProcessTestCase.test_specific_shellc G s� t j t j t j � } z@ t j t j d d g d d d t j d t j d t j �} Wd t j t j | � X| j j d � t
| | � | � | S) Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
time.sleep(30)
� close_fdsTr" r# r rZ ) r( �SIGINT�default_int_handlerr r$ r% r& r' r# r� �getattr)r �methodr r) r, r r r
�
_kill_process s z"POSIXProcessTestCase._kill_process�netbsd�openbsdz"Due to known OS bug (issue #16762)c
G sz t j t j d d g d d d t j d t j d t j �} | j j d � t j d � t | | � | � | j
� d S) Nz-cz�if 1:
import sys, time
sys.stdout.write('x\n')
sys.stdout.flush()
r Tr" r# r rZ ) r r$ r% r& r' r# r� r r r� rh )r r� r r, r r r
�_kill_dead_process, s
z'POSIXProcessTestCase._kill_dead_processc C sQ | j d t j � } | j � \ } } | j d | � | j | j � d � d S)N� send_signals KeyboardInterruptr )r� r( r� rh rA rJ r )r r, r� r r r r
�test_send_signalC s z%POSIXProcessTestCase.test_send_signalc C sO | j d � } | j � \ } } | j | d � | j | j � t j � d S)Nr r� )r� rh r r r r( �SIGKILL)r r, r� r r r r
� test_killI s z POSIXProcessTestCase.test_killc C sO | j d � } | j � \ } } | j | d � | j | j � t j � d S)N� terminater� )r� rh r r r r( �SIGTERM)r r, r� r r r r
�test_terminateO s z#POSIXProcessTestCase.test_terminatec C s | j d t j � d S)Nr� )r� r( r� )r r r r
�test_send_signal_deadU s z*POSIXProcessTestCase.test_send_signal_deadc C s | j d � d S)Nr )r� )r r r r
�test_kill_deadY 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 sO g } xB | D]: } t j | � } t j | � } | j | | | f � q
W| S)N)r[ �get_inheritable�dupr� )r Zsave_fdsr r
� inheritable�savedr r r
� _save_fdsa s
z POSIXProcessTestCase._save_fdsc C sA x: | D]2 \ } } } t j | | d | �t j | � q Wd S)Nr� )r[ �dup2r+ )r r r
r� r� r r r
� _restore_fdsi s z!POSIXProcessTestCase._restore_fdsc C s� d } | j | � } x- | D]% \ } } } | d k r | } Pq q Wz� x | D] } t j | � qO Wt j t j d d g d | d t j d t j �j � \ } } t j
| � } | j | | f d � Wd | j | � Xd 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' rh r r r r� ) r r r" � saved_fdsr
r� r� rk rl r r r
�check_close_std_fdsn s"
z(POSIXProcessTestCase.check_close_std_fdsc C s | j d g � d S)Nr )r� )r r r r
�test_close_fd_0� s z$POSIXProcessTestCase.test_close_fd_0c C s | j d g � d S)NrZ )r� )r r r r
�test_close_fd_1� s z$POSIXProcessTestCase.test_close_fd_1c C s | j d g � d S)Nr" )r� )r r r r
�test_close_fd_2� s z$POSIXProcessTestCase.test_close_fd_2c C s | j d d g � d S)Nr rZ )r� )r r r r
�test_close_fds_0_1� s z'POSIXProcessTestCase.test_close_fds_0_1c C s | j d d g � d S)Nr r" )r� )r r r r
�test_close_fds_0_2� s z'POSIXProcessTestCase.test_close_fds_0_2c C s | j d d g � d S)NrZ r" )r� )r r r r
�test_close_fds_1_2� s z'POSIXProcessTestCase.test_close_fds_1_2c C s | j d d d g � d S)Nr rZ r" )r� )r r r r
�test_close_fds_0_1_2� s z)POSIXProcessTestCase.test_close_fds_0_1_2c C s� t j d � } t j d � } z= t j d � t j d � t j t j d d g � j � Wd t j | d � t j | d � t j | � t j | � Xd S)z=Issue #15798: Popen should work when stdio fds are available.r rZ z-cz*print('AssertionError:0:CLOEXEC failure.')N) 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 s� d d � t d � D� } zyd d � | D� } x! | D] \ } } t j | � q6 Wt j | d d � t j | d d d � | j t d � � } zt x* t | � D] \ } } t j | | � q� Wt j t
j d d g d
| d d | d d
| d �} | j � Wd | j
| � Xx! | D] } t j | d d � q Wt j | d d � } t j t j | d d � � } | j | d � | j | d � Wd x | D] } t j | � q�WXd S)Nc S s g | ] } t j � � q Sr )rH r )rY r� r r r
rZ � s z?POSIXProcessTestCase.test_remapping_std_fds.<locals>.<listcomp>rU c S s g | ] \ } } | � q Sr r )rY r
rr r r r
rZ � s rZ s STDINr z-czZimport sys; got = sys.stdin.read();sys.stdout.write("got %s"%got); sys.stderr.write("err")r" r# r" r i s got STDINs err)r� r[ �unlinkrK r� r� � enumerater� r r$ r% r& r r� r� r r r r+ )
r �temps�temp_fdsr
rr r� �temp_fdr, rk rl r r r
�test_remapping_std_fds� s4
z+POSIXProcessTestCase.test_remapping_std_fdsc
C s� d d � t d � D� } d d � | D� } zJx! | D] \ } } t j | � q6 W| j t d � � } z� x* t | � D] \ } } t j | | � qx Wt j | d � t j | d d � t j t
j d d g d | d
| d | �}
|
j � x! | D] } t j | d d � q� Wt j
| d � } t j t j
| d � � } Wd | j | � X| j | d
� | j | d � Wd x | D] } t j | � q�WXd S)Nc S s g | ] } t j � � q Sr )rH r )rY r� r r r
rZ � s z7POSIXProcessTestCase.check_swap_fds.<locals>.<listcomp>rU c S s g | ] \ } } | � q Sr r )rY r
rr r r r
rZ � 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� rK r� r r$ r% r& r r� r r r� r r+ )
r Zstdin_noZ stdout_noZ stderr_nor� r� r
rr r� r� r, rk rl r r r
�check_swap_fds� s4
z#POSIXProcessTestCase.check_swap_fdsc C sv | j d d d � | j d d d � | j d d d � | j d d d � | j d d d � | j d d d � d S)Nr rZ r" )r� )r r r r
�
test_swap_fds s z"POSIXProcessTestCase.test_swap_fdsc C s� d d � } y# t j t j d d g d | �Wn� t k
r| } z+ | j t j � | j t | � d � WYd d } ~ Xn[ t j k
r� } z+ | j
t j � | j t | � d � WYd d } ~ Xn X| j d � d S) Nc S s t d � � d S)Nu
surrogate:�)rQ r r r r
�prepare s zCPOSIXProcessTestCase.test_surrogates_error_message.<locals>.preparez-cr� rN u
surrogate:�z!Exception occurred in preexec_fn.z1Expected ValueError or subprocess.SubprocessError) r r2 r% r& rQ r� rU r r� rT ZassertIsNotNonerR )r r� rl r r r
� test_surrogates_error_message s ()z2POSIXProcessTestCase.test_surrogates_error_messagec C sf x_d D]W\ } } | j d d � } d t | � } t j j � } | | | <d | d <t j j d
� r | j d d � } n | } t j
t j d | g d
| �} | j d � } | j
| j d � t | � � | j d d � } d t | � } t j j � } | | | <t j
t j d | g d
| �} | j d � } | j
| j d � t | � � q Wd S)N�test� abc�� test��42r rp z&import os; print(ascii(os.getenv(%s)))�C�LC_ALL�aix�latin1z-crY s
z'import os; print(ascii(os.getenvb(%s)))�ztestr� �r� r� )r� r� )r� rA r[ r\ r] r% r6 r� r� r r@ r& ri r r )r �key�valueZ
encoded_value�scriptrY Z
decoded_valuer# r r r
�test_undecodable_env+ s.
z)POSIXProcessTestCase.test_undecodable_envc C s2 t j t j � } t j j t j � \ } } t j | � } t j | d d g � } | j | d � d | d } t j | d d �} | j | d � t j j
� } | | d <t j | d d g d | �} | j | d � t j j
� } t j | � | d
<t j | d d g d | �} | j | d � d S) Nz-cr� r � 's ' -c passrx T�PATHrY s PATH) r[ �fsencoder% r& rr r~ r r2 r r\ r] �environb)r Z abs_programrr ZprogramZexitcoder� rY Zenvbr r r
�test_bytes_programN s
z'POSIXProcessTestCase.test_bytes_programc
C s# t j d d d �} t j d d d �} t j t j | g d t j d t j d t j d d �} | j | j d
� t j t j | g d t j d d �} | j � \ } } t t
t | j d � � � } t | j
j � | j j � | j j � g � } | j | | @d | | | @f � d S)
Nzinput_reader.py�subdir�subprocessdataz fd_status.pyr" r# r r Fr� � ,z5Expected no fds from %r to be open in child, found %r)r �findfiler r$ r% r& r' rJ rh �set�maprI r~ r" r� r# r r ) r Zsleeper� fd_status�p1�p2rB �errorZ
result_fdsZ unwanted_fdsr r r
�test_pipe_cloexech s
z&POSIXProcessTestCase.test_pipe_cloexecc
s� t j d d d �} t j d d d �} d } | d d } t j t j | g d t j d t j d
d �� t j t j | | g d � j d t j d
d �� | j � j � | j � j � � f d d
� } � f d d � } | j | � | j | � � j
j | � � j
j � t
j
� j g g g d � \ } } } | j | d � | j � j j � | � � j j � � j j � d S)Nzqcat.pyr� r� zqgrep.pys zxcvbnr� s
r" r# r Fc s' y � j � Wn t k
r" Yn Xd S)N)r� �ProcessLookupErrorr )r� r r
�kill_p1� s
zBPOSIXProcessTestCase.test_pipe_cloexec_real_tools.<locals>.kill_p1c s' y � j � Wn t k
r" Yn Xd 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# rJ r r" rK 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
sh t j d d d �} t j � } | j t j | d � | j t j | d � t | � � xL t d � D]> } t j t j t j
� } | j t j | � � j | � qh Wx � D] } t j | d � q� Wt
j t j | g d t
j d d
�} | j � \ } } t t t | j d � � � } | j | � @� d � t
j t j | g d t
j d d �} | j � \ } } t t t | j d � � � } | j | � @d
� | j d | d � t � f d d � t d � D� � } t
j t j | g d t
j d d d f �} | j � \ } } t t t | j d � � � } | j | | @� @d � | j d | d � d S)Nz fd_status.pyr� r� r rZ r Tr# r Fr� zSome fds were closedzSome fds were left openzSubprocess failedc 3 s | ] } � j � Vq d S)N)r� )rY r� )r r r
� <genexpr>� s z6POSIXProcessTestCase.test_close_fds.<locals>.<genexpr>r[ �pass_fdsz'Some fds not in pass_fds were left open)r r� r[ r� rJ r+ r� r� r� �devnullrX �add�set_inheritabler r$ r% r& r' rh r� rI r~ r r rA )
r r� r r� r
r, rB �ignored�
remaining_fdsZ 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 j d d d �} t j t j d t j d | � g d t j �} | j � \ } } | j
� } | j t | � d d d | �t
t t | d
j � j d � � � } t
t t | d j � j d � � � } | j | | @d d
�d S)z;Confirm that issue21618 is fixed (may fail under valgrind).z 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# r" r z(expected exactly two lines of output:
%rr r� rZ zSome fds were left open.N)r r� r r$ r% r& r� r� r' rh �
splitlinesr r� r� r� rI r r~ r )r r� r, rB Z
unused_stderrZ output_linesZ
opened_fdsr� r r r
�%test_close_fds_when_max_fd_is_lowered� s
0 ((z:POSIXProcessTestCase.test_close_fds_when_max_fd_is_loweredr� rD c C s� t j d d d �} t � } x� t d � D]u } t j � } | j t j | d � | j t j | d � t j | d d � t j | d d � | j | � q+ Wx| D]� } t
j t j
| g d t
j d d d
| f �} | j � \ } } t t t | j d � � � } | | h }
| j | | d � | j | |
@d
� | j t � �8 } | j t
j t j
d d g d d d
| f �� Wd QX| j d t | j � � q� Wd S)Nz fd_status.pyr� r� rD r rZ Tr# r r� r� zfd to be passed not passedzfd to be closed passedz-czimport sys; sys.exit(0)Fzoverriding close_fds)r r� r� r� r[ r� rJ r+ r� �updater r$ r% r& r' rh r� rI r~ rA r Z assertWarns�RuntimeWarningr2 r� Zwarning) r r� r r� r r
r, rB r� r� Z to_be_closed�contextr r r
�
test_pass_fds+ s0
z"POSIXProcessTestCase.test_pass_fdsc
C sH t j d d d �} t j � \ } } | j t j | � | j t j | � t j | d � t j | d � | | f } t j | g } | t t
t | � � 7} t j
| d t j d d d | �} | j � \ } } t t
t | j d � � � } | j | t | � d
| � | j t j | � d � | j t j | � d � d S) Nz fd_status.pyr� r� TFr# r r� r� z output=%a)r r� r[ r� rJ r+ r� r% r& �listr� r� r r$ r' rh r� rI r~ r r� )
r r� r� Znon_inheritabler� r r, rB r� r r r r
�test_pass_fds_inheritableM s" z.POSIXProcessTestCase.test_pass_fds_inheritablec
C sS t j t j d � �8 } t j t j d d g d | d | �} | j � Wd QXd 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_fdg s z:POSIXProcessTestCase.test_stdout_stdin_are_single_inout_fdc
C sS t j t j d � �8 } t j t j d d g d | d | �} | j � Wd QXd 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_fdm s z;POSIXProcessTestCase.test_stdout_stderr_are_single_inout_fdc
C sS t j t j d � �8 } t j t j d d g d | d | �} | j � Wd QXd 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_fds s z:POSIXProcessTestCase.test_stderr_stdin_are_single_inout_fdc C sx t j d d d �} t j t j | g d t j d t j �} | j � \ } } | j d | 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' rh r r= r� )r Zsigchild_ignorer, r# r r r r
� test_wait_when_sigchild_ignoredy s z4POSIXProcessTestCase.test_wait_when_sigchild_ignoredc
C s� t j d � } t j t j d d g d t j d d �} | j } | j | j � zF | j
| j d � d � | j | | j
| g g g d � d � Wd | j � Xd S)
Nr� z-cz$import sys;sys.stdout.write("apple")r# r. r r� s applg )r �
import_moduler r$ r% r& r' r# rJ r+ r r� rA r� r )r r� r, �fr r r
�test_select_unbuffered� s -z+POSIXProcessTestCase.test_select_unbufferedc C s� t j t j d d g d t j d t j �} | j | j j � | j | j j � t | � } | j
} ~ | j | d d � t j D� � d S)Nz-cz import sys, time;time.sleep(0.2)r# r c S s g | ] } t | � � q Sr )�id)rY �or r r
rZ � s zEPOSIXProcessTestCase.test_zombie_fast_process_del.<locals>.<listcomp>)
r r$ r% r& r' rJ r# r+ r r� �pidrA r
)r r, �identr� r r r
� test_zombie_fast_process_del� s z1POSIXProcessTestCase.test_zombie_fast_process_delc C s- t j t j d d g d t j d t j �} | j | j j � | j | j j � t | � } | j
} ~ t j | t
j � | j | d d � t j D� � t j d � | j t � �4 } t j d g d t j d t j �� } Wd QXWd QX| j t t j | d � | j | d
d � t j D� � d S) Nz-czimport time;time.sleep(3)r# r c S s g | ] } t | � � q Sr )r� )rY r� r r r
rZ � s zJPOSIXProcessTestCase.test_leak_fast_process_del_killed.<locals>.<listcomp>g�������?r� r c S s g | ] } t | � � q Sr )r� )rY r� r r r
rZ � s )r r$ r% r& r' rJ r# r+ r r� r� r[ r r( r� rA r
r r r6 r� �waitpidr\ )r r, r� r� r> r r r r
�!test_leak_fast_process_del_killed� s&
z6POSIXProcessTestCase.test_leak_fast_process_del_killedc
s� t j d d d �} t j d � � | j t j � � t j t j | g d t j
d d d � f d d
� �} | j � \ } } t t
t | j d � � � } | j � | � d S) Nz fd_status.pyr� r� rZ r# r TrN c s t j d � � S)NrZ )r[ r� r )r
r r
rQ � s zCPOSIXProcessTestCase.test_close_fds_after_preexec.<locals>.<lambda>r� )r r� r[ r� rJ r+ r r$ r% r& r' rh r� r� rI r~ r\ )r r� r, rB r� r� r )r
r
� test_close_fds_after_preexec� s z1POSIXProcessTestCase.test_close_fds_after_preexecc C s d d l } t j � } z� d d � } t j � d } x� d d g d d g f d g d d d g f d g d g d d g f d g d g d d f f D]e \ } } } } | j t � �B | j | | d g | | d d d d d
d d d
d d | � Wd QXq� WWd | st j � n Xd S)Nr c S s d S)Nr r r r r
rQ � s z5POSIXProcessTestCase.test_fork_exec.<locals>.<lambda>�exec�{ s exes envs argTrZ r" rU r� ���r� r� r� )rU ra rb rd r6 r` Z fork_execrc ) r rU Z
gc_enabled�funcZexecutable_listr Zexe_listr{ Zenv_listr r r
�test_fork_exec� s(
+ z#POSIXProcessTestCase.test_fork_exec)r� r� )Nr r r r rC rD rE rF rG rK rM rR rV r r$ r] r0 r1 r[ rr r r^ rf r% r6 rm rt rf rw rx ry 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� �stat�st_devr� r Zrequires_mac_verr� r� r� r� r� r� r� r� r� r� Z cpython_onlyr� r r )r> r
r; � s�
+ %!
- 0 # ) 0!Q " r; zWindows specific testsc @ s� e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d S) �Win32ProcessTestCasec C sM d } d } t j � } | | _ | | _ t j t j d d g d | �d S)NrZ rU z-czimport sys; sys.exit(0)ru )r Z STARTUPINFOZdwFlagsZ wShowWindowr2 r% r& )r ZSTARTF_USESHOWWINDOWZ SW_MAXIMIZEru r r r
�test_startupinfo� s z%Win32ProcessTestCase.test_startupinfoc C s4 d } t j j d � t j t j d d | �d S)N� z' a DOS box should flash briefly ...
z# -c "import time; time.sleep(0.25)"rv )r% r rK r r2 r& )r ZCREATE_NEW_CONSOLEr r r
�test_creationflags s
z'Win32ProcessTestCase.test_creationflagsc C sc | j t t j t j d d g d d d � �| j t t j t j d d g d t j d d �d S) Nz-czimport sys; sys.exit(47)rN c S s d S)NrZ r r r r r
rQ s z8Win32ProcessTestCase.test_invalid_args.<locals>.<lambda>r# r T)r6 rQ r r2 r% r& r' )r r r r
rf s
z&Win32ProcessTestCase.test_invalid_argsc C s5 t j t j d d g d d �} | j | d � d S)Nz-czimport sys; sys.exit(47)r Tr1 )r r2 r% r& r )r r3 r r r
r� s z#Win32ProcessTestCase.test_close_fdsc C sp t j j � } d | d <t j d g d d d t j d | �} | j | j j � | j d | j j
� � d S) N�physalisrX r� rx rZ r# rY s physalis) r[ r\ r] r r$ r' rJ r# r+ rA r� )r r^ r, r r r
rw ' s
z(Win32ProcessTestCase.test_shell_sequencec C sm t j j � } d | d <t j d d d d t j d | �} | j | j j � | j d | j j
� � d S) Nr� rX r� rx rZ r# rY s physalis) r[ r\ r] r r$ r' rJ r# r+ rA r� )r r^ r, r r r
rx 1 s
z&Win32ProcessTestCase.test_shell_stringc C s* t j t j d � } | j | d � d S)Nz -c "import sys; sys.exit(47)"r1 )r r2 r% r& r )r r3 r r r
ry ; s
z%Win32ProcessTestCase.test_call_stringc G s� t j t j d d g d t j d t j d t j �} | j | j j � | j | j j � | j | j j � | j j
d � t | | � | � | j � \ } } | j
| d � | j � } | 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 rZ r� r )r r$ r% r& r' rJ r# r+ r r" r� r� rh r r rJ )r r� r r, r� r r= r r r
r� A s z"Win32ProcessTestCase._kill_processc G s� t j t j d d g d t j d t j d t j �} | j | j j � | j | j j � | j | j j � | j j
d � t j d � t
| | � | � | j � \ } } | j | d � | j � } | 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 rZ r� �* )r r$ r% r& r' rJ r# r+ r r" r� r r r� rh r r r )r r� r r, r� r r3 r r r
r� X s
z'Win32ProcessTestCase._kill_dead_processc C s | j d t j � d S)Nr� )r� r( r� )r r r r
r� q s z%Win32ProcessTestCase.test_send_signalc C s | j d � d S)Nr )r� )r r r r
r� t s z Win32ProcessTestCase.test_killc C s | j d � d S)Nr� )r� )r r r r
r� w s z#Win32ProcessTestCase.test_terminatec C s | j d t j � d S)Nr� )r� r( r� )r r r r
r� z s z*Win32ProcessTestCase.test_send_signal_deadc C s | j d � d S)Nr )r� )r r r r
r� } s z#Win32ProcessTestCase.test_kill_deadc C s | j d � d S)Nr� )r� )r r r r
r� � s z(Win32ProcessTestCase.test_terminate_deadN)r r r r� r� rf r� rw rx ry r� r� r� r� r� r� r� r� r r r r
r� � s
r� c @ s e Z d Z d d � Z d S)� CommandTestsc C s� | j t j d � d � | j t j d � d � d } zZ t j � } t j j | d � } t j t rn d n d | � \ } } | j
| d � Wd | d k r� t j | � n Xd S)Nz
echo xyzzy�xyzzyr Zfooztype zcat )r r ) r r Z getoutputZgetstatusoutputrH r� r[ rr rs r� rJ �rmdir)r �dirr} ZstatusrB r r r
�test_getoutput� s zCommandTests.test_getoutputN)r r r r r r r r
r� � s r� Z PollSelectorz!Test needs selectors.PollSelectorc @ s( e Z d Z d d � Z d d � Z d S)�ProcessTestCaseNoPollc C s) t j | _ t j t _ t j | � d S)N)r �_PopenSelector�
orig_selector� selectorsZSelectSelectorr! r )r r r r
r � s zProcessTestCaseNoPoll.setUpc C s | j t _ t j | � 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 c @ s. e Z d Z e j e d � d d � � Z d S)�HelperFunctionTestsz(errno and EINTR make no sense on windowsc s� g � � f d d � } | j d t j | d d � � | j d g � � | j d t j | d � � | j d d
d g � � d S) Nc sD � j | � t � � d k r4 t t j d � � n t t | � � S)Nr" z fake interrupted system call)r� r� r� r� ZEINTR�tuple�reversed)r )� record_callsr r
� fake_os_func� s
z?HelperFunctionTests.test_eintr_retry_call.<locals>.fake_os_func�� r� � )r
r� )r� r
)r )r� r
)r )r )r r Z_eintr_retry_call)r r r )r r
�test_eintr_retry_call� s z)HelperFunctionTests.test_eintr_retry_callN)r r r r0 r1 r� r r r r r
r � s r zWindows-specific testsc sp e Z d Z � f d d � Z � f d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z � S)�CommandsWithSpacesc sU t � j � t j d d � \ } } | j � | _ t j | d � t j | � d S)Nz.pyzte stsV import sys;sys.stdout.write('%d %s' % (len(sys.argv), [a.lower () for a in sys.argv]))) r< r rH r �lowerrr r[ rK r+ )r r� rr )r> r r
r � s
zCommandsWithSpaces.setUpc s! t j | j � t � j � d S)N)r[ r rr r< r )r )r> r r
r � s zCommandsWithSpaces.tearDownc O s_ t j | d <t j | | � } | j | j j � | j | j j � j d � d | j � d S)Nr# �mbcsz2 [%r, 'ab cd'])
r r' r$ rJ r# r+ r r� r� rr )r r r r, r r r
� with_spaces� s
z CommandsWithSpaces.with_spacesc C s* | j d t j | j d f d d �d S)Nz"%s" "%s" "%s"zab cdrx rZ )r r% r& rr )r r r r
� test_shell_string_with_spaces� s z0CommandsWithSpaces.test_shell_string_with_spacesc C s& | j t j | j d g d d �d S)Nzab cdrx rZ )r r% r& rr )r r r r
� test_shell_sequence_with_spaces� s z2CommandsWithSpaces.test_shell_sequence_with_spacesc C s$ | j d t j | j d f � d S)Nz"%s" "%s" "%s"zab cd)r r% r& rr )r r r r
� test_noshell_string_with_spaces� s z2CommandsWithSpaces.test_noshell_string_with_spacesc C s | j t j | j d g � d S)Nzab cd)r r% r& rr )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 r r )r> r
r � s r c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �ContextManagerTestsc
C s� t j t j d d g d t j d t j ��9 } | j | j j � d � | j | j j � d � Wd QX| j
| j j � | 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 s; t j t j d d g � � } Wd QX| j | j d � d S)Nz-czimport sys; sys.exit(100)�d )r r$ r% r& r r= )r r r r r
�test_returncode� s z#ContextManagerTests.test_returncodec
C sQ t j t j d d g d t j ��' } | j d � | j | j d � Wd QXd S)Nz-cz2import sys;sys.exit(sys.stdin.read() == 'context')r" s contextrZ )r r$ r% r& r' rh r r= )r r r r r
r� � s
z*ContextManagerTests.test_communicate_stdinc C sI | j t � �4 } t j d g d t j d t j �� } Wd QXWd QXd S)Nr� r# r )r6 rv r r$ r' )r r> r r r r
rf
s
z%ContextManagerTests.test_invalid_argsc C s� t j t j d d g d t j d t j d �} | j � } | j j d t j � | j
| j � | j t
| j d d d � | j | j d � | j | j j � d S) z9Broken pipe error should not prevent wait() (Issue 21619)z-cr� r" r. r" r! Nr )r r$ r% r& r' r r� � __enter__r" rK r� r= r6 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 r r� rf r r r r r
r � s
r c C s9 t t t t t t t t f } t j | � t j
� d S)N) r! r; r� r� r r r r r Z run_unittestr )Z
unit_testsr r r
� test_main
s
r �__main__).r0 r� r r r r% r( r) Zlocaler[ r� rH r �rer r4 �warningsr� r� ra r� r rk r6 r� r� ZTestCaser � Exceptionr r$ r r! r1 r; r8 r� r� r9 r r r r r r �mainr r r r
�<module> sl
� � � � � � � � � � +6