o ip8@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)Zauto_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__r0r0`C:\Users\psant\Downloads\PAF backend\school\venv\Lib\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-boolr/r0r0r0r1r26s  r2c@s&eZdZdZdededefddZdS)Workerz) An object that can run actions. caseresultreturncsdS)z" Run a test case. Nr0)selfr6r7r0r0r1runCsz Worker.runN)r*r+r,r-r rr2r:r0r0r0r1r5>sr5_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)r9ZforceGarbageCollectionr0r0r1__init__Ts  zWorkerProtocol.__init__testCaser8c 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)Z consumeErrorsFz Result reporting for {id} failed)failureidz5Additionally, reporting the reporting failure failed.r3) r?ZgatherReportingResultsr>Z loadByNamer!r@r:rloggerrCZaddErrorFallible BaseException)r9rBresultsr6ZsuiteZ allSucceededr3r7r0r0r1r:Ys6   zWorkerProtocol.runcCst|ddiS)zb Set up the worker, moving into given directory for tests to run in them. r3TN)oschdir)r9 directoryr0r0r1starts zWorkerProtocol.startN)F)r*r+r,r-rrErArRun responderr.r2r:StartrKr0r0r0r1r<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=)superrAr%_streams)r9Z boxReceiverlocator __class__r0r1rAs zLocalWorkerAMP.__init__cCsd|jiS)NstreamId)rQopenr9r0r0r1 streamOpenszLocalWorkerAMP.streamOpencCs|j||iSr=)rQwrite)r9rUdatar0r0r1 streamWriteszLocalWorkerAMP.streamWritecCs|j|jddiS)z0 Add a success to the reporter. r3TN)r? addSuccess _testCase)r9testNamer0r0r1r\szLocalWorkerAMP.addSuccesserror errorClassframesr8c 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#N)rrrangelenraappendint)r9r_r`raZ errorTyperCir0r0r1 _buildFailures $zLocalWorkerAMP._buildFailurer^ 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|]}|dqSrmdecode.0framer0r0r1  z+LocalWorkerAMP.addError..r3TN) joinrQfinishrqrir(r?addErrorr])r9r^r`rjrkr_rarCr0r0r1rys 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. rlrmcSrnrorprrr0r0r1rurvz-LocalWorkerAMP.addFailure..r3TN) rwrQrxrqrir(r? addFailurer])r9r^rzr{rkZfailrarCr0r0r1r|s zLocalWorkerAMP.addFailurecC|j|j|ddiS)z- Add a skip to the reporter. r3TN)r?addSkipr])r9r^reasonr0r0r1r~zLocalWorkerAMP.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. rlrmNz r3T)rwrQrxrqr"r?addExpectedFailurer])r9r^rjrr_Z_todor0r0r1rs z!LocalWorkerAMP.addExpectedFailurecCr})z< Add an unexpected success to the reporter. r3TN)r?addUnexpectedSuccessr])r9r^rr0r0r1rrz#LocalWorkerAMP.addUnexpectedSuccesscCs"|j|d|jddiS)z4 Print test output from the worker.  r3TN) _testStreamrYflush)r9outr0r0r1 testWrites zLocalWorkerAMP.testWriterBr7c sV||_||_|j||}z|jtj|dIdHW|j|S|j|w)z Run a test. )rBN)r]r?Z startTestrD callRemoterrLZstopTest)r9rBr7Z testCaseIdr0r0r1r:(s zLocalWorkerAMP.runcCs ||_dS)z? Set the stream used to log output from tests. N)r)r9streamr0r0r1 setTestStream5s zLocalWorkerAMP.setTestStream)NN)(r*r+r,r-rAr$rMrXr&r[rZ AddSuccessr\r(r.rrriZAddErrorrgrr4ryZ AddFailurer|ZAddSkipr~ZAddExpectedFailurerrZAddUnexpectedSuccessrZ TestWriterr rr2r:r __classcell__r0r0rSr1rOsx              rOc@eZdZdZdS)LocalWorkerAddressz A L{IAddress} implementation meant to provide stub addresses for L{ITransport.getPeer} and L{ITransport.getHost}. Nr*r+r,r-r0r0r0r1r<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)r9 transportr0r0r1rAKs zLocalWorkerTransport.__init__cCs|jt|dS)z, Forward data to transport. NrZ writeToChildrr9rZr0r0r1rYNszLocalWorkerTransport.writecCs|D] }|jt|qdS)zP Emulate C{writeSequence} by iterating data in the C{sequence}. Nr)r9sequencerZr0r0r1 writeSequenceTsz"LocalWorkerTransport.writeSequencecCs|jdS)z' Closes the transport. N)rloseConnectionrWr0r0r1r[sz#LocalWorkerTransport.loseConnectioncCtSz: Return a L{LocalWorkerAddress} instance. NrrWr0r0r1getHostazLocalWorkerTransport.getHostcCrrrrWr0r0r1getPeergrzLocalWorkerTransport.getPeerN) r*r+r,r-rArYrrrrr0r0r0r1rDs rc@r) NotRunningzN An operation was attempted on a worker process which is not running. Nrr0r0r0r1rnsrc@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)r9rrrr0r0r1rAs zLocalWorker.__init__r8NcsJ|jdur t|j}|jtz|IdHWdSty$YdSw)z3 Cause the worker process to exit. N)rrrZ closeChildFDrr)r9rr0r0r1exits   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)ZignoreExistingDirectoryzout.logwzerr.log)rJcSsdSr=r0)xr0r0r1sz,LocalWorker.connectionMade..N)rZmakeConnectionrrrmakedirschildrV_outLog_errLogrrrrrNpathZ addErrback)r9dr0r0r1connectionMadeszLocalWorker.connectionMadecCs|j|jd|_dS)zk On connection lost, close the log files that we're managing for stdin and stdout. N)rcloserrr9rr0r0r1connectionLosts   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)rrrcallbackrr0r0r1 processEndeds  zLocalWorker.processEndedcC|j|dS)z8 Send data received from stdout to log. N)rrYrr0r0r1 outReceivedszLocalWorker.outReceivedcCr)z* Write error data to log. N)rrYrr0r0r1 errReceivedszLocalWorker.errReceivedcCs*|tkr |j|dSt|||dS)zT Handle data received on the specific pipe for the C{_ampProtocol}. N)rrZ dataReceivedrchildDataReceived)r9ZchildFDrZr0r0r1rszLocalWorker.childDataReceived)r8N)r*r+r,r-rOrrrrArrrrrrrrr0r0r0r1rts      r)Fr-rHtypingrrrrrrrr Zunittestr Zzope.interfacer attrsr Ztyping_extensionsr rZtwisted.internet.deferrrZtwisted.internet.errorrZtwisted.internet.interfacesrrZtwisted.internet.protocolrZtwisted.loggerrZtwisted.protocols.amprZtwisted.python.failurerZtwisted.python.filepathrZtwisted.python.reflectrZtwisted.trial._distrrrrZ"twisted.trial._dist.workerreporterrZtwisted.trial.reporterrZtwisted.trial.runnerr r!Ztwisted.trial.unittestr"rr$r%r& Exceptionr(r2r5r;Z WorkerActionr<rOrrrrr0r0r0r1sJ(               C-)