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