tf8|dZddlmZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZddlmZmZddlmZdd lmZmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"m#Z#dZ$e%ej&Z'ej()ej*e'd<GddejZ+Gdde#Z,dS)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! PYTHONPATHcFeZdZUdZdZded<ded<d d Zd Zd Zd Z dS)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 transportreturnNonecttj|_tj|_i|_dSN)rDeferred onConnection onCompletiondataselfs V/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/test/test_stdio.py__init__z&StandardIOTestProcessProtocol.__init__=s-27.2B2B27.2B2B&( c:|jddSr)r!callbackr$s r&connectionMadez,StandardIOTestProcessProtocol.connectionMadeBs ""4(((((r(c|j|d|z|j|<|j&|jdc}|_||dSdS)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.childDataReceivedEs_ )--c22U: $   *%)%8$ "At" JJt      + *r(c:|j|dSr)r"r*)r%reasons r& processEndedz*StandardIOTestProcessProtocol.processEndedOs ""6*****r()rr) __name__ __module__ __qualname____doc__r__annotations__r'r+r1r4r(r&rr's$37N6666    )))) )))+++++r(rceZdZejr eddZd&dZd'dZdZ d(d)dZ d*dZ d*dZ d*dZ dZdZdZd Zd!Zd"Zd#Zeejd$d%ZdS)+StandardInputOutputTests win32processNzIOn windows, spawnProcess is not available in the absence of win32process.protorsibling str | bytesargsstrkwrrr ct|tr|}tjdd|zt jjgt|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%r>r?rArCprocargss r& _spawnProcessz&StandardInputOutputTests._spawnProcessZs( gu % % 'nn&&G N  g %   (  JJ  5w''4 3>8  1: >@   r(r0defer.Deferred[None]r*Callable[[Failure], object]cFfd}fd}|||S)Nc8d|dS)Nz%Process terminated with non-Failure: )fail)resultr%s r&cbz4StandardInputOutputTests._requireFailure..cb}s$ IIHfHH I I I I Ir(c|Srr:)errr*s r&ebz4StandardInputOutputTests._requireFailure..ebs8C== r() addCallbacks)r%r0r*rVrYs` ` r&_requireFailurez(StandardInputOutputTests._requireFailurezsQ J J J J J ! ! ! ! !~~b"%%%r(ctjdztj}dfd}||S)z Verify that a protocol connected to L{StandardIO} can disconnect itself using C{transport.loseConnection}. zChild process logging to sstdio_test_loseconnc(t5}|D]+}tjd|z, dddn #1swxYwYdj|tjdS)NzChild logged: ) openrmsgrstripfailIfInr#trapr ProcessDone)r3fline errorLogFilepr%s r&r4zBStandardInputOutputTests.test_loseConnection..processEndedsl## >q>>DG,t{{}}<====> > > > > > > > > > > > > > > > MM!QV $ $ $ KK) * * * * *s/A  AA)mktemprr`rr"rOr[)r%r0r4rgrhs` @@r&test_loseConnectionz,StandardInputOutputTests.test_loseConnections {{}}  +l:;;; ) + + N 14lCCC + + + + + + +##A|444r(Fexampleoutboolc*|}ttj_dfd }j|d}|j|}||||S)NignoredrrrPcj}rjnj|Sr)r"r closeStdout closeStdin)ror0rlrhs r&cbByteszGStandardInputOutputTests.exampleOutputsAndZeroExitCode..cbBytessBA ) '')))) &&(((Hr(cD|tjdSr)rcrrd)r3s r&r4zLStandardInputOutputTests.exampleOutputsAndZeroExitCode..processEndeds KK) * * * * *r()rorrrP)rirrr r addCallbackr[rO)r%rkrlrgrsr4r0rhs ` @r&exampleOutputsAndZeroExitCodez6StandardInputOutputTests.exampleOutputsAndZeroExitCodes{{}} ) + + >++        $$W--- + + +  !1< @ @ 1g|444r(c,|dS)z When stdin is closed and the protocol connected to it implements L{IHalfCloseableProtocol}, the protocol's C{readConnectionLost} method is called. stdio_test_halfcloservr$s r&test_readConnectionLostz0StandardInputOutputTests.test_readConnectionLosts 112HIIIr(c,|dS) 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_buggyryr$s r&test_buggyReadConnectionLostz5StandardInputOutputTests.test_buggyReadConnectionLosts 112NOOOr(c0|ddS)r| stdio_test_halfclose_buggy_writeT)rlryr$s r&test_buggyWriteConnectionLostz6StandardInputOutputTests.test_buggyWriteConnectionLosts& 11 .D2   r(ct dtdn.#t$r!}t t |d}~wwxYwfd}j|S)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)usePTYNcjdtdjd|t jdS)z Asserts that the parent received the bytes written by the child immediately after the child starts. r^z Received z) from child, did not find expected bytes.N) assertTruer#endswithUNIQUE_LAST_WRITE_STRINGrcrrdr3rhr%s r&r4zEStandardInputOutputTests.test_lastWriteReceived..processEndeds` OOq ""#;<<OAFOOO    KK) * * * * *r()rrOr ValueErrorrrBr[r")r%er4rhs` @r&test_lastWriteReceivedz/StandardInputOutputTests.test_lastWriteReceiveds * + + #   *,DT      # # #3q66"" " # + + + + + +##ANLAAAs1 AAActj}dfd}||S)z Verify that the transport of a protocol connected to L{StandardIO} has C{getHost} and C{getPeer} methods. sstdio_test_hostpeercjd\}}|||tjdS)Nr^)r# splitlinesrrcrrd)r3hostpeerrhr%s r&r4z?StandardInputOutputTests.test_hostAndPeer..processEndeds]--//JD$ OOD ! ! ! OOD ! ! ! KK) * * * * *r(rr"rOr[r%r0r4rhs` @r&test_hostAndPeerz)StandardInputOutputTests.test_hostAndPeerse * + + N 14555 + + + + + + ##A|444r(ctj}dfd}||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_writecjdd|tjdSNr^sok! assertEqualr#rcrrdrs r&r4z9StandardInputOutputTests.test_write..processEnded:   QVAY / / / KK) * * * * *r(rrs` @r& test_writez#StandardInputOutputTests.test_writese * + + N 11222 + + + + + +##A|444r(ctj}dfd}||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_writeseqcjdd|tjdSrrrs r&r4zAStandardInputOutputTests.test_writeSequence..processEnded rr(rrs` @r&test_writeSequencez+StandardInputOutputTests.test_writeSequencese * + + N 14555 + + + + + +##A|444r(c|}t|d5}tdD]}|d|fz dddn #1swxYwY|S)Nwbi%d )rir_rangewrite)r%junkPathjunkFileis r& _junkPathz"StandardInputOutputTests._junkPath&s;;== (D ! ! /X4[[ / /w!~.... / / / / / / / / / / / / / / / /s,AA"%A"c tj}gttdfddjfd}||S)z Verify that the transport of a protocol connected to L{StandardIO} is a working L{IProducer} provider. dcr^dfzdtjdddSdS)Nrg{Gz?)appendpoprr callLater)ignr+proctoWritewrittens r&r+z>StandardInputOutputTests.test_producer..connectionMade8sh >w'++--)99::: 72;'''!$===== > >r(sstdio_test_producercjdddt fz|t jdS)Nr^r(z*Connection lost with %d writes left to go.)rr#join assertFalselenrcrrd)r3rhr%rrs r&r4z.processEndedBss   QVAY(9(9 : : :   EWW    KK) * * * * *r()rr"rKrrOr!rur[)r%r0r4r+rhrrrs` @@@@@r& test_producerz&StandardInputOutputTests.test_producer-s * + + NuSzz"" > > > > > > > > !!!%;<< "">222 + + + + + + + +##A|444r(ctj}dfd}||S)z Verify that the transport of a protocol connected to L{StandardIO} is a working L{IConsumer} provider. sstdio_test_consumerctd5}jd|dddn #1swxYwY|t jdS)Nrbr^)r_rr#readrcrrd)r3rerrhr%s r&r4z.processEndedWsh%% 6  AFFHH555 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 KK) * * * * *s4AAA)rr"rrOr[)r%r0r4rrhs` @@r& test_consumerz&StandardInputOutputTests.test_consumerKs| * + + N>>## 14h??? + + + + + + + ##A|444r(zpStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.c tj}t|}tj dx_}|j t| }tj s[tj\}}tj |tj |||d<t!j|fi|d t%j  fd t)jd  fd}|||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]8}|krdSd|fztjddS)N%dr)loseConnectionrr r)value connectioncounthowManyspins r&rzAStandardInputOutputTests.test_normalFileStandardOut..spinsm  G##--///FF  %!1222  a & & & & &r(rc tdzddt DdS)Nr^r(c3"K|] }d|fzV dS)rNr:).0rs r& zVStandardInputOutputTests.test_normalFileStandardOut..cbLost..s(+Q+QQEQDL+Q+Q+Q+Q+Q+Qr()rnext getContentrr)r3rrpathr%s r&cbLostzCStandardInputOutputTests.test_normalFileStandardOut..cbLostsu   T%[['A+ 6 6 6   !!388+Q+Q%..+Q+Q+Q#Q#Q     r()rr rrFilePathrir_normal addCleanupclosedictfilenor isWindowsospiper StandardIO itertoolsrr rru) r% onConnLostr>rkwargsrwrrrrrrs ` @@@@@r&test_normalFileStandardOutz3StandardInputOutputTests.test_normalFileStandardOut^s^%% / ;; //#yy. f  %%%V]]__---!## 799DAq OOBHa ( ( ( OOBHa ( ( (F7O%e66v66 !! ' ' ' ' ' ' ' ' !T"""         v&&&r() r>rr?r@rArBrCrrr )r0rPr*rQrrP)F)rkrBrlrmrrP)rrP)r5r6r7rrrskiprOr[rjrvrzr~rrrrrrrrrrr:r(r&r<r<Ssx n = = E '     @ & & & &555.).2JJJJPPPP    %B%B%BN555"555 555 555<555& V = 33  333r(r<)-r8 __future__rrrrHtypingrrunittestrtwisted.internetrrr r r twisted.internet.interfacesr r twisted.internet.protocolrtwisted.pythonrrtwisted.python.failurertwisted.python.reflectrtwisted.python.runtimertwisted.test.test_tcprtwisted.trial.unittestrrrrenvironrMpathseprrrr<r:r(r&rs #""""" CCCCCCCCCCCCCCJJJJJJJJ555555((((((((******000000++++++AAAAAA55555555: D   *//#(33 ,)+)+)+)+)+H$<)+)+)+XCCCCCxCCCCCr(