o b$@sdZddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZdd lmZdd lmZmZmZmZdd lmZdd lmZmZdd lmZGddde ZGddde ZeeGdddZeeGdddZ Gddde Z!dS)z: This module implements the worker classes. @since: 12.3 N) implementer)Deferred)IAddress ITransport)ProcessProtocol)AMP)Failure) namedObject)_WORKER_AMP_STDIN_WORKER_AMP_STDOUTmanagercommandsworkercommands)WorkerReporter) TestLoader TrialSuite)Todoc@sBeZdZdZd ddZddZejeddZ ej e d S) WorkerProtocolz5 The worker-side trial distributed protocol. FcCst|_t||_||_dSN)r_loaderr_result_forceGarbageCollection)selfforceGarbageCollectionrraddErrorr/)rr0r9r:r6r<rrrr@ezLocalWorkerAMP.addErrorcCr?)z0 Add a failure to the reporter. rT)r>r addFailurer/)rr0fail failClassr6r<rrrrBorAzLocalWorkerAMP.addFailurecC|j|j|ddiS)z- Add a skip to the reporter. rT)raddSkipr/)rr0reasonrrrrFyzLocalWorkerAMP.addSkipcCs"t|}|j|j||ddiS)z: Add an expected failure to the reporter. rT)rraddExpectedFailurer/)rr0r9todo_todorrrrIsz!LocalWorkerAMP.addExpectedFailurecCrE)z< Add an unexpected success to the reporter. rT)raddUnexpectedSuccessr/)rr0rJrrrrLrHz#LocalWorkerAMP.addUnexpectedSuccesscCs"|j|d|jddiS)z4 Print test output from the worker.  rT) _testStreamwriteflush)routrrr testWrites zLocalWorkerAMP.testWritecCs|j|j|S)zL Stop the current running test case, forwarding the result. )rstopTestr/)rresultrrr _stopTestszLocalWorkerAMP._stopTestcCs<||_||_|j||}|jtj|d}||jS)z Run a test. )r) r/r startTestid callRemoter r* addCallbackrU)rrrT testCaseIddrrrrs   zLocalWorkerAMP.runcCs ||_dS)z? Set the stream used to log output from tests. N)rN)rstreamrrr setTestStreams zLocalWorkerAMP.setTestStreamN)r&r'r(r)r.r AddSuccessr+r>r@AddErrorrB AddFailurerFAddSkiprIAddExpectedFailurerLAddUnexpectedSuccessrR TestWriterUrr]rrrrr-As(        r-c@seZdZdZdS)LocalWorkerAddressz A L{IAddress} implementation meant to provide stub addresses for L{ITransport.getPeer} and L{ITransport.getHost}. N)r&r'r(r)rrrrresrec@s@eZdZdZddZddZddZdd Zd d Zd d Z dS)LocalWorkerTransportzi A stub transport implementation used to support L{AMP} over a L{ProcessProtocol} transport. cCs ||_dSr) _transport)r transportrrrrs zLocalWorkerTransport.__init__cCs|jt|dS)z, Forward data to transport. Nrg writeToChildr rdatarrrrOszLocalWorkerTransport.writecCs|D] }|jt|qdS)zP Emulate C{writeSequence} by iterating data in the C{sequence}. Nri)rsequencerlrrr writeSequencesz"LocalWorkerTransport.writeSequencecCs|jdS)z' Closes the transport. N)rgloseConnectionrrrrrosz#LocalWorkerTransport.loseConnectioncCtSz: Return a L{LocalWorkerAddress} instance. rerprrrgetHostzLocalWorkerTransport.getHostcCrqrrrsrprrrgetPeerruzLocalWorkerTransport.getPeerN) r&r'r(r)rrOrnrortrvrrrrrfs rfc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS) LocalWorkeraX Local process worker protocol. This worker runs as a local process and communicates via stdin/out. @ivar _ampProtocol: The L{AMP} protocol instance used to communicate with the worker. @ivar _logDirectory: The directory where logs will reside. @ivar _logFile: The name of the main log file for tests output. cCs||_||_||_t|_dSr) _ampProtocol _logDirectory_logFiler endDeferred)r ampProtocol logDirectorylogFilerrrrs zLocalWorker.__init__cCs|jt|jtj|jst|jt tj |jdd|_ t tj |jdd|_ t tj |j|j dddd|_|j|j|j}|jjtj|d}|d d d S) zL When connection is made, create the AMP protocol instance. zout.logwbzerr.logwzutf-8strict)encodingerrors)r$cSsdSrr)xrrrsz,LocalWorker.connectionMade..N)rxmakeConnectionrfrhr"pathexistsrymakedirsopenjoin_outLog_errLogrz_testLogr]rXr r, addErrback)rr}r[rrrconnectionMades zLocalWorker.connectionMadecCs"|j|j|jdS)zk On connection lost, close the log files that we're managing for stdin and stdout. N)rcloserrrrGrrrconnectionLosts  zLocalWorker.connectionLostcCs&|||j||j|dS)z When the process closes, call C{connectionLost} for cleanup purposes and forward the information to the C{_ampProtocol}. N)rrxr{callbackrrrr processEndeds  zLocalWorker.processEndedcC|j|dS)z8 Send data received from stdout to log. N)rrOrkrrr outReceived(szLocalWorker.outReceivedcCr)z* Write error data to log. N)rrOrkrrr errReceived/szLocalWorker.errReceivedcCs*|tkr |j|dSt|||dS)zT Handle data received on the specific pipe for the C{_ampProtocol}. N)r rx dataReceivedrchildDataReceived)rchildFDrlrrrr5szLocalWorker.childDataReceivedN) r&r'r(r)rrrrrrrrrrrrws    rw)"r)r"zope.interfacertwisted.internet.deferrtwisted.internet.interfacesrrtwisted.internet.protocolrtwisted.protocols.amprtwisted.python.failurertwisted.python.reflectr twisted.trial._distr r r r "twisted.trial._dist.workerreporterrtwisted.trial.runnerrrtwisted.trial.unittestrrr-rerfrwrrrrs(         w)