o >hp8@sdZddlZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZmZddlmZmZdd lmZdd lmZmZdd lmZdd lmZdd l m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)m*Z*m+Z+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7m8Z8m9Z9eddGddde:Z;GdddeZee=gee>fZ?Gd!d"d"e!Z@Gd#d$d$e!ZAeeGd%d&d&ZBeeGd'd(d(ZCGd)d*d*e:ZDGd+d,d,eZEdS)-z: This module implements the worker classes. @since: 12.3 N)Any AwaitableCallableDictListOptionalTextIOTypeVar)TestCase) implementer)frozen)Protocol TypedDict)Deferred DeferredList) ProcessDone)IAddress ITransport)ProcessProtocol)Logger)AMP)Failure)FilePath) namedObject)_WORKER_AMP_STDIN_WORKER_AMP_STDOUTmanagercommandsworkercommands)WorkerReporter) TestResult) TestLoader TrialSuite)Todo) StreamOpenStreamReceiver StreamWriteF)auto_excc@eZdZUdZeed<dS)WorkerExceptionz An exception was reported by a test running in a worker process. @ivar message: An error message describing the exception. messageN)__name__ __module__ __qualname____doc__str__annotations__r1r1z/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_dist/worker.pyr)+s  r)c@r() RunResultzB Represent the result of a L{workercommands.Run} command. successN)r+r,r-r.boolr0r1r1r1r2r36s  r3c@s&eZdZdZdededefddZdS)Workerz) An object that can run actions. caseresultreturncsdS)z" Run a test case. Nr1)selfr7r8r1r1r2runCsz Worker.runN)r+r,r-r.r rr3r;r1r1r1r2r6>sr6_Tc@sJeZdZdZeZd ddZejj de de fddZ ej j d d Zd S) WorkerProtocolz5 The worker-side trial distributed protocol. FcCst|_t||_||_dSN)r _loaderr_result_forceGarbageCollection)r:forceGarbageCollectionr1r1r2__init__Ts  zWorkerProtocol.__init__testCaser9c s|j}|j|}t|g|j}||jWdn1s$wYd}t|ddIdHD]-\}}|r;q4d}|jj d||dz |j ||IdHWq4t ya|j dYq4wd|iS) z* Run a test case by name. NT) consumeErrorsFz Result reporting for {id} failed)failureidz5Additionally, reporting the reporting failure failed.r4) r@gatherReportingResultsr? loadByNamer!rAr;rloggerrFaddErrorFallible BaseException)r:rDresultsr7suite allSucceededr4r8r1r1r2r;Ys6   zWorkerProtocol.runcCst|ddiS)zb Set up the worker, moving into given directory for tests to run in them. r4T)oschdir)r: directoryr1r1r2starts zWorkerProtocol.startN)F)r+r,r-r.rrJrCrRun responderr/r3r;StartrSr1r1r1r2r=Ms ,r=c sDeZdZdZd)fdd ZejddZejddZ e j jd d Z d e d ed eedefddZe jjded edededeeeff ddZe jjdededededeeeff ddZe jjddZe jjdededeedeeeffddZe jjdd Ze j jd!d"Z!d#e"d$e#de$fd%d&Z%d'd(Z&Z'S)*LocalWorkerAMPz7 Local implementation of the manager commands. Ncst||t|_dSr>)superrCr%_streams)r: boxReceiverlocator __class__r1r2rCs zLocalWorkerAMP.__init__cCsd|jiS)NstreamId)rYopenr:r1r1r2 streamOpenszLocalWorkerAMP.streamOpencCs|j||iSr>)rYwrite)r:r^datar1r1r2 streamWriteszLocalWorkerAMP.streamWritecCs|j|jddiS)z0 Add a success to the reporter. r4T)r@ addSuccess _testCase)r:testNamer1r1r2reszLocalWorkerAMP.addSuccesserror errorClassframesr9c CsZt|}t||}tdt|dD]}|j||||dt||dggfq|S)a Helper to build a C{Failure} with some traceback. @param error: An C{Exception} instance. @param errorClass: The class name of the C{error} class. @param frames: A flat list of strings representing the information need to approximatively rebuild C{Failure} frames. @return: A L{Failure} instance with enough information about a test error. rr#)rrrangelenrjappendint)r:rhrirj errorTyperFir1r1r2 _buildFailures $zLocalWorkerAMP._buildFailurerg errorStreamIdframesStreamIdcCsXd|j|d}dd|j|D}|t|||}|j|j|ddiS)a Add an error to the reporter. @param errorStreamId: The identifier of a stream over which the text of this error was previously completely sent to the peer. @param framesStreamId: The identifier of a stream over which the lines of the traceback for this error were previously completely sent to the peer. @param error: A message describing the error. utf-8cSg|]}|dqSrwdecode.0framer1r1r2  z+LocalWorkerAMP.addError..r4T) joinrYfinishr{rsr)r@addErrorrf)r:rgrirtrurhrjrFr1r1r2rs zLocalWorkerAMP.addError failStreamId failClasscCsXd|j|d}dd|j|D}|t|||}|j|j|ddiS)az Add a failure to the reporter. @param failStreamId: The identifier of a stream over which the text of this failure was previously completely sent to the peer. @param framesStreamId: The identifier of a stream over which the lines of the traceback for this error were previously completely sent to the peer. rvrwcSrxryrzr|r1r1r2rrz-LocalWorkerAMP.addFailure..r4T) rrYrr{rsr)r@ addFailurerf)r:rgrrrufailrjrFr1r1r2rs zLocalWorkerAMP.addFailurecC|j|j|ddiS)z- Add a skip to the reporter. r4T)r@addSkiprf)r:rgreasonr1r1r2rzLocalWorkerAMP.addSkiptodocCsFd|j|d}t|durdn|}|j|j||ddiS)z Add an expected failure to the reporter. @param errorStreamId: The identifier of a stream over which the text of this error was previously completely sent to the peer. rvrwNz r4T)rrYrr{r"r@addExpectedFailurerf)r:rgrtrrh_todor1r1r2rs z!LocalWorkerAMP.addExpectedFailurecCr)z< Add an unexpected success to the reporter. r4T)r@addUnexpectedSuccessrf)r:rgrr1r1r2rrz#LocalWorkerAMP.addUnexpectedSuccesscCs"|j|d|jddiS)z4 Print test output from the worker.  r4T) _testStreamrbflush)r:outr1r1r2 testWrites zLocalWorkerAMP.testWriterDr8c sV||_||_|j||}z|jtj|dIdHW|j|S|j|w)z Run a test. )rDN)rfr@ startTestrG callRemoterrTstopTest)r:rDr8 testCaseIdr1r1r2r;(s zLocalWorkerAMP.runcCs ||_dS)z? Set the stream used to log output from tests. N)r)r:streamr1r1r2 setTestStream5s zLocalWorkerAMP.setTestStream)NN)(r+r,r-r.rCr$rUrar&rdr AddSuccessrer)r/rrrsAddErrorrprr5r AddFailurerAddSkiprAddExpectedFailurerrAddUnexpectedSuccessr TestWriterr rr3r;r __classcell__r1r1r\r2rWsx              rWc@eZdZdZdS)LocalWorkerAddressz A L{IAddress} implementation meant to provide stub addresses for L{ITransport.getPeer} and L{ITransport.getHost}. Nr+r,r-r.r1r1r1r2r<src@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: transportr1r1r2rCKs zLocalWorkerTransport.__init__cCs|jt|dS)z, Forward data to transport. Nr writeToChildrr:rcr1r1r2rbNszLocalWorkerTransport.writecCs|D] }|jt|qdS)zP Emulate C{writeSequence} by iterating data in the C{sequence}. Nr)r:sequencercr1r1r2 writeSequenceTsz"LocalWorkerTransport.writeSequencecCs|jdS)z' Closes the transport. N)rloseConnectionr`r1r1r2r[sz#LocalWorkerTransport.loseConnectioncCtSz: Return a L{LocalWorkerAddress} instance. rr`r1r1r2getHostazLocalWorkerTransport.getHostcCrrrr`r1r1r2getPeergrzLocalWorkerTransport.getPeerN) r+r,r-r.rCrbrrrrr1r1r1r2rDs rc@r) NotRunningzN An operation was attempted on a worker process which is not running. Nrr1r1r1r2rnsrc@sneZdZdZdedeedefddZdd d Z d d Z d dZ de ddfddZ ddZddZddZdS) LocalWorkeraL 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 main log file for tests output. ampProtocol logDirectorylogFilecCs||_||_||_t|_dSr>) _ampProtocol _logDirectory_logFiler endDeferred)r:rrrr1r1r2rCs zLocalWorker.__init__r9NcsJ|jdur t|j}|jtz|IdHWdSty$YdSw)z3 Cause the worker process to exit. N)rrr closeChildFDrr)r:rr1r1r2exits   zLocalWorker.exitcCs~|jt|j|jjdd|jdd|_|jdd|_ |j |j |jj t j|jjd}|ddd S) zL When connection is made, create the AMP protocol instance. T)ignoreExistingDirectoryzout.logwzerr.log)rRcSsdSr>r1)xr1r1r2sz,LocalWorker.connectionMade..N)rmakeConnectionrrrmakedirschildr__outLog_errLogrrrrrVpath addErrback)r:dr1r1r2connectionMadeszLocalWorker.connectionMadecCs|j|jd|_dS)zk On connection lost, close the log files that we're managing for stdin and stdout. N)rcloserrr:rr1r1r2connectionLosts   zLocalWorker.connectionLostrcCs&|||j||j|dS)z When the process closes, call C{connectionLost} for cleanup purposes and forward the information to the C{_ampProtocol}. N)rrrcallbackrr1r1r2 processEndeds  zLocalWorker.processEndedcC|j|dS)z8 Send data received from stdout to log. N)rrbrr1r1r2 outReceivedszLocalWorker.outReceivedcCr)z* Write error data to log. N)rrbrr1r1r2 errReceivedszLocalWorker.errReceivedcCs*|tkr |j|dSt|||dS)zT Handle data received on the specific pipe for the C{_ampProtocol}. N)rr dataReceivedrchildDataReceived)r:childFDrcr1r1r2rszLocalWorker.childDataReceived)r9N)r+r,r-r.rWrrrrCrrrrrrrrr1r1r1r2rts      r)Fr.rPtypingrrrrrrrr unittestr zope.interfacer attrsr typing_extensionsr rtwisted.internet.deferrrtwisted.internet.errorrtwisted.internet.interfacesrrtwisted.internet.protocolrtwisted.loggerrtwisted.protocols.amprtwisted.python.failurertwisted.python.filepathrtwisted.python.reflectrtwisted.trial._distrrrr"twisted.trial._dist.workerreporterrtwisted.trial.reporterrtwisted.trial.runnerr r!twisted.trial.unittestr"rr$r%r& Exceptionr)r3r6r< WorkerActionr=rWrrrrr1r1r1r2sJ(               C-)