File: //lib/python3/dist-packages/twisted/trial/_dist/__pycache__/worker.cpython-38.pyc
U
W[�# � @ s d Z ddlZddlmZ ddlmZ ddlmZmZ ddl m
Z
ddl m Z ddl
mZ dd lmZ dd
lmZ dd lmZmZ dd lmZmZ dd
lmZmZ ddlm Z G dd� de �Z G dd� de �Z ee�G dd� de ��Z ee�G dd� de ��Z!G dd� de�Z"dS )z:
This module implements the worker classes.
@since: 12.3
� N)� implementer)�ProcessProtocol)�
ITransport�IAddress)�Deferred)�AMP)�Failure)� namedObject)�Todo)�
TrialSuite�
TestLoader)�workercommands�managercommands)�_WORKER_AMP_STDIN�_WORKER_AMP_STDOUT)�WorkerReporterc @ sB e Zd ZdZd
dd�Zdd� Zej�e� dd� Z ej
�e � d S ) �WorkerProtocolz5
The worker-side trial distributed protocol.
Fc C s t � | _t| �| _|| _d S �N)r �_loaderr �_result�_forceGarbageCollection)�selfZforceGarbageCollection� r �</usr/lib/python3/dist-packages/twisted/trial/_dist/worker.py�__init__# s
zWorkerProtocol.__init__c C s. | j �|�}t|g| j�}|�| j� ddiS )z*
Run a test case by name.
�successT)r Z
loadByNamer r �runr )r �testCaseZcaseZsuiter r r r ) s zWorkerProtocol.runc C s t �|� ddiS )zb
Set up the worker, moving into given directory for tests to run in
them.
r T)�os�chdir)r � directoryr r r �start5 s
zWorkerProtocol.startN)F) �__name__�
__module__� __qualname__�__doc__r r r
�Run� responderr! �Startr r r r r s
r c @ s� e Zd ZdZdd� Zej�e� dd� Zdd� Z ej
�e � dd � Z ej �e � d
d � Z
ej�e
� d d
� Zej�e� dd� Zej�e� dd� Zej�e� dd� Zdd� Zdd� ZdS )�LocalWorkerAMPz7
Local implementation of the manager commands.
c C s | j �| j� ddiS )z0
Add a success to the reporter.
r T)r �
addSuccess� _testCase)r �testNamer r r r* F s zLocalWorkerAMP.addSuccessc C sZ t |�}t||�}tdt|�d�D ]2}|j�|| ||d t||d �g g f� q"|S )a�
Helper to build a C{Failure} with some traceback.
@param error: An C{Exception} instance.
@param error: 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.
r � � � )r r �range�len�frames�append�int)r �error�
errorClassr2 Z errorType�failure�ir r r �
_buildFailureP s
$�z LocalWorkerAMP._buildFailurec C s& | � |||�}| j�| j|� ddiS )z/
Add an error to the reporter.
r T)r9 r �addErrorr+ )r r, r5 r6 r2 r7 r r r r: f s zLocalWorkerAMP.addErrorc C s& | � |||�}| j�| j|� ddiS )z0
Add a failure to the reporter.
r T)r9 r �
addFailurer+ )r r, ZfailZ failClassr2 r7 r r r r; q s zLocalWorkerAMP.addFailurec C s | j �| j|� ddiS )z-
Add a skip to the reporter.
r T)r �addSkipr+ )r r, �reasonr r r r< | s zLocalWorkerAMP.addSkipc C s" t |�}| j�| j||� ddiS )z:
Add an expected failure to the reporter.
r T)r
r �addExpectedFailurer+ )r r, r5 �todoZ_todor r r r> � s z!LocalWorkerAMP.addExpectedFailurec C s | j �| j|� ddiS )z<
Add an unexpected success to the reporter.
r T)r �addUnexpectedSuccessr+ )r r, r? r r r r@ � s z#LocalWorkerAMP.addUnexpectedSuccessc C s" | j �|d � | j �� ddiS )z4
Print test output from the worker.
�
r T)� _testStream�write�flush)r �outr r r � testWrite� s
zLocalWorkerAMP.testWritec C s | j �| j� |S )zL
Stop the current running test case, forwarding the result.
)r ZstopTestr+ )r �resultr r r � _stopTest� s zLocalWorkerAMP._stopTestc C s< || _ || _| j�|� |�� }| jtj|d�}|�| j�S )z
Run a test.
)r ) r+ r Z startTest�id�
callRemoter
r& Z addCallbackrH )r r rG Z
testCaseId�dr r r r � s zLocalWorkerAMP.runc C s
|| _ dS )z?
Set the stream used to log output from tests.
N)rB )r �streamr r r �
setTestStream� s z LocalWorkerAMP.setTestStreamN)r" r# r$ r% r* r Z
AddSuccessr' r9 r: ZAddErrorr; Z
AddFailurer< ZAddSkipr> ZAddExpectedFailurer@ ZAddUnexpectedSuccessrF Z TestWriterH r rM r r r r r) A s& r) c @ s e Zd ZdZdS )�LocalWorkerAddressz
A L{IAddress} implementation meant to provide stub addresses for
L{ITransport.getPeer} and L{ITransport.getHost}.
N)r" r# r$ r% r r r r rN � s rN c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�LocalWorkerTransportzi
A stub transport implementation used to support L{AMP} over a
L{ProcessProtocol} transport.
c C s
|| _ d S r )�
_transport)r � transportr r r r � s z LocalWorkerTransport.__init__c C s | j �t|� dS )z,
Forward data to transport.
N�rP Z writeToChildr �r �datar r r rC � s zLocalWorkerTransport.writec C s |D ]}| j �t|� qdS )zP
Emulate C{writeSequence} by iterating data in the C{sequence}.
NrR )r ZsequencerT r r r �
writeSequence� s z"LocalWorkerTransport.writeSequencec C s | j �� dS )z'
Closes the transport.
N)rP �loseConnection�r r r r rV � s z#LocalWorkerTransport.loseConnectionc C s t � S �z:
Return a L{LocalWorkerAddress} instance.
�rN rW r r r �getHost� s z LocalWorkerTransport.getHostc C s t � S rX rY rW r r r �getPeer� s z LocalWorkerTransport.getPeerN)
r" r# r$ r% r rC rU rV rZ r[ r r r r rO � s rO c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
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.
c C s || _ || _|| _t� | _d S r )� _ampProtocol�
_logDirectory�_logFiler � endDeferred)r Z ampProtocol� logDirectoryZlogFiler r r r s zLocalWorker.__init__c C s� | j �t| j�� tj�| j�s,t�| j� t tj�
| jd�d�| _ t tj�
| jd�d�| _ t tj�
| j| j
�d�| _| j �| j� | j}| j jtj|d�}|�dd� � dS ) zL
When connection is made, create the AMP protocol instance.
zout.log�wbzerr.log�w)r c S s d S r r )�xr r r �<lambda> � z,LocalWorker.connectionMade.<locals>.<lambda>N)r] ZmakeConnectionrO rQ r �path�existsr^ �makedirs�open�join�_outLog�_errLogr_ �_testLogrM rJ r
r( Z
addErrback)r ra rK r r r �connectionMade s �
�zLocalWorker.connectionMadec C s" | j �� | j�� | j�� dS )zk
On connection lost, close the log files that we're managing for stdin
and stdout.
N)rl �closerm rn �r r= r r r �connectionLost# s
zLocalWorker.connectionLostc C s& | � |� | j� |� | j�|� dS )z�
When the process closes, call C{connectionLost} for cleanup purposes
and forward the information to the C{_ampProtocol}.
N)rr r] r` �callbackrq r r r � processEnded- s
zLocalWorker.processEndedc C s | j �|� dS )z8
Send data received from stdout to log.
N)rl rC rS r r r � outReceived7 s zLocalWorker.outReceivedc C s | j �|� dS )z*
Write error data to log.
N)rm rC rS r r r � errReceived? s zLocalWorker.errReceivedc C s( |t kr| j�|� nt�| ||� dS )zT
Handle data received on the specific pipe for the C{_ampProtocol}.
N)r r] Z dataReceivedr �childDataReceived)r ZchildFDrT r r r rw F s z LocalWorker.childDataReceivedN) r" r# r$ r% r ro rr rt ru rv rw r r r r r\ � s
r\ )#r% r Zzope.interfacer Ztwisted.internet.protocolr Ztwisted.internet.interfacesr r Ztwisted.internet.deferr Ztwisted.protocols.ampr Ztwisted.python.failurer Ztwisted.python.reflectr Ztwisted.trial.unittestr
Ztwisted.trial.runnerr r Ztwisted.trial._distr
r r r Z"twisted.trial._dist.workerreporterr r r) �objectrN rO r\ r r r r �<module> s* # /