9Mc@sdZeZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlmZmZddlmZddlmZdd lmZdd lmZmZdd lmZmZdd lmZdd lm Z m!Z!ddlm"Z"ddl#m$Z$e$jj% rMddlm&Z&ndefdYZ'de fdYZ(de(fdYZ)e*j+e)j,de(fdYZ-e*j+e-j,defdYZ.de/fdYZ0defdYZ1dS(s2 Tests for implementations of L{IReactorProcess}. iN(tTestCasetSkipTest(tReactorBuilder(tset(tmsgterr(tplatform(tFilePath(tutils(tIReactorProcesstIProcessTransport(tDeferredtsucceed(tProcessProtocol(t ProcessDonetProcessTerminated(t_signals(truntime(tprocesst _ShutdownCallbackProcessProtocolcBs eZdZdZdZRS(sh An L{IProcessProtocol} which fires a Deferred when the process it is associated with ends. cCs ||_dS(N(t whenFinished(tselfR((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt__init__#scCs|jjddS(N(RtcallbacktNone(Rtreason((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt processEnded's(t__name__t __module__t__doc__RR(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs tProcessTestsBuilderBasecBsheZdZegZdZdZeedddkrKde_ ndZ dZ dZ RS( s Base class for L{IReactorProcess} tests which defines some tests which can be applied to PTY or non-PTY uses of C{spawnProcess}. Subclasses are expected to set the C{usePTY} attribute to C{True} or C{False}. cst}t|}|jj|tjtjddgd|j}|jtj ||j fd|j dS(s L{IReactorProcess.spawnProcess} connects the protocol passed to it to a transport which provides L{IProcessTransport}. s-cttusePTYcs jS(N(tstop(tignored(treactor(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytNsN( R Rt buildReactort spawnProcesstsyst executableR t assertTrueR t providedByt addCallbackt runReactor(Rtendedtprotocolt transport((R#sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_processTransportInterface7s    cs|j|jrd}ni}tjfd}tjtj|t}jt |t j t j ddgd|jd|j dj s|jdng}|j|j|rdS|jfd |j|j|dS( s If, before the reactor is started with L{IReactorCore.run}, a process is started with L{IReactorProcess.spawnProcess} and terminates, the process is reaped once the reactor is started. csjdS(N(R(targs(tsignaled(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pythandleress-cRR tchildFDsixs,Timed out waiting for child process to exit.Ncs jS(N(R!(R"(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$s(R%R Rt threadingtEventtsignaltSIGCHLDR R&RR'R(twaittisSettfailR+tappendR,R)(RR4R3R-tresult((R#R2sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_spawnProcessEarlyIsReapedRs*     "   R8s;Platform lacks SIGCHLD, early-spawnProcess test can't work.csdttdtd}dtffdY}jjj|tjtjd|gdj fd}j |j t j fd j d S( s The C{reason} argument passed to L{IProcessProtocol.processExited} is a L{ProcessTerminated} instance if the child process exits with a signal. tTERMtSIGsEimport sys sys.stdout.write('x') sys.stdout.flush() sys.stdin.read() tExitercs8eZfdZdZfdZdZRS(cs(td||f|jjdS(NschildDataReceived(%d, %r)(RR/t signalProcess(Rtfdtdata(tsigName(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytchildDataReceivedscSstd|fdS(NschildConnectionLost(%d)(R(RRC((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytchildConnectionLostscs%td|fj|gdS(NsprocessExited(%r)(RR(RR(texited(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt processExitedscSstd|fdS(NsprocessEnded(%r)(R(RR((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs(RRRFRGRIR((RERH(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRAs  s-cR cs~|\}|jt|j}tjrTj|jdj|j dn&j|jj|j ddS(Ni( ttrapRtvalueRt isWindowstassertIdenticalR7Rt assertEqualtexitCode(t.0tfailureR(RtsigNum(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytcbExiteds   cs jS(N(R!(tign(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$sN(tgetattrR7R R R%tcallWhenRunningR&R'R(R R+t addErrbackRR,(RtsourceRARS((RERHRRRR#sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_processExitedWithSignals     cs|jddg}ttdd d k }jj}||krb| rbtd|ntjjdkrtdngfd}j ||j |j dgd S( s If a child process exits while a system call is in progress, the system call should not be interfered with. In particular, it should not fail with EINTR. Older versions of Twisted installed a SIGCHLD handler on POSIX without using the feature exposed by the SA_RESTART flag to sigaction(2). The most noticable problem this caused was for blocking reads and writes to sometimes fail with EINTR. t Glib2Reactort Gtk2Reactort siginterrupts(%s is not supported without siginterruptt_installHandlerUsingSignals@_signals._installHandlerUsingSignal doesn't support this featurecs[zItjdtjf}tjdtjf}j|jWdjXdS(Ns$%s -c "import time; time.sleep(0.1)"s1%s -c "import time; time.sleep(0.5); print 'Foo'"(tostpopenR'R(R<treadR!(tf1tf2(R#R=(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytfs  sFoo N( R%RUR7Rt __class__RRRtinstallHandlerRVR,RN(RtskippedReactorsthasSigInterrupttreactorClassNameRc((R#R=sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt'test_systemCallUninterruptedByChildExits     c sddlm}|dkr+tdntt}|jjjj}dd|jfddd f}fd jd tffd Y}|}j j |t j t j d ddj |gdjjdS(s A spawned process has only stdin, stdout and stderr open (file descriptor 3 is also reported as open, because of the call to 'os.listdir()'). i(t platformTypetposixs$Test only applies to POSIX platformss import sysssys.path.insert(0, '%s')s$from twisted.internet import processs-sys.stdout.write(str(process._listOpenFDs()))ssys.stdout.flush()csjd|dS(Ns [0, 1, 2, 3](t assertEquals(toutput(R(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt checkOutputstProtocolcs,eZdZdZfdZRS(cSs g|_dS(N(Rm(R((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRscSs|jj|dS(N(RmR<(RRD((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt outReceivedscs,zdj|jWdjXdS(NR(tjoinRmR!(RR(RnR#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs(RRRRpR((R#Rn(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRos  s-Wignores-cs R N(ttwisted.python.runtimeRjRRt__file__tparenttpathR%R RVR&R'R(RqR R,(RRjtherettopRXRotproto((R#RRnsF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_openFileDescriptorss&       N(RRRR trequiredInterfacesR0R>RUR7RtskipRYRiRy(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR,s   6  E .tProcessTestsBuildercBs}eZdZeZeejdjZ e j r?dZ ndZ dZ dZdZdZdZd ZRS( sp Builder defining tests relating to L{IReactorProcess} for child processes which do not have a PTY. sprocess_helper.pytwindowsRcstitd6td6td6dtffdY}d}|jjj|tjtjd|gd|jfd }j|fd }j|d }j|j t fd }j||j d S(s L{IProcessProtocol.childConnectionLost} is called each time a file descriptor associated with a child process is closed. iiitClosercs&eZfdZfdZRS(csj|dS(N(R(RR/(t connected(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytmakeConnectionFscs|jddS(N(RR(RtchildFD(tlost(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRGIs(RRRRG((RR(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR~Esswimport os, sys while 1: line = sys.stdin.readline().strip() if not line: break os.close(int(line)) s-cR cs'jddjfdS(Ns2 icsS(N((RT(R/(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$[s(twriteR+(R/(R(R/sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt cbConnectedYs cs'jddjfdS(Ns1 icsS(N((RT(R/(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$`s(RR+(R/(R(R/sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt lostSecond^s cSs|jddS(Ns (R(R/((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt lostFirstcscsjdS(N(R!(R"(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytcbEndedhsN( R R R%RVR&R'R(R R+RWRR,(RR~RXRRRR((R#RRsF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_childConnectionLost=s$ $       cstgdtffdY}jjj|tjtjjdjgdj fd}j |j t j fdj dS(s L{IProcessProtocol.processEnded} is called after the child process exits and L{IProcessProtocol.childConnectionLost} is called for each of its file descriptors. tEndercs8eZdZfdZdZfdZRS(cSs%td||f|jjdS(NschildDataReceived(%d, %r)(RR/tloseConnection(RRCRD((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRFzscs"td|fj|dS(NschildConnectionLost(%d)(RR<(RR(R(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRG~scSstd|fdS(NsprocessExited(%r)(R(RR((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRIscs%td|fj|gdS(NsprocessEnded(%r)(RR(RR(R-(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs(RRRFRGRIR((RR-(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRys  tchildR cs?|\}|jtjttdddgdS(Niii(RJRRNR(RPRQ(RR(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs cs jS(N(R!(RT(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$sN(R R R%RVR&R'R(tkeepStdioOpenProgramtkeepStdioOpenArgR R+RWRR,(RRR((R#RRR-sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_processEndedps      csttgdtffdY}jjj|tjtjjdjgdj fd}j |fd}j |j t j fdj dS( s L{IProcessProtocol.processExited} is called when the child process exits, even if file descriptors associated with the child are still open. tWaitercs2eZdZfdZfdZRS(cSstd||fdS(NschildDataReceived(%d, %r)(R(RRCRD((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRFscsDtd|fj|tdkr@jdndS(NschildConnectionLost(%d)i(RR<tlenRR(RR(RtallLost(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRGs cs2td|fj|g|jjdS(NsprocessExited(%r)(RRR/R(RR(RH(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRIs(RRRFRGRI((RRRH(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRs RR cs;|\}|jttdfjgS(NscbExited; lost = %s(RJRRRN(RPRQ(RRR(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRSs  cs)jttdddgdS(Niii(RNR(R"(RR(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt cbAllLostscs jS(N(R!(RT(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$sN(R R R%RVR&R'R(RRR R+RWRR,(RRRSR((R#RRRRHsF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyttest_processExiteds"  "      cCsU|j}t|d}|jtjj|tj|jtjj|S(sj Write the given list of lines to a text file and return the absolute path to it. twt( tmktemptfileRR^tlinesepRqtcloseRutabspath(Rt sourceLinestscriptt scriptFile((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytmakeSourceFiles    csdjdtjfddfdgtjdjfdfdfd }j|jd S( s Spawning a process with an executable which is a script starting with an interpreter definition line (#!) uses that interpreter to evaluate the script. sthis is the shebang outputs#!%ss import sysssys.stdout.write('%s')ssys.stdout.flush()icsZ|\}}}td|||fj|j|dj|ddS(NscbProcessExited((%r, %r, %d))Ri(RRN(RPtoutRtcode(RtSHEBANG_OUTPUT(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytcbProcessExiteds  csj|S(N(R!(t passthrough(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytshutdowns cs@tjd}|j|j|jtdS(NR#(RtgetProcessOutputAndValuetaddBothR+RWR(td(R#RRR(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytstarts  N(RR'R(R^tchmodR%RVR,(RR((R#RRRRRsF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt test_shebangs     csddddddgjdjtttdd jfd fd fd }j|jd S(sq Arguments given to spawnProcess are passed to the child process as originally intended. simport sys, os try: import msvcrt msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) except ImportError: pass for arg in sys.argv[1:]: sys.stdout.write(arg + chr(0)) sys.stdout.flush()thellot"s |<>^&s "\\"hello\\"s"foo\ bar baz\""Riics-|jd}|jj|dS(Nt(tsplittpopRl(Rm(RR1(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytprocessFinisheds csj|S(N(R!(R=(R#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR s csFtd}|jfd|j|jdS(Ncs#tjtjdgdS(Ns-cR#(RtgetProcessOutputR'R((tdummy(R#R1RX(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR$&s(R RR+R(R(R1RRXRR#(sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt spawnChild$s  N(R<RqtmaptchrtxrangeR%RVR,(RR((RR#RR1RXRsF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt test_processCommandLineArgumentss  (  (RRRtFalseR RRstsiblingRuRRRLRRRRRRR(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR|-s   3 . 2 %tPTYProcessTestsBuildercBsEeZdZeZejr'dZnejrCidd6Z nRS(si Builder defining tests relating to L{IReactorProcess} for child processes which have a PTY. s"PTYs are not supported on Windows.s#OS X's poll() does not support PTYss(twisted.internet.pollreactor.PollReactor( RRRtTrueR RRLR{tisMacOSXRf(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyR1s   tPotentialZombieWarningTestscBseZdZdZRS(sE Tests for L{twisted.internet.error.PotentialZombieWarning}. cCsvddlm}|j|j|jg}|j|ddt|j|ddd|jt|ddS( s Accessing L{PotentialZombieWarning} via the I{PotentialZombieWarning} attribute of L{twisted.internet.error} results in a deprecation warning being emitted. i(terroritcategorytmessagestwisted.internet.error.PotentialZombieWarning was deprecated in Twisted 10.0.0: There is no longer any potential for zombie process.iN(ttwisted.internetRtPotentialZombieWarningt flushWarningsttest_deprecatedRltDeprecationWarningR(RRtwarnings((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRFs (RRRR(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRBstFakeResourceModulecBseZdZdZdZRS(sf Fake version of L{resource} which hard-codes a particular rlimit for maximum open files. icCs ddgS(s Mocks just the getrlimit method of the resource module, simulating what happens on OS-X when a ridiculous number is returned. il((Rtno((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyt getrlimit`s(RRRt RLIMIT_NOFILER(((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pyRZstFDDetectorTestcBseZdZejjr$dZneZeZ eZ eZ dZ dZdZdZdZdZdZdZd Zd Zd Zd Zd ZdZdZRS(sj Tests for _FDDetector class in twisted.internet.process, which detects which function to drop in place for the _listOpenFDs method. @ivar devfs: A flag indicating whether the filesystem fake will indicate that /dev/fd exists. @ivar accurateDevFDResults: A flag indicating whether the /dev/fd fake returns accurate open file information. @ivar procfs: A flag indicating whether the filesystem fake will indicate that /proc//fd exists. @ivar openedFile: A flag indicating whether to include an extra file descriptor in the result of listing /dev/fd. sHProcess support does not need to detect open file descriptors on WindowscCsdS(s> Fake os.getpid, always return the same thing i{((R((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytgetpidscCs|dkr+|jr"dddgStn|dkr|jr|jsVdddgS|jroddddgSdddgSqtndS(s Fake os.listdir, depending on what mode we're in to simulate behaviour. @param arg: the directory to list s /proc/123/fdt0t1t2s/dev/fdt3N(tprocfstOSErrortdevfstaccurateDevFDResultst openedFile(Rtarg((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytlistdirs         cCs t|_dS(s This is a mock for L{open}. It just keeps track of the fact that a file has opened so an extra file descriptor is found in /dev/fd. N(RR(Rtfnametmode((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytopenfilescCs7yddl}Wntk r)d}nX||_dS(sF Save the resource module so we can restore it later. iN(tresourcet ImportErrorRtsavedResourceModule(RR((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytsaveResourceModules   cCs!dtjd<|j|jdS(sp Make the L{resource} module unimportable for the remainder of the current test method. RN(RR'tmodulest addCleanuptreplaceResourceModule(R((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pythideResourceModules cCs$ttjd<|j|jdS(sc Make a L{FakeResourceModule} instance importable at the L{resource} name. RN(RR'RRR(R((sF/usr/lib/python2.7/dist-packages/twisted/internet/test/test_process.pytrevealResourceModulescCs|jtjds80