o пb║1у@s╬dZddlZddlZddlZddlmZddlmZmZm Z m Z mZddlm Z mZddlmZddlmZddlmZdd lmZmZd ZeejГZejаejбed<Gdd Дd e jГZ GddДdeГ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. щN)┌skipIf)┌defer┌error┌protocol┌reactor┌stdio)┌filepath┌log)┌ requireModule)┌platform)┌ConnectionLostNotifyingProtocol)┌SkipTest┌TestCasesxyz123abc Twisted is great!┌ PYTHONPATHc@s4eZdZdZdZddДZddДZddДZd d ДZdS)┌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. NcCstаб|_tаб|_i|_dSйN)r┌Deferred┌onConnection┌onCompletion┌dataй┌selfйr·9/usr/lib/python3/dist-packages/twisted/test/test_stdio.py┌__init__8s z&StandardIOTestProcessProtocol.__init__cCs|jаdбdSr)r┌callbackrrrr┌connectionMade=єz,StandardIOTestProcessProtocol.connectionMadecCsB|jа|dб||j|<|jdur|jd}|_|а|бdSdS)zУ Record all bytes received from the child process in the C{data} dictionary. Fire C{onDataReceived} if it is not L{None}. єN)r┌get┌onDataReceivedr)r┌name┌bytes┌drrr┌childDataReceived@s ■z/StandardIOTestProcessProtocol.childDataReceivedcCs|jа|бdSr)rr)r┌reasonrrr┌processEndedJrz*StandardIOTestProcessProtocol.processEnded) ┌__name__┌ __module__┌__qualname__┌__doc__r rrr$r&rrrrr#s rc@sТeZdZeабredГdurdZddДZddДZdd ДZ d dДZ dd ДZddДZddДZ ddДZddДZddДZddДZeeабdГddДГZdS)┌StandardInputOutputTests┌win32processNzIOn windows, spawnProcess is not available in the absence of win32process.cOs:tjdd|tjjgt|Г}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)r┌proto┌sibling┌args┌kwrrr┌ _spawnProcessVs№√z&StandardInputOutputTests._spawnProcesscs$ЗfddД}ЗfddД}|а||бS)NcsИаd|ЫЭбdS)Nz%Process terminated with non-Failure: )┌fail)┌resultrrr┌cbqsz4StandardInputOutputTests._requireFailure..cbcsИ|ГSrr)┌err)rrr┌ebtsz4StandardInputOutputTests._requireFailure..eb)┌addCallbacks)rr#rr;r=r)rrr┌_requireFailurepsz(StandardInputOutputTests._requireFailurecsLИабЙtаdИбtГЙИj}ИаИdИбЗЗЗfddД}Иа||бS)zД Verify that a protocol connected to L{StandardIO} can disconnect itself using C{transport.loseConnection}. ·Child process logging to sstdio_test_loseconncsbtИГП}|D]}tаd|аббqWdГn1swYИаdИjб|аtjбdS)NzChild logged: щ) ┌openr ┌msg┌rstrip┌failIfInr┌trapr┌ProcessDone)r%┌f┌lineй┌errorLogFile┌prrrr&Дs zBStandardInputOutputTests.test_loseConnection..processEnded)┌mktempr rCrrr8r?йrr#r&rrJr┌test_loseConnectionys z,StandardInputOutputTests.test_loseConnectioncsf|аб}tаd|бtГЙtабИ_ЗfddД}Иjа|бddД}|аИ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. r@csИj}Иjаб|Sr)r┌ transport┌ closeStdin)┌ignoredr#йrLrr┌cbBytesЪs zAStandardInputOutputTests.test_readConnectionLost..cbBytescSs|аtjбdSr)rFrrGйr%rrrr&бrzFStandardInputOutputTests.test_readConnectionLost..processEndedsstdio_test_halfclose) rMr rCrrrr ┌addCallbackr?r8)rrKrTr&r#rrSr┌test_readConnectionLostПs z0StandardInputOutputTests.test_readConnectionLostc s^tГЙzИjИdtddНWnty }ztt|ГГВd}~wwЗЗfddД}ИаИ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)┌usePTYNcs2ИаИjdаtбdИjЫdЭб|аtjбdS)zЙ Asserts that the parent received the bytes written by the child immediately after the child starts. rAz Received z) from child, did not find expected bytes.N)┌ assertTruer┌endswith┌UNIQUE_LAST_WRITE_STRINGrFrrGrUйrLrrrr&├s ■zEStandardInputOutputTests.test_lastWriteReceived..processEnded)rr8r[┌ ValueErrorr ┌strr?r)r┌er&rr\r┌test_lastWriteReceivedйs А■z/StandardInputOutputTests.test_lastWriteReceivedcє2tГЙИj}ИаИdбЗЗfddД}Иа||бS)zГ Verify that the transport of a protocol connected to L{StandardIO} has C{getHost} and C{getPeer} methods. sstdio_test_hostpeercs6Иjdаб\}}Иа|бИа|б|аtjбdS)NrA)r┌ splitlinesrYrFrrG)r%┌host┌peerr\rrr&┘s z?StandardInputOutputTests.test_hostAndPeer..processEndedйrrr8r?rNrr\r┌test_hostAndPeer╨s z)StandardInputOutputTests.test_hostAndPeercra)zР Verify that the C{write} method of the transport of a protocol connected to L{StandardIO} sends bytes to standard out. sstdio_test_writecє"ИаИjddб|аtjбdSйNrAsok!й┌assertEqualrrFrrGrUr\rrr&ыєz9StandardInputOutputTests.test_write..processEndedrerNrr\r┌ test_writeсє z#StandardInputOutputTests.test_writecra)zШ Verify that the C{writeSequence} method of the transport of a protocol connected to L{StandardIO} sends bytes to standard out. sstdio_test_writeseqcrgrhrirUr\rrr&√rkzAStandardInputOutputTests.test_writeSequence..processEndedrerNrr\r┌test_writeSequenceёrmz+StandardInputOutputTests.test_writeSequencecCsV|аб}t|dГП}tdГD] }|аd|fбqWdГ|S1s$wY|S)N┌wbiє%d )rMrB┌range┌write)r┌junkPath┌junkFile┌irrr┌ _junkPaths ¤z"StandardInputOutputTests._junkPathcsdtГЙИj}gЙttdГГЙЗЗЗЗfddДЙИаИdбЙИjаИбЗЗЗЗfddД}Иа||бS)zА Verify that the transport of a protocol connected to L{StandardIO} is a working L{IProducer} provider. щdcs<ИrИаdИабfбИаИdбtаdИdбdSdS)Nrpщ g{оGсzД?)┌append┌poprrr┌ callLater)┌ign)r┌proc┌toWrite┌writtenrrrs ¤z>StandardInputOutputTests.test_producer..connectionMadesstdio_test_producercs>ИаИjddаИббИаИdtИГfб|аtjбdS)NrArz*Connection lost with %d writes left to go.)rjr┌join┌assertFalse┌lenrFrrGrU)rLrr~rrrr&s z.processEnded)rrr1rqr8rrVr?rNr)rrLr}rr~rr┌ test_producersz&StandardInputOutputTests.test_producercs>tГЙИj}ИабЙИаИdИбЗЗЗfddД}Иа||бS)zА Verify that the transport of a protocol connected to L{StandardIO} is a working L{IConsumer} provider. sstdio_test_consumercsPtИdГП}ИаИjd|аббWdГn1swY|аtjбdS)N┌rbrA)rBrjr┌readrFrrG)r%rHйrsrLrrrr&2s z.processEnded)rrrvr8r?rNrrЖr┌ test_consumer&sz&StandardInputOutputTests.test_consumerzpStandardIO does not accept stdout as an argument to Windows. Testing redirection to a file is therefore harder.cs▄tаб}t|Г}tаИаббЙИаdбИ_}Иа|j бt |абdН}tа бs@tаб\}}Иаtj |бИаtj |б||d<tj|fi|дОЙdЙtабЙЗЗЗЗfddДЙtаdИбЗЗЗЗfdd Д}|а|б|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. ro)┌stdout┌stdinщcs@ИD]}|ИkrИабdSИаd|fбtаdИбdS)Nє%dr)┌loseConnectionrrrr{)┌value)┌ connection┌count┌howMany┌spinrrrС]szAStandardInputOutputTests.test_normalFileStandardOut..spinrcs<ИаtИГИdбИаИабdаddДtИГDГббdS)NrArcssБ|]}d|fVqdS)rЛNr)┌.0rurrr┌ msАzVStandardInputOutputTests.test_normalFileStandardOut..cbLost..)rj┌next┌ getContentrАrqrU)rПrР┌pathrrr┌cbLostjs zCStandardInputOutputTests.test_normalFileStandardOut..cbLost)rrrr┌FilePathrMrB┌normal┌ addCleanup┌close┌dict┌filenor┌ isWindows┌os┌piper┌ StandardIO┌ itertoolsrПrr{rV)r┌ onConnLostr4rЩ┌kwargs┌r┌wrЧr)rОrПrРrЦrrСr┌test_normalFileStandardOut9s& z3StandardInputOutputTests.test_normalFileStandardOut)r'r(r)rrЮr ┌skipr8r?rOrWr`rfrlrnrvrГrЗrrзrrrrr+Ns( '■r+)"r*rвrЯr.┌unittestr┌twisted.internetrrrrr┌twisted.pythonrr ┌twisted.python.reflectr ┌twisted.python.runtimer┌twisted.test.test_tcpr┌twisted.trial.unittestr rr[rЬ┌environr3┌pathseprАrЦ┌ProcessProtocolrr+rrrr┌s +