File: //lib/python3/dist-packages/twisted/spread/test/__pycache__/test_pbfailure.cpython-38.pyc
U
W[I< � @ sl d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
ddlmZ G dd � d e�ZG d
d � d e�ZG d d
� d
e j�ZG dd� de j�ZG dd� de je je j�ZG dd� de je j�Ze �ee� e �ee� e j� e� G dd� de j �Z G dd� de j �Z G dd� dej!�Z"G dd� de"�Z#G d d � d e#�Z$G d d � d e%�Z&G d d!� d!ej!�Z'd"S )#z!
Tests for error handling in PB.
� )�reactor�defer)�log)�NativeStringIO)�qual)�pb�flavors�jelly)�unittestc @ s e Zd ZdZdS )�AsynchronousExceptionz�
Helper used to test remote methods which return Deferreds which fail with
exceptions which are not L{pb.Error} subclasses.
N��__name__�
__module__� __qualname__�__doc__� r r �D/usr/lib/python3/dist-packages/twisted/spread/test/test_pbfailure.pyr s r c @ s e Zd ZdZdS )�SynchronousExceptionzm
Helper used to test remote methods which raise exceptions which are not
L{pb.Error} subclasses.
Nr r r r r r s r c @ s e Zd ZdZdS )�AsynchronousErrorz�
Helper used to test remote methods which return Deferreds which fail with
exceptions which are L{pb.Error} subclasses.
Nr r r r r r s r c @ s e Zd ZdZdS )�SynchronousErrorzi
Helper used to test remote methods which raise exceptions which are
L{pb.Error} subclasses.
Nr r r r r r ' s r c @ s e Zd ZdS )�
JellyErrorN�r
r r r r r r r / s r c @ s e Zd ZdS )�
SecurityErrorNr r r r r r 4 s r c @ sh e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z ddd�Z ddd�Z
dS )�
SimpleRootc C s t �td��S )zD
Fail asynchronously with a non-pb.Error exception.
z remote asynchronous exception)r �failr ��selfr r r � remote_asynchronousException@ s z'SimpleRoot.remote_asynchronousExceptionc C s t d��dS )zC
Fail synchronously with a non-pb.Error exception.
z remote synchronous exceptionN)r r r r r �remote_synchronousExceptionG s z&SimpleRoot.remote_synchronousExceptionc C s t �td��S )z@
Fail asynchronously with a pb.Error exception.
zremote asynchronous error)r r r r r r r �remote_asynchronousErrorN s z#SimpleRoot.remote_asynchronousErrorc C s t d��dS )z?
Fail synchronously with a pb.Error exception.
zremote synchronous errorN)r r r r r �remote_synchronousErrorU s z"SimpleRoot.remote_synchronousErrorc C s G dd� dt j�}|d��dS )z>
Fail with error that is not known to client.
c @ s e Zd ZdS )z4SimpleRoot.remote_unknownError.<locals>.UnknownErrorNr r r r r � UnknownError` s r! zI'm not known to client!N)r �Error)r r! r r r �remote_unknownError\ s z SimpleRoot.remote_unknownErrorc C s | � � d S �N)�
raiseJellyr r r r � remote_jellye s zSimpleRoot.remote_jellyc C s | � � d S r$ )�
raiseSecurityr r r r �remote_securityi s zSimpleRoot.remote_securityc C s" t �� }|�| j� |�d � |S r$ )r �Deferred� addCallbackr% �callback�r �dr r r �remote_deferredJellym s
z SimpleRoot.remote_deferredJellyc C s" t �� }|�| j� |�d � |S r$ )r r) r* r' r+ r, r r r �remote_deferredSecurityt s
z"SimpleRoot.remote_deferredSecurityNc C s t d��d S )NzI'm jellyable!)r �r Zresultsr r r r% { s zSimpleRoot.raiseJellyc C s t d��d S )Nz I'm secure!)r r0 r r r r' s zSimpleRoot.raiseSecurity)N)N)r
r r r r r r r# r& r( r. r/ r% r' r r r r r ? s
r c @ s e Zd ZdZdZdd� ZdS )�SaveProtocolServerFactoryzd
A L{pb.PBServerFactory} that saves the latest connected client in
C{protocolInstance}.
Nc C s
|| _ dS )z3
Keep track of the given protocol.
N)�protocolInstance)r Zprotocolr r r �clientConnectionMade� s z.SaveProtocolServerFactory.clientConnectionMade)r
r r r r2 r3 r r r r r1 � s r1 c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�PBConnTestCaser c C s | � � | �� d S r$ )� _setUpServer� _setUpClientr r r r �setUp� s zPBConnTestCase.setUpc C s. t t� �| _| j| j_tjd| jdd�| _d S )Nr � 127.0.0.1)Z interface)r1 r �
serverFactory�unsafeTracebacksr Z listenTCP�
serverPortr r r r r5 � s
zPBConnTestCase._setUpServerc C s, | j �� j}t�� | _t�d|| j�| _d S )Nr8 ) r; ZgetHostZportr ZPBClientFactory�
clientFactoryr Z
connectTCP�clientConnector)r ZportNor r r r6 � s
zPBConnTestCase._setUpClientc C s0 | j jd k r| j jj�� t�| �� | �� g�S r$ )r9 r2 Z transportZloseConnectionr Z
gatherResults�_tearDownServer�_tearDownClientr r r r �tearDown� s �zPBConnTestCase.tearDownc C s t �| jj�S r$ )r Z
maybeDeferredr; Z
stopListeningr r r r r> � s z PBConnTestCase._tearDownServerc C s | j �� t�d �S r$ )r= Z
disconnectr Zsucceedr r r r r? � s
z PBConnTestCase._tearDownClientN)
r
r r r: r7 r5 r6 r@ r> r? r r r r r4 � s r4 c @ s� e Zd ZejjZdd� Zdd� Zdd� Z dd� Z
d d
� Z d d � Z d
d� Z
d$dd�Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� ZdS )%�PBFailureTestsc s6 ���fdd�� �j �� }� �fdd�}|�|� |S )Nc sF | � � � ��| jd� �r6��� �}��t|�d� | j| j| jfS )N�Traceback unavailable
� )�trap�compare� traceback�flushLoggedErrors� assertEqual�len�type�value)�err�errs)�
exceptionType�flushr r r �eb� s
z)PBFailureTests._exceptionTest.<locals>.ebc s | � ��}|�� � |S r$ )�
callRemote�
addErrback)�rootr- )rP �methodr r �
gotRootObject� s
z4PBFailureTests._exceptionTest.<locals>.gotRootObject�r< �
getRootObjectr* )r rT rN rO r- rU r )rP rN rO rT r r �_exceptionTest� s
z PBFailureTests._exceptionTestc C s | � dtd�S )z�
Test that a Deferred returned by a remote method which already has a
Failure correctly has that error passed back to the calling side.
ZasynchronousExceptionT)rX r r r r r �test_asynchronousException� s
�z)PBFailureTests.test_asynchronousExceptionc C s | � dtd�S )zw
Like L{test_asynchronousException}, but for a method which raises an
exception synchronously.
�synchronousExceptionT)rX r r r r r �test_synchronousException� s
�z(PBFailureTests.test_synchronousExceptionc C s | � dtd�S )z�
Like L{test_asynchronousException}, but for a method which returns a
Deferred failing with an L{pb.Error} subclass.
ZasynchronousErrorF)rX r r r r r �test_asynchronousError� s
�z%PBFailureTests.test_asynchronousErrorc C s | � dtd�S )z}
Like L{test_asynchronousError}, but for a method which synchronously
raises a L{pb.Error} subclass.
�synchronousErrorF)rX r r r r r �test_synchronousError� s
�z$PBFailureTests.test_synchronousErrorc C s | � ||� |S r$ )rH )r �result�expectedResultr r r �_success� s zPBFailureTests._successc C s |j | j||fd� |S )N)Z callbackArgs)Z addCallbacksra )r Z
remoteCallr` rP r r r �_addFailingCallbacks� s
�z#PBFailureTests._addFailingCallbacksNc s, �j �� }� ����fdd�}|�|� |S )z�
Call the given remote method and attach the given errback to the
resulting Deferred. If C{exc} is not None, also assert that one
exception of that type was logged.
c s8 �� | ����� �}�d k r4��fdd�}|�|� |S )Nc s �� t��� ��d� | S )NrC )rH rI rG )rL )�excr r r �
gotFailure s z@PBFailureTests._testImpl.<locals>.gotRootObj.<locals>.gotFailure)rb rQ ZaddBoth)�objZfailureDeferredrd �rP rc �expectedrT r r r �
gotRootObj s
z,PBFailureTests._testImpl.<locals>.gotRootObjrV )r rT rg rP rc Z rootDeferredrh r rf r � _testImpl� s
zPBFailureTests._testImplc s � fdd�}� � dd|�S )z�
Test that an exception which is a subclass of L{pb.Error} has more
information passed across the network to the calling side.
c s, | � t� � �| jt� � �| j| j� dS )N�+ �rD r �assertNotIsInstancerJ �str�assertIsInstancerK �r r r r � failureJelly s
z6PBFailureTests.test_jellyFailure.<locals>.failureJellyr rj �ri )r rp r r r �test_jellyFailure s z PBFailureTests.test_jellyFailurec s � fdd�}� � dd|�S )z�
Test that a Deferred which fails with a L{pb.Error} is treated in
the same way as a synchronously raised L{pb.Error}.
c s, | � t� � �| jt� � �| j| j� dS )N� rk ro r r r �failureDeferredJelly" s
zFPBFailureTests.test_deferredJellyFailure.<locals>.failureDeferredJellyZ
deferredJellyrs rq )r rt r r r �test_deferredJellyFailure s z(PBFailureTests.test_deferredJellyFailurec s � fdd�}� � dd|�S )z�
A non-jellyable L{pb.Error} subclass raised by a remote method is
turned into a Failure with a type set to the FQPN of the exception
type.
c s � � | jd� dS )Ns3 twisted.spread.test.test_pbfailure.SynchronousError� �rH rJ ro r r r �failureUnjellyable0 s
�zBPBFailureTests.test_unjellyableFailure.<locals>.failureUnjellyabler] rv rq )r rx r r r �test_unjellyableFailure* s z&PBFailureTests.test_unjellyableFailurec s � fdd�}� � dd|�S )z�
Test that an exception which is a subclass of L{pb.Error} but not
known on the client side has its type set properly.
c s � � | jd� dS )Ns/ twisted.spread.test.test_pbfailure.UnknownError�� rw ro r r r �failureUnknown= s
�z:PBFailureTests.test_unknownFailure.<locals>.failureUnknownZ unknownErrorrz rq )r r{ r r r �test_unknownFailure8 s z"PBFailureTests.test_unknownFailurec s � fdd�}� � dd|�S )z�
Test that even if an exception is not explicitly jellyable (by being
a L{pb.Jellyable} subclass), as long as it is an L{pb.Error}
subclass it receives the same special treatment.
c s, | � t� � �| jt� � �| j| j� dS )N�� �rD r rl rJ rm rn rK ro r r r �failureSecurityJ s
z<PBFailureTests.test_securityFailure.<locals>.failureSecurityZsecurityr} rq )r r r r r �test_securityFailureD s z#PBFailureTests.test_securityFailurec s � fdd�}� � dd|�S )z�
Test that a Deferred which fails with a L{pb.Error} which is not
also a L{pb.Jellyable} is treated in the same way as a synchronously
raised exception of the same type.
c s, | � t� � �| jt� � �| j| j� dS )N� r~ ro r r r �failureDeferredSecurityX s
zEPBFailureTests.test_deferredSecurity.<locals>.failureDeferredSecurityZdeferredSecurityr� rq )r r� r r r �test_deferredSecurityR s �z$PBFailureTests.test_deferredSecurityc s � fdd�}� � dd|t�S )z�
Test that attempting to call a method which is not defined correctly
results in an AttributeError on the calling side.
c s | � tj� � �| jd� dS )NrB �� )rD r Z NoSuchMethodrE rF ro r r r �
failureNoSuchf s z>PBFailureTests.test_noSuchMethodFailure.<locals>.failureNoSuchZnosuchr� )ri �AttributeError)r r� r r r �test_noSuchMethodFailurea s z'PBFailureTests.test_noSuchMethodFailurec s6 � j �� }dd� }|�|� � fdd�}|�|� |S )z�
Test that a copied failure received from a PB call can be logged
locally.
Note: this test needs some serious help: all it really tests is that
log.err(copiedFailure) doesn't raise an exception.
c S s
| � d�S )NrZ )rQ )ZrootObjr r r � connectedw s z;PBFailureTests.test_copiedFailureLogging.<locals>.connectedc s( t �| � � �t�}� �t|�d� d S )N� )r rL rG r rH rI )ZfailurerM r r r � exception{ s
z;PBFailureTests.test_copiedFailureLogging.<locals>.exception)r< rW r* rR )r r- r� r� r r r �test_copiedFailureLoggingm s
z(PBFailureTests.test_copiedFailureLoggingc s� t �td��}t�tj|t� d��}g � � �fdd�}|� }|�d� ��t|j |� ��
t � �d� � d }��
|j t
t��d�� ��
|jd � ��
|jd
� dS ) z�
L{pb.CopiedFailure.throwExceptionIntoGenerator} will throw a
L{RemoteError} into the given paused generator at the point where it
last yielded.
�foo�Zinvokerc
3 sH z
d V W n. t jk
r8 } z� �| � W 5 d } ~ X Y n X ��d� d S )NzRemoteError not raised)r Z RemoteError�appendr )rc �r� r r r �
generatorFunc� s
zFPBFailureTests.test_throwExceptionIntoGenerator.<locals>.generatorFuncNrC r �ascii)r� rB )r �CopyableFailurer� r �unjelly� DummyInvoker�sendZ assertRaises�
StopIterationZthrowExceptionIntoGeneratorrH rI Z
remoteTyper �encode�argsZremoteTraceback)r �original�copyr� �genrc r r� r � test_throwExceptionIntoGenerator� s
z/PBFailureTests.test_throwExceptionIntoGenerator)N)r
r r r
�TestCaserH rE rX rY r[ r\ r^ ra rb ri rr ru ry r| r� r� r� r� r� r r r r rA � s$
rA c @ s e Zd ZejjZdZdS )�PBFailureUnsafeTestsrC N)r
r r r
r� Z failIfEqualsrE r: r r r r r� � s r� c @ s e Zd ZdZdZdS )r� zZ
A behaviorless object to be used as the invoker parameter to
L{jelly.jelly}.
N)r
r r r ZserializingPerspectiver r r r r� � s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �FailureJellyingTestsz:
Tests for the interaction of jelly and failures.
c C sn t �t� �}| �|�t�t� | �|�t�t� t�tj|t� d��}| �|�t�t� | �|�t�t� dS )z�
An unjellied L{CopyableFailure} has a check method which behaves the
same way as the original L{CopyableFailure}'s check method.
r� N� r r� �ZeroDivisionErrorZassertIsZcheck�ArithmeticErrorr r� r� )r r� �copiedr r r �test_unjelliedFailureCheck� s � �z/FailureJellyingTests.test_unjelliedFailureCheckc C s� t �t� �}| �|�t�t� | �|�t�t� t�tj|t� d��}t �|�}t�tj|t� d��}| �|�t�t� | �|�t�t� dS )aX
The object which results from jellying a L{CopyableFailure}, unjellying
the result, creating a new L{CopyableFailure} from the result of that,
jellying it, and finally unjellying the result of that has a check
method which behaves the same way as the original L{CopyableFailure}'s
check method.
r� Nr� )r r� Z
copiedOnceZ
derivativeZ copiedTwicer r r � test_twiceUnjelliedFailureCheck� s* ��
� � �z4FailureJellyingTests.test_twiceUnjelliedFailureCheckc C sZ t �td��}t�tj|t� d��}t� }|�|� tt�}d� |�}| �
||� � � dS )a
When L{CopiedFailure.printTraceback} is used to print a copied failure
which was unjellied from a L{CopyableFailure} with C{unsafeTracebacks}
set to C{False}, the string representation of the exception value is
included in the output.
z some reasonr� z.Traceback from remote host -- {}: some reason
N) r r� � Exceptionr r� r� r ZprintTracebackr �formatrH �getvalue)r r� r� �outputr� ZexpectedOutputr r r � test_printTracebackIncludesValue� s
�z5FailureJellyingTests.test_printTracebackIncludesValueN)r
r r r r� r� r� r r r r r� � s r� N)(r Ztwisted.internetr r Ztwisted.pythonr Ztwisted.python.compatr Ztwisted.python.reflectr Ztwisted.spreadr r r Z
twisted.trialr
r� r r r" r r Z JellyableZ
RemoteCopyr r ZsetUnjellyableForClassZglobalSecurityZallowInstancesOfZRootr ZPBServerFactoryr1 r� r4 rA r� �objectr� r� r r r r �<module> s. E& g