tfp8dZddlZddlmZmZmZmZmZmZm 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/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7m8Z8m9Z9edGdde:Z;GddeZ<GddeZ=e d Z>ee=gee>fZ?Gd!d"e!Z@Gd#d$e!ZAeeGd%d&ZBeeGd'd(ZCGd)d*e:ZDGd+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_excceZdZUdZeed<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__Y/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/twisted/trial/_dist/worker.pyr*r*+s% LLLLLr3r*ceZdZUdZeed<dS) RunResultzB Represent the result of a L{workercommands.Run} command. successN)r,r-r.r/boolr1r2r3r4r6r66s%MMMMMr3r6c&eZdZdZdededefdZdS)Workerz) An object that can run actions. caseresultreturnc KdS)z" Run a test case. Nr2)selfr;r<s r4runz Worker.runCs r3N)r,r-r.r/r r r6r@r2r3r4r:r:>sG h  y      r3r:_TceZdZdZeZd dZejj de de fdZ ej j dZdS) WorkerProtocolz5 The worker-side trial distributed protocol. Fcbt|_t||_||_dSN)r!_loaderr_result_forceGarbageCollection)r?forceGarbageCollections r4__init__zWorkerProtocol.__init__Ts*!|| %d++ '=$$$r3testCaser=cK|j5}|j|}t |g|j}||jdddn #1swxYwYd}t|dd{VD]s\}}|rd}|j d|| |j ||d{VJ#t$r|j dYpwxYwd|iS) z* Run a test case by name. NT) consumeErrorsFz Result reporting for {id} failed)failureidz5Additionally, reporting the reporting failure failed.r7) rGgatherReportingResultsrF loadByNamer"rHr@rloggerrNaddErrorFallible BaseException)r?rKresultsr;suite allSucceededr7r<s r4r@zWorkerProtocol.runYs \ 0 0 2 2 $g<**844Dvt'CDDE IIdl # # # $ $ $ $ $ $ $ $ $ $ $ $ $ $ $  %1'%N%N%NNNNNNN  OGV !L K  2     l33 !    ##K  <((s$A A33A7:A7?!C!!$DDc2tj|ddiS)zb Set up the worker, moving into given directory for tests to run in them. r7T)oschdir)r? directorys r4startzWorkerProtocol.starts 4  r3N)F)r,r-r.r/rrRrJrRun responderr0r6r@Startr\r2r3r4rCrCMsVXXF>>>> !*)#*))*)*)*)"!*)X#!!$#!!!r3rCc BeZdZdZdfd ZejdZejdZ e j jdZ de ded eed efd Ze jjd eded eded eeeff dZe jjd edededed eeeff dZe jjdZe jjd ed edeed eeeffdZe jjdZe j jdZ!de"de#d e$fdZ%dZ&xZ'S)LocalWorkerAMPz7 Local implementation of the manager commands. Ncrt||t|_dSrE)superrJr&_streams)r? boxReceiverlocator __class__s r4rJzLocalWorkerAMP.__init__s. g...&(( r3c8d|jiS)NstreamId)rdopenr?s r4 streamOpenzLocalWorkerAMP.streamOpensDM..0011r3c<|j||iSrE)rdwrite)r?ridatas r4 streamWritezLocalWorkerAMP.streamWrites Hd+++ r3cH|j|jddiS)z0 Add a success to the reporter. r7T)rG addSuccess _testCase)r?testNames r4rrzLocalWorkerAMP.addSuccesss' ///4  r3error errorClassframesr=c t|}t||}tdt|dD]F}|j||||dzt ||dzggfG|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$)rrrangelenrwappendint)r?rurvrw errorTyperNis r4 _buildFailurezLocalWorkerAMP._buildFailures& ++ %++q#f++q))  A N ! !F1q5M3va!e}+=+=r2F    r3rt errorStreamIdframesStreamIdcZd|j|d}d|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. r3utf-8c8g|]}|dSrdecode.0frames r4 z+LocalWorkerAMP.addError..1   &+ELL ! !   r3r7T) joinrdfinishrrr*rGaddErrorrs)r?rtrvrrrurwrNs r4rzLocalWorkerAMP.addErrors(--m<<==DDWMM  /3}/C/CN/S/S   $$_U%;%;ZPP dng6664  r3 failStreamId failClasscZd|j|d}d|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. r3rc8g|]}|dSrrrs r4rz-LocalWorkerAMP.addFailure..rr3r7T) rrdrrrr*rG addFailurers)r?rtrrrfailrwrNs r4rzLocalWorkerAMP.addFailures$xx ,,\::;;BB7KK  /3}/C/CN/S/S   $$_T%:%:IvNN 8884  r3cJ|j|j|ddiS)z- Add a skip to the reporter. r7T)rGaddSkiprs)r?rtreasons r4rzLocalWorkerAMP.addSkips) T^V4444  r3todocd|j|d}t |dn|}|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. r3rNz r7T)rrdrrr#rGaddExpectedFailurers)r?rtrrru_todos r4rz!LocalWorkerAMP.addExpectedFailureso--m<<==DDWMMDL[[d;; ''uEEE4  r3cJ|j|j|ddiS)z< Add an unexpected success to the reporter. r7T)rGaddUnexpectedSuccessrs)r?rtrs r4rz#LocalWorkerAMP.addUnexpectedSuccesss) ))$.$???4  r3cv|j|dz|jddiS)z4 Print test output from the worker.  r7T) _testStreamrnflush)r?outs r4 testWritezLocalWorkerAMP.testWrites? sTz***    4  r3rKr<cBK||_||_|j||} |t j|d{V |j|S#|j|wxYw)z Run a test. )rKN)rsrG startTestrO callRemoterr]stopTest)r?rKr< testCaseIds r4r@zLocalWorkerAMP.run(s"  x((([[]]  ,);jQQQQQQQQ Q L ! !( + + + +DL ! !( + + + +s &BBc||_dS)z? Set the stream used to log output from tests. N)r)r?streams r4 setTestStreamzLocalWorkerAMP.setTestStream5s"r3)NN)(r,r-r.r/rJr%r^rlr'rpr AddSuccessrrr*r0rrrAddErrorr~rr8r AddFailurerAddSkiprAddExpectedFailurerrAddUnexpectedSuccessr TestWriterr r r6r@r __classcell__)rgs@r4raras))))))222)!!*)!S   6'!!! !  ! c4i !!!('!>)!!! !  ! c4i !!!*)!4&!!'&!'1 ! !,/ !7?} ! c4i ! ! !21 !)3!!43!(!!)(! ,( ,J ,9 , , , ,"""""""r3raceZdZdZdS)LocalWorkerAddressz A L{IAddress} implementation meant to provide stub addresses for L{ITransport.getPeer} and L{ITransport.getHost}. Nr,r-r.r/r2r3r4rr<sr3rc6eZdZdZdZdZdZdZdZdZ dS) LocalWorkerTransportzi A stub transport implementation used to support L{AMP} over a L{ProcessProtocol} transport. c||_dSrE) _transport)r? transports r4rJzLocalWorkerTransport.__init__Ks #r3cF|jt|dS)z, Forward data to transport. Nr writeToChildrr?ros r4rnzLocalWorkerTransport.writeNs# $$%6=====r3cP|D]"}|jt|#dS)zP Emulate C{writeSequence} by iterating data in the C{sequence}. Nr)r?sequenceros r4 writeSequencez"LocalWorkerTransport.writeSequenceTs= B BD O ( ():D A A A A B Br3c8|jdS)z' Closes the transport. N)rloseConnectionrks r4rz#LocalWorkerTransport.loseConnection[s &&(((((r3ctSz: Return a L{LocalWorkerAddress} instance. rrks r4getHostzLocalWorkerTransport.getHosta"###r3ctSrrrks r4getPeerzLocalWorkerTransport.getPeergrr3N) r,r-r.r/rJrnrrrrr2r3r4rrDs{ $$$>>> BBB))) $$$ $$$$$r3rceZdZdZdS) NotRunningzN An operation was attempted on a worker process which is not running. Nrr2r3r4rrnsr3rcheZdZdZdedeedefdZddZ d Z d Z d e ddfd Z d ZdZdZdS) 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 logDirectorylogFilecV||_||_||_t|_dSrE) _ampProtocol _logDirectory_logFiler endDeferred)r?rrrs r4rJzLocalWorker.__init__s- () +3::r3r=NcK|jt|j}|jt |d{VdS#t $rYdSwxYw)z3 Cause the worker process to exit. N)rrr closeChildFDrr)r?rs r4exitzLocalWorker.exits~ > !,, &  ##$5666              DD sA AAc&|jt|j|jd|jdd|_|jdd|_ |j |j |j tj|jj}|ddS) zL When connection is made, create the AMP protocol instance. T)ignoreExistingDirectoryzout.logwzerr.log)r[cdSrEr2)xs r4z,LocalWorker.connectionMade..str3N)rmakeConnectionrrrmakedirschildrj_outLog_errLogrrrrr_path addErrback)r?ds r4connectionMadezLocalWorker.connectionMades (()=dn)M)MNNN ##D#AAA)// ::??DD )// ::??DD  '' 666   ( (  (- )   ^^$$$$$r3cx|j|jd|_dS)zk On connection lost, close the log files that we're managing for stdin and stdout. N)rcloserrr?rs r4connectionLostzLocalWorker.connectionLosts7  r3rc|||j||j|dS)z When the process closes, call C{connectionLost} for cleanup purposes and forward the information to the C{_ampProtocol}. N)rrrcallbackrs r4 processEndedzLocalWorker.processEndedsL F### ((000 !!&)))))r3c:|j|dS)z8 Send data received from stdout to log. N)rrnrs r4 outReceivedzLocalWorker.outReceiveds 4     r3c:|j|dS)z* Write error data to log. N)rrnrs r4 errReceivedzLocalWorker.errReceiveds  4     r3c|tkr|j|dStj|||dS)zT Handle data received on the specific pipe for the C{_ampProtocol}. N)rr dataReceivedrchildDataReceived)r?childFDros r4rzLocalWorker.childDataReceivedsH ( ( (   * *4 0 0 0 0 0  -dGT B B B B Br3)r=N)r,r-r.r/rarrr rJrrrrrrrrr2r3r4rrts   6# 6sm 6 6 6 6 6    %%%"*7*t****!!!!!! CCCCCr3r)Fr/rYtypingrrrrrrr r unittestr zope.interfacer attrsr typing_extensionsrrtwisted.internet.deferrrtwisted.internet.errorrtwisted.internet.interfacesrrtwisted.internet.protocolrtwisted.loggerrtwisted.protocols.amprtwisted.python.failurertwisted.python.filepathrtwisted.python.reflectrtwisted.trial._distrrrr"twisted.trial._dist.workerreporterrtwisted.trial.reporterr twisted.trial.runnerr!r"twisted.trial.unittestr#rr%r&r' Exceptionr*r6r:rA WorkerActionrCrarrrrr2r3r4rs   RRRRRRRRRRRRRRRRRRRR&&&&&&1111111199999999......<<<<<<<<555555!!!!!!%%%%%%******,,,,,,...... >=====------77777777'''''';;;;;;;;;;i      X    WT]])B-/0 @!@!@!@!@!S@!@!@!Fi"i"i"i"i"Si"i"i"X X Z&$&$&$&$&$&$&$&$R ]C]C]C]C]C/]C]C]C]C]Cr3