Your IP : 216.73.216.213
�
f fNE � @ 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 n 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 � n 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/python34/lib64/python3.4/test/test_asyncio/test_subprocess.py�_start s z TestSubprocessTransport._startN)�__name__�
__module__� __qualname__r r r r r r s r c @ s7 e Z d Z d d � Z d d d � Z d d � Z d S)�SubprocessTransportTestsc C s# | j � | _ | j | j � d S)N)Z
new_test_loop�loop�set_event_loop)r r r r �setUp% 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_exited3 s$
z)SubprocessTransportTests.test_proc_exited)r r r r r$ r7 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' 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 � � Dd H} | j j | � | j j � Dd H| j j � | j j
� Dd H} | j � Dd H} | | f S)Nr r r ) r& �create_subprocess_execr �PIPEr r �write�drainr6 r �read�wait)�data�proc�exitcode)r r r r �runS s
z.SubprocessMixin.test_stdin_stdout.<locals>.runs some datag N@r r )� PROGRAM_CATr& � coroutine�wait_forr r' r( )r rB �taskrA r r )r r r �test_stdin_stdoutP 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 � � Dd H} | j | � Dd H\ } } | j | f S)Nr r r )r& r9 r r: r � communicate�
returncode)r? r@ r r )r r r r rB n s z-SubprocessMixin.test_communicate.<locals>.runs some datag N@r r )rC r& rD rE r r' r( )r rB rF rA r r )r r r �test_communicatek 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 �creater@ rA 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& rL r r' r> r( )r rM r@ rA 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& r9 r r' r5 r> �sys�platform�assertIsInstance�intr( r2 �SIGKILL)r r rM r@ rI 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 )rQ r& r9 r r' r4 r> rR rS rT rU r( r2 r3 )r r rM r@ rI 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 � Dd H} � j | d � | j t j � | j � Dd H} | S)Ns sleeping
)r �readliner( r1 r2 �SIGHUPr> )r@ �linerI )r r r r1 � s
z5SubprocessMixin.test_send_signal.<locals>.send_signal) rR �
executabler& r9 r r: r r' rD r( r2 rZ )r �coder rM r@ r1 rI 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& r9 rR r\ r r: r r' )r �
large_datarM r@ 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 QX| j j | j
� � d S)Nc s s& | j j | � | j j � Dd Hd S)N)r r; r< )r@ r? r r r � write_stdin� s z;SubprocessMixin.test_stdin_broken_pipe.<locals>.write_stdin) rb r& rD r �disable_loggerr/ �BrokenPipeError�ConnectionResetErrorr r' r> )r r@ ra rc �coror r r �test_stdin_broken_pipe� s
z&SubprocessMixin.test_stdin_broken_pipec
C sX | j � \ } } t j � � | j j | j | � � Wd QX| j j | j � � d S)N)rb r rd r r' rH r> )r r@ ra 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 �Dd H} | j
j d � } | j � Dd H\ } } | | f S)
N�
z
import syszsys.stdout.write("x" * %s)zsys.stdout.flush()c ? sB � | | � Dd 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 rl )
�joinr rq r& rD r9 rR r\ r r: Z
_transportZget_pipe_transportrH )r] rr r@ Zstdout_transportr r )rs r �size)rq r �test_pause_reading� s z>SubprocessMixin.test_pause_reading.<locals>.test_pause_readingr_ ) r& rD r r' r( r* rn r+ ro )r rv r r# r )rs r ru r rv � 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 �Dd H} | j | � Dd H\ } } | j � Dd H} | | f S) Nz5import sys; data = sys.stdin.read(); print(len(data))z-cr r r Z close_fdsFr ) r& r9 rR r\ r r: r rH r> )�messager] r@ r r rA )r r r � len_message s z?SubprocessMixin.test_stdin_not_inheritable.<locals>.len_messages abc� 3r )r& rD r r' r( �rstrip)r rx �outputrA r )r r �test_stdin_not_inheritable
s z*SubprocessMixin.test_stdin_not_inheritablec s2 t j � f d d � � } � j j | � � d S)Nc 3 s� t j d � j t � Dd H} � j j | j � � } � j j | j � y
| Dd HWn t j k
rl Yn X| j � | j � | j � Dd Hd S)Nr )
r& r9 r rQ � create_taskr> � call_soon�cancel�CancelledErrorr5 )r@ rF )r r r � cancel_wait% s
z=SubprocessMixin.test_cancel_process_wait.<locals>.cancel_wait)r& rD r r' )r r� r )r r �test_cancel_process_wait" s z(SubprocessMixin.test_cancel_process_waitc sE t j � f d d � � } t j � � � j j | � � Wd QXd S)Nc 3 sf t j d � j t � } � j j | � } � j j | j � y
| Dd HWn t j k
ra Yn Xd S)Nr )r& r9 r rQ r} r~ r r� )rg rF )r r r �cancel_make_transport= s
zYSubprocessMixin.test_cancel_make_subprocess_transport_exec.<locals>.cancel_make_transport)r& rD r rd r r' )r r� r )r r �*test_cancel_make_subprocess_transport_exec< s
z:SubprocessMixin.test_cancel_make_subprocess_transport_execc sU t j � f d d � � } t j � �( � j j | � � t j � j � Wd QXd S)Nc 3 sf � j j t j t � } � j j | � } � j j | j � y
| Dd HWn t j k
ra Yn Xd S)N) r �subprocess_execr& �SubprocessProtocolrQ r} r~ r r� )rg rF )r r r r� O s
zDSubprocessMixin.test_cancel_post_init.<locals>.cancel_make_transport)r& rD r rd r r' � run_briefly)r r� r )r r �test_cancel_post_initN s
z%SubprocessMixin.test_cancel_post_initc sw t j � f d d � � } t j � � � j j | � � \ } } Wd QX� j | � � j | � t j � j � d S)Nc 3 s� � j j t j t � } | Dd H\ } } d � � � f d d � } | j d � } | j � | | _ | j � } | j � | j � Dd H| � f S)NFc s d � � � d S)NTr r )� kill_called� orig_killr r r5 i s zKSubprocessMixin.test_close_kill_running.<locals>.kill_running.<locals>.killr )
r r� r& r� rQ �get_extra_infor5 r) r6 Z_wait)rM r# r" r5 r@ rI )r )r� r� r � kill_runningb s
z=SubprocessMixin.test_close_kill_running.<locals>.kill_running) r& rD r rd r r' r. r* r� )r r� rI �killedr )r r �test_close_kill_runninga s
!
z'SubprocessMixin.test_close_kill_runningc s� t j � f d d � � } t j � �* � j j | � � } t j � j � Wd QX| \ } } } � j | � � j | � � j | � d S)Nc 3 s� � j j t j t � } | Dd H\ } } | j d � } | j � | j � t j � | _ | j
� } | j � } | j � | | | j j
f S)Nr )r r� r& r� rQ r� r5 r> r r Zpollr) r6 r+ )rM r# r" r@ �proc_returncodeZtransport_returncode)r r r r� � s
zCSubprocessMixin.test_close_dont_kill_finished.<locals>.kill_running)
r& rD r rd r r' r� ZassertIsNotNoner. r- )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 QX| j | g � Wd QXWd QXd S) Nr zasyncio.windows_utils.Popenzsubprocess.Popenz-cr` r �recordT)rR rS r Zpatch�ZeroDivisionErrorZ side_effectr& r9 r\ r �warnings�catch_warningsr/ r' r( )r �target�popen�excrM Zwarnsr r r �test_popen_error� s z SubprocessMixin.test_popen_errorN)r r r rG rJ rN rP rW rX �unittestZskipIfr^ rb rh ri rv r| r� r� r� r� r� r� r r r r r8 N s$ . "r8 c @ s" e Z d Z d Z d d � Z d S)�SubprocessWatcherMixinNc C sk t j � } | j � | _ | j | j � | j � } | j | j � | j | � | j | j d � d S)N) r& Zget_event_loop_policyZnew_event_loopr r �WatcherZ attach_loopZset_child_watcherZ
addCleanup)r ZpolicyZwatcherr 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)�SubprocessFastWatcherTestsN)r r r r ZFastChildWatcherr� r r r r r� � s r� c @ s e Z d Z d d � Z d S)�SubprocessProactorTestsc C s# t j � | _ | j | j � d S)N)r& ZProactorEventLoopr r )r r r r r � s z SubprocessProactorTests.setUpN)r r r r r r r r r� � s r� �__main__) r2 rR r� r� r r& r r r r r � ImportErrorr rS r r\ rQ rt rC ZBaseSubprocessTransportr ZTestCaser r8 r� r� r� r� r �mainr r r r �<module> s@
*� g