File: //usr/lib/python3/dist-packages/twisted/test/__pycache__/test_defgen.cpython-310.pyc
o
�b�) � @ s� d Z ddlmZmZ ddlmZmZmZmZm Z ddl
m Z ddl m
Z
ddlmZ dd� Zd d
� ZG d d � d e�ZG d
d� d�Zdd� ZG dd� dee
j�ZG dd� dee
j�ZG dd� de
j�ZdS )zI
Tests for L{twisted.internet.defer.deferredGenerator} and related APIs.
� )�defer�reactor)�Deferred�deferredGenerator�inlineCallbacks� returnValue�waitForDeferred)�runWithWarningsSuppressed)�unittest)�suppressc 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 � | � � �| 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 �
testBasics6 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 � testBuggy? s zBaseDefgenTests.testBuggyc C r )z9Test that a generator which never yields results in None.N)� _genNothingr! r" r# r r r � testNothingF s zBaseDefgenTests.testNothingc C r )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 �testHandledTerminalFailureK 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 � deferredGeneratorResultDeferredr r r � testHandledTerminalAsyncFailureS s
z/BaseDefgenTests.testHandledTerminalAsyncFailurec C r )zj
Make sure we don't blow the stack when yielding immediately
available deferreds.
r )�_genStackUsager! r" r# r r r �testStackUsage] � z BaseDefgenTests.testStackUsagec C r )zg
Make sure we don't blow the stack when yielding immediately
available values.
r )�_genStackUsage2r! r" r# r r r �testStackUsage2d r3 z BaseDefgenTests.testStackUsage2N) r r r �__doc__r% r( r* r, r0 r2 r5 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 �deprecatedDeferredGeneratorl 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 n ty; } z| �t|�d� W Y d }~nd }~ww dV d S �Nr r r )r r � getResultr" r r �str)r$ �x�ow�er r r r � 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 � � d S �Nr r# r r r r) � s �z"DeferredGeneratorTests._genNothingc c s> � t t�td���}|V z|�� W d S ty Y d S w �Nr- )r r �failr r> �r$ r@ r r r r+ � s � �z1DeferredGeneratorTests._genHandledTerminalFailurec c s4 � t |�}|V z|�� W d S ty Y d S w rF )r r> r )r$ r r@ r r r r. � s � �z6DeferredGeneratorTests._genHandledTerminalAsyncFailurec c s6 � t d�D ]}tt�d��}|V |�� }qdV d S �Ni� rD r )�ranger r �succeedr> rI r r r r1 � s �
z%DeferredGeneratorTests._genStackUsagec c s � t d�D ]}dV qdV d S rJ )rK rI r r r r4 � 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 rF �r r r r r � _genDeferred� s � zADeferredGeneratorTests.testDeferredYielding.<locals>._genDeferred)r; r& � TypeError)r$ rN r r r �testDeferredYielding� s z+DeferredGeneratorTests.testDeferredYieldingz5twisted.internet.defer.waitForDeferred was deprecatedr7 N)r r r r; r r' r) r+ r. r1 r4 rP r9 r r r r r r< ~ 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 sd � t � V }| �|d� zt� V W n ty+ } z| �t|�d� W Y d }~nd }~ww td� d S r= )r r" r r r? r )r$ r@ rB r r r r � s � �� z InlineCallbacksTests._genBasicsc c s � t � V dd d S rC rM r# r r r r' � s � z InlineCallbacksTests._genBuggyc c rE rF r r# r r r r) � s �z InlineCallbacksTests._genNothingc c s. � z t �td��V W d S ty Y d S w rG )r rH r r# r r r r+ � s � �z/InlineCallbacksTests._genHandledTerminalFailurec c s$ � z|V W d S t y Y d S w rF )r )r$ r r r r r. � s � �z4InlineCallbacksTests._genHandledTerminalAsyncFailurec c s( � t d�D ]}t�d�V qtd� d S rJ )rK r rL r rI r r r r1 s � z#InlineCallbacksTests._genStackUsagec c s" � t d�D ]}dV qtd� d S rJ )rK r rI r r r r4
s � z$InlineCallbacksTests._genStackUsage2c C � 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 � � z8InlineCallbacksTests.testYieldNonDeferred.<locals>._testrT �r r! r" )r$ rV r r r �testYieldNonDeferred s z)InlineCallbacksTests.testYieldNonDeferredc C rR )z9Ensure a standard python return results in a None result.c s s � dV d S rS r r r r r � _noReturn( s �z9InlineCallbacksTests.testReturnNoValue.<locals>._noReturnNrX )r$ rZ r r r �testReturnNoValue% � z&InlineCallbacksTests.testReturnNoValuec C rR )z Ensure that returnValue works.c s s � dV t d� d S )NrT � rU r r r r �_return3 rW z5InlineCallbacksTests.testReturnValue.<locals>._returnr] rX )r$ r^ r r r �testReturnValue0 r\ z$InlineCallbacksTests.testReturnValuec C �, 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 rS r r r r r �_noYieldA s z>InlineCallbacksTests.test_nonGeneratorReturn.<locals>._noYieldr N�r �assertInr? � assertRaisesrO �r$ ra r r r �test_nonGeneratorReturn; � z,InlineCallbacksTests.test_nonGeneratorReturnc C r` )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 rS rU r r r r ra N s zCInlineCallbacksTests.test_nonGeneratorReturnValue.<locals>._noYieldr Nrb re r r r � test_nonGeneratorReturnValueH rg z1InlineCallbacksTests.test_nonGeneratorReturnValueN)r r r r r r' r) r+ r. r1 r4 rY r[ r_ rf rh r r r r rQ � s(
rQ 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 rF r r r r r �decoratedFunctiona s �
z[DeprecateDeferredGeneratorTests.test_deferredGeneratorDeprecated.<locals>.decoratedFunctionrD r �categoryr8 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$ rj �warningsr r r rm \ s
�z@DeprecateDeferredGeneratorTests.test_deferredGeneratorDeprecatedc C sX t � }t|� | �| jg�}| �t|�d� | �|d d t� | �|d d d� dS )z3
L{waitForDeferred} is deprecated.
rD r rk r8 z�twisted.internet.defer.waitForDeferred was deprecated in Twisted 15.0.0; please use twisted.internet.defer.inlineCallbacks insteadN)r r rl � test_waitForDeferredDeprecatedr" rn ro )r$ r rp r r r rq o s
�z>DeprecateDeferredGeneratorTests.test_waitForDeferredDeprecatedN)r r r r6 rm rq r r r r ri V s ri N) r6 �twisted.internetr r �twisted.internet.deferr r r r r �twisted.python.utilr �
twisted.trialr
�twisted.trial.utilr r9 r r � Exceptionr r r; �TestCaser< rQ �SynchronousTestCaseri r r r r �<module> s >W