U 2ܟj@5@sdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z ddl mZddlmZddlmZddlmZmZGd d d eZGd d d eZGd ddeZdS)zD Test running processes with the APIs in L{twisted.internet.utils}. N)skipIf)error interfacesreactorutils)Deferred)platform)SuppressedWarningsTests)SynchronousTestCaseTestCasec@seZdZdZeeddkr dZdZdZ e j Z ddZ ddZdd Zd d Zd d ZddZeedddZddZddZddZddZddZddZdd Zd!d"Zd#d$ZdS)%ProcessUtilsTestszt Test running a process using L{getProcessOutput}, L{getProcessValue}, and L{getProcessOutputAndValue}. Nz)reactor doesn't implement IReactorProcessc CsB|}t|d}|tj|tjW5QRXtj|S)zj Write the given list of lines to a text file and return the absolute path to it. wt)mktempopenwriteoslinesepjoinpathabspath)selfZ sourceLinesscript scriptFiler.cbFailed)rrrr r'IOErrorr!)rrr$r*rr)rtest_outputWithErrorIgnored?s   z-ProcessUtilsTests.test_outputWithErrorIgnoredcCs8|dddddg}tj|jd|gdd}||jd S) z If a C{True} value is supplied for the C{errortoo} parameter to L{getProcessOutput}, the returned L{Deferred} fires with the child's stderr output as well as its stdout output. rzsys.stdout.write("foo")sys.stdout.flush()zsys.stderr.write("foo")sys.stderr.flush()rT)Zerrortoosfoofoorr#rrrtest_outputWithErrorCollectedRs z/ProcessUtilsTests.test_outputWithErrorCollectedcCs,|dg}t|jd|g}||jdS)z| The L{Deferred} returned by L{getProcessValue} is fired with the exit status of the child process. zraise SystemExit(1)r)rrgetProcessValuer r!r"r#rrr test_valuegs zProcessUtilsTests.test_valuecs:ddddg}fdd}tjd|g}||S)a The L{Deferred} returned by L{getProcessOutputAndValue} fires with a three-tuple, the elements of which give the data written to the child's stdout, the data written to the child's stderr, and the exit status of the child. rz*sys.stdout.buffer.write(b'hello world!\n')z,sys.stderr.buffer.write(b'goodbye world!\n')z sys.exit(1)cs2|\}}}|d|d|ddS)Ns hello world! sgoodbye world! r0r"Z out_err_codeoutr(coder)rrgotOutputAndValues   z@ProcessUtilsTests.test_outputAndValue..gotOutputAndValuerrrgetProcessOutputAndValuer r!rrr7r$rr)rtest_outputAndValueqs z%ProcessUtilsTests.test_outputAndValuez"Windows doesn't have real signals.csJddddddg}fdd}tjd |g}|t}||S) z If the child process exits because of a signal, the L{Deferred} returned by L{getProcessOutputAndValue} fires a L{Failure} of a tuple containing the child's stdout, stderr, and the signal which caused it to exit. zimport sys, os, signalz"sys.stdout.write('stdout bytes\n')z"sys.stderr.write('stderr bytes\n')r-r.z$os.kill(os.getpid(), signal.SIGKILL)cs4|\}}}|d|d|tjdS)Ns stdout bytes s stderr bytes )r"signalSIGKILL)Z out_err_sigr5r(sigr)rrr7s   z>ProcessUtilsTests.test_outputSignal..gotOutputAndValuer)rrr9r r'tupler!r:rr)rtest_outputSignals   z#ProcessUtilsTests.test_outputSignalcCsVtj|}t||ddg}||jd|g|d}|||t |S)Nimport os, syszsys.stdout.write(os.getcwd())r)r) rrrrmakedirsrr r!encodesysgetfilesystemencoding)rutilFunccheckdirrr$rrr _pathTests zProcessUtilsTests._pathTestcCs|tj|jS)z L{getProcessOutput} runs the given command with the working directory given by the C{path} parameter. )rIrrr"r)rrrtest_getProcessOutputPathsz+ProcessUtilsTests.test_getProcessOutputPathcsfdd}tj|S)z~ L{getProcessValue} runs the given command with the working directory given by the C{path} parameter. cs|ddSNrr3resultZignoredr)rrrGsz9ProcessUtilsTests.test_getProcessValuePath..check)rIrr1rrGrr)rtest_getProcessValuePaths z*ProcessUtilsTests.test_getProcessValuePathcsfdd}tj|S)z L{getProcessOutputAndValue} runs the given command with the working directory given by the C{path} parameter. cs&|\}}}|||ddSrKr3Zout_err_statusrHr5r(statusr)rrrGs  zBProcessUtilsTests.test_getProcessOutputAndValuePath..check)rIrr9rNrr)r!test_getProcessOutputAndValuePaths z3ProcessUtilsTests.test_getProcessOutputAndValuePathcCstj|}t||dddg}|tjtt|t t dj }t |t j t jB|tj ||||jd|g}|||t|S)NrAzcdir = os.getcwd()zsys.stdout.write(cdir).r)rrrrrBrZ addCleanupchdirgetcwdstatS_IMODEst_modechmodS_IXUSRS_IRUSRr r!rCrDrE)rrFrGrHrZ originalModer$rrr_defaultPathTests  z"ProcessUtilsTests._defaultPathTestcCs|tj|jS)a If no value is supplied for the C{path} parameter, L{getProcessOutput} runs the given command in the same working directory as the parent process and succeeds even if the current working directory is not accessible. )r\rrr"r)rrr test_getProcessOutputDefaultPathsz2ProcessUtilsTests.test_getProcessOutputDefaultPathcsfdd}tj|S)a If no value is supplied for the C{path} parameter, L{getProcessValue} runs the given command in the same working directory as the parent process and succeeds even if the current working directory is not accessible. cs|ddSrKr3rLr)rrrGsz@ProcessUtilsTests.test_getProcessValueDefaultPath..check)r\rr1rNrr)rtest_getProcessValueDefaultPaths z1ProcessUtilsTests.test_getProcessValueDefaultPathcsfdd}tj|S)a  If no value is supplied for the C{path} parameter, L{getProcessOutputAndValue} runs the given command in the same working directory as the parent process and succeeds even if the current working directory is not accessible. cs&|\}}}|||ddSrKr3rPr)rrrGs  zIProcessUtilsTests.test_getProcessOutputAndValueDefaultPath..check)r\rr9rNrr)r(test_getProcessOutputAndValueDefaultPaths z:ProcessUtilsTests.test_getProcessOutputAndValueDefaultPathcsDddg}dtjjd|gd}fdd}|||S)z Standard input can be made available to the child process by passing bytes for the `stdinBytes` parameter. rz"sys.stdout.write(sys.stdin.read())sThese are the bytes to see.r) stdinBytescs&|\}}}|d|dSrK)ZassertInr"r4rr`rrr7(s  zPProcessUtilsTests.test_get_processOutputAndValueStdin..gotOutputAndValuer8)rrr$r7rrar#test_get_processOutputAndValueStdins z5ProcessUtilsTests.test_get_processOutputAndValueStdin) __name__ __module__ __qualname____doc__rZIReactorProcessrskipoutputvaluerD executabler rr%r,r/r2r;rrZ isWindowsr@rIrJrOrRr\r]r^r_rbrrrrr s.         r c@seZdZdZddZdS)SuppressWarningsTestsz. Tests for L{utils.suppressWarnings}. csgfdd}|td|dd}t|dtddf}|d |td |d|td |d |td d S)zs L{utils.suppressWarnings} decorates a function so that the given warnings are suppressed. cs||fdSr&)append)rakwrMrr showwarning@sz@SuppressWarningsTests.test_suppressWarnings..showwarningrpcSst|dSr&)warningswarn)msgrrrfEsz6SuppressWarningsTests.test_suppressWarnings..f)ignorezThis is messagemessagezSanity check messager0zUnignored messageN)patchrqrZsuppressWarningsdictr"len)rrprtgrrortest_suppressWarnings9s z+SuppressWarningsTests.test_suppressWarningsN)rcrdrerfr}rrrrrk4srkc@s*eZdZdZeejZddZddZdS)DeferredSuppressedWarningsTestsz` Tests for L{utils.runWithWarningsSuppressed}, the version that supports Deferreds. cshdifdifg}t||fddtddtd|dgdd |Dd S) z If the function called by L{utils.runWithWarningsSuppressed} returns a C{Deferred}, the warning filters aren't removed until the Deferred fires. ruz.*foo.*ruz.*bar.*csSr&rrrorrlzGDeferredSuppressedWarningsTests.test_deferredCallback.. ignore foo ignore foo 2cSsg|] }|dqSrvr.0wrrr pszIDeferredSuppressedWarningsTests.test_deferredCallback..N)rrunWithWarningsSuppressedrqrrcallbackr" flushWarnings)rfiltersrrortest_deferredCallbackds   z5DeferredSuppressedWarningsTests.test_deferredCallbackcsxdifdifg}t||fdd}tdt|ddtd|dgdd |Dd S) z If the function called by L{utils.runWithWarningsSuppressed} returns a C{Deferred}, the warning filters aren't removed until the Deferred fires with an errback. rrcsSr&rrrorrrzrzFDeferredSuppressedWarningsTests.test_deferredErrback..rcSs |tSr&)ZtrapZeroDivisionError)rtrrrr}rrcSsg|] }|dqSrvrrrrrrszHDeferredSuppressedWarningsTests.test_deferredErrback..N) rrrqrrZerrbackrZ addErrbackr"r)rrr$rrortest_deferredErrbackrs   z4DeferredSuppressedWarningsTests.test_deferredErrbackN) rcrdrerf staticmethodrrrrrrrrr~Zs r~)rfrr<rVrDrqZunittestrZtwisted.internetrrrrZtwisted.internet.deferrZtwisted.python.runtimerZtwisted.python.test.test_utilr Ztwisted.trial.unittestr r r rkr~rrrrs     &