Vh8dZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZddlmZmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"m#Z#dZ$e%ejLZ'ejPjSejTe'd<Gddej*Z+Gdde#Z,y)z Tests for L{twisted.internet.stdio}. @var properEnv: A copy of L{os.environ} which has L{bytes} keys/values on POSIX platforms and native L{str} keys/values on Windows. ) annotationsN)AnyCallable)skipIf)defererrorprotocolreactorstdio)IProcessTransportIReactorProcess)ProcessProtocol)filepathlog)Failure) requireModule)platform)ConnectionLostNotifyingProtocol)SkipTestTestCasesxyz123abc Twisted is great! PYTHONPATHcDeZdZUdZdZded<ded<d dZdZd Zd Z y) StandardIOTestProcessProtocola Test helper for collecting output from a child process and notifying something when it exits. @ivar onConnection: A L{defer.Deferred} which will be called back with L{None} when the connection to the child process is established. @ivar onCompletion: A L{defer.Deferred} which will be errbacked with the failure associated with the child process exiting when it exits. @ivar onDataReceived: A L{defer.Deferred} which will be called back with this instance whenever C{childDataReceived} is called, or L{None} to suppress these callbacks. @ivar data: A C{dict} mapping file descriptors to strings containing all bytes received from the child process on each file descriptor. Nzdefer.Deferred[None] | NoneonDataReceivedr transportcvtj|_tj|_i|_yN)rDeferred onConnection onCompletiondataselfs G/home/dcms/DCMS/lib/python3.12/site-packages/twisted/test/test_stdio.py__init__z&StandardIOTestProcessProtocol.__init__=s'27..2B27..2B&( c:|jjdyr)rcallbackr"s r$connectionMadez,StandardIOTestProcessProtocol.connectionMadeBs ""4(r&c|jj|d|z|j|<|j&|jdc}|_|j|yy)z Record all bytes received from the child process in the C{data} dictionary. Fire C{onDataReceived} if it is not L{None}. r&N)r!getrr()r#namebytesds r$childDataReceivedz/StandardIOTestProcessProtocol.childDataReceivedEsW ))--c2U: $    *%)%8%8$ "At" JJt  +r&c:|jj|yr)r r()r#reasons r$ processEndedz*StandardIOTestProcessProtocol.processEndedOs ""6*r&)returnNone) __name__ __module__ __qualname____doc__r__annotations__r%r)r/r2r&r$rr's.$37N/6  ) )+r&rc eZdZejr eddZ ddZ ddZdZ d ddZ ddZ dd Z dd Z d Zd Zd ZdZdZdZdZeejddZy)StandardInputOutputTests win32processNzIOn windows, spawnProcess is not available in the absence of win32process.c$t|tr|j}tjdd|zt j jgt|z}tt j|tj|fdti|S)ak Launch a child Python process and communicate with it using the given ProcessProtocol. @param proto: A L{ProcessProtocol} instance which will be connected to the child process. @param sibling: The basename of a file containing the Python program to run in the child process. @param *args: strings which will be passed to the child process on the command line as C{argv[2:]}. @param **kw: additional arguments to pass to L{reactor.spawnProcess}. @return: The L{IProcessTransport} provider for the spawned process. z-mz twisted.test.env) isinstancer-decodesys executabler __class__r6listr spawnProcess properEnv)r#protosiblingargskwprocargss r$ _spawnProcessz&StandardInputOutputTests._spawnProcessZs( gu %nn&G NN  g %    ( (  J  5w'44 3>>8 1: >@  r&c>fd}fd}|j||S)Nc.jd|y)Nz%Process terminated with non-Failure: )fail)resultr#s r$cbz4StandardInputOutputTests._requireFailure..cb}s II=fZH Ir&c|Srr:)errr(s r$ebz4StandardInputOutputTests._requireFailure..ebs C= r&) addCallbacks)r#r.r(rRrUs` ` r$_requireFailurez(StandardInputOutputTests._requireFailurezs! J !~~b"%%r&cjtjdztj}j dfd}j ||S)z Verify that a protocol connected to L{StandardIO} can disconnect itself using C{transport.loseConnection}. zChild process logging to sstdio_test_loseconnct5}|D](}tjd|jz* dddj dj |j tjy#1swYExYw)NzChild logged: ) openrmsgrstripfailIfInr!trapr ProcessDone)r1fline errorLogFilepr#s r$r2zBStandardInputOutputTests.test_loseConnection..processEndedsol# >q>DGG,t{{}<=> > MM!QVV $ KK)) *  > >s .BB )mktemprr\rr rMrW)r#r.r2rcrds` @@r$test_loseConnectionz,StandardInputOutputTests.test_loseConnections` {{}  +l:; ) + NN 14lC +##A|44r&c|j}ttj_dfd }jj |d}|j j|}|j|||S)Ncj}rjj|Sjj|Sr)r r closeStdout closeStdin)ignoredr.outrds r$cbByteszGStandardInputOutputTests.exampleOutputsAndZeroExitCode..cbBytess=A '')H &&(Hr&cB|jtjyr)r_rr`)r1s r$r2zLStandardInputOutputTests.exampleOutputsAndZeroExitCode..processEndeds KK)) *r&)rkr4r3defer.Deferred[None])rerrrr addCallbackrWrM)r#examplerlrcrmr2r.rds ` @r$exampleOutputsAndZeroExitCodez6StandardInputOutputTests.exampleOutputsAndZeroExitCodesx{{} ) + >>+  $$W- +  !1!1< @ 1g|4r&c$|jdS)z When stdin is closed and the protocol connected to it implements L{IHalfCloseableProtocol}, the protocol's C{readConnectionLost} method is called. stdio_test_halfcloserrr"s r$test_readConnectionLostz0StandardInputOutputTests.test_readConnectionLosts 112HIIr&c$|jdS) When stdin is closed and the protocol connnected to it implements L{IHalfCloseableProtocol} but its C{readConnectionLost} method raises an exception its regular C{connectionLost} method will be called. stdio_test_halfclose_buggyrur"s r$test_buggyReadConnectionLostz5StandardInputOutputTests.test_buggyReadConnectionLosts 112NOOr&c(|jddS)rx stdio_test_halfclose_buggy_writeT)rlrur"s r$test_buggyWriteConnectionLostz6StandardInputOutputTests.test_buggyWriteConnectionLosts! 11 .D2  r&ct jdtdfd}j j|S#t$r}t t |d}~wwxYw)z Verify that a write made directly to stdout using L{os.write} after StandardIO has finished is reliably received by the process reading that stdout. sstdio_test_lastwriteT)usePTYNcjjdjtdjd|j t j y)z Asserts that the parent received the bytes written by the child immediately after the child starts. rZz Received z) from child, did not find expected bytes.N) assertTruer!endswithUNIQUE_LAST_WRITE_STRINGr_rr`r1rdr#s r$r2zEStandardInputOutputTests.test_lastWriteReceived..processEndedsM OOq ""#;<AFF:%NO  KK)) *r&)rrMr ValueErrorrstrrWr )r#er2rds` @r$test_lastWriteReceivedz/StandardInputOutputTests.test_lastWriteReceivedso * + #   *,DT   +##ANNLAA #3q6" " #sA A+A&&A+ctj}jdfd}j||S)z Verify that the transport of a protocol connected to L{StandardIO} has C{getHost} and C{getPeer} methods. sstdio_test_hostpeercjdj\}}j|j||jtj y)NrZ)r! splitlinesrr_rr`)r1hostpeerrdr#s r$r2z?StandardInputOutputTests.test_hostAndPeer..processEndedsG--/JD$ OOD ! OOD ! KK)) *r&rr rMrWr#r.r2rds` @r$test_hostAndPeerz)StandardInputOutputTests.test_hostAndPeersA * + NN 145 + ##A|44r&ctj}jdfd}j||S)z Verify that the C{write} method of the transport of a protocol connected to L{StandardIO} sends bytes to standard out. sstdio_test_writecjjdd|jtjyNrZsok! assertEqualr!r_rr`rs r$r2z9StandardInputOutputTests.test_write..processEnded-   QVVAY / KK)) *r&rrs` @r$ test_writez#StandardInputOutputTests.test_writesA * + NN 112 +##A|44r&ctj}jdfd}j||S)z Verify that the C{writeSequence} method of the transport of a protocol connected to L{StandardIO} sends bytes to standard out. sstdio_test_writeseqcjjdd|jtjyrrrs r$r2zAStandardInputOutputTests.test_writeSequence..processEnded rr&rrs` @r$test_writeSequencez+StandardInputOutputTests.test_writeSequencesA * + NN 145 +##A|44r&c|j}t|d5}tdD]}|jd|fz ddd|S#1swY|SxYw)Nwbi%d )rer[rangewrite)r#junkPathjunkFileis r$ _junkPathz"StandardInputOutputTests._junkPath&s\;;= (D ! /X4[ /w!~. / / /s &AActj}gttdfdj dj j fd}j||S)z Verify that the transport of a protocol connected to L{StandardIO} is a working L{IProducer} provider. dcrOjdjfzjdtjddyy)Nrg{Gz?)appendpoprr callLater)ignr)proctoWritewrittens r$r)z>StandardInputOutputTests.test_producer..connectionMade8sGw'++-)99: 72;'!!$=r&sstdio_test_producercjjddjjdt fz|j t jy)NrZr&z*Connection lost with %d writes left to go.)rr!join assertFalselenr_rr`)r1rdr#rrs r$r2z.processEndedBsW   QVVAY(9 :   EWW  KK)) *r&)rr rErrMrrprW)r#r.r2r)rdrrrs` @@@@@r$ test_producerz&StandardInputOutputTests.test_producer-so * + NNuSz" > !!!%;< "">2 +##A|44r&ctj}jjdfd}j ||S)z Verify that the transport of a protocol connected to L{StandardIO} is a working L{IConsumer} provider. sstdio_test_consumerctd5}jjd|jddd|j t j y#1swY)xYw)NrbrZ)r[rr!readr_rr`)r1rarrdr#s r$r2z.processEndedWsTh% 6  AFFH5 6 KK)) * 6 6s .A$$A-)rr rrMrW)r#r.r2rrds` @@r$ test_consumerz&StandardInputOutputTests.test_consumerKsO * + NN>># 14h? + ##A|44r&zpStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.c tj}t|}tjj  j dx_}j|jt|j}tjs\tj\}}jtj|jtj|||d<t!j"|fi|d t%j&  fd t)j*d  fd}|j-||S)aE If L{StandardIO} is created with a file descriptor which refers to a normal file (ie, a file from the filesystem), L{StandardIO.write} writes bytes to that file. In particular, it does not immediately consider the file closed or call its protocol's C{connectionLost} method. r)stdoutstdincD]/}|k(rjyjd|fzntjdy)N%dr)loseConnectionrr r)value connectioncounthowManyspins r$rzAStandardInputOutputTests.test_normalFileStandardOut..spinsS G#--/  %!12     a &r&rc jtdzjjdjdt Dy)NrZr&c3(K|] }d|fz yw)rNr:).0rs r$ zVStandardInputOutputTests.test_normalFileStandardOut..cbLost..s+QQEQDL+Qs)rnext getContentrr)r1rrpathr#s r$cbLostzCStandardInputOutputTests.test_normalFileStandardOut..cbLostsH   T%['A+ 6   !388+Q%.+Q#Q r&)rrrrFilePathrer[normal addCleanupclosedictfilenor isWindowsospiper StandardIO itertoolsrr rrp) r# onConnLostrHrkwargsrwrrrrrrs ` @@@@@r$test_normalFileStandardOutz3StandardInputOutputTests.test_normalFileStandardOut^s ^^% / ;  /#yy. f  %V]]_-!!#779DAq OOBHHa ( OOBHHa (F7O%%e6v6 ! ' !T"  v&r&) rHrrIz str | bytesrJrrKrr3r )r.ror(zCallable[[Failure], object]r3ro)F)rqrrlboolr3ro)r3ro)r5r6r7rrrskiprMrWrfrrrvrzr}rrrrrrrrrr:r&r$r<r<Ssx n = E '  $ /: CF NQ  @ &% &1L &  &5.).!% 2JP %BN5"5 5 5<5&  = 3  3r&r<)-r8 __future__rrrrBtypingrrunittestrtwisted.internetrrr r r twisted.internet.interfacesr r twisted.internet.protocolrtwisted.pythonrrtwisted.python.failurertwisted.python.reflectrtwisted.python.runtimertwisted.test.test_tcprtwisted.trial.unittestrrrrenvironrGpathseprrrr<r:r&r$rs # CCJ5(*0+A5:   **//#((3 ,)+H$<$<)+XCxCr&