\d18dZddlZddlZddlZddlmZddlmZmZm Z m Z m Z ddl m Z mZddlmZddlmZddlmZdd lmZmZd ZeejZejejed <Gd d e jZ GddeZ!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. N)skipIf)defererrorprotocolreactorstdio)filepathlog) requireModule)platform)ConnectionLostNotifyingProtocol)SkipTestTestCasesxyz123abc Twisted is great! PYTHONPATHc.eZdZdZdZdZdZdZdZdS)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. Ncttj|_tj|_i|_dSN)rDeferred onConnection onCompletiondataselfs 7lib/python3.11/site-packages/twisted/test/test_stdio.py__init__z&StandardIOTestProcessProtocol.__init__8s-!N,,!N,, c:|jddSr)rcallbackrs rconnectionMadez,StandardIOTestProcessProtocol.connectionMade=s ""4(((((rc|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}. rN)rgetonDataReceivedr)rnamebytesds rchildDataReceivedz/StandardIOTestProcessProtocol.childDataReceived@s_ )--c22U: $   *%)%8$ "At" JJt      + *rc:|j|dSr)rr)rreasons r processEndedz*StandardIOTestProcessProtocol.processEndedJs ""6*****r) __name__ __module__ __qualname____doc__r#rr r'r*rrrr#sa$N )))+++++rrceZdZejr eddZdZdZdZ dZ dZ d Z d Z d Zd Zd ZdZeejddZdS)StandardInputOutputTests win32processNzIOn windows, spawnProcess is not available in the absence of win32process.ctjdd|ztjjgt |z}tj|tj|fdti|S)a_ 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. s-ms twisted.test.env)sys executabler __class__r,list spawnProcess properEnv)rprotosiblingargskws r _spawnProcessz&StandardInputOutputTests._spawnProcessVs[& N  w &   (  JJ  #E3>4UUYURTUUUrcFfd}fd}|||S)Nc8d|dS)Nz%Process terminated with non-Failure: )fail)resultrs rcbz4StandardInputOutputTests._requireFailure..cbqs$ IIHfHH I I I I Irc|Srr/)errrs rebz4StandardInputOutputTests._requireFailure..ebts8C== r) addCallbacks)rr&rrDrGs` ` r_requireFailurez(StandardInputOutputTests._requireFailurepsQ J J J J J ! ! ! ! !~~b"%%%rctjdztj}dfd}||S)z Verify that a protocol connected to L{StandardIO} can disconnect itself using C{transport.loseConnection}. Child process logging to sstdio_test_loseconnc(t5}|D]+}tjd|z, dddn #1swxYwYdj|tjdS)NzChild logged: ) openr msgrstripfailIfInrtrapr ProcessDone)r)fline errorLogFileprs rr*zBStandardInputOutputTests.test_loseConnection..processEndedsl## >q>>DG,t{{}}<====> > > > > > > > > > > > > > > > MM!QV $ $ $ KK) * * * * *s/A  AA)mktempr rOrrr?rI)rr&r*rVrWs` @@rtest_loseConnectionz,StandardInputOutputTests.test_loseConnectionys {{}}  +l:;;; ) + + N 14lCCC + + + + + + +##A|444rcR|}tjd|ztt j_fd}j|d}|j|}| d||S)z When stdin is closed and the protocol connected to it implements L{IHalfCloseableProtocol}, the protocol's C{readConnectionLost} method is called. rKcHj}j|Sr)r transport closeStdin)ignoredr&rWs rcbByteszAStandardInputOutputTests.test_readConnectionLost..cbBytess#A K " " $ $ $HrcD|tjdSr)rRrrS)r)s rr*zFStandardInputOutputTests.test_readConnectionLost..processEndeds KK) * * * * *rsstdio_test_halfclose) rXr rOrrrr# addCallbackrIr?)rrVr_r*r&rWs @rtest_readConnectionLostz0StandardInputOutputTests.test_readConnectionLosts {{}}  +l:;;; ) + + >++      $$W--- + + +  !1< @ @ 15|DDDrct 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. rMz Received z) from child, did not find expected bytes.N) assertTruerendswithUNIQUE_LAST_WRITE_STRINGrRrrSr)rWrs rr*zEStandardInputOutputTests.test_lastWriteReceived..processEndeds` OOq ""#;<<OAFOOO    KK) * * * * *r)rr?rh ValueErrorrstrrIr)rer*rWs` @rtest_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)NrM)r splitlinesrfrRrrS)r)hostpeerrWrs rr*z?StandardInputOutputTests.test_hostAndPeer..processEndeds]--//JD$ OOD ! ! ! OOD ! ! ! KK) * * * * *rrrr?rIrr&r*rWs` @rtest_hostAndPeerz)StandardInputOutputTests.test_hostAndPeerse * + + N 14555 + + + + + + ##A|444rctj}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|tjdSNrMsok! assertEqualrrRrrSris rr*z9StandardInputOutputTests.test_write..processEnded:   QVAY / / / KK) * * * * *rrsrts` @r test_writez#StandardInputOutputTests.test_writese * + + N 11222 + + + + + +##A|444rctj}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|tjdSrxryris rr*zAStandardInputOutputTests.test_writeSequence..processEndedr{rrsrts` @rtest_writeSequencez+StandardInputOutputTests.test_writeSequencese * + + N 14555 + + + + + +##A|444rc|}t|d5}tdD]}|d|fz dddn #1swxYwY|S)Nwbi%d )rXrNrangewrite)rjunkPathjunkFileis r _junkPathz"StandardInputOutputTests._junkPaths;;== (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 rr z>StandardInputOutputTests.test_producer..connectionMadesh >w'++--)99::: 72;'''!$===== > >rsstdio_test_producercjdddt fz|t jdS)NrMrz*Connection lost with %d writes left to go.)rzrjoin assertFalselenrRrrS)r)rWrrrs rr*z.processEndedss   QVAY(9(9 : : :   EWW    KK) * * * * *r)rrr8rr?rrarI)rr&r*r rWrrrs` @@@@@r test_producerz&StandardInputOutputTests.test_producers * + + NuSzz"" > > > > > > > > !!!%;<< "">222 + + + + + + + +##A|444rctj}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)NrbrM)rNrzrreadrRrrS)r)rTrrWrs rr*z.processEnded2sh%% 6  AFFHH555 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 KK) * * * * *s4AAA)rrrr?rI)rr&r*rrWs` @@r test_consumerz&StandardInputOutputTests.test_consumer&s| * + + N>>## 14h??? + + + + + + + ##A|444rzpStandardIO 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)loseConnectionrrr)value connectioncounthowManyspins rrzAStandardInputOutputTests.test_normalFileStandardOut..spin]sm  G##--///FF  %!1222  a & & & & &rrc tdzddt DdS)NrMrc3"K|] }d|fzV dS)rNr/).0rs r zVStandardInputOutputTests.test_normalFileStandardOut..cbLost..ms(+Q+QQEQDL+Q+Q+Q+Q+Q+Qr)rznext getContentrr)r)rrpathrs rcbLostzCStandardInputOutputTests.test_normalFileStandardOut..cbLostjsu   T%[['A+ 6 6 6   !!388+Q+Q%..+Q+Q+Q#Q#Q     r)rrr r FilePathrXrNnormal addCleanupclosedictfilenor isWindowsospiper StandardIO itertoolsrrrra) r onConnLostr;rkwargsrwrrrrrrs ` @@@@@rtest_normalFileStandardOutz3StandardInputOutputTests.test_normalFileStandardOut9s^%% / ;; //#yy. f  %%%V]]__---!## 799DAq OOBHa ( ( ( OOBHa ( ( (F7O%e66v66 !! ' ' ' ' ' ' ' ' !T"""         v&&&r)r+r,r-r rr skipr?rIrYrbrmrur|rrrrrrr/rrr1r1Ns)x n = = E ' VVV4&&&555,4%B%B%BN555"555 555 555<555& V = 33  333rr1)"r.rrr5unittestrtwisted.internetrrrrrtwisted.pythonr r twisted.python.reflectr twisted.python.runtimer twisted.test.test_tcpr twisted.trial.unittestrrrhrenvironr:pathseprrProcessProtocolrr1r/rrrsp CCCCCCCCCCCCCC((((((((000000++++++AAAAAA55555555: D   *//#(33 ,(+(+(+(+(+H$<(+(+(+Vcccccxcccccr