← Back
Editing: 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
Save File
Cancel