Your IP : 216.73.216.213
��YfN � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z d d l m Z d d l m Z d d l m Z y d d l m
Z
Wn" e k
r� d d l m Z
Yn Xe j
d k r� d d l m Z e j d
d g Z e j d
d j d
d d f � g Z Gd d � d e j � Z Gd d � d e j � Z Gd d � d � Z e j
d k r�Gd d � d e � Z Gd d � d e e j � Z Gd d � d e e j � Z n Gd d � d e e j � Z e d k r�e j � d S) � N)�mock)�base_subprocess)�
subprocess)�
test_utils)�support)� test_support�win32)� unix_eventsz-cz import time; time.sleep(3600)�;z
import sysz data = sys.stdin.buffer.read()z sys.stdout.buffer.write(data)c @ s e Z d Z d d � Z d S)�TestSubprocessTransportc O s7 t j � | _ d | j _ d | j _ d | j _ d S)N)r �Mock�_proc�stdin�stdout�stderr)�self�args�kwargs� r �F/opt/alt/python35/lib64/python3.5/test/test_asyncio/test_subprocess.py�_start s z TestSubprocessTransport._startN)�__name__�
__module__� __qualname__r r r r r r s r c s= e Z d Z � f d d � Z d d d � Z d d � Z � S)�SubprocessTransportTestsc s0 t � j � | j � | _ | j | j � d S)N)�super�setUpZ
new_test_loop�loop�set_event_loop)r )� __class__r r r % s
z SubprocessTransportTests.setUpNc C s[ t j � } d | j _ d | j _ t | j | d g d d d d d d | �} | | f S)NF�testr �waiter)r r �connection_madeZ
_is_coroutine�process_exitedr r )r r! �protocol� transportr r r �create_transport+ s z)SubprocessTransportTests.create_transportc C s@ t j d | j � } | j | � \ } } | j d � | j j | � | j | j � d � | j | j j
� | j | j j
� | j | j j
� | j | j j
d d � | j | j � � | j | j � | j | j � | j | j � | j t | j t j � | j t | j � | j t | j � | j � d S)Nr � r )N) �asyncioZFuturer r& Z_process_exited�run_until_complete� assertEqual�get_returncode�
assertTruer" �calledr# Zconnection_lost� call_args� assertFalseZ
is_closing� assertIsNoneZ_loopr
Z _protocol� assertRaises�ProcessLookupError� send_signal�signal�SIGTERM� terminate�kill�close)r r! r% r$ r r r �test_proc_exited4 s$
z)SubprocessTransportTests.test_proc_exited)r r r r r& r9 r r )r r r $ 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 e j
d
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 d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ S),�SubprocessMixinc s� t � t j � � f d d � � } | d � } t j | d d � j �} � j j | � \ } } � j | d � � j | d � d S)Nc 3 s� t j � d t j d t j d � j � Ed H} | j j | � | j j � Ed H| j j � | j j
� Ed H} | j � Ed H} | | f S)Nr r r ) r( �create_subprocess_execr �PIPEr r �write�drainr8 r �read�wait)�data�proc�exitcode)r r r r �runT s
z.SubprocessMixin.test_stdin_stdout.<locals>.runs some datag N@r r )� PROGRAM_CATr( � coroutine�wait_forr r) r* )r rD �taskrC r r )r r r �test_stdin_stdoutQ s z!SubprocessMixin.test_stdin_stdoutc s� t � t j � � f d d � � } | d � } t j | d d � j �} � j j | � \ } } � j | d � � j | d � d S)Nc 3 sV t j � d t j d t j d � j � Ed H} | j | � Ed H\ } } | j | f S)Nr r r )r( r; r r< r � communicate�
returncode)rA rB r r )r r r r rD o s z-SubprocessMixin.test_communicate.<locals>.runs some datag N@r r )rE r( rF rG r r) r* )r rD rH rC r r )r r r �test_communicatel s
z SubprocessMixin.test_communicatec C sV t j d d | j �} | j j | � } | j j | j � � } | j | d � d S)Nzexit 7r � )r( �create_subprocess_shellr r) r@ r* )r �createrB rC r r r �
test_shell s
zSubprocessMixin.test_shellc C s\ t j d d d d | j �} | j j | � } | j j | j � � } | j | d � d S)Nzexit 8Zstart_new_sessionTr � )r( rN r r) r@ r* )r rO rB rC r r r �test_start_new_session� s z&SubprocessMixin.test_start_new_sessionc C s� t } t j | d | j � } | j j | � } | j � | j j | j � � } t j d k rt | j | t
� n | j t j
| � d S)Nr r )�PROGRAM_BLOCKEDr( r; r r) r7 r@ �sys�platform�assertIsInstance�intr* r4 �SIGKILL)r r rO rB rK r r r � test_kill� s
zSubprocessMixin.test_killc C s� t } t j | d | j � } | j j | � } | j � | j j | j � � } t j d k rt | j | t
� n | j t j
| � d S)Nr r )rS r( r; r r) r6 r@ rT rU rV rW r* r4 r5 )r r rO rB rK r r r �test_terminate� s
z SubprocessMixin.test_terminateTz"Temporarily skipped (rhbz#1088233)c s� d } t j d | g } t j | d t j d � j � } � j j | � } t j � f d d � � } � j j | | � � } � j t
j | � d S)Nz<import time; print("sleeping", flush=True); time.sleep(3600)z-cr r c 3 sI | j j � Ed H} � j | d � | j t j � | j � Ed H} | S)Ns sleeping
)r �readliner* r3 r4 �SIGHUPr@ )rB �linerK )r r r r3 � s
z5SubprocessMixin.test_send_signal.<locals>.send_signal) rT �
executabler( r; r r< r r) rF r* r4 r\ )r �coder rO rB r3 rK r )r r �test_send_signal� s
z SubprocessMixin.test_send_signalc C sS d t j } t j t j d d d t j d | j �} | j j | � } | | f S)N� xz-c�passr r )
r Z
PIPE_MAX_SIZEr( r; rT r^ r r< r r) )r �
large_datarO rB r r r �prepare_broken_pipe_test� s
z(SubprocessMixin.prepare_broken_pipe_testc C s� | j � \ } } t j d d � � } | | | � } t j � �$ | j t t f | j j | � Wd QRX| j j | j
� � d S)Nc s s& | j j | � | j j � Ed Hd S)N)r r= r> )rB rA r r r � write_stdin� s z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) rd r( rF r �disable_loggerr1 �BrokenPipeError�ConnectionResetErrorr r) r@ )r rB rc re �coror r r �test_stdin_broken_pipe� s
z&SubprocessMixin.test_stdin_broken_pipec
C sY | j � \ } } t j � � | j j | j | � � Wd QRX| j j | j � � d S)N)rd r rf r r) rJ r@ )r rB rc r r r �#test_communicate_ignore_broken_pipe� s
z3SubprocessMixin.test_communicate_ignore_broken_pipec s� d � � d d � t j � � � f d d � � } � j j | � � \ } } � j | d � � � j | j j � � j | j j � d S)N�
� � c 3 s� d j d d � d f � } � j j � t j � f d d � � } | � j _ t j t j d | d t j j d t j j d
� d � j �Ed H} | j
j d � } | j � Ed H\ } } | | f S)
N�
z
import syszsys.stdout.write("x" * %s)zsys.stdout.flush()c ? sB � | | � Ed H\ } } t j � | _ t j � | _ | | f S)N)r r �
pause_reading�resume_reading)r �kwr% r$ )�connect_read_piper r �connect_read_pipe_mock� s z^SubprocessMixin.test_pause_reading.<locals>.test_pause_reading.<locals>.connect_read_pipe_mockz-cr r �limitr rn )
�joinr rs r( rF r; rT r^ r r< Z
_transportZget_pipe_transportrJ )r_ rt rB Zstdout_transportr r )ru r �size)rs r �test_pause_reading� s z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingra ) r( rF r r) r* r, rp r- rq )r rx r r% r )ru r rw r rx � s !$z"SubprocessMixin.test_pause_readingc sc t j � f d d � � } � j j | d � � \ } } � j | j � d � � j | d � d S)Nc 3 s� d } t j t j d | d t j j d t j j d t j j d d d � j �Ed H} | j | � Ed H\ } } | j � Ed H} | | f S) Nz5import sys; data = sys.stdin.read(); print(len(data))z-cr r r � close_fdsFr ) r( r; rT r^ r r< r rJ r@ )�messager_ rB r r rC )r r r � len_message s z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_messages abc� 3r )r( rF r r) r* �rstrip)r r{ �outputrC r )r r �test_stdin_not_inheritable s z*SubprocessMixin.test_stdin_not_inheritablec s` t j � f d d � � } � j j | � � \ } } � j | j � d � � j | d � d S)Nc 3 s� d } t j t j d | d t j j d t j j d t j j d d d � j �Ed H} | j d � Ed H\ } } | j � Ed H} | | f S)
Nz5import sys; data = sys.stdin.read(); print(len(data))z-cr r r ry Fr � ) r( r; rT r^ r r< r rJ r@ )r_ rB r r rC )r r r � empty_input$ s z5SubprocessMixin.test_empty_input.<locals>.empty_input� 0r )r( rF r r) r* r} )r r� r~ rC r )r r �test_empty_input# s z SubprocessMixin.test_empty_inputc s2 t j � f d d � � } � j j | � � d S)Nc 3 s� t j t d � j � Ed H} � j j | j � � } � j j | j � y
| Ed HWn t j k
rl Yn X| j � | j � | j � Ed Hd S)Nr )
r( r; rS r � create_taskr@ � call_soon�cancel�CancelledErrorr7 )rB rH )r r r � cancel_wait9 s
z=SubprocessMixin.test_cancel_process_wait.<locals>.cancel_wait)r( rF r r) )r r� r )r r �test_cancel_process_wait6 s z(SubprocessMixin.test_cancel_process_waitc sF t j � f d d � � } t j � � � j j | � � Wd QRXd S)Nc 3 sf t j t d � j � } � j j | � } � j j | j � y
| Ed HWn t j k
ra Yn Xd S)Nr )r( r; rS r r� r� r� r� )ri rH )r r r �cancel_make_transportQ s
zYSubprocessMixin.test_cancel_make_subprocess_transport_exec.<locals>.cancel_make_transport)r( rF r rf r r) )r r� r )r r �*test_cancel_make_subprocess_transport_execP s
z:SubprocessMixin.test_cancel_make_subprocess_transport_execc sV t j � f d d � � } t j � �( � j j | � � t j � j � Wd QRXd S)Nc 3 sf � j j t j t � } � j j | � } � j j | j � y
| Ed HWn t j k
ra Yn Xd S)N) r �subprocess_execr( �SubprocessProtocolrS r� r� r� r� )ri rH )r r r r� c s
zDSubprocessMixin.test_cancel_post_init.<locals>.cancel_make_transport)r( rF r rf r r) � run_briefly)r r� r )r r �test_cancel_post_initb s
z%SubprocessMixin.test_cancel_post_initc sx t j � f d d � � } t j � � � j j | � � \ } } Wd QRX� j | � � j | � t j � j � d S)Nc 3 s� � j j t j t � } | Ed H\ } } d � � � f d d � } | j d � } | j � | | _ | j � } | j � | j � Ed H| � f S)NFc s d � � � d S)NTr r )� kill_called� orig_killr r r7 } s zKSubprocessMixin.test_close_kill_running.<locals>.kill_running.<locals>.killr )
r r� r( r� rS �get_extra_infor7 r+ r8 Z_wait)rO r% r$ r7 rB rK )r )r� r� r � kill_runningv s
z=SubprocessMixin.test_close_kill_running.<locals>.kill_running) r( rF r rf r r) r0 r, r� )r r� rK �killedr )r r �test_close_kill_runningu s
"
z'SubprocessMixin.test_close_kill_runningc
s� t j � f d d � � } t j � �* � j j | � � } t j � j � Wd QRX| \ } } } � j | � � j | � � j | � t
j d k r� t � t
� r� t j � j j � d S)Nc 3 s� � j j t j t � } | Ed H\ } } | j d � } | j � | j � t j � | _ | j
� } | j � } | j � | | | j j
f S)Nr )r r� r( r� rS r� r7 r@ r r Zpollr+ r8 r- )rO r% r$ rB �proc_returncodeZtransport_returncode)r r r r� � s
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_runningr )r( rF r rf r r) r� ZassertIsNotNoner0 r/ rT rU �
isinstance�SubprocessFastWatcherTestsZget_child_watcherZ
_callbacks�clear)r r� �resultr� Ztransport_return_coder� r )r r � test_close_dont_kill_finished� s
z-SubprocessMixin.test_close_dont_kill_finishedc C s� t j d k r d } n d } t j | � �� } t } | | _ t j t j d d d | j �} t
j d d � �> } | j | � � | j j
| � Wd QRX| j | g � Wd QRXWd QRXd S) Nr zasyncio.windows_utils.Popenzsubprocess.Popenz-crb r �recordT)rT rU r Zpatch�ZeroDivisionErrorZ side_effectr( r; r^ r �warnings�catch_warningsr1 r) r* )r �target�popen�excrO Zwarnsr r r �test_popen_error� s z SubprocessMixin.test_popen_errorc s2 t j � f d d � � } � j j | � � d S)Nc 3 s� d j d d d d d g � } t j t j d | d t j j d � j �} | Ed H} x@ | j j d
� Ed H} | r� t j
d d � j �Ed HqY PqY Wd S) Nro z
import syszfor _ in range(64):z sys.stdout.write("x" * 4096)zsys.stdout.flush()z sys.exit(1)z-cr r i g333333�?) rv r( r; rT r^ r r< r r r? Zsleep)r_ ZfutZprocessrA )r r r �execute� s zDSubprocessMixin.test_read_stdout_after_process_exit.<locals>.execute)r( rF r r) )r r� r )r r �#test_read_stdout_after_process_exit� s z3SubprocessMixin.test_read_stdout_after_process_exitN)r r r rI rL rP rR rY rZ �unittestZskipIfr` rd rj rk rx r r� r� r� r� r� r� r� r� r r r r r: O s( . ) r: c s( e Z d Z d Z � f d d � Z � S)�SubprocessWatcherMixinNc sx t � j � t j � } | j � | _ | j | j � | j � } | j | j � | j | � | j
| j d � d S)N) r r r( Zget_event_loop_policyZnew_event_loopr r �WatcherZ attach_loopZset_child_watcherZ
addCleanup)r ZpolicyZwatcher)r r r r � s
z SubprocessWatcherMixin.setUp)r r r r� r r r )r r r� � s r� c @ s e Z d Z e j Z d S)�SubprocessSafeWatcherTestsN)r r r r ZSafeChildWatcherr� r r r r r� � s r� c @ s e Z d Z e j Z d S)r� N)r r r r ZFastChildWatcherr� r r r r r� � s r� c s" e Z d Z � f d d � Z � S)�SubprocessProactorTestsc s0 t � j � t j � | _ | j | j � d S)N)r r r( ZProactorEventLoopr r )r )r r r r s
z SubprocessProactorTests.setUp)r r r r r r )r r r� s r� �__main__) r4 rT r� r� r r( r r r r r � ImportErrorr rU r r^ rS rv rE ZBaseSubprocessTransportr ZTestCaser r: r� r� r� r� r �mainr r r r �<module> s@
+� �