File: //lib/python3/dist-packages/twisted/internet/test/__pycache__/test_defer_yieldfrom.cpython-310.pyc
o
�b� � @ sP d Z ddlZddlmZmZmZmZ ddlmZ ddl m
Z
G dd� de
�Z dS )z/
Tests for C{yield from} support in Deferreds.
� N)�Deferred�ensureDeferred�fail�succeed)�Clock)�TestCasec @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�YieldFromTestsz@
Tests for using Deferreds in conjunction with PEP-380.
c C sJ dd� }|� }| � |tj� t|�}| � |t� | �|�}| �|d� dS )zM
L{ensureDeferred} will turn a coroutine into a L{Deferred}.
c s � � t d�} | E d H }|S �N�foo�r ��d�res� r �L/usr/lib/python3/dist-packages/twisted/internet/test/test_defer_yieldfrom.py�run � �
z/YieldFromTests.test_ensureDeferred.<locals>.runr N)�assertIsInstance�types�
GeneratorTyper r �successResultOf� assertEqual��selfr �rr r r r r �test_ensureDeferred s
z"YieldFromTests.test_ensureDeferredc sX � fdd�}dd� � |� }| � |tj� t�|�}| � |t� | �|�}| �|d� dS )zU
L{Deferred.fromCoroutine} will turn a coroutine into a L{Deferred}.
c 3 s$ � t d�} | E d H � � E d H }|S )N�barr r
��run2r r r 0 s
�
z6YieldFromTests.test_DeferredfromCoroutine.<locals>.runc s r r
r r
r r r r 6 r z7YieldFromTests.test_DeferredfromCoroutine.<locals>.run2r N)r r r r �
fromCoroutiner r r r r r �test_DeferredfromCoroutine+ s
z)YieldFromTests.test_DeferredfromCoroutinec C s. dd� }t �|� �}| �|�}| �|d� dS )ze
L{Deferred.fromCoroutine} allows a function to C{yield from} a
L{Deferred}.
c s r r
r r
r r r r M r z&YieldFromTests.test_basic.<locals>.runr N)r r r r �r r r r r r r �
test_basicG s
zYieldFromTests.test_basicc C sD dd� }t �|� �}| �|�}| �t|j�t� | �|jjd� dS )��
An exception in a generator scheduled with L{Deferred.fromCoroutine}
will cause the returned L{Deferred} to fire with a failure.
c s s � t d�} | E d H td��)Nr �Oh no!)r �
ValueError�r r r r r \ s �
z*YieldFromTests.test_exception.<locals>.run)r% N)r r �failureResultOfr �type�valuer& �argsr" r r r �test_exceptionV s
z YieldFromTests.test_exceptionc s� t � � g �� �fdd��� ��fdd�}t�|� �}� �d� | ��ddg� � �d� | ��g d �� � �d� | ��g d �� � �d� | ��g d
�� | �|�}| �|d � d S )
r$ c 3 s: � �� d� t� } � �d| jd � | E d H �� d� dS )N� � � �Yay!��appendr � callLater�callbackr' )�reactor�sectionsr r �runonen s �
z+YieldFromTests.test_twoDeep.<locals>.runonec 3 sP � �� d� �� E d H } �� d� t� }� �d|jd � |E d H �� d� | S )Nr. � � r1 )�resultr �r5 r7 r6 r r r v s �
z(YieldFromTests.test_twoDeep.<locals>.rung�������?r. r- 皙�����?)r. r- r/ r8 )r. r- r/ r8 r9 r0 N)r r r �advancer r r" r r; r � test_twoDeepf s
zYieldFromTests.test_twoDeepc s. � fdd�}� � t�|� ��}� �|d� dS )zT
Yielding from an already failed Deferred will raise the exception.
c
3 sP � z t td��E d H W dS ty' } z� �| jd� W Y d } ~ dS d } ~ ww )N�Boom)r? r. r )r r& r r+ )�e�r r r �test� s ����z)YieldFromTests.test_reraise.<locals>.testr. N)r r r r )r rB r r rA r � test_reraise� s zYieldFromTests.test_reraisec sB t � � � fdd�}t�|� �}� �d� | �|�}| �|d� dS )zi
Yielding from a paused & chained Deferred will give the result when it
has one.
c 3 sH � t � } t � � | �� fdd�� | �d � ��d� jd� | E d H }|S )Nc s � S )Nr )�ignored��d2r r �<lambda>� s z;YieldFromTests.test_chained.<locals>.test.<locals>.<lambda>r �bye)r � addCallbackr4 r3 r
�r5 rE r rB � s �
z)YieldFromTests.test_chained.<locals>.testr< rH N)r r r r= r r )r rB r r r rJ r � test_chained� s
zYieldFromTests.test_chainedN) �__name__�
__module__� __qualname__�__doc__r r! r# r, r> rC rK r r r r r s + r ) rO r �twisted.internet.deferr r r r �twisted.internet.taskr �twisted.trial.unittestr r r r r r �<module> s