File: //lib/python3/dist-packages/twisted/test/__pycache__/test_defgen.cpython-38.pyc
U
W[�) � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z m
Z
m Z ddlm Z m
Z
ddlmZ ddlmZ dd lmZ d
d � Zd d
� ZG dd� de�ZG dd� d�Zdd� ZG dd� deej�ZG dd� deej�Z G dd� dej �Z dS )zI
Tests for L{twisted.internet.defer.deferredGenerator} and related APIs.
� )�division�absolute_import)�reactor)�unittest)�waitForDeferred�deferredGenerator�Deferred)�inlineCallbacks� returnValue)�defer)�suppress)�runWithWarningsSuppressedc C s t � } t�d| jd� | S )Nr �hi)r r � callLater�callback��d� r �:/usr/lib/python3/dist-packages/twisted/test/test_defgen.py�getThing s r c s" t � � � fdd�} t�d| � � S )Nc s � � td�� d S )N�OMG)�errback�ZeroDivisionErrorr r r r �CRAP s zgetOwie.<locals>.CRAPr )r r r )r r r r �getOwie s r c @ s e Zd ZdS )�TerminalExceptionN)�__name__�
__module__� __qualname__r r r r r $ s r c @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�BaseDefgenTestsa
This class sets up a bunch of test cases which will test both
deferredGenerator and inlineCallbacks based generators. The subclasses
DeferredGeneratorTests and InlineCallbacksTests each provide the actual
generator implementations tested.
c C s | � � �| jd�S )z�
Test that a normal deferredGenerator works. Tests yielding a
deferred which callbacks, as well as a deferred errbacks. Also
ensures returning a final value works.
�WOOSH)�
_genBasics� addCallback� assertEqual��selfr r r �
testBasics/ s zBaseDefgenTests.testBasicsc C s | � | �� t�S )zp
Ensure that a buggy generator properly signals a Failure
condition on result deferred.
)�
assertFailure� _genBuggyr r$ r r r � testBuggy8 s zBaseDefgenTests.testBuggyc C s | � � �| jd�S )z9Test that a generator which never yields results in None.N)� _genNothingr" r# r$ r r r � testNothing? s zBaseDefgenTests.testNothingc C s | � � �| jd�S )z�
Create a Deferred Generator which yields a Deferred which fails and
handles the exception which results. Assert that the Deferred
Generator does not errback its Deferred.
N)�_genHandledTerminalFailurer" r# r$ r r r �testHandledTerminalFailureD s z*BaseDefgenTests.testHandledTerminalFailurec C s. t �� }| �|�}|�td�� |�| jd�S )z~
Just like testHandledTerminalFailure, only with a Deferred which fires
asynchronously with an error.
�Handled Terminal FailureN)r r � _genHandledTerminalAsyncFailurer r r" r# )r% r Z deferredGeneratorResultDeferredr r r � testHandledTerminalAsyncFailureL s
�z/BaseDefgenTests.testHandledTerminalAsyncFailurec C s | � � �| jd�S )zj
Make sure we don't blow the stack when yielding immediately
available deferreds.
r )�_genStackUsager" r# r$ r r r �testStackUsageW s z BaseDefgenTests.testStackUsagec C s | � � �| jd�S )zg
Make sure we don't blow the stack when yielding immediately
available values.
r )�_genStackUsage2r" r# r$ r r r �testStackUsage2^ s z BaseDefgenTests.testStackUsage2N) r r r �__doc__r& r) r+ r- r0 r2 r4 r r r r r ' s r c C s t tdd�gt| �S )z�
Calls L{deferredGenerator} while suppressing the deprecation warning.
@param f: Function to call
@return: Return value of function.
z7twisted.internet.defer.deferredGenerator was deprecated��message)r
�SUPPRESSr )�fr r r �deprecatedDeferredGeneratorg s
�r: c @ s| e Zd Zedd� �Zedd� �Zedd� �Zedd� �Zed d
� �Zd d � Z ee �Z d
d� Z
ee
�Z
dd� Z e dd�gZ
dS )�DeferredGeneratorTestsc
c s~ t t� �}|V |�� }| �|d� t t� �}|V z |�� W n2 tk
rr } z| �t|�d� W 5 d }~X Y nX dV d S �Nr r r )r r � getResultr# r r �str)r% �xZow�er r r r! x s
"z!DeferredGeneratorTests._genBasicsc c s t t� �V dd d S �N� r )r r r$ r r r r( � s z DeferredGeneratorTests._genBuggyc c s d S �Nr r$ r r r r* � s z"DeferredGeneratorTests._genNothingc c s> t t�td���}|V z |�� W n tk
r8 Y nX d S �Nr. )r r �failr r= �r% r? r r r r, � s z1DeferredGeneratorTests._genHandledTerminalFailurec c s4 t |�}|V z |�� W n tk
r. Y nX d S rC )r r= r )r% r r? r r r r/ � s z6DeferredGeneratorTests._genHandledTerminalAsyncFailurec c s4 t d�D ] }tt�d��}|V |�� }qdV d S �Ni� rB r )�ranger r �succeedr= rF r r r r1 � s
z%DeferredGeneratorTests._genStackUsagec c s t d�D ]
}dV qdV d S rG )rH rF r r r r3 � s z&DeferredGeneratorTests._genStackUsage2c C s dd� }t |�}| �|� t�S )zZ
Ensure that yielding a Deferred directly is trapped as an
error.
c s s t � V d S rC �r r r r r � _genDeferred� s zADeferredGeneratorTests.testDeferredYielding.<locals>._genDeferred)r: r' � TypeError)r% rK r r r �testDeferredYielding� s z+DeferredGeneratorTests.testDeferredYieldingz5twisted.internet.defer.waitForDeferred was deprecatedr6 N)r r r r: r! r( r* r, r/ r1 r3 rM r8 r r r r r r; u s"
�r; c @ s� e Zd Zdd� Zee�Zdd� Zee�Zdd� Zee�Zdd� Zee�Zd d
� Zee�Zd d � Z ee �Z d
d� Z
ee
�Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�InlineCallbacksTestsc
c s` t � V }| �|d� z t� V W n2 tk
rR } z| �t|�d� W 5 d }~X Y nX td� d S r< )r r# r r r> r
)r% r? r@ r r r r! � s "z InlineCallbacksTests._genBasicsc c s t � V dd d S rA rJ r$ r r r r( � s z InlineCallbacksTests._genBuggyc c s d S rC r r$ r r r r* � s z InlineCallbacksTests._genNothingc c s. zt �td��V W n tk
r( Y nX d S rD )r rE r r$ r r r r, � s z/InlineCallbacksTests._genHandledTerminalFailurec c s$ z
|V W n t k
r Y nX d S rC )r )r% r r r r r/ � s
z4InlineCallbacksTests._genHandledTerminalAsyncFailurec c s& t d�D ]}t�d�V qtd� d S rG )rH r rI r
rF r r r r1 � s z#InlineCallbacksTests._genStackUsagec c s t d�D ]
}dV qtd� d S rG )rH r
rF r r r r3 s z$InlineCallbacksTests._genStackUsage2c C s dd� }t |�}|� �| jd�S )z�
Ensure that yielding a non-deferred passes it back as the
result of the yield expression.
@return: A L{twisted.internet.defer.Deferred}
@rtype: L{twisted.internet.defer.Deferred}
c s s dV t d� d S �N� �r
r r r r �_test s z8InlineCallbacksTests.testYieldNonDeferred.<locals>._testrP �r r" r# )r% rR r r r �testYieldNonDeferred s z)InlineCallbacksTests.testYieldNonDeferredc C s dd� }t |�}|� �| jd�S )z9Ensure a standard python return results in a None result.c s s
dV d S rO r r r r r � _noReturn s z9InlineCallbacksTests.testReturnNoValue.<locals>._noReturnNrS )r% rU r r r �testReturnNoValue s z&InlineCallbacksTests.testReturnNoValuec C s dd� }t |�}|� �| jd�S )z Ensure that returnValue works.c s s dV t d� d S )NrP � rQ r r r r �_return# s z5InlineCallbacksTests.testReturnValue.<locals>._returnrW rS )r% rX r r r �testReturnValue! s z$InlineCallbacksTests.testReturnValuec C s, dd� }t |�}| �dt| �t|��� dS )z�
Ensure that C{TypeError} with a message about L{inlineCallbacks} is
raised when a non-generator returns something other than a generator.
c S s dS rO r r r r r �_noYield0 s z>InlineCallbacksTests.test_nonGeneratorReturn.<locals>._noYieldr N�r ZassertInr> Z assertRaisesrL �r% rZ r r r �test_nonGeneratorReturn+ s
�z,InlineCallbacksTests.test_nonGeneratorReturnc C s, dd� }t |�}| �dt| �t|��� dS )z�
Ensure that C{TypeError} with a message about L{inlineCallbacks} is
raised when a non-generator calls L{returnValue}.
c S s t d� d S rO rQ r r r r rZ = s zCInlineCallbacksTests.test_nonGeneratorReturnValue.<locals>._noYieldr Nr[ r\ r r r � test_nonGeneratorReturnValue8 s
�z1InlineCallbacksTests.test_nonGeneratorReturnValueN)r r r r! r r( r* r, r/ r1 r3 rT rV rY r] r^ r r r r rN � s* �
rN c @ s e Zd ZdZdd� Zdd� ZdS )� DeprecateDeferredGeneratorTestszY
Tests that L{DeferredGeneratorTests} and L{waitForDeferred} are
deprecated.
c C sV t dd� �}| �| jg�}| �t|�d� | �|d d t� | �|d d d� dS ) z5
L{deferredGenerator} is deprecated.
c s s
d V d S rC r r r r r �decoratedFunctionP s z[DeprecateDeferredGeneratorTests.test_deferredGeneratorDeprecated.<locals>.decoratedFunctionrB r �categoryr7 z�twisted.internet.defer.deferredGenerator was deprecated in Twisted 15.0.0; please use twisted.internet.defer.inlineCallbacks insteadN)r �
flushWarnings� test_deferredGeneratorDeprecatedr# �len�DeprecationWarning)r% r` �warningsr r r rc L s
�z@DeprecateDeferredGeneratorTests.test_deferredGeneratorDeprecatedc C sX t � }t|� | �| jg�}| �t|�d� | �|d d t� | �|d d d� dS )z3
L{waitForDeferred} is deprecated.
rB r ra r7 z�twisted.internet.defer.waitForDeferred was deprecated in Twisted 15.0.0; please use twisted.internet.defer.inlineCallbacks insteadN)r r rb � test_waitForDeferredDeprecatedr# rd re )r% r rf r r r rg ] s
�z>DeprecateDeferredGeneratorTests.test_waitForDeferredDeprecatedN)r r r r5 rc rg r r r r r_ F s r_ N) r5 Z
__future__r r Ztwisted.internetr Z
twisted.trialr Ztwisted.internet.deferr r r r r
r Ztwisted.trial.utilr r8 Ztwisted.python.utilr
r r � Exceptionr r r: ZTestCaser; rN ZSynchronousTestCaser_ r r r r �<module> s
@Xy