o pjn=@sRdZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZm Z ddl!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)GdddeZ*Gddde)Z+Gddde)Z,Gddde#Z-GdddZ.Gddde)Z/dS)z) Test for distributed trial worker side. N)BytesIOStringIO) verifyObject)failsucceed)ConnectionDone)IAddress ITransport)CONNECTION_DONE)AMP)Failure)FilePath)fullyQualifiedName)trial)StringTransport)managercommandsworkercommands) LocalWorkerLocalWorkerAMPLocalWorkerTransportWorkerProtocol) TestResult)TestCasec@seZdZdZdS)FakeAMPz A fake amp protocol. N)__name__ __module__ __qualname____doc__rrF/usr/lib/python3/dist-packages/twisted/trial/_dist/test/test_worker.pyr#src@s(eZdZdZddZddZddZdS) WorkerProtocolTestsz& Tests for L{WorkerProtocol}. cCs@t|_t|_t|_|j|jt|_|j|jdS)zN Set up a transport, a result stream and a protocol instance. N)rserverTransportclientTransportrservermakeConnectionrclientselfrrrsetUp.s zWorkerProtocolTests.setUpcsdjjtjdd}fdd}||jjj jj j |S)z Calling the L{workercommands.Run} command on the client returns a response with C{success} sets to C{True}. doesntexist)testCasecs|ddSNsuccess) assertTrueresultr&rrcheck@z+WorkerProtocolTests.test_run..check) r% callRemoterRun addCallbackr# dataReceivedr"valueclearr!)r'dr0rr&rtest_run9s    zWorkerProtocolTests.test_runcCsFtjtjj}|tj||jd|tjtjj|dS)z@ The C{start} command changes the current path. z..N) ospathrealpathcurdir addCleanupchdirr#startassertNotEqual)r'r=rrr test_startJs zWorkerProtocolTests.test_startN)rrrrr(r9rBrrrrr )s  r c@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)LocalWorkerAMPTestszR Test case for distributed trial's manager-side local worker AMP protocol cCst|_t|_|j|jt|_t|_t|_ |j |jt }d|_ |d |j t |j|_|j|j|j|jdS)Nztwisted.doesnexisttests)rmanagerTransportr managerAMPr$rr/workerTransportr workerrOptionstestNameappend _getSuite_testspopr*runr7)r'configrrrr(YszLocalWorkerAMPTests.setUpcCs2|j|j|j|j|jdS)z Sends data from C{self.workerTransport} to C{self.managerAMP}, and then data from C{self.managerTransport} back to C{self.worker}. N)rFr5rGr6r7rHrEr&rrrpumpTransportsjs z"LocalWorkerAMPTests.pumpTransportscs@g|jjtj|jd}|fdd||dS)z* Run a test, and succeed. )rJc|dSr+rKr.resultsrrzz5LocalWorkerAMPTests.test_runSuccess..N)rHr2r AddSuccessrJr4rQr-r'r8rrTrtest_runSuccessss z#LocalWorkerAMPTests.test_runSuccesscs^g|jjtj|jddd}|fdd|||j|j j dd| dS)z2 Run a test, and fail expectedly. error todoReason)rJr[todocrRr+rSr.rTrrrVrWz=LocalWorkerAMPTests.test_runExpectedFailure..rN) rHr2rAddExpectedFailurerJr4rQ assertEqualr*r/expectedFailuresr-rYrrTrtest_runExpectedFailuresz+LocalWorkerAMPTests.test_runExpectedFailurechgtt}|jjtj|jd|gd}|fdd|| |j |j j dd| dS)z5 Run a test, and encounter an error. r[rJr[ errorClassframescrRr+rSr.rTrrrVrWz3LocalWorkerAMPTests.test_runError..rN)r ValueErrorrHr2rAddErrorrJr4rQr_r*r/errorsr-r'rdr8rrTr test_runErrorz!LocalWorkerAMPTests.test_runErrorcsgtt}|jjtj|jd|gdd}|fdd|| |j |j j dd| ddd ggfg|j j dd j |d S) z L{LocalWorkerAMP._buildFailure} recreates the C{Failure.frames} from the C{frames} argument passed to C{AddError}. r[)file.py invalid code3rccrRr+rSr.rTrrrVrWz=LocalWorkerAMPTests.test_runErrorWithFrames..rrlrmN)rrfrHr2rrgrJr4rQr_r*r/rhrer-rirrTrtest_runErrorWithFramess z+LocalWorkerAMPTests.test_runErrorWithFramescrb)z' Run a test, and fail. r)rJr failClassrecrRr+rSr.rTrrrVrWz5LocalWorkerAMPTests.test_runFailure..rN)r RuntimeErrorrHr2r AddFailurerJr4rQr_r*r/failuresr-)r'rrr8rrTrtest_runFailurerkz#LocalWorkerAMPTests.test_runFailurec\g|jjtj|jdd}|fdd|||j|j j dd| dS)z* Run a test, but skip it. reason)rJrxcrRr+rSr.rTrrrVrWz2LocalWorkerAMPTests.test_runSkip..rN) rHr2rAddSkiprJr4rQr_r*r/skipsr-rYrrTr test_runSkip z LocalWorkerAMPTests.test_runSkipcrw)z7 Run a test, and succeed unexpectedly. r])rJr]crRr+rSr.rTrrrVrWzALocalWorkerAMPTests.test_runUnexpectedSuccesses..rN) rHr2rAddUnexpectedSuccessrJr4rQr_r*r/unexpectedSuccessesr-rYrrTrtest_runUnexpectedSuccessesr|z/LocalWorkerAMPTests.test_runUnexpectedSuccessescsdgt}|j|tj}|jj|dd}|fdd|| d| | dS)zb L{LocalWorkerAMP.testWrite} writes the data received to its test stream. z Some output)outcrRr+rSr.rTrrrVrWz4LocalWorkerAMPTests.test_testWrite..z Some output N) rrF setTestStreamr TestWriterHr2r4rQr_getvaluer-)r'streamcommandr8rrTrtest_testWrites z"LocalWorkerAMPTests.test_testWritecsbtgfdd}||j_Gfdddt}|j|j|}||jg||jS)z| L{LocalWorkerAMP.run} calls C{stopTest} on its test result once the C{Run} commands has succeeded. cstSN)r)rr*r.rrfakeCallRemotesz=LocalWorkerAMPTests.test_stopAfterRun..fakeCallRemotecseZdZfddZdS)z=LocalWorkerAMPTests.test_stopAfterRun..StopTestResultcs|dSrrS)r'teststoppedrrstopTest szFLocalWorkerAMPTests.test_stopAfterRun..StopTestResult.stopTestN)rrrrrrrrStopTestResult sr) objectrFr2rrOr*r_r4assertIdentical)r'rrr8r)r/rrtest_stopAfterRuns z%LocalWorkerAMPTests.test_stopAfterRunN)rrrrr(rQrZrarjrqrvr{rrrrrrrrCTs   rCc@s eZdZdZdZdZddZdS)SpyDataLocalWorkerAMPz A fake implementation of L{LocalWorkerAMP} that records the received data and doesn't automatically dispatch any command.. rcCs|j|7_dSr dataString)r'datarrrr5r1z"SpyDataLocalWorkerAMP.dataReceivedN)rrrridrr5rrrrrs  rc@s(eZdZdZdZdZddZddZdS) FakeTransportz> A fake process transport implementation for testing. rrcCs|j|7_dSrr)r'fdrrrr writeToChild(r1zFakeTransport.writeToChildcCs|jd7_dS)Nrp)callsr&rrrloseConnection+r1zFakeTransport.loseConnectionN)rrrrrrrrrrrrr s  rc@sxeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZddZddZdS)LocalWorkerTestsz? Tests for L{LocalWorker} and L{LocalWorkerTransport}. cOsHt|i|}|t||jj||jj||jj|S)z Create a L{LocalWorker}, connect it to a transport, and ensure its log files are closed. @param args: See L{LocalWorker} @param kwargs: See L{LocalWorker} @return: a L{LocalWorker} instance )rr$rr>_testLogclose_outLog_errLog)r'argskwargsrHrrrtidyLocalWorker4s  z LocalWorkerTests.tidyLocalWorkercCsV|tdd}t|_|dd|dd|d|jj|d|jdS)z L{LocalWorker.childDataReceived} forwards the received data to linked L{AMP} protocol if the right file descriptor, otherwise forwards to C{ProcessProtocol.childDataReceived}. .test.logsfoorpsbarN) rrrrchildDataReceivedr_ _ampProtocolrrr' localWorkerrrrtest_childDataReceivedFs   z'LocalWorkerTests.test_childDataReceivedcCst}t|}|d}t}|tjtj|ttjdt ||j d}| t ||j j||jjz |dW|jn|jw|dtjd|dS)z L{LocalWorker} write the log data with local newlines but in UTF-8 encoding regardless of the default encoding. r)CasciiuHere comes the ☉!sHere comes the ☉!rN)rr mktempchildlocale getlocaler> setlocaleLC_ALLrr;r$rrrr testWriterr_r:linesepencode getContent)r'amptempDirlogFile currentLocalerHrrr_test_unicodeLogFileUTF8Ss"    z)LocalWorkerTests._test_unicodeLogFileUTF8cC<|tdd}t|_d}|||||jdS)zb L{LocalWorker.outReceived} logs the output into its C{_outLog} log file. rr+The quick brown fox jumps over the lazy dogN)rrrr outReceivedr_rr'rrrrrtest_outReceivedr  z!LocalWorkerTests.test_outReceivedcCr)zb L{LocalWorker.errReceived} logs the errors into its C{_errLog} log file. rrrN)rrrr errReceivedr_rrrrrtest_errReceived}rz!LocalWorkerTests.test_errReceivedcCs.t}t|}d}|||||jdS)zi L{LocalWorkerTransport.write} forwards the written data to the given transport. rN)rrwriter_rr' transportlocalTransportrrrr test_writes  zLocalWorkerTests.test_writecCs4t}t|}d}|||d||jdS)zq L{LocalWorkerTransport.writeSequence} forwards the written data to the given transport. )s The quick sbrown fox jumps sover the lazy dogrN)rr writeSequencer_joinrrrrrtest_writeSequences  z#LocalWorkerTests.test_writeSequencecCs(t}t|}|||jddS)zj L{LocalWorkerTransport.loseConnection} forwards the call to the given transport. rpN)rrrr_r)r'rrrrrtest_loseConnectionsz$LocalWorkerTests.test_loseConnectioncCsH|tdd}|d||jj||jj||jjdS)zG L{LocalWorker.connectionLost} closes the log streams. rrN)rrconnectionLostr-rclosedrrrrrrtest_connectionLosts  z$LocalWorkerTests.test_connectionLostcCsvt}t}t|dd}|||tt||jj ||j j ||j j | d|j ||jtS)zs L{LocalWorker.processEnded} calls C{connectionLost} on itself and on the L{AMP} protocol. rrN)rrrr$ processEndedr r r-rrrrrr assertFailure endDeferredr)r'rprotocolrrrrtest_processEndeds  z"LocalWorkerTests.test_processEndedcCs4td}|tt||tt|dS)zy L{LocalWorkerTransport.getPeer} and L{LocalWorkerTransport.getHost} return L{IAddress} objects. N)rr-rrgetPeergetHostr'rrrrtest_addressesszLocalWorkerTests.test_addressescCstd}|tt|dS)zk L{LocalWorkerTransport} implements L{ITransport} to be able to be used by L{AMP}. N)rr-rr rrrrtest_transportszLocalWorkerTests.test_transportcCs8dd}t}||_||dd|g|tdS)z L{LocalWorker} swallows the exceptions returned by the L{AMP} protocol start method, as it generates unnecessary errors. cSs ttdS)Noops)rrs)r directoryrrrfailCallRemotes z8LocalWorkerTests.test_startError..failCallRemoterrN)rr2rr_flushLoggedErrorsrs)r'rrrrrtest_startErrors z LocalWorkerTests.test_startErrorN)rrrrrrrrrrrrrrrrrrrrrr/s        r)0rrr:iorrzope.interface.verifyrtwisted.internet.deferrrtwisted.internet.errorrtwisted.internet.interfacesrr twisted.internet.mainr twisted.protocols.ampr twisted.python.failurer twisted.python.filepathr twisted.python.reflectrtwisted.scriptsrtwisted.test.proto_helpersrtwisted.trial._distrrtwisted.trial._dist.workerrrrrtwisted.trial.reporterrtwisted.trial.unittestrrr rCrrrrrrrs4           +@