Your IP : 216.73.216.213
3
��2aOO � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl
m Z d dl
Z
d dl
mZ d dl
mZ d dl
mZ d dl
mZ yd dlmZ W n ek
r� d dl
mZ Y nX yd d lmZ W nJ ek
�r0 yd d lmZ W n" ek
�r* d d lmZ Y nX Y nX ejd5kZejd6kZ e
j d
d� �Z ej dd� �Z d7dd�Z!G dd� d�Z"G dd� d�Z#dd� Z$e
j%e&ed�d�G d d � d e#ej'��Z(e
j%e&ed�d�e$G d d � d e#ej'���Z)e
j%e&ed�d�G d d!� d!e#ej'��Z*e
j%e&ed�d�G d"d#� d#e#ej'��Z+G d$d%� d%e#ej'�Z,e$G d&d'� d'e#ej'��Z-G d(d)� d)ej'�Z.G d*d+� d+�Z/G d,d-� d-e/ej'�Z0G d.d/� d/e/ej'�Z1G d0d1� d1ej'�Z2G d2d3� d3ej'�Z3e4d4k�r�e
j5� dS )8� N)�mock)�
coroutines)�futures)�tasks)�
test_utils)�support)� test_support)�assert_python_ok� � � c C s d S )N� r
r
r
�//opt/alt/python36/lib64/python3.6/test_tasks.py�coroutine_function' s r c c s* t j}|j}z| |_d V W d ||_X d S )N)�asyncior �_DEBUG)Zenabledr Z old_debugr
r
r �set_coroutine_debug, s
r Fc C sR |rd| }nd| }|d k r@|d }d| |||d |d f S d| ||f S d S )Nz%sz %s, defined� z%coro=<%s() %s at %s> created at %s:%sr zcoro=<%s() %s at %s>���r
)�qualname�state�src�source_traceback� generator�framer
r
r �format_coroutine9 s
r c @ s e Zd Zdd� Zdd� ZdS )�Dummyc C s dS )Nz<Dummy>r
)�selfr
r
r �__repr__H s zDummy.__repr__c G s d S )Nr
)r �argsr
r
r �__call__K s zDummy.__call__N)�__name__�
__module__� __qualname__r r r
r
r
r r F s r c sB e Zd ZdZdZdd� Zdd� Z� fdd�Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zejed�dd� �Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Z d-d.� Z d/d0� Z d1d2� Z d3d4� Z d5d6� Z!d7d8� Z"d9d:� Z#d;d<� Z$d=d>� Z%d?d@� Z&dAdB� Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+dKdL� Z,dMdN� Z-dOdP� Z.dQdR� Z/dSdT� Z0dUdV� Z1dWdX� Z2dYdZ� Z3d[d\� Z4d]d^� Z5d_d`� Z6dadb� Z7dcdd� Z8dedf� Z9dgdh� Z:didj� Z;dkdl� Z<dmdn� Z=dodp� Z>dqdr� Z?dsdt� Z@dudv� ZAdwdx� ZBdydz� ZCd{d|� ZDd}d~� ZEdd�� ZFd�d�� ZGd�d�� ZHd�d�� ZId�d�� ZJd�d�� ZKd�d�� ZLd�d�� ZMd�d�� ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd�d�� ZVd�d�� ZWd�d�� ZXd�d�� ZYejeZd��d�d�� �Z[e\j]d��d�d�� �Z^e\j]d��d�d�� �Z_d�d�� Z`d�d�� Zad�d�� Zbd�d�� Zcd�d�� Zdd�d�� Zed�d�� Zfd�d�� Zge\j]d��d�d�� �Zh� ZiS )��
BaseTaskTestsNc C s | j j||d�S )N)�loop)� __class__�Task)r r% �coror
r
r �new_taskT s zBaseTaskTests.new_taskc C s | j j|d�S )N)r% )r& �Future)r r% r
r
r �
new_futureW s zBaseTaskTests.new_futurec s6 t � j� � j� � _� jj� j� � fdd�� j_d S )Nc s � j � j�S )N)r+ r% r
)r r
r �<lambda>^ s z%BaseTaskTests.setUp.<locals>.<lambda>)�super�setUp�
new_test_loopr% �set_task_factoryr) Z
create_future)r )r& )r r r. Z s
zBaseTaskTests.setUpc sR G dd� d�� t j� fdd����jjt j��fdd�td�D �d�ji�� d S ) Nc @ s e Zd Zdd� ZdS )z1BaseTaskTests.test_task_del_collect.<locals>.Evilc S s t j� d S )N)�gcZcollect)r r
r
r �__del__b s z9BaseTaskTests.test_task_del_collect.<locals>.Evil.__del__N)r! r" r# r2 r
r
r
r �Evila s r3 c s � � S )Nr
r
)r3 r
r �rune s z0BaseTaskTests.test_task_del_collect.<locals>.runc s g | ]}�j �j� � ��qS r
)r) r% )�.0�_)r4 r r
r �
<listcomp>k s z7BaseTaskTests.test_task_del_collect.<locals>.<listcomp>�d r% )r � coroutiner% �run_until_complete�gather�range)r r
)r3 r4 r r �test_task_del_collect` s z#BaseTaskTests.test_task_del_collectc C s\ t j� }| j|�}t jdd� �}z,| jtd�� | jj||�� W d Q R X W d |j� X d S )Nc s s | E d H d S )Nr
)�futr
r
r r4 r s z1BaseTaskTests.test_other_loop_future.<locals>.runz Task .* got Future .* attached) r �new_event_loopr+ r9 �assertRaisesRegex� RuntimeErrorr% r: �close)r �
other_loopr> r4 r
r
r �test_other_loop_futuren s
z$BaseTaskTests.test_other_loop_futurec sL t j� fdd��}t j|� | jd�� | jtd�� | jj� � W d Q R X d S )Nc 3 s � E d H d S )Nr
r
)�taskr
r �test~ s z6BaseTaskTests.test_task_awaits_on_itself.<locals>.test)r% zTask cannot await on itself)r r9 �
ensure_futurer% r@ rA r: )r rF r
)rE r �test_task_awaits_on_itself} s
z(BaseTaskTests.test_task_awaits_on_itselfc C s� t jdd� �}| j| j|� �}| jj|� | j|j� � | j|j� d� | j |j
| j� t j � }| j |� | j||� �}| j |j
|� |j|� |j
� d S )Nc S s dS )N�okr
r
r
r
r �notmuch� s z.BaseTaskTests.test_task_class.<locals>.notmuchrI )r r9 r) r% r: �
assertTrue�done� assertEqual�result�assertIs�_loopr? �set_event_looprB )r rJ �tr% r
r
r �test_task_class� s
z BaseTaskTests.test_task_classc C s� t jdd� �}t j|� | jd�}| jj|� | j|j� � | j|j� d� | j |j
| j� t j � }| j |� t j|� |d�}| j |j
|� |j|� |j
� d S )Nc S s dS )NrI r
r
r
r
r rJ � s z;BaseTaskTests.test_ensure_future_coroutine.<locals>.notmuch)r% rI )r r9 rG r% r: rK rL rM rN rO rP r? rQ rB )r rJ rR r% r
r
r � test_ensure_future_coroutine� s
z*BaseTaskTests.test_ensure_future_coroutinec C s� | j | j�}|jd� tj|�}| jj|� | j|j� � | j|j � d� | j
||� tj � }| j |� | j
t�� tj||d�}W d Q R X |j� tj|| jd�}| j
||� d S )N�ko)r% )r+ r% �
set_resultr rG r: rK rL rM rN rO r? rQ � assertRaises�
ValueErrorrB )r Zf_orig�fr% r
r
r �test_ensure_future_future� s
z'BaseTaskTests.test_ensure_future_futurec C s� t jdd� �}| j| j|� �}t j|�}| jj|� | j|j� � | j|j � d� | j
||� t j � }| j |� | j
t�� t j||d�}W d Q R X |j� t j|| jd�}| j
||� d S )Nc S s dS )NrI r
r
r
r
r rJ � s z6BaseTaskTests.test_ensure_future_task.<locals>.notmuchrI )r% )r r9 r) r% rG r: rK rL rM rN rO r? rQ rW rX rB )r rJ Zt_origrR r% r
r
r �test_ensure_future_task� s
z%BaseTaskTests.test_ensure_future_taskzneed python 3.5 or laterc C sP G dd� d�}t jdd� �}t j� }| j|� t j||� �|d�}|j|� d S )Nc @ s e Zd Zdd� Zdd� ZdS )z6BaseTaskTests.test_ensure_future_awaitable.<locals>.Awc S s
|| _ d S )N)r( )r r( r
r
r �__init__� s z?BaseTaskTests.test_ensure_future_awaitable.<locals>.Aw.__init__c s s | j E d H S )N)r( )r r
r
r � __await__� s z@BaseTaskTests.test_ensure_future_awaitable.<locals>.Aw.__await__N)r! r" r# r\ r] r
r
r
r �Aw� s r^ c S s dS )NrI r
r
r
r
r r( � s z8BaseTaskTests.test_ensure_future_awaitable.<locals>.coro)r% )r r9 r? rQ rG r: )r r^ r( r% r> r
r
r � test_ensure_future_awaitable� s
z*BaseTaskTests.test_ensure_future_awaitablec C s$ | j t�� tjd� W d Q R X d S )NrI )rW � TypeErrorr rG )r r
r
r �test_ensure_future_neither� s z(BaseTaskTests.test_ensure_future_neitherc C s: | j | j�}| jtd�� | j|tj|�� W d Q R X d S )Nz#function is deprecated, use ensure_)r+ r% ZassertWarnsRegex�DeprecationWarningrO r �async)r rY r
r
r �test_async_warning� s z BaseTaskTests.test_async_warningc sR d � t j�fdd���t j� �fdd���t j� ��fdd��}�jj|� � d S )Nc 3 s � � E d H d S )Nr
r
)�barr
r �foo� s z)BaseTaskTests.test_get_stack.<locals>.fooc sd � j dd�} z�j| d jjd� W d d } X tj� }� jd|d� |jd� |j� }�j |d� d S )Nr )�limitr rf )rg �filezfoo\(\) running)
Z get_stackrM �f_code�co_name�io�StringIO� print_stack�seek�read� assertRegex)rY rh �tb)�Tr r
r re � s
z)BaseTaskTests.test_get_stack.<locals>.barc 3 s t j�� �jd�� � E d H d S )N)r% )r rG r% r
)rr rf r r
r �runner s z,BaseTaskTests.test_get_stack.<locals>.runner)r r9 r% r: )r rs r
)rr re rf r r �test_get_stack� s
z BaseTaskTests.test_get_stackc C sv | j jd� tjdd� �}| j|jd� tr:| j|jd� | j|j t� t
j |�\}}d||f }|� }t j
srtrxd}nd}| j|jd� tr�| j|j|� | j| j |�}|jt� � t|d||jdd �}| jt|�d
| � |j� | jt|�d | � | jtj| j j|� t|d ||j�}| jt|�d
| � | j| j |� �}| j j|� t|d ||j�}| jt|�d| � d S )NFc s s g E d H dS )N�abcr
r
r
r
r rJ s
z-BaseTaskTests.test_task_repr.<locals>.notmuchrJ z#\w+.test_task_repr.<locals>.notmuchz%s:%s�runningT)r z <Task pending %s cb=[<Dummy>()]>z#<Task cancelling %s cb=[<Dummy>()]>rL z<Task cancelled %s>z <Task finished %s result='abc'>)r% � set_debugr r9 rM r! �PY35rp r# r" r �get_function_sourcer r r) �add_done_callbackr r �_source_traceback�repr�cancelrW �CancelledErrorr: ) r rJ �filename�linenor �gen�
coro_qualnamerR r( r
r
r �test_task_repr sL
z BaseTaskTests.test_task_reprc C sz | j jd� tjdd� �}| j|jd� tr:| j|jd� | j|j t� |� }t
j sXtrbd}d}nd }}| j|j|� tr�| j|j|� t
j r�t
j r�t j
|�\}}|jd }d||||d |d f }n|j} d
|| j| jf }| jt|�d | � | j| j |�}
|
jt� � t
j �r(d t j
|� } n|j} d | j| jf } t|d
| |
jt
j d�}| jt|
�d| � | j j|
� d S )NFc S s dS )N�{ r
r
r
r
r rJ N s z<BaseTaskTests.test_task_repr_coro_decorator.<locals>.notmuchrJ z4\w+.test_task_repr_coro_decorator\.<locals>\.notmuchr( r z0%s() running, defined at %s:%s, created at %s:%sr z%s() running at %s:%sz<CoroWrapper %s>z%s:%srv )r z <Task pending %s cb=[<Dummy>()]>r )r% rw r r9 rM r! rx rp r# r" r r r ry r{ �gi_code� co_filename�co_firstlinenor| r) rz r r r: ) r rJ r� Z coro_namer� r r� r r( �coderR r r
r
r � test_task_repr_coro_decoratorK sN
z+BaseTaskTests.test_task_repr_coro_decoratorc C s| | j jd� tjdd� �}| j| j �}| j| j ||��}tj| j � | jt |�dt
j t |�� � |j d � | j j
|� d S )NFc s s
| E d H S )Nr
)r> r
r
r �wait_for� s z7BaseTaskTests.test_task_repr_wait_for.<locals>.wait_forz<Task .* wait_for=%s>)r% rw r r9 r+ r) r � run_brieflyrp r| �re�escaperV r: )r r� r> rE r
r
r �test_task_repr_wait_for� s
z%BaseTaskTests.test_task_repr_wait_forc C s� t d��V | jjd� tjdd� �}tjtj|d��}| jj|d��}d|_| j |j
j � W d Q R X t |j
�}d}| j
||� d S )NTc s s t jd�E d H d S )Nr )r �sleep)�x�yr
r
r �func� s z>BaseTaskTests.test_task_repr_partial_corowrapper.<locals>.funcr � FzV<CoroWrapper \w+.test_task_repr_partial_corowrapper\.<locals>\.func\(1\)\(\) running, )r r% rw r r9 � functools�partial� create_taskZ_log_destroy_pending�
addCleanupZ_cororB r| rp )r r� Z partial_funcrE Z coro_repr�expectedr
r
r �"test_task_repr_partial_corowrapper� s
z0BaseTaskTests.test_task_repr_partial_corowrapperc sN t j� �fdd��}t jdd� �� t jdd� ��|� }| j| jj|�d� d S )Nc 3 s � � E d H } �� E d H }| | S )Nr
)�a�b)�inner1�inner2r
r �outer� s z-BaseTaskTests.test_task_basics.<locals>.outerc S s dS )N�* r
r
r
r
r r� � s z.BaseTaskTests.test_task_basics.<locals>.inner1c S s dS )Ni� r
r
r
r
r r� � s z.BaseTaskTests.test_task_basics.<locals>.inner2i )r r9 rM r% r: )r r� rR r
)r� r� r �test_task_basics� s
z BaseTaskTests.test_task_basicsc s� �fdd�}�j |�� tj� fdd��}�j� |� �}� j|j� �jtj�� � j|� W d Q R X �j |j
� � �j |j � � �j |j� � d S )Nc 3 s d V } � j d| � dV d S )Ng $@r )�assertAlmostEqual)�when)r r
r r� � s z&BaseTaskTests.test_cancel.<locals>.genc 3 s t jd� d�E d H dS )Ng $@)r% � )r r� r
)r% r
r rE � s z'BaseTaskTests.test_cancel.<locals>.task)
r/ r r9 r) � call_soonr} rW r~ r: rK rL � cancelled� assertFalse)r r� rE rR r
)r% r r � test_cancel� s
zBaseTaskTests.test_cancelc C st t jdd� �}| j| j|� �}tj| j� |j� | jt j| jj |� | j
|j � � | j
|j � � | j
|j� � d S )Nc s s d V d V dS )Nr� r
r
r
r
r rE � s z-BaseTaskTests.test_cancel_yield.<locals>.task)r r9 r) r% r r� r} rW r~ r: rK rL r� r� )r rE rR r
r
r �test_cancel_yield� s z BaseTaskTests.test_cancel_yieldc s� | j | j�� tj� fdd��}| j| j|� �}tj| j� � j� | jtj �� | jj
|� W d Q R X | j � j � � | j |j � � d S )Nc 3 s � E d H dS )Nr� r
r
)rY r
r rE � s
z4BaseTaskTests.test_cancel_inner_future.<locals>.task)
r+ r% r r9 r) r r� r} rW r~ r: rK r� )r rE rR r
)rY r �test_cancel_inner_future� s z&BaseTaskTests.test_cancel_inner_futurec s� | j | j�� tj� fdd��}| j| j|� �}tj| j� � j� |j� | jtj �� | jj
|� W d Q R X | j |j � � | j � j
� � | j |j
� � d S )Nc 3 s � E d H dS )Nr� r
r
)rY r
r rE s
zBBaseTaskTests.test_cancel_both_task_and_inner_future.<locals>.task)r+ r% r r9 r) r r� r} rW r~ r: rK rL r� )r rE rR r
)rY r �&test_cancel_both_task_and_inner_future� s z4BaseTaskTests.test_cancel_both_task_and_inner_futurec s� | j | j�� | j | j��tj� �fdd��}| j| j|� �}tj| j� | j|j� � � j d � tj| j� | j|j�� |j
� | j �j � � | jj
|�}| j|d� | j|j � � d S )Nc 3 s4 � E d H y�E d H W n t jk
r. dS X d S )Nr� )r r~ r
)�fut1�fut2r
r rE s
z5BaseTaskTests.test_cancel_task_catching.<locals>.taskr� )r+ r% r r9 r) r r� rO � _fut_waiterrV r} rK r� r: rM r� )r rE rR �resr
)r� r� r �test_cancel_task_catching s
z'BaseTaskTests.test_cancel_task_catchingc s� | j | j�� | j | j��| j | j��tj� ��fdd��}| j| j|� �}tj| j� | j|j� � � j d � tj| j� | j|j�� |j
� | j �j � � tj| j� | j|j�� �j d� | jj
|�}| j|d� | j�j � � | j|j � � d S )Nc 3 s> � E d H y�E d H W n t jk
r. Y nX �E d H } | S )N)r r~ )r� )r� r� �fut3r
r rE 1 s
z5BaseTaskTests.test_cancel_task_ignoring.<locals>.taskr� )r+ r% r r9 r) r r� rO r� rV r} rK r� r: rM r� )r rE rR r� r
)r� r� r� r �test_cancel_task_ignoring, s&
z'BaseTaskTests.test_cancel_task_ignoringc st t j� � �j� � t j� ��fdd��}�j� |� ���jt j� j�� �j�j � � �j
�j � �j
�j � � d S )Nc 3 s, �j � �j�j� tjd� d�E d H dS )Nr8 )r% r� )r} rK � _must_cancelr r� r
)r% r rR r
r rE O s z4BaseTaskTests.test_cancel_current_task.<locals>.task)
r r? rQ r9 r) rW r~ r: rK rL r� r� r} )r rE r
)r% r rR r �test_cancel_current_taskK s
z&BaseTaskTests.test_cancel_current_taskc sr t j� }� j|� t j� �fdd��}� j||� ��� jt j|j�� � j�j � � � j
�j � � j
�j � � d S )Nc s �j � � j�j� dS )Nr� )r} rK r� r
)r rR r
r rE c s z.BaseTaskTests.test_cancel_at_end.<locals>.task)
r r? rQ r9 r) rW r~ r: rK rL r� r� r} )r r% rE r
)r rR r �test_cancel_at_end^ s
z BaseTaskTests.test_cancel_at_endc s� t j� ��j�fdd�� d ���j��� � fdd�������fdd�}�j�|� ���jt j�� �j�� W d Q R X �j�j � � �j�j � � �j� j � � d S )Nc s � j � S )N)rB r
)r% r
r r, x s z8BaseTaskTests.test_cancel_awaited_task.<locals>.<lambda>c � s � I d H d S )Nr
r
)r> r
r �nested} s z6BaseTaskTests.test_cancel_awaited_task.<locals>.nestedc � s4 �j � �� ��tjd�I d H �j� �I d H d S )Nr )r) r r� r} r
)r% r� � nested_taskr rE r
r r( � s z4BaseTaskTests.test_cancel_awaited_task.<locals>.coro)
r r? r� r+ r) rW r~ r: rK r� )r r( r
)r> r% r� r� r rE r �test_cancel_awaited_taskp s
z&BaseTaskTests.test_cancel_awaited_taskc s� �fdd�}�j |�� d�g �tj� ��fdd��}�j� |� �}�jt��}� j|� W d Q R X �jt|j �d� �j
|j � � �j�d� �j d� j
� � x�D ] }|j� q�W |j� �jtj� j|� d S ) Nc 3 s@ d V } � j d| � dV } � j d| � dV } � j d| � dV d S )Ng�������?g�������?g333333�?)r� )r� )r r
r r� � s z:BaseTaskTests.test_stop_while_run_in_complete.<locals>.genr c 3 sL xF�dk rF�j tjd� d�� �d E d H �d7 ��dkr� j� qW d S )N�
g�������?)r% r r� r )�appendr r� �stopr
)r% �waitersr� r
r rE � s
z;BaseTaskTests.test_stop_while_run_in_complete.<locals>.taskz+Event loop stopped before Future completed.r� g333333�?)r/ r r9 r) rW rA r: rM �str� exceptionr� rL r� �timerB r} r~ )r r� rE rR �cm�wr
)r% r r� r� r � test_stop_while_run_in_complete� s"
z-BaseTaskTests.test_stop_while_run_in_completec
s� �fdd�}�j |��d � tj� �fdd��}�j�|� �}�jtj�� �jtj|d�d�� W d Q R X �j|j � � �j|j
� � �j d�j � � �j
� d� d S )Nc 3 s. d V } � j d| � dV } � j d| � dV } d S )Ng�������?r g�������?)r� )r� )r r
r r� � s
z(BaseTaskTests.test_wait_for.<locals>.genc
3 s( d� zt jd�d�E d H W d d� X dS )NTg�������?)r% FrL )r r� r
)� foo_runningr% r
r rf � s
z(BaseTaskTests.test_wait_for.<locals>.foog�������?)r% F)r/ r r9 r) rW � TimeoutErrorr: r� rK rL r� r� r� rM )r r� rf r> r
)r� r% r r �
test_wait_for� s
zBaseTaskTests.test_wait_forc C s> | j � }tjdd� �}|jtj|� d |d��}| j|d� d S )Nc S s dS )NrL r
r
r
r
r r( � s z2BaseTaskTests.test_wait_for_blocking.<locals>.coro)�timeoutr% rL )r/ r r9 r: r� rM )r r% r( r� r
r
r �test_wait_for_blocking� s
z$BaseTaskTests.test_wait_for_blockingc s� �fdd�}�j |�� tj� fdd��}tj� � z<�j� |� �}�jtj�� � jtj|d�� W d Q R X W d tjd � X �j d� j
� � �j |j � � �j |j
� � d S )Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?r g{�G�z�?)r� )r� )r r
r r� � s
z9BaseTaskTests.test_wait_for_with_global_loop.<locals>.genc 3 s t jd� d�E d H dS )Ng�������?)r% rL )r r� r
)r% r
r rf � s z9BaseTaskTests.test_wait_for_with_global_loop.<locals>.foog{�G�z�?)r/ r r9 rQ r) rW r� r: r� r� r� rK rL r� )r r� rf r> r
)r% r r � test_wait_for_with_global_loop� s
z,BaseTaskTests.test_wait_for_with_global_loopc C sV dd� }| j |�}| j|�}tj|d|d�}|jd|jd� |j|�}| j|d� d S )Nc s s dV dV dV d S )Ng�������?r
r
r
r
r r� s z7BaseTaskTests.test_wait_for_race_condition.<locals>.geng�������?)r� r% g�������?rI )r/ r+ r r� �
call_laterrV r: rM )r r� r% r> rE r� r
r
r � test_wait_for_race_condition s
z*BaseTaskTests.test_wait_for_race_conditionc s� �fdd�}�j |���j�tjd�d��� �j�tjd�d���tj� ���fdd��}�j�j�|� ��}�j|d� �jd�j� � �j�j�|� ��}�jd�j� � �j|d� d S ) Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?r g333333�?)r� )r� )r r
r r� s
z$BaseTaskTests.test_wait.<locals>.geng�������?)r% g333333�?c 3 sB t j�� g�d�E d H \} }�j| t� �g�� �j|t� � dS )N)r% r� )r �waitrM �set)rL �pending)r� r� r% r r
r rf # s z$BaseTaskTests.test_wait.<locals>.foor� ) r/ r) r r� r9 r: rM r� r� )r r� rf r� r
)r� r� r% r r � test_wait s
zBaseTaskTests.test_waitc s� �fdd�}�j |�}�j|tjd|d��� �j|tjd|d���tj� ��fdd��}tj|� |j�j||� ��}�j|d� d S ) Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng{�G�z�?r g� ��Q��?)r� )r� )r r
r r� 5 s
z5BaseTaskTests.test_wait_with_global_loop.<locals>.geng{�G�z�?)r% g� ��Q��?c 3 s> t j�� g�E d H \} }�j| t� �g�� �j|t� � dS )Nr� )r r� rM r� )rL r� )r� r� r r
r rf A s z5BaseTaskTests.test_wait_with_global_loop.<locals>.foor� )r/ r) r r� r9 rQ r: rM )r r� r% rf r� r
)r� r� r r �test_wait_with_global_loop3 s
z(BaseTaskTests.test_wait_with_global_loopc C sv t jdd� �}|d�}| j| jt j|||d�g| jd��}| jj|�\}}| j|� | jtdd� |D ��ddh� d S )Nc S s | S )Nr
)�sr
r
r r( O s z:BaseTaskTests.test_wait_duplicate_coroutines.<locals>.cororF �spam)r% c s s | ]}|j � V qd S )N)rN )r5 rY r
r
r � <genexpr>[ s z?BaseTaskTests.test_wait_duplicate_coroutines.<locals>.<genexpr>) r r9 r) r% r� r: r� rM r� )r r( �crE rL r� r
r
r � test_wait_duplicate_coroutinesN s
z,BaseTaskTests.test_wait_duplicate_coroutinesc C sb | j t| jjtjt� | jd�� tjd| jd�}tj|gd| jd�}| j t| jj|� |j� d S )N)r% g $@r )� return_whenr% r ) rW rX r% r: r r� r� r� rB )r Z
sleep_coroZ wait_coror
r
r �test_wait_errors] s z BaseTaskTests.test_wait_errorsc s� � fdd�}� j |�}� j|tjd|d��}� j|tjd|d��}� j|tj||gtj|d��}|j|�\}}� j|h|� � j|h|� � j|j � � � j
|j � � � j |j � � � j
d|j� � |jd� |jtj||g|d�� d S )Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng $@r g�������?)r� )r� )r r
r r� l s
z4BaseTaskTests.test_wait_first_completed.<locals>.geng $@)r% g�������?)r� r% r� )r/ r) r r� r� �FIRST_COMPLETEDr: rM r� rL rK � assertIsNonerN r� r� � advance_time)r r� r% r� r� rE rL r� r
)r r �test_wait_first_completedj s"
z'BaseTaskTests.test_wait_first_completedc C s� t jdd� �}t jdd� �}| j| j|� �}| j| j|� �}| j| jt j||gt j| jd��}| jj|�\}}| j||h|� | j|j � � | j
|j � � | j|j � � | j
|j � � d S )Nc s s
d V d S )Nr
r
r
r
r �coro1� s z2BaseTaskTests.test_wait_really_done.<locals>.coro1c s s d V d V d S )Nr
r
r
r
r �coro2� s z2BaseTaskTests.test_wait_really_done.<locals>.coro2)r� r% ) r r9 r) r% r� r� r: rM rK rL r� rN )r r� r� r� r� rE rL r� r
r
r �test_wait_really_done� s z#BaseTaskTests.test_wait_really_donec s� � fdd�}� j |�}� j|tjd|d��}tjdd� �}� j||� �}� j|tj||gtj|d��}|j|�\}}� j|h|� � j|h|� � j d|j
� � |j d � |jtj||g|d�� d S )
Nc 3 s d V } � j d| � dV d S )Ng $@r )r� )r� )r r
r r� � s z4BaseTaskTests.test_wait_first_exception.<locals>.geng $@)r% c S s t d��d S )N�err)�ZeroDivisionErrorr
r
r
r �exc� s z4BaseTaskTests.test_wait_first_exception.<locals>.exc)r� r% r r� ) r/ r) r r� r9 r� �FIRST_EXCEPTIONr: rM r� r� r� ) r r� r% r� r� r� rE rL r� r
)r r �test_wait_first_exception� s
z'BaseTaskTests.test_wait_first_exceptionc s� �fdd�}�j |�� �j� tjd� d��}tj� fdd��}�j� |� �}tj||gtj� d�}� j|�\}}�j|h|� �j|h|� �j d� j
� � � j d � � jtj||g� d�� d S )
Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng $@r g{�G�z�?)r� )r� )r r
r r� � s
z<BaseTaskTests.test_wait_first_exception_in_wait.<locals>.geng $@)r% c 3 s t jd� d�E d H td��d S )Ng{�G�z�?)r% r� )r r� r� r
)r% r
r r� � s z<BaseTaskTests.test_wait_first_exception_in_wait.<locals>.exc)r� r% g{�G�z�?r� ) r/ r) r r� r9 r� r� r: rM r� r� r� )r r� r� r� r� rE rL r� r
)r% r r �!test_wait_first_exception_in_wait� s
z/BaseTaskTests.test_wait_first_exception_in_waitc s� �fdd�}�j |���j�tjd�d��� tj�fdd��}�j�|� ��tj� ���fdd��}�j�j�|� �� �jd �j� � �j�j�|� �� �jd �j� � d S )
Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?r g333333�?)r� )r� )r r
r r� � s
z3BaseTaskTests.test_wait_with_exception.<locals>.geng�������?)r% c 3 s t jd� d�E d H td��d S )Ng333333�?)r% Zreally)r r� r� r
)r% r
r �sleeper� s z7BaseTaskTests.test_wait_with_exception.<locals>.sleeperc 3 s` t j�� g�d�E d H \} }�jt| �d� �j|t� � tdd� | D ��}�jt|�d� d S )N)r% r� c s s | ]}|j � d k r|V qd S )N)r� )r5 rY r
r
r r� � s zFBaseTaskTests.test_wait_with_exception.<locals>.foo.<locals>.<genexpr>r )r r� rM �lenr� )rL r� �errors)r� r� r% r r
r rf � s
z3BaseTaskTests.test_wait_with_exception.<locals>.foog333333�?)r/ r) r r� r9 r: r� r� )r r� r� rf r
)r� r� r% r r �test_wait_with_exception� s
z&BaseTaskTests.test_wait_with_exceptionc s� �fdd�}�j |���j�tjd�d��� �j�tjd�d���tj� ���fdd��}�j�j�|� �� �jd�j� � �jd � �jtj � �g�d�� d S )
Nc 3 s@ d V } � j d| � dV } � j d| � dV } � j d| � dV d S )Ng�������?r g333333�?g)\��(�?)r� )r� )r r
r r� s z1BaseTaskTests.test_wait_with_timeout.<locals>.geng�������?)r% g333333�?c 3 sF t j�� gd�d�E d H \} }�j| t� g�� �j|t�g�� d S )Ng)\��(�?)r� r% )r r� rM r� )rL r� )r� r� r% r r
r rf s z1BaseTaskTests.test_wait_with_timeout.<locals>.foog)\��(�?r� )
r/ r) r r� r9 r: r� r� r� r� )r r� rf r
)r� r� r% r r �test_wait_with_timeout s
z$BaseTaskTests.test_wait_with_timeoutc s� � fdd�}� j |�}� j|tjd|d��}� j|tjd|d��}|jtj||gd|d��\}}� j|t|g�� � j|t|g�� � jd|j � � |j
d� |jtj||g|d�� d S )Nc 3 s@ d V } � j d| � dV } � j d| � dV } � j d| � dV d S )Ng�������?r g333333�?)r� )r� )r r
r r� $ s z8BaseTaskTests.test_wait_concurrent_complete.<locals>.geng�������?)r% g333333�?)r� r% r� ) r/ r) r r� r: r� rM r� r� r� r� )r r� r% r� r� rL r� r
)r r � test_wait_concurrent_complete" s
z+BaseTaskTests.test_wait_concurrent_completec s� dd� }| j |��d�_t� �d�tj���fdd��}|dd�� |dd ��|d
d ��tj� ���fd d
��}�j| j�|� ��}| jd
�j� � | j d|d d� k� | j d |d d� k� | j
|d d � �j| j�|� ��}| jd
�j� � d S )Nc s s dV dV dV dV d S )Nr g{�G�z�?r
r
r
r
r r� ? s z,BaseTaskTests.test_as_completed.<locals>.geng �?Fc 3 sF t j| �d�E d H � j|� � rBd� krBd� krBd��jd� |S )N)r% r� r� Tg�Q� ��?)r r� �addr� )�dtr� )� completedr% � time_shiftedr
r r� K s
z0BaseTaskTests.test_as_completed.<locals>.sleeperg{�G�z�?r� r� g333333�?r� c 3 s6 g } x,t j��� g�d�D ]}| j|E d H � qW | S )N)r% )r � as_completedr� )�valuesrY )r� r� r� r% r
r rf Y s z,BaseTaskTests.test_as_completed.<locals>.foor� ) r/ Zslow_callback_durationr� r r9 r: r) r� r� rK rM )r r� r� rf r� r
)r� r� r� r� r% r� r �test_as_completed= s"
z BaseTaskTests.test_as_completedc s� dd� }| j |��tjdd�d�� tjdd�d��tj� ��fdd ��}�j| j�|� ��}| jt|�d
|� | j|d d� | j|d d d
� | j|d d tj � | j
d
�j � � �j d� �jtj
� �g�d�� d S )Nc s s d V dV dV dV d S )Nr g�������?r
r
r
r
r r� l s z9BaseTaskTests.test_as_completed_with_timeout.<locals>.geng�������?r� )r% g333333�?r� c 3 s� g } xzt j� �gd�d�D ]b}| r,�jd� y |E d H }| jd|f� W q t jk
rz } z| jd|f� W Y d d }~X qX qW | S )Ng� ��Q��?)r� r% g{�G�z�?r r� )r r� r� r� r� )r� rY �vr� )r� r� r% r
r rf w s
$z9BaseTaskTests.test_as_completed_with_timeout.<locals>.foor� r r g� ��Q��?r� )r r� )r/ r r� r9 r: r) rM r� �assertIsInstancer� r� r� r� r� )r r� rf r� r
)r� r� r% r � test_as_completed_with_timeoutj s
z,BaseTaskTests.test_as_completed_with_timeoutc sP dd� }�j |��tjdd�d�� tj� ��fdd��}�j�j�|� �� d S )Nc s s d V dV dV d S )Nr g{�G�z�?r
r
r
r
r r� � s z@BaseTaskTests.test_as_completed_with_unused_timeout.<locals>.geng{�G�z�?r� )r% c 3 s6 x0t j� gd�d�D ]} | E d H }�j|d� qW d S )Nr )r� r% r� )r r� rM )rY r� )r� r% r r
r rf � s
z@BaseTaskTests.test_as_completed_with_unused_timeout.<locals>.foo)r/ r r� r9 r: r) )r r� rf r
)r� r% r r �%test_as_completed_with_unused_timeout� s
z3BaseTaskTests.test_as_completed_with_unused_timeoutc C s� dd� }| j |�}tjdd|d�}tjdd|d�}||h}ttj||d��}| jt|�d� |j|d �}| j|d� | jd|j � � |j
d� |j|d
�}| j|d� | jd|j � � d S ) Nc s s dV dV dV d S )Nr g�������?r
r
r
r
r r� � s z9BaseTaskTests.test_as_completed_reverse_wait.<locals>.geng�������?r� )r% g�������?r� r� r r ) r/ r r� �listr� rM r� r: r� r� r� ) r r� r% r� r� �fs�futsr� r� r
r
r � test_as_completed_reverse_wait� s
z,BaseTaskTests.test_as_completed_reverse_waitc
s� � fdd�}� j |�}tjdd|d�}tjdd|d�}||h}ttj||d��}� jt|�d� tj||d�}|j|�\}} � jt dd � |D ��ddh� d S )
Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?r )r� )r� )r r
r r� � s
z7BaseTaskTests.test_as_completed_concurrent.<locals>.geng�������?r� )r% r� r� c s s | ]}|j � V qd S )N)rN )r5 rY r
r
r r� � s z=BaseTaskTests.test_as_completed_concurrent.<locals>.<genexpr>)
r/ r r� r� r� rM r� r� r: r� )
r r� r% r� r� r� r� �waiterrL r� r
)r r � test_as_completed_concurrent� s
z*BaseTaskTests.test_as_completed_concurrentc sn t jdd� �� t j� �fdd��}�j�j|� �}�jj|� |j� }�jt|�ddh� �jt|�d� d S )Nc S s | S )Nr
)r� r
r
r r( � s zBBaseTaskTests.test_as_completed_duplicate_coroutines.<locals>.coroc 3 sD g } � d�}x2t j||� d�g�jd�D ]}| j|E d H � q(W | S )N�hamr� )r% )r r� r% r� )rN r� rY )r( r r
r rs � s zDBaseTaskTests.test_as_completed_duplicate_coroutines.<locals>.runnerr� r� r� ) r r9 r) r% r: rN rM r� r� )r rs r> rN r
)r( r r �&test_as_completed_duplicate_coroutines� s z4BaseTaskTests.test_as_completed_duplicate_coroutinesc sv �fdd�}�j |�� tj� fdd��}�j� |dd��}� j|� �j|j� � �j|j� d� �j d� j
� � d S )Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?g�������?)r� )r� )r r
r r� � s
z%BaseTaskTests.test_sleep.<locals>.genc 3 s6 t j| d � d�E d H t j| d |� d�E d H }|S )Nr� )r% )r r� )r� �argr� )r% r
r r� � s z)BaseTaskTests.test_sleep.<locals>.sleeperg�������?�yeah) r/ r r9 r) r: rK rL rM rN r� r� )r r� r� rR r
)r% r r �
test_sleep� s
zBaseTaskTests.test_sleepc s� �fdd�}�j |�}�j|tjdd|d��}d � |j�� �fdd�}||_tj|� �j� j� |j � tj|� �j
� j� d S )Nc 3 s d V } � j d| � dV d S )Ng $@r )r� )r� )r r
r r� � s z,BaseTaskTests.test_sleep_cancel.<locals>.geng $@r� )r% c s �| |f|�� � � S )Nr
)Zdelay�callbackr )�handle�orig_call_laterr
r r� s z3BaseTaskTests.test_sleep_cancel.<locals>.call_later) r/ r) r r� r� r r� r� Z
_cancelledr} rK )r r� r% rR r� r
)r r r r �test_sleep_cancel� s
z BaseTaskTests.test_sleep_cancelc sj �fdd�}�j |�� tj� fdd���tj� ��fdd��}|� }�j� j|�d� �jd� j� � d S ) Nc 3 s. d V } � j d| � dV } � j d| � dV d S )Ng�������?r i� )r� )r� )r r
r r� s
z9BaseTaskTests.test_task_cancel_sleeping_task.<locals>.genc 3 s t j| � d�E d H d S )N)r% )r r� )r� )r% r
r r� s z;BaseTaskTests.test_task_cancel_sleeping_task.<locals>.sleepc 3 sL �j � �d��} � jd| j� y| E d H W n tjk
rB dS X dS d S )Ni� g�������?r� zslept in)r) r� r} r r~ )r� )r% r r� r
r �doit$ s z:BaseTaskTests.test_task_cancel_sleeping_task.<locals>.doitr� g�������?)r/ r r9 rM r: r� r� )r r� r Zdoerr
)r% r r� r � test_task_cancel_sleeping_task s
z,BaseTaskTests.test_task_cancel_sleeping_taskc s� | j | j�� tj� fdd��}| j| j|� �}tj| j� | j|j� � |j � tj| j� | j
tj | jj |� | j
|j� | j� j� � d S )Nc 3 s � E d H d S )Nr
r
)r> r
r r( 6 s z:BaseTaskTests.test_task_cancel_waiter_future.<locals>.coro)r+ r% r r9 r) r r� rO r� r} rW r~ r: r� rK r� )r r( rE r
)r> r � test_task_cancel_waiter_future3 s z,BaseTaskTests.test_task_cancel_waiter_futurec C sF t jdd� �}|� }| j| j|�}|jd� | jt|j� |j� d S )Nc S s dS )NrU r
r
r
r
r rJ F s z:BaseTaskTests.test_step_in_completed_task.<locals>.notmuchrI ) r r9 r) r% rV rW �AssertionError�_steprB )r rJ r� rE r
r
r �test_step_in_completed_taskE s
z)BaseTaskTests.test_step_in_completed_taskc C s& t jdd� �}| jt| jj|� � d S )Nc s s d V dV dS )Nr rU r
r
r
r
r rJ R s z/BaseTaskTests.test_step_result.<locals>.notmuch)r r9 rW rA r% r: )r rJ r
r
r �test_step_resultQ s z BaseTaskTests.test_step_resultc s� G dd� dt j�}|| jd�� d �t j� �fdd��}| j| j|� �}tj| j� | j� j� t � }� j
|� tj| j� | j |�� | j|j � � | j
|j� � d S )Nc s( e Zd Z� fdd�Z� fdd�Z� ZS )z2BaseTaskTests.test_step_result_future.<locals>.Futc s d| _ t� j||� d S )NF)�cb_addedr- r\ )r r �kwds)r& r
r r\ _ s z;BaseTaskTests.test_step_result_future.<locals>.Fut.__init__c s d| _ t� j|� d S )NT)r
r- rz )r �fn)r& r
r rz c s zDBaseTaskTests.test_step_result_future.<locals>.Fut.add_done_callback)r! r" r# r\ rz �
__classcell__r
r
)r& r �Fut^ s r )r% c 3 s � E d H �d S )Nr
r
)r> rN r
r �wait_for_futurej s z>BaseTaskTests.test_step_result_future.<locals>.wait_for_future)r r* r% r9 r) r r� rK r
�objectrV rO rL r� rN )r r r rR r� r
)r> rN r �test_step_result_future[ s
z%BaseTaskTests.test_step_result_futurec C sN t jdd� �}| j| j|� �}| jt|j� | j|j� � | j |j
� t� d S )Nc S s
t � �d S )N)�
BaseExceptionr
r
r
r �notmutch{ s z<BaseTaskTests.test_step_with_baseexception.<locals>.notmutch) r r9 r) r% rW r r rK rL r� r� )r r rE r
r
r � test_step_with_baseexceptionz s
z*BaseTaskTests.test_step_with_baseexceptionc s� �fdd�}�j |��tj�fdd���t� � tj� �fdd��}�j�|� �}tj�� |j� �j|j � � �j
ttj�� �j |j � � �j|j � � �j
|j� � � d S )Nc 3 s d V } � j d| � dV d S )Ng $@r )r� )r� )r r
r r� � s z;BaseTaskTests.test_baseexception_during_cancel.<locals>.genc 3 s t jd� d�E d H d S )Nr� )r% )r r� r
)r% r
r r� � s z?BaseTaskTests.test_baseexception_during_cancel.<locals>.sleeperc 3 s0 y�� E d H W n t jk
r* � �Y nX d S )N)r r~ r
)�base_excr� r
r r � s z@BaseTaskTests.test_baseexception_during_cancel.<locals>.notmutch)r/ r r9 r r) r r� r} r� rL rW rK r� rO r� )r r� r rE r
)r r% r r� r � test_baseexception_during_cancel� s
z.BaseTaskTests.test_baseexception_during_cancelc C sf dd� }| j tj|�� dd� }| j tj|�� tjdd� �}| jtj|�� | j tjtj� �� d S )Nc S s d S )Nr
r
r
r
r r � s z2BaseTaskTests.test_iscoroutinefunction.<locals>.fnc s s
d V d S )Nr
r
r
r
r �fn1� s z3BaseTaskTests.test_iscoroutinefunction.<locals>.fn1c s s
d V d S )Nr
r
r
r
r �fn2� s z3BaseTaskTests.test_iscoroutinefunction.<locals>.fn2)r� r �iscoroutinefunctionr9 rK r �Mock)r r r r r
r
r �test_iscoroutinefunction� s z&BaseTaskTests.test_iscoroutinefunctionc sX | j | j�� tj� fdd��}|� }| jt�� | jj|� W d Q R X | j� j� � d S )Nc 3 s
� V d S )Nr
r
)r> r
r r � s z?BaseTaskTests.test_yield_vs_yield_from.<locals>.wait_for_future) r+ r% r r9 rW rA r: r� rL )r r rE r
)r> r �test_yield_vs_yield_from� s z&BaseTaskTests.test_yield_vs_yield_fromc s< t jdd� �� t j� fdd��}|� }| jt| jj|� d S )Nc s s
d V d S )Nr
r
r
r
r r( � s z>BaseTaskTests.test_yield_vs_yield_from_generator.<locals>.coroc
3 s � � } z
| V W d | j � X d S )N)rB )r� )r( r
r r � s
zIBaseTaskTests.test_yield_vs_yield_from_generator.<locals>.wait_for_future)r r9 rW rA r% r: )r r rE r
)r( r �"test_yield_vs_yield_from_generator� s z0BaseTaskTests.test_yield_vs_yield_from_generatorc C sP t jdd� �}| jt j|�� |� }| jt j|�� | jj|�}| j|d� d S )Nc S s dS )NrF r
r
r
r
r r� � s z;BaseTaskTests.test_coroutine_non_gen_function.<locals>.funcrF )r r9 rK r � iscoroutiner% r: rM )r r� r( r� r
r
r � test_coroutine_non_gen_function� s z-BaseTaskTests.test_coroutine_non_gen_functionc sz | j | j�� tj� fdd��}tj� fdd��}| j| j|� �}| j| j|� �}| jj|�}| j|d� | j|j� � d S )Nc s � S )Nr
r
)r> r
r r� � s zIBaseTaskTests.test_coroutine_non_gen_function_return_future.<locals>.funcc s � j d� d S )NrF )rV r
)r> r
r r( � s zIBaseTaskTests.test_coroutine_non_gen_function_return_future.<locals>.cororF ) r+ r% r r9 r) r: rM r� rN )r r� r( �t1�t2r� r
)r> r �-test_coroutine_non_gen_function_return_future� s z;BaseTaskTests.test_coroutine_non_gen_function_return_futurec sj �j j� �j� j�jd�� tj� ��fdd��}�j�j|�j����jj�� �j� j�jd�� d S )N)r% c sX �j � j| d��k� tj| � z&�j� jd ��� �j� j� �� W d tjd � X d S )N)r% )rK � current_taskr rQ rO )r% )r' r rE r
r r( � s
z-BaseTaskTests.test_current_task.<locals>.coro) r& r' r� r# r% r r9 r) r: )r r( r
)r' r rE r �test_current_task� s z BaseTaskTests.test_current_taskc s� �j j� �j� j�jd�� �j�j���j�j��tj� ����fdd��}tj� ����fdd��}�j�j|�j����j�j|�j����jj tj
��f�jd�� �j� j�jd�� d S )N)r% c 3 sD �j � j| d��k� �E d H �j � j| d��k� �jd� d S )N)r% T)rK r# rV )r% )r' r� r� r �task1r
r r� s
zFBaseTaskTests.test_current_task_with_interleaving_tasks.<locals>.coro1c 3 sD �j � j| d��k� �jd� �E d H �j � j| d��k� d S )N)r% T)rK r# rV )r% )r' r� r� r �task2r
r r� s
zFBaseTaskTests.test_current_task_with_interleaving_tasks.<locals>.coro2) r& r' r� r# r% r+ r r9 r) r: r� )r r� r� r
)r' r� r� r r% r& r �)test_current_task_with_interleaving_tasks
s z7BaseTaskTests.test_current_task_with_interleaving_tasksc s� d��j �j��tj���fdd��� tj� �fdd��}tj|� �jd�}tj�j� |j� �jj|� �j �d� �j
�j � � d S )Nr c 3 s> y�E d H W n t jk
r. � d7 � � Y n X �jd� d S )Nr zgot past sleep() in inner())r r~ �failr
)�proofr r� r
r �inner/ s z<BaseTaskTests.test_yield_future_passes_cancel.<locals>.innerc 3 s< y� � E d H W n t jk
r. �d7 �Y n
X �d7 �d S )Nr8 r� )r r~ r
)r* r) r
r r� : s
z<BaseTaskTests.test_yield_future_passes_cancel.<locals>.outer)r% �e ) r+ r% r r9 rG r r� r} r: rM rK r� )r r� rY r
)r* r) r r� r � test_yield_future_passes_cancel* s
z-BaseTaskTests.test_yield_future_passes_cancelc s� d��j �j��tj��fdd��� tj� ��fdd��}tj|� �jd�}tj�j� |j� �jtj �jj
|� �j d � tj�j� �j �d� d S )Nr c 3 s �E d H � d7 � d S )Nr r
r
)r) r� r
r r* Q s
zCBaseTaskTests.test_yield_wait_does_not_shield_cancel.<locals>.innerc 3 s* t j� � g�jd�E d H \} }�d7 �d S )N)r% r8 )r r� r% )�d�p)r* r) r r
r r� W s zCBaseTaskTests.test_yield_wait_does_not_shield_cancel.<locals>.outer)r% r )
r+ r% r r9 rG r r� r} rW r~ r: rV rM )r r� rY r
)r* r) r r� r �&test_yield_wait_does_not_shield_cancelK s
z4BaseTaskTests.test_yield_wait_does_not_shield_cancelc C s< | j | j�}tj|�}|jd� | jj|�}| j|d� d S )Nr� )r+ r% r �shieldrV r: rM )r r* r� r� r
r
r �test_shield_resultf s
z BaseTaskTests.test_shield_resultc C sT | j | j�}tj|�}tj| j� td�}|j|� tj| j� | j|j � |� d S )Nr� )
r+ r% r r0 r r� rA �
set_exceptionrO r� )r r* r� r� r
r
r �test_shield_exceptionm s
z#BaseTaskTests.test_shield_exceptionc C sH | j | j�}tj|�}tj| j� |j� tj| j� | j|j� � d S )N) r+ r% r r0 r r� r} rK r� )r r* r� r
r
r �test_shield_cancelv s
z BaseTaskTests.test_shield_cancelc C s8 | j | j�}|jd� | jjtj|��}| j|d� d S )Nr� )r+ r% rV r: r r0 rM )r r> r� r
r
r �test_shield_shortcut~ s
z"BaseTaskTests.test_shield_shortcutc
s� d��j �j��tj��fdd��� tj� ��fdd��}tj|� �jd�}tj�j� |j� �jtj �� �jj
|� W d Q R X �j d � tj�j� �j �d� d S )Nr c 3 s �E d H � d7 � d S )Nr r
r
)r) r� r
r r* � s
z/BaseTaskTests.test_shield_effect.<locals>.innerc 3 s$ t j� � �jd�E d H �d7 �d S )N)r% r8 )r r0 r% r
)r* r) r r
r r� � s z/BaseTaskTests.test_shield_effect.<locals>.outer)r% r )
r+ r% r r9 rG r r� r} rW r~ r: rV rM )r r� rY r
)r* r) r r� r �test_shield_effect� s
z BaseTaskTests.test_shield_effectc C s� | j | j�}| j | j�}tj||| jd�}tj|| jd�}tj| j� |j� tj| j� | j|j � � |j
d� |j
d� tj| j� | j |j � ddg� d S )N)r% r r� )
r+ r% r r; r0 r r� r} rK r� rV rM rN )r �child1�child2�parentr� r
r
r �test_shield_gather� s
z BaseTaskTests.test_shield_gatherc C s� | j | j�}| j | j�}tj|| jd�}tj|| jd�}tj||| jd�}tj| j� |j� tj| j� | j|j � tj
� | j |j � � | j |j � � |j
d� |j
d� tj| j� d S )N)r% r r� )r+ r% r r0 r; r r� r} r� r� r~ rK r� rV )r r7 r8 r� r� r9 r
r
r �test_gather_shield� s
z BaseTaskTests.test_gather_shieldc C sZ | j | j�}| jt| jjtj|| jd�� t� }| jt| jjtj|| jd�� |j� d S )N)r% ) r+ r% rW r` r: r r� r rB )r r> r( r
r
r � test_as_completed_invalid_args� s z,BaseTaskTests.test_as_completed_invalid_argsc C sx | j | j�}| jt| jjtj|| jd�� t� }| jt| jjtj|| jd�� |j� | jt | jjtjg | jd�� d S )N)r% )
r+ r% rW r` r: r r� r rB rX )r r> r( r
r
r �test_wait_invalid_args� s z$BaseTaskTests.test_wait_invalid_argsc sD � fdd�}t d�� |� W d Q R X t d�� |� W d Q R X d S )Nc s� t j� ��fdd��} �j�j�� � jd � | � ��jt j��� �j�jt j
� �j �j � �j�j
t j� �jj�� �j�j� d S )Nc 3 s �j �j� � E d H d S )N)rK �
gi_runningr
)r> r� r r
r r( � s zKBaseTaskTests.test_corowrapper_mocks_generator.<locals>.check.<locals>.coro)r r9 r+ r% rV rK r r� �gi_frame�types� FrameTyper� r> r� �CodeTyper: r� )r( )r )r> r� r �check� s
z=BaseTaskTests.test_corowrapper_mocks_generator.<locals>.checkFT)r )r rC r
)r r � test_corowrapper_mocks_generator� s
z.BaseTaskTests.test_corowrapper_mocks_generatorc st t d��b tj�fdd��}tj� �fdd���tjdd� ��� j� j|� �}� jj|�}� j|d � W d Q R X d S ) NTc 3 s � � E d H S )Nr
r
)r! r
r r s z5BaseTaskTests.test_yield_from_corowrapper.<locals>.t1c 3 s( � j � j�} � j� j�| �� | E d H S )N)r+ r% r) )rY )r �t3r
r r! s z5BaseTaskTests.test_yield_from_corowrapper.<locals>.t2c S s | j d� d S )Nr r� r
)r r� r
)rV )rY r
r
r rE
s z5BaseTaskTests.test_yield_from_corowrapper.<locals>.t3r r� r
)r r� r
)r r r9 r) r% r: rM )r r rE �valr
)r r! rE r �test_yield_from_corowrapper� s
z)BaseTaskTests.test_yield_from_corowrapperc s8 dd� � � fdd�}| j |d�d � | j |d�d� d S )
Nc s s
d V } | S )Nr
)r� r
r
r rf s z;BaseTaskTests.test_yield_from_corowrapper_send.<locals>.fooc sZ t jj� � �}|jd � y|j| � W n& tk
rL } z
|jd S d }~X n
X td��d S )Nr zStopIteration was expected)r r � CoroWrapper�send�
StopIterationr r )r� �cwZex)rf r
r �call s
z<BaseTaskTests.test_yield_from_corowrapper_send.<locals>.callr r� r� )r r� )r r� )rM )r rL r
)rf r � test_yield_from_corowrapper_send s
z.BaseTaskTests.test_yield_from_corowrapper_sendc C s0 t j� }dd� }tjj|� �}||d< d |_d S )Nc s s g E d H d S )Nr
r
r
r
r rf & s z3BaseTaskTests.test_corowrapper_weakref.<locals>.foorK )�weakref�WeakValueDictionaryr r rH r� )r Zwdrf rK r
r
r �test_corowrapper_weakref$ s
z&BaseTaskTests.test_corowrapper_weakrefc C s� dd� }t d�}tjj|� �}|jd � | j||j|�� tjj|� �}|jd � | j||jt |�� tjj|� �}|jd � |jt d�}| j|t � | j|j d� tjj|� �}|jd � |jt dd �}| j|t � | j|j d� d S )Nc s s@ d } x6y
| V } W q t k
r6 } z
|} W Y d d }~X qX qW d S )N)� Exception)�value�er
r
r rf - s
z1BaseTaskTests.test_corowrapper_throw.<locals>.foorf )rf )rf )
rQ r r rH rI rO �throwr� rM r )r rf r� rK r
r
r �test_corowrapper_throw+ s$
z$BaseTaskTests.test_corowrapper_throwzneed python 3.4 or laterc s* � j j}tj� fdd��}tj� }� jjd� � jj|� |� j�}tj |� jd�}� j
|j � jd�|h� tj � j� z*� j
|j � |h� � j
|j d �|h� W d tj d � X � jj
� � j
t� jj�d� |jjd= d }|j}d }tj� � j
|j � jd�t� � |j� jdtj|d�� |j� d S ) Nc 3 s � j | �}|E d H td��d S )Nzcode never reached)r+ rQ )r% �future)r r
r �kill_meO s
z>BaseTaskTests.test_log_destroyed_pending_task.<locals>.kill_meT)r% r rV z%Task was destroyed but it is pending!)�messagerE r )r& r' r r9 r r r% rw �set_exception_handlerrG rM � all_tasksrQ Z _run_oncer� �_readyr? �f_localsr{ r �
gc_collectr� Zassert_called_with�ANYZ
reset_mock)r r' rW Z mock_handlerr( rE r r
)r r � test_log_destroyed_pending_taskJ s4
z-BaseTaskTests.test_log_destroyed_pending_taskzasyncio.base_events.loggerc sT t j� ��j�� t jdd� �� t j� ��fdd��}�j|� � �j|jj� d S )Nc S s t �d S )N)r` r
r
r
r r( � s zBBaseTaskTests.test_tb_logger_not_called_after_cancel.<locals>.coroc 3 s2 �j �� � �} tjd�d�E d H | j� d } d S )Ng�������?)r% )r) r r� r} )rE )r( r% r r
r rs � s zDBaseTaskTests.test_tb_logger_not_called_after_cancel.<locals>.runner)r r? rQ r9 r: r� �error�called)r �m_logrs r
)r( r% r r �&test_tb_logger_not_called_after_cancel s
z4BaseTaskTests.test_tb_logger_not_called_after_cancelzasyncio.coroutines.loggerc
C s� t d�� tjdd� �}W d Q R X t}tj� jd }|� tj� | j |j
j � |j
j d d }t
j|�\}}dtj|j�tj|�|tj|�|f }| j|tj|tj�� d S )NTc S s d S )Nr
r
r
r
r � coro_noop� s z=BaseTaskTests.test_coroutine_never_yielded.<locals>.coro_noopr� r z�^<CoroWrapper %s\(?\)? .* at %s:%s, .*> was never yielded from\nCoroutine object created at \(most recent call last, truncated to \d+ last lines\):\n.*\n File "%s", line %s, in test_coroutine_never_yielded\n coro_noop\(\)$)r r r9 �__file__�sys� _getframe�f_linenor r] rK r` ra � call_argsr ry r� r� r# rp �compile�DOTALL) r rb rd Z tb_filename� tb_linenorX Z
func_filenameZ func_linenoZregexr
r
r � test_coroutine_never_yielded� s
z*BaseTaskTests.test_coroutine_never_yieldedc sD � fdd�}t d�� |� W d Q R X t d�� |� W d Q R X d S )Nc s, t jdd� �} � jj| � �}� j|d� d S )Nc S s t jdd� �} | � S )Nc S s dS )Nr r
r
r
r
r �
inner_coro� s ziBaseTaskTests.test_return_coroutine_from_coroutine.<locals>.check.<locals>.outer_coro.<locals>.inner_coro)r r9 )rn r
r
r �
outer_coro� s zUBaseTaskTests.test_return_coroutine_from_coroutine.<locals>.check.<locals>.outer_coror )r r9 r% r: rM )ro rN )r r
r rC � s zABaseTaskTests.test_return_coroutine_from_coroutine.<locals>.checkFT)r )r rC r
)r r �$test_return_coroutine_from_coroutine� s
z2BaseTaskTests.test_return_coroutine_from_coroutinec C sh | j jd� | j| j t� �}tj� jd }| j|jt � | j
|jd d d� t |df� | j j |� d S )NTr r� r
�test_task_source_traceback���)
r% rw r) r rf rg rh r� r{ r� rM re r: )r rE r� r
r
r rq � s z(BaseTaskTests.test_task_source_tracebackc sz t j� � �j� j� t j� �fdd��}� j|� �}� jt j||� d��}� j|j� �j t j
� j |� �j |j
� � d S )Nc 3 s �j � �} | E d H d S )N)r+ )r> )r% r r
r �blocking_coroutine� s
z?BaseTaskTests._test_cancel_wait_for.<locals>.blocking_coroutine)r% )r r? r� rB r9 r� r� r� r} rW r~ r: rK r� )r r� rs rE r� r
)r% r r �_test_cancel_wait_for� s
z#BaseTaskTests._test_cancel_wait_forc C s | j d � d S )N)rt )r r
r
r � test_cancel_blocking_wait_for� s z+BaseTaskTests.test_cancel_blocking_wait_forc C s | j d� d S )Ng N@)rt )r r
r
r �test_cancel_wait_for� s z"BaseTaskTests.test_cancel_wait_forc s� t j� }| j|j� | j|���fdd�}t j|� |d�}t j||d��d � � �fdd�}�j|� �jd� |j �� | j
� d� | j �j � � | j
�j
� dg� d S )Nc 3 s
� E d H S )Nr
r
)r> r
r �
child_coro� s z6BaseTaskTests.test_cancel_gather_1.<locals>.child_coro)r% c s �j � � d S )N)r} )r6 )�
cancel_result� gather_taskr
r �cancelling_callback� s z?BaseTaskTests.test_cancel_gather_1.<locals>.cancelling_callbackr� F)r r? r� rB r+ r; rG rz rV r: rM r� r� rN )r r% rw Z
gather_futurerz r
)rx r> ry r �test_cancel_gather_1� s
z"BaseTaskTests.test_cancel_gather_1c s@ t j� � �j� j� � fdd��� ��fdd�}� j|� � d S )Nc � sB d} x8| d7 } t jt jd� d�d� d�I d H | dkrd S qW d S )Nr g�������?)r% T)�return_exceptionsr% r )r r; r� )r� )r% r
r rF s z0BaseTaskTests.test_cancel_gather_2.<locals>.testc � s^ �j � �� �} tjd� d�I d H | j� y| I d H W n tjk
rN Y n X �jd� d S )Ng�������?)r% z1gather did not propagate the cancellation request)r) r r� r} r~ r( )Zqwe)r% r rF r
r �main s z0BaseTaskTests.test_cancel_gather_2.<locals>.main)r r? r� rB r: )r r} r
)r% r rF r �test_cancel_gather_2 s
z"BaseTaskTests.test_cancel_gather_2c s4 t jdd� �� t j� �fdd��}�jj|� � d S )Nc S s dd d S )Nr r r
r
r
r
r rf * s z3BaseTaskTests.test_exception_traceback.<locals>.fooc 3 s* �j �j� � �} d V �j| j� j� d S )N)r) r% ZassertIsNotNoner� �
__traceback__)rE )rf r r
r r} . s z4BaseTaskTests.test_exception_traceback.<locals>.main)r r9 r% r: )r r} r
)rf r r �test_exception_traceback' s z&BaseTaskTests.test_exception_tracebackc C s� dd� }|| j _tjdd� �}| j|jj� | jt��* |� }z| j | j |� W d |j
� X W d Q R X | j |jj� |jj d d }| j
d|� | j| jj| j �t� � d S )Nc W s t �d S )N)rX )r� r r
r
r r� 8 s z8BaseTaskTests.test_error_in_call_soon.<locals>.call_soonc S s d S )Nr
r
r
r
r r( < s z3BaseTaskTests.test_error_in_call_soon.<locals>.coror z$Task was destroyed but it is pending)r% r� r r9 r� r` ra rW rX r) rB rK ri �assertInrM r' rZ r� )r rb r� r( r� rX r
r
r �test_error_in_call_soon6 s z%BaseTaskTests.test_error_in_call_soon)jr! r" r# r' r* r) r+ r. r= rD rH rS rT rZ r[ �unittest�
skipUnlessrx r_ ra rd rt r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r r r" r$ r' r, r/ r1 r3 r4 r5 r6 r: r; r< r= rD rG rM rP rU �PY34r_ r Zpatchrc rm rp rq rt ru rv r{ r~ r� r� r
r
r
)r& r r$ O s� ;D $)!
" -%
"
! ' 4 r$ c C sv | j }| j}|d ks |d kr | S G dd� d�}G dd� d||�}G dd� d||�}dd� }|| _ || _|| _d | _| S ) Nc s4 e Zd Z� fdd�Z� fdd�Z� fdd�Z� ZS )z(add_subclass_tests.<locals>.CommonFuturec s" t jdd� �| _t� j||� d S )Nc S s dS )Nr r
r
r
r
r r, Y s zCadd_subclass_tests.<locals>.CommonFuture.__init__.<locals>.<lambda>)� collections� defaultdict�callsr- r\ )r r �kwargs)r& r
r r\ X s z1add_subclass_tests.<locals>.CommonFuture.__init__c s | j d d7 < t� j� S )N�_schedule_callbacksr )r� r- r� )r )r& r
r r� \ s z<add_subclass_tests.<locals>.CommonFuture._schedule_callbacksc s | j d d7 < t� j|� S )Nrz r )r� r- rz )r r )r& r
r rz ` s z:add_subclass_tests.<locals>.CommonFuture.add_done_callback)r! r" r# r\ r� rz r
r
r
)r& r � CommonFutureW s r� c s( e Zd Z� fdd�Z� fdd�Z� ZS )z add_subclass_tests.<locals>.Taskc s | j d d7 < t� j|� S )Nr r )r� r- r )r r )r& r
r r e s z&add_subclass_tests.<locals>.Task._stepc s | j d d7 < t� j|� S )N�_wakeupr )r� r- r� )r r )r& r
r r� i s z(add_subclass_tests.<locals>.Task._wakeup)r! r" r# r r� r
r
r
)r& r r' d s r' c @ s e Zd ZdS )z"add_subclass_tests.<locals>.FutureN)r! r" r# r
r
r
r r* m s r* c s~ �j �jd�� � �fdd�}�j|� �jd�}�jj|�}�j|d� �jt|j�ddddd�� �jt� j�ddd�� d S ) N)r% c � s �j j� fdd�� � I d H S )Nc s
� j d�S )Nr� )rV r
)r> r
r r, t s zYadd_subclass_tests.<locals>.test_subclasses_ctask_cfuture.<locals>.func.<locals>.<lambda>)r% r� r
)r> r r
r r� s s zGadd_subclass_tests.<locals>.test_subclasses_ctask_cfuture.<locals>.funcr� r� r )r r� rz r� )rz r� )r* r% r' r: rM �dictr� )r r� rE rN r
)r> r r � test_subclasses_ctask_cfuturep s
z9add_subclass_tests.<locals>.test_subclasses_ctask_cfuture)r' r* r� rq )�clsZBaseTaskZ
BaseFuturer� r' r* r� r
r
r �add_subclass_testsP s
r� �_CFuturez requires the C _asyncio modulec @ s2 e Zd Zeedd�Zeedd�Zej dd� �Z
dS )�CTask_CFuture_Tests�_CTaskNr� c C s� t jtd�}tjdd� �}| j| j|� �}| jj|� |� }x.td�D ]"}|j |� | jd� | jj|� qFW | j
|� | ddd� d S ) N�gettotalrefcountc S s d S )Nr
r
r
r
r r( � s z@CTask_CFuture_Tests.test_refleaks_in_task___init__.<locals>.coror8 )r% r r� )Zdelta) r Z
get_attributerf r r9 r) r% r: r< r\ r� )r r� r( rE Z refs_before�ir
r
r � test_refleaks_in_task___init__� s z2CTask_CFuture_Tests.test_refleaks_in_task___init__) r! r" r# �getattrr r' r r* r Z
refcount_testr� r
r
r
r r� � s r� c @ s$ e Zd Zeedd�Zeedd�ZdS )�CTask_CFuture_SubclassTestsr� Nr� )r! r" r# r� r r' r r* r
r
r
r r� � s r� c @ s e Zd Zeedd�ZejZdS )�CTask_PyFuture_Testsr� N) r! r" r# r� r r' r � _PyFuturer* r
r
r
r r� � s r� c @ s e Zd ZejZeedd�ZdS )�PyTask_CFuture_Testsr� N) r! r" r# r �_PyTaskr' r� r r* r
r
r
r r� � s r� c @ s e Zd ZejZejZdS )�PyTask_PyFuture_TestsN) r! r" r# r r� r' r r� r* r
r
r
r r� � s r� c @ s e Zd ZejZejZdS )� PyTask_PyFuture_SubclassTestsN) r! r" r# r r� r' r r� r* r
r
r
r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�GenericTaskTestsc C s | j ttjtj�� d S )N)rK �
issubclassr r' r* )r r
r
r �test_future_subclass� s z%GenericTaskTests.test_future_subclassc C sR y dd l }W n tk
r Y n.X y dd l}W n tk
rL | jd� Y nX d S )Nr z_asyncio module is missing)�
_functools� ImportError�_asyncior( )r r� r� r
r
r � test_asyncio_module_compiled� s z-GenericTaskTests.test_asyncio_module_compiledN)r! r" r# r� r� r
r
r
r r� � s r� c sT e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�GatherTestsBasec s2 t � j� | j� | _| j� | _| j| jdd� d S )NF)�cleanup)r- r. r/ �one_looprC rQ )r )r& r
r r. � s
zGatherTestsBase.setUpc C s x|j rtj|� qW d S )N)r[ r r� )r r% r
r
r � _run_loop� s zGatherTestsBase._run_loopc s� � fdd�t d�D �\}}}tj� j|||�|�}tj� }|j|� |jd� |jd� � j� j � � j
|j d� � j |j
� � |jd� � j� j � |j|� � j
|j� dddg� d S )Nc s g | ]}t j� jd ��qS ))r% )r r* r� )r5 r� )r r
r r7 � s z2GatherTestsBase._check_success.<locals>.<listcomp>r
r r� F)r< r r; � wrap_futuresr � MockCallbackrz rV r� r� rM ra r� rL �assert_called_once_withrN )r r� r� r� r� r> �cbr
)r r �_check_success� s
z GatherTestsBase._check_successc C s | j � | j dd� d S )NF)r| )r� )r r
r
r � test_success� s z GatherTestsBase.test_successc C s | j dd� d S )NT)r| )r� )r r
r
r � test_result_exception_success� s z-GatherTestsBase.test_result_exception_successc s� � fdd�t d�D �\}}}}}tj� j|||||�� }tj� }|j|� t� }|jd� |j |� � j
� j � � j |j
� � |j|� � j|j� |� |jd� |j� |j t� � |j� d S )Nc s g | ]}t j� jd ��qS ))r% )r r* r� )r5 r� )r r
r r7 s z6GatherTestsBase.test_one_exception.<locals>.<listcomp>r r r
)r< r r; r� r r� rz r� rV r2 r� r� rK rL r� rO r� r} rA ) r r� r� r� r- rS r> r� r� r
)r r �test_one_exception s
z"GatherTestsBase.test_one_exceptionc s� � fdd�t d�D �\}}}}tj� j||||�ddi�}tj� }|j|� t� }t� }|j d� |j
|� |j d� � j � j � � j
|j� � |j
|� � j � j � � j|j� � |j|� � j|j� dd||g� d S )Nc s g | ]}t j� jd ��qS ))r% )r r* r� )r5 r� )r r
r r7 s z:GatherTestsBase.test_return_exceptions.<locals>.<listcomp>r r| Tr r
)r< r r; r� r r� rz r� rA rV r2 r� r� r� rL rK r� rM rN ) r r� r� r� r- r> r� r� Zexc2r
)r r �test_return_exceptions s"
z&GatherTestsBase.test_return_exceptionsc C s� t jjt jjtj��}djd �}tdd||d�\}}}| j|j� d� td|d|d �\}}}| j|j� d� td|d
|d �\}}}| j|j� d � tdd|d
|d �\}}}| j|j� d� d S )
N�
�import asyncio.coroutines� print(asyncio.coroutines._DEBUG)z-Ez-c)�
PYTHONPATHs False� )ZPYTHONASYNCIODEBUGr� �1s True)r� r� ) �os�path�dirnamer re �joinr rM �rstrip)r Zaio_pathr� �sts�stdout�stderrr
r
r �test_env_var_debug' s& z"GatherTestsBase.test_env_var_debug) r! r" r# r. r� r� r� r� r� r� r� r
r
r
)r& r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�FutureGatherTestsc G s |S )Nr
)r r r
r
r r� F s z FutureGatherTests.wrap_futuresc C s� t j| j� | jt jd � t j|� }| j|t j� | j|j| j� | j | j� | j
|j � � | j |j
� g � t j|d| ji�}| j|j| j� d S )Nr% )r rQ r� r� r; r� r* rO rP r� rK rL rM rN rC )r Z seq_or_iterr> r
r
r �_check_empty_sequenceI s
z'FutureGatherTests._check_empty_sequencec C s2 | j g � | j f � | j t� � | j td�� d S )Nr� )r� r� �iter)r r
r
r � test_constructor_empty_sequenceU s
z1FutureGatherTests.test_constructor_empty_sequencec C sh t j| jd�}t j| jd�}| jt�� t j||� W d Q R X | jt�� t j|| jd� W d Q R X d S )N)r% )r r* r� rC rW rX r; )r r� r� r
r
r �%test_constructor_heterogenous_futures[ s z7FutureGatherTests.test_constructor_heterogenous_futuresc s� � fdd�t d�D �}tj|� }� j|j� j� � j� j� � j|j� � tj|d� ji�}� j|j� j� � j� j� � j|j� � d S )Nc s g | ]}t j� jd ��qS ))r% )r r* rC )r5 r� )r r
r r7 d s zIFutureGatherTests.test_constructor_homogenous_futures.<locals>.<listcomp>r
r% ) r< r r; rO rP rC r� r� rL )r Zchildrenr> r
)r r �#test_constructor_homogenous_futuresc s
z5FutureGatherTests.test_constructor_homogenous_futuresc s� � fdd�t d�D �\}}}}}tj|||||�}tj� }|j|� |jd� |j� � j� j � � j
|j � � |j |� � j
|j� � � j|j� tj� |jd� |j� |jt� � |j� d S )Nc s g | ]}t j� jd ��qS ))r% )r r* r� )r5 r� )r r
r r7 o s z;FutureGatherTests.test_one_cancellation.<locals>.<listcomp>r r r
)r< r r; r r� rz rV r} r� r� rK rL r� r� r� r� r� r~ r2 rA )r r� r� r� r- rS r> r� r
)r r �test_one_cancellationn s
z'FutureGatherTests.test_one_cancellationc s � fdd�t d�D �\}}}}}}tj||||||dd�}tj� }|j|� |jd� t� } |j| � |j � � j
� j � � j |j
� � |jd� |j � t� }
|j|
� � j j|�} � j| d tj� � j| d tj� d | d< | d < � j| d| d dd |
g� |j|� d S )
Nc s g | ]}t j� jd ��qS ))r% )r r* r� )r5 r� )r r
r r7 � s zLFutureGatherTests.test_result_exception_one_cancellation.<locals>.<listcomp>� T)r| r r
r� r )r< r r; r r� rz rV r� r2 r} r� r� r� rL rA r: r� r~ rM r� ) r r� r� r� r- rS rY r> r� ZzdeZrter� r
)r r �&test_result_exception_one_cancellation� s*
z8FutureGatherTests.test_result_exception_one_cancellationN)
r! r" r# r� r� r� r� r� r� r� r
r
r
r r� D s r� c sD e Zd Z� fdd�Zdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )
�CoroutineGatherTestsc s t � j� tj| j� d S )N)r- r. r rQ r� )r )r& r
r r. � s
zCoroutineGatherTests.setUpc G s4 g }x*|D ]"}t j|fdd��}|j|� � q
W |S )Nc s s
| E d H S )Nr
)r> r
r
r r( � s z/CoroutineGatherTests.wrap_futures.<locals>.coro)r r9 r� )r r Zcorosr> r( r
r
r r� � s
z!CoroutineGatherTests.wrap_futuresc C s� t jdd� �}|� }|� }t j||�}| j|j| j� | jj|� | j| jdd� |� }|� }t j||| jd�}| j|j| j� | jj|� d S )Nc S s dS )Nru r
r
r
r
r r( � s zBCoroutineGatherTests.test_constructor_loop_selection.<locals>.coroF)r� )r% ) r r9 r; rO rP r� r: rQ rC )r r( Zgen1Zgen2r> Zgen3Zgen4r� r
r
r � test_constructor_loop_selection� s z4CoroutineGatherTests.test_constructor_loop_selectionc C sX t jdd� �}|d�}t j|||d�|| jd�}| j| j� | j|j� ddddg� d S )Nc S s | S )Nr
)r� r
r
r r( � s z<CoroutineGatherTests.test_duplicate_coroutines.<locals>.cororu �def)r% )r r9 r; r� r� rM rN )r r( r� r> r
r
r �test_duplicate_coroutines� s
z.CoroutineGatherTests.test_duplicate_coroutinesc
s d�t j�jd��t j��fdd��}t j|� �jd�� t j|� �jd��d �t j� ����fdd��}t j|� �jd�}tj�j� �j|j� � �j t j
�� �jj |� W d Q R X �j �j� � �j�j
� � �j� j
� � �j�j
� � tj�j� �j�d� d S )Nr )r% c 3 s �E d H � d7 � d S )Nr r
r
)r) r� r
r r* � s
z?CoroutineGatherTests.test_cancellation_broadcast.<locals>.innerc 3 s( t j� ��jd���E d H �d7 �d S )N)r% r8 )r r; r� r
)r7 r8 �gathererr) r r
r r� � s
z?CoroutineGatherTests.test_cancellation_broadcast.<locals>.outer)r r* r� r9 rG r r� rK r} rW r~ r: r� r� rM )r r* r� rY r
)r7 r8 r� r) r r� r �test_cancellation_broadcast� s$ z0CoroutineGatherTests.test_cancellation_broadcastc s� t jdd� ��t j�jd�� t j�jd��t j� ���fdd��}t j|� �jd�}tj�j� � jd � tj�j� �jd � tj�j� �j|j � t
� d S )Nc s s | E d H t d��d S )Nzshould not be ignored)rA )rY r
r
r r* � s
z:CoroutineGatherTests.test_exception_marking.<locals>.inner)r% c 3 s$ t j�� �����jd�E d H d S )N)r% )r r; r� r
)r� r� r* r r
r r� � s z:CoroutineGatherTests.test_exception_marking.<locals>.outer) r r9 r* r� rG r r� rV r� r� rA )r r� rY r
)r� r� r* r r �test_exception_marking� s
z+CoroutineGatherTests.test_exception_marking)
r! r" r# r. r� r� r� r� r� r
r
r
)r& r r� � s "r� c s^ e Zd Z� fdd�Zejddd��Zddd�Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z � Z
S )�RunCoroutineThreadsafeTestsc s$ t � j� tj� | _| j| j� d S )N)r- r. r r? r% rQ )r )r& r
r r. � s
z!RunCoroutineThreadsafeTests.setUpFc c sH t jd| jd�E d H |r"td��|r@t jjj| j�j� d V || S )Ng�������?)r% zFail!)r r� r% rA r r' r# r} )r r� r� r( r} r
r
r r�
s z RunCoroutineThreadsafeTests.addNc
C sV | j dd||d�}tj|| j�}|r4| jj|jd � z
|j|�S |j� pN|j� X d S )Nr r� )r( r} ) r� r Zrun_coroutine_threadsafer% Zcall_soon_threadsaferI rN rL r} )r r( r} r� � advance_coror( rV r
r
r �target
s
z"RunCoroutineThreadsafeTests.targetc C s, | j jd | j�}| j j|�}| j|d� d S )Nr
)r% �run_in_executorr� r: rM )r rV rN r
r
r � test_run_coroutine_threadsafe!
s z9RunCoroutineThreadsafeTests.test_run_coroutine_threadsafec
C sH | j jd | jd�}| jt��}| j j|� W d Q R X | jd|jj� d S )NTzFail!) r% r� r� rW rA r: r� r� r )r rV � exc_contextr
r
r �,test_run_coroutine_threadsafe_with_exception'
s zHRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_exceptionc
st � fdd�}� j jd |�}� jtj�� � j j|� W d Q R X tj� j � x$tjj � j �D ]}� j
|j � � qZW d S )Nc s � j dd�S )Nr )r� )r� r
)r r
r r, 2
s zXRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeout.<locals>.<lambda>) r% r� rW r r� r: r r� r' rZ rK rL )r r� rV rE r
)r r �*test_run_coroutine_threadsafe_with_timeout/
s zFRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeoutc sB � fdd�}� j jd |�}� jtj�� � j j|� W d Q R X d S )Nc s � j dd�S )NT)r} )r� r
)r r
r r, >
s zZRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_cancelled.<locals>.<lambda>)r% r� rW r r~ r: )r r� rV r
)r r �,test_run_coroutine_threadsafe_task_cancelled;
s zHRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_cancelledc s� � j jd � fdd��}� j jdd� � tj� }� j j|� � jt��}� j j|� W d Q R X � j d|j
j d � � j t
|j�d� |j\\}}}� j |d |j
� d S )Nc s � j dd�S )NT)r� )r� r
)r r
r r, H
s zbRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_factory_exception.<locals>.<lambda>c S s t S )N)�
wrong_name)r% r( r
r
r r, J
s r� r r r� )r% r� r0 r r� rY rW � NameErrorr: r� r� r rM r� Zcall_args_listri )r rV r� r� r% �contextr� r
)r r �4test_run_coroutine_threadsafe_task_factory_exceptionC
s zPRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_factory_exception)FF)FFNF)r! r" r# r. r r9 r� r� r� r� r� r� r� r
r
r
)r& r r� � s
r� c s0 e Zd Z� fdd�Z� fdd�Zdd� Z� ZS )�
SleepTestsc s" t � j� tj� | _tjd � d S )N)r- r. r r? r% rQ )r )r& r
r r. Y
s
zSleepTests.setUpc s | j j� d | _ t� j� d S )N)r% rB r- �tearDown)r )r& r
r r� ^
s
zSleepTests.tearDownc sD d��fdd�� t j� ��fdd��}�jj|� � �j�d� d S )Nr c s � | 7 � d S )Nr
)�num)rN r
r �
inc_resultf
s z.SleepTests.test_sleep_zero.<locals>.inc_resultc 3 sJ �j j� d� �j�d� tjd�j dd�E d H } �j�d� � | � d S )Nr r r� )r% rN )r% r� rM r r� )r� )r� rN r r
r r( j
s
z(SleepTests.test_sleep_zero.<locals>.coro� )r r9 r% r: rM )r r( r
)r� rN r r �test_sleep_zeroc
s
zSleepTests.test_sleep_zero)r! r" r# r. r� r� r
r
r
)r& r r� X
s r� �__main__)r
r )r
r )F)6r� �
contextlibr� r1 rk r� r� rf r� r@ r� rN r r r r r r rF r r� r Ztest.support.script_helperr Ztest.script_helperZasyncio.test_support� version_infor� rx r9 r �contextmanagerr r r r$ r� r� �hasattrZTestCaser� r� r� r� r� r� r� r� r� r� r� r� r! r} r
r
r
r �<module> s�
E dTc]