Your IP : 216.73.216.213
���] - � @ su d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m
Z
d d l Z d d l m Z d d l m
Z
y d d l m Z Wn" e k
r� d d l m Z Yn Xy d d l m Z WnK e k
rNy d d l m Z Wn" e k
rId d l m Z Yn XYn Xe j d d f k Z e j d d
f k Z e j d d � � Z e j d
d � � Z d d d � Z Gd d � d � Z Gd d � d e
j � Z Gd d � d � Z! Gd d � d e! e
j � Z" Gd d � d e! e
j � Z# Gd d � d e
j � Z$ Gd d � d e
j � Z% e& d k rqe j' � d S)!� N)�mock)�
coroutines)�
test_utils)�support)� test_support)�assert_python_ok� � � c C s d S)N� r r r �//opt/alt/python35/lib64/python3.5/test_tasks.py�coroutine_function# s r
c c s5 t j } | j } z | | _ d VWd | | _ Xd S)N)�asyncior �_DEBUG)Zenabledr Z old_debugr r r �set_coroutine_debug( s r Fc C sg | r d | } n
d | } | d k rR | d } d | | | | d | d f Sd | | | f Sd 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_coroutine5 s
r c @ s( e Z d Z d d � Z d d � Z d S)�Dummyc C s d S)Nz<Dummy>r )�selfr r r �__repr__D s zDummy.__repr__c G s d S)Nr )r �argsr r r �__call__G s zDummy.__call__N)�__name__�
__module__� __qualname__r r r r r r r B s r c sl e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z e
j e d � d d � � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* d+ � Z d, d- � Z d. d/ � Z d0 d1 � Z d2 d3 � Z d4 d5 � Z d6 d7 � Z d8 d9 � Z! d: d; � Z" d< d= � Z# d> d? � Z$ d@ dA � Z% dB dC � Z& dD dE � Z' dF dG � Z( dH dI � Z) dJ dK � Z* dL dM � Z+ dN dO � Z, dP dQ � Z- dR dS � Z. dT dU � Z/ dV dW � Z0 dX dY � Z1 dZ d[ � Z2 d\ d] � Z3 d^ d_ � Z4 d` da � Z5 db dc � Z6 dd de � Z7 df dg � Z8 dh di � Z9 dj dk � Z: dl dm � Z; dn do � Z< dp dq � Z= dr ds � Z> dt du � Z? dv dw � Z@ dx dy � ZA dz d{ � ZB d| d} � ZC d~ d � ZD d� d� � ZE d� d� � ZF d� d� � ZG d� d� � ZH d� d� � ZI d� d� � ZJ d� d� � ZK d� d� � ZL d� d� � ZM d� d� � ZN d� d� � ZO d� d� � ZP d� d� � ZQ d� d� � ZR d� d� � ZS e
j eT d� � d� d� � � ZU eV jW d� � d� d� � � ZX eV jW d� � d� d� � � ZY d� d� � ZZ d� d� � Z[ d� d� � Z\ d� d� � Z] d� d� � Z^ d� d� � Z_ � S)�� TaskTestsc s t � j � | j � | _ d S)N)�super�setUp�
new_test_loop�loop)r )� __class__r r r$ M s
zTaskTests.setUpc C sy t j � } t j d | � } t j d d � � } z4 | j t d � � | j j | | � � Wd QRXWd | j � Xd S)Nr& c s s
| Ed Hd S)Nr )�futr r r �runU s z-TaskTests.test_other_loop_future.<locals>.runz Task .* got Future .* attached) r �new_event_loop�Future� coroutine�assertRaisesRegex� RuntimeErrorr& �run_until_complete�close)r �
other_loopr( r) r r r �test_other_loop_futureQ s
!z TaskTests.test_other_loop_futurec sd t j � f d d � � } t j | � d | j �� | j t d � � | j j � � Wd QRXd S)Nc 3 s
� Ed Hd S)Nr r )�taskr r �testa s z2TaskTests.test_task_awaits_on_itself.<locals>.testr& zTask cannot await on itself)r r, �
ensure_futurer& r- r. r/ )r r4 r )r3 r �test_task_awaits_on_itself` s
z$TaskTests.test_task_awaits_on_itselfc C s� t j d d � � } t j | � d | j �} | j j | � | j | j � � | j | j � d � | j | j
| j � t j � } | j | � t j | � d | �} | j | j
| � | j | � | j
� d S)Nc S s d S)N�okr r r r r �notmuchl s z*TaskTests.test_task_class.<locals>.notmuchr& r7 )r r, �Taskr& r/ �
assertTrue�done� assertEqual�result�assertIs�_loopr* �set_event_loopr0 )r r8 �tr& r r r �test_task_classk s
zTaskTests.test_task_classc C s� t j d d � � } t j | � d | j �} | j j | � | j | j � � | j | j � d � | j | j
| j � t j � } | j | � t j | � d | �} | j | j
| � | j | � | j
� d S)Nc S s d S)Nr7 r r r r r r8 } s z7TaskTests.test_ensure_future_coroutine.<locals>.notmuchr& r7 )r r, r5 r& r/ r: r; r< r= r> r? r* r@ r0 )r r8 rA r& r r r � test_ensure_future_coroutine| s
z&TaskTests.test_ensure_future_coroutinec C s� t j d | j � } | j d � t j | � } | j j | � | j | j � � | j | j � d � | j
| | � t j � } | j | � | j
t � � t j | d | �} Wd QRX| j � t j | d | j �} | j
| | � d S)Nr& �ko)r r+ r& �
set_resultr5 r/ r: r; r< r= r> r* r@ � assertRaises�
ValueErrorr0 )r Zf_orig�fr& r r r �test_ensure_future_future� s
z#TaskTests.test_ensure_future_futurec C s t j d d � � } t j | � d | j �} t j | � } | j j | � | j | j � � | j | j � d � | j
| | � t j � } | j | � | j
t � � t j | d | �} Wd QRX| j � t j | d | j �} | j
| | � d S)Nc S s d S)Nr7 r r r r r r8 � s z2TaskTests.test_ensure_future_task.<locals>.notmuchr& r7 )r r, r9 r& r5 r/ r: r; r< r= r> r* r@ rF rG r0 )r r8 Zt_origrA r& r r r �test_ensure_future_task� s
z!TaskTests.test_ensure_future_taskzneed python 3.5 or laterc C sp Gd d � d � } t j d d � � } t j � } | j | � t j | | � � d | �} | j | � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z2TaskTests.test_ensure_future_awaitable.<locals>.Awc S s
| | _ d S)N)�coro)r rK r r r �__init__� s z;TaskTests.test_ensure_future_awaitable.<locals>.Aw.__init__c s s | j Ed HS)N)rK )r r r r � __await__� s z<TaskTests.test_ensure_future_awaitable.<locals>.Aw.__await__N)r r r! rL rM r r r r �Aw� s rN c S s d S)Nr7 r r r r r rK � s z4TaskTests.test_ensure_future_awaitable.<locals>.coror& )r r, r* r@ r5 r/ )r rN rK r& r( r r r � test_ensure_future_awaitable� s
z&TaskTests.test_ensure_future_awaitablec C s( | j t � � t j d � Wd QRXd S)Nr7 )rF � TypeErrorr r5 )r r r r �test_ensure_future_neither� s z$TaskTests.test_ensure_future_neitherc C sL t j d | j � } | j t d � � | j | t j | � � Wd QRXd S)Nr& z#function is deprecated, use ensure_)r r+ r& ZassertWarnsRegex�DeprecationWarningr> �async)r rH r r r �test_async_warning� s
z TaskTests.test_async_warningc sw d � t j � f d d � � � t j � � f d d � � � t j � � � f d d � � } � j j | � � d S)Nc 3 s � � Ed Hd S)Nr r )�barr r �foo� s z%TaskTests.test_get_stack.<locals>.fooc
s� � j d d � } z � j | d j j d � Wd d } Xt j � } � j d d d | � | j d � | j � } � j | d � d S)N�limitr r rV �filezfoo\(\) running)
Z get_stackr< �f_code�co_name�io�StringIO� print_stack�seek�read� assertRegex)rH rX �tb)�Tr r r rU � s
z%TaskTests.test_get_stack.<locals>.barc 3 s( t j � � d � j �� � Ed Hd S)Nr& )r r5 r& r )rb rV r r r �runner� s z(TaskTests.test_get_stack.<locals>.runner)r r, r& r/ )r rc r )rb rU rV r r �test_get_stack� s
!zTaskTests.test_get_stackc C s | j j d � t j d d � � } | j | j d � t rQ | j | j d � | j | j t � t j
| � \ } } d | | f } | � } t j s� t r� d } n d } | j | j d � t r� | j | j | � t j
| d | j �} | j t � � t | d | | j d d �} | j t | � d
| � | j � | j t | � d | � | j t j | j j | � t | d | | j � } | j t | � d
| � t j
| � d | j �} | j j | � t | d | | j � } | j t | � d | � d S)NFc s s
g Ed Hd S)N�abcr r r r r r8 � s z)TaskTests.test_task_repr.<locals>.notmuchr8 z%s:%sr& �runningr Tz <Task pending %s cb=[<Dummy>()]>z#<Task cancelling %s cb=[<Dummy>()]>r; z<Task cancelled %s>z <Task finished %s result='abc'>)r& � set_debugr r, r< r �PY35r! r r �get_function_sourcer r r9 �add_done_callbackr r �_source_traceback�repr�cancelrF �CancelledErrorr/ ) r r8 �filename�linenor �gen�
coro_qualnamerA rK r r r �test_task_repr� sL
zTaskTests.test_task_reprc C s | j j d � t j d d � � } | j | j d � t rQ | j | j d � | j | j t � | � } t j
s| t r� d } d } n
d } } | j | j | � t r� | j | j | � t j
rUt j
rt j | � \ } } | j
d } d | | | | d | d f } n" | j } d | | j | j f } | j t | � d
| � t j | d | j �}
|
j t � � t j
r�d t j | � } n | j } d | j | j f } t | d
| |
j
d t j
�} | j t |
� d | � | j j |
� d S)NFc S s d S)N�{ r r r r r r8 1 s z8TaskTests.test_task_repr_coro_decorator.<locals>.notmuchr8 rK r z0%s() running, defined at %s:%s, created at %s:%sr z%s() running at %s:%sz<CoroWrapper %s>r& z%s:%srf r z <Task pending %s cb=[<Dummy>()]>r )r& rg r r, r< r rh r! r r r r ri rk �gi_code� co_filename�co_firstlinenorl r9 rj r r r/ ) r r8 rq Z coro_namerr ro rp r rK �coderA r r r r � test_task_repr_coro_decorator. sN
z'TaskTests.test_task_repr_coro_decoratorc C s� | j j d � t j d d � � } t j d | j � } t j | | � d | j �} t j | j � | j t | � d t
j t | � � � | j d � | j j
| � d S)NFc s s | Ed HS)Nr )r( r r r �wait_foru s z3TaskTests.test_task_repr_wait_for.<locals>.wait_forr& z<Task .* wait_for=%s>)r& rg r r, r+ r9 r � run_brieflyr` rl �re�escaperE r/ )r rz r( r3 r r r �test_task_repr_wait_forr s
z!TaskTests.test_task_repr_wait_forc C s� t d � �y | j j d � t j d d � � } t j t j | d � � } | j j | d � � } d | _ | j | j
j � Wd QRXt | j
� } d } | j
| j | � | � d S)NTc s s t j d � Ed Hd S)Nr )r �sleep)�x�yr r r �func� s z:TaskTests.test_task_repr_partial_corowrapper.<locals>.funcr � FzV<CoroWrapper TaskTests.test_task_repr_partial_corowrapper.<locals>.func(1)() running, )r r& rg r r, � functools�partial� create_taskZ_log_destroy_pending�
addCleanupZ_coror0 rl r: �
startswith)r r� Z partial_funcr3 Z coro_repr�expectedr r r �"test_task_repr_partial_corowrapper� s
z,TaskTests.test_task_repr_partial_corowrapperc sq t j � � f d d � � } t j d d � � � t j d d � � � | � } | j | j j | � d � d S)Nc 3 s$ � � Ed H} � � Ed H} | | S)Nr )�a�b)�inner1�inner2r r �outer� s z)TaskTests.test_task_basics.<locals>.outerc S s d S)N�* r r r r r r� � s z*TaskTests.test_task_basics.<locals>.inner1c S s d S)Ni� r r r r r r� � s z*TaskTests.test_task_basics.<locals>.inner2i )r r, r< r& r/ )r r� rA r )r� r� r �test_task_basics� s
zTaskTests.test_task_basicsc s� � f d d � } � j | � � t j � f d d � � } t j | � d � �} � j | j � � j t j � � � j | � Wd QRX� j | j
� � � j | j � � � j | j � � d S)Nc 3 s d V} � j d | � d Vd S)Ng $@r )�assertAlmostEqual)�when)r r r rq � s z"TaskTests.test_cancel.<locals>.genc 3 s t j d d � �Ed Hd S)Ng $@r& � )r r r )r& r r r3 � s z#TaskTests.test_cancel.<locals>.taskr& )
r% r r, r9 � call_soonrm rF rn r/ r: r; � cancelled� assertFalse)r rq r3 rA r )r& r r � test_cancel� s zTaskTests.test_cancelc C s� t j d d � � } t j | � d | j �} t j | j � | j � | j t j | j j | � | j
| j � � | j
| j � � | j
| j � � d S)Nc s s d Vd Vd S)Nr� r r r r r r3 � s z)TaskTests.test_cancel_yield.<locals>.taskr& )r r, r9 r& r r{ rm rF rn r/ r: r; r� r� )r r3 rA r r r �test_cancel_yield� s
zTaskTests.test_cancel_yieldc s� t j d | j � � t j � f d d � � } t j | � d | j �} t j | j � � j � | j t j � � | j j
| � Wd QRX| j � j � � | j | j � � d S)Nr& c 3 s
� Ed Hd S)Nr� r r )rH r r r3 � s z0TaskTests.test_cancel_inner_future.<locals>.task)
r r+ r& r, r9 r r{ rm rF rn r/ r: r� )r r3 rA r )rH r �test_cancel_inner_future� s
z"TaskTests.test_cancel_inner_futurec s� t j d | j � � t j � f d d � � } t j | � d | j �} t j | j � � j � | j � | j t j � � | j j
| � Wd QRX| j | j � � | j � j
� � | j | j
� � d S)Nr& c 3 s
� Ed Hd S)Nr� r r )rH r r r3 � s z>TaskTests.test_cancel_both_task_and_inner_future.<locals>.task)r r+ r& r, r9 r r{ rm rF rn r/ r: r; r� )r r3 rA r )rH r �&test_cancel_both_task_and_inner_future� s
z0TaskTests.test_cancel_both_task_and_inner_futurec s t j d | j � � t j d | j � � t j � � f d d � � } t j | � d | j �} t j | j � | j | j � � � j d � t j | j � | j | j � � | j
� | j � j � � | j j
| � } | j | d � | j | j � � d S)Nr& c 3 s6 � Ed Hy
� Ed HWn t j k
r1 d SYn Xd S)Nr� )r rn r )�fut1�fut2r r r3 � s
z1TaskTests.test_cancel_task_catching.<locals>.taskr� )r r+ r& r, r9 r r{ r> � _fut_waiterrE rm r: r� r/ r< r� )r r3 rA �resr )r� r� r �test_cancel_task_catching� s
z#TaskTests.test_cancel_task_catchingc sg t j d | j � � t j d | j � � t j d | j � � t j � � � f d d � � } t j | � d | j �} t j | j � | j | j � � � j d � t j | j � | j | j � � | j
� | j � j � � t j | j � | j | j � � � j d � | j j
| � } | j | d � | j � j � � | j | j � � d S)Nr& c 3 s= � Ed Hy
� Ed HWn t j k
r- Yn X� Ed H} | S)N)r rn )r� )r� r� �fut3r r r3 s
z1TaskTests.test_cancel_task_ignoring.<locals>.taskr� )r r+ r& r, r9 r r{ r> r� rE rm r: r� r/ r< r� )r r3 rA r� r )r� r� r� r �test_cancel_task_ignoring s& !
z#TaskTests.test_cancel_task_ignoringc s� t j � � � j � � t j � � � f d d � � } t j | � d � �� � j t j � j � � � j � j � � � j
� j � � j
� j � � d S)Nc 3 s6 � j � � j � j � t j d d � �Ed Hd S)N�d r& r� )rm r: � _must_cancelr r r )r& r rA r r r3 1 s
z0TaskTests.test_cancel_current_task.<locals>.taskr& )
r r* r@ r, r9 rF rn r/ r: r; r� r� rm )r r3 r )r& r rA r �test_cancel_current_task- s
!z"TaskTests.test_cancel_current_taskc s� t j � } � j | � t j � � f d d � � } t j | � d | �� � j t j | j � � � j � j � � � j
� j � � j
� j � � d S)Nc s � j � � j � j � d S)Nr� )rm r: r� r )r rA r r r3 E s
z*TaskTests.test_cancel_at_end.<locals>.taskr& )
r r* r@ r, r9 rF rn r/ r: r; r� r� rm )r r& r3 r )r rA r �test_cancel_at_end@ s
z TaskTests.test_cancel_at_endc s � f d d � } � j | � � d � g � t j � � � f d d � � } t j | � d � �} � j t � � } � j | � Wd QRX� j t | j � d � � j
| j � � � j � d � � j d � j
� � x � D] } | j � q� W| j � � j t j � j | � d S)
Nc 3 sN d V} � j d | � d V} � j d | � d V} � j d | � d Vd S)Ng�������?g�������?g333333�?)r� )r� )r r r rq T s z6TaskTests.test_stop_while_run_in_complete.<locals>.genr c 3 s` xY � d k r[ � j t j d d � �� � d Ed H� d 7� � d k r � j � q Wd S)N�
g�������?r& r r� r )�appendr r �stopr )r& �waitersr� r r r3 b s
z7TaskTests.test_stop_while_run_in_complete.<locals>.taskr& z+Event loop stopped before Future completed.r� g333333�?)r% r r, r9 rF r. r/ r< �str� exceptionr� r; r� �timer0 rm rn )r rq r3 rA �cm�wr )r& r r� r� r � test_stop_while_run_in_completeR s" !
z)TaskTests.test_stop_while_run_in_completec
s� � f d d � } � j | � � d � t j � � f d d � � } t j | � d � �} � j t j � �$ � j t j | d d � �� Wd QRX� j | j � � � j | j
� � � j d � j � � � j
� d � d S)Nc 3 s9 d V} � j d | � d V} � j d | � d V} d S)Ng�������?r g�������?)r� )r� )r r r rq } s
z$TaskTests.test_wait_for.<locals>.genc
3 s0 d � z t j d d � �Ed HWd d � Xd S)NTg�������?r& Fr; )r r r )� foo_runningr& r r rV � s
z$TaskTests.test_wait_for.<locals>.foor& g�������?F)r% r r, r9 rF � TimeoutErrorr/ rz r: r; r� r� r� r< )r rq rV r( r )r� r& r r �
test_wait_for{ s
&zTaskTests.test_wait_forc C s\ | j � } t j d d � � } | j t j | � d d d | �� } | j | d � d S)Nc S s d S)Nr; r r r r r rK � s z.TaskTests.test_wait_for_blocking.<locals>.coro�timeoutr& r; )r% r r, r/ rz r< )r r& rK r� r r r �test_wait_for_blocking� s z TaskTests.test_wait_for_blockingc s� � f d d � } � j | � � t j � f d d � � } t j � � zO t j | � d � �} � j t j � � � j t j | d � � Wd QRXWd t j d � X� j d � j
� � � j | j � � � j | j
� � d S)Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?r g{�G�z�?)r� )r� )r r r rq � s
z5TaskTests.test_wait_for_with_global_loop.<locals>.genc 3 s t j d d � �Ed Hd S)Ng�������?r& r; )r r r )r& r r rV � s z5TaskTests.test_wait_for_with_global_loop.<locals>.foor& g{�G�z�?)r% r r, r@ r9 rF r� r/ rz r� r� r: r; r� )r rq rV r( r )r& r r � test_wait_for_with_global_loop� s
$z(TaskTests.test_wait_for_with_global_loopc C s� d d � } | j | � } t j d | � } t j | d d d | �} | j d | j d � | j | � } | j | d � d S)Nc s s d Vd Vd Vd S)Ng�������?r r r r r rq � s z3TaskTests.test_wait_for_race_condition.<locals>.genr& r� g�������?g�������?r7 )r% r r+ rz �
call_laterrE r/ r< )r rq r& r( r3 r� r r r � test_wait_for_race_condition� s z&TaskTests.test_wait_for_race_conditionc s � f d d � } � j | � � t j t j d d � �d � �� t j t j d d � �d � �� t j � � � � f d d � � } � j t j | � d � �� } � j | d � � j d � j � � � j t j | � d � �� } � j d � j � � � j | d � d S) Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?r g333333�?)r� )r� )r r r rq � s
z TaskTests.test_wait.<locals>.geng�������?r& g333333�?c 3 sY t j � � g d � �Ed H\ } } � j | t � � g � � � j | t � � d S)Nr& r� )r �waitr< �set)r; �pending)r� r� r& r r r rV � s & z TaskTests.test_wait.<locals>.foor� ) r% r r9 r r, r/ r< r� r� )r rq rV r� r )r� r� r& r r � test_wait� s $$$!!zTaskTests.test_waitc s� � f d d � } � j | � } t j t j d d | �d | �� t j t j d d | �d | �� t j � � � f d d � � } t j | � | j t j | � d | �� } � j | d � d S) Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng{�G�z�?r g� ��Q��?)r� )r� )r r r rq � s
z1TaskTests.test_wait_with_global_loop.<locals>.geng{�G�z�?r& g� ��Q��?c 3 sS t j � � g � Ed H\ } } � j | t � � g � � � j | t � � d S)Nr� )r r� r< r� )r; r� )r� r� r r r rV � s z1TaskTests.test_wait_with_global_loop.<locals>.foor� )r% r r9 r r, r@ r/ r< )r rq r& rV r� r )r� r� r r �test_wait_with_global_loop� s $$!
z$TaskTests.test_wait_with_global_loopc C s� t j d d � � } | d � } t j t j | | | d � g d | j �d | j �} | j j | � \ } } | j | � | j t d d � | D� � d d h � d S)Nc S s | S)Nr )�sr r r rK
s z6TaskTests.test_wait_duplicate_coroutines.<locals>.coror4 �spamr& c s s | ] } | j � Vq d S)N)r= )�.0rH r r r � <genexpr> s z;TaskTests.test_wait_duplicate_coroutines.<locals>.<genexpr>) r r, r9 r� r& r/ r� r< r� )r rK �cr3 r; r� r r r � test_wait_duplicate_coroutines s '
z(TaskTests.test_wait_duplicate_coroutinesc C s� | j t | j j t j t � d | j �� t j d d | j �} t j | g d d d | j �} | j t | j j | � | j � d S)Nr& g $@� return_whenr r ) rF rG r& r/ r r� r� r r0 )r Z
sleep_coroZ wait_coror r r �test_wait_errors s ! zTaskTests.test_wait_errorsc sY � f d d � } � j | � } t j t j d d | �d | �} t j t j d d | �d | �} t j t j | | g d t j d | �d | �} | j | � \ } } � j | h | � � j | h | � � j | j � � � j
| j � � � j | j � � � j
d | j � � | j d � | j t j | | g d | �� d S)Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng $@r g�������?)r� )r� )r r r rq * s
z0TaskTests.test_wait_first_completed.<locals>.geng $@r& g�������?r� r� )r% r r9 r r� �FIRST_COMPLETEDr/ r< r� r; r: � assertIsNoner= r� r� � advance_time)r rq r& r� r� r3 r; r� r )r r �test_wait_first_completed( s" $$
z#TaskTests.test_wait_first_completedc C s t j d d � � } t j d d � � } t j | � d | j �} t j | � d | j �} t j t j | | g d t j d | j �d | j �} | j j | � \ } } | j | | h | � | j | j � � | j
| j � � | j | j � � | j
| j � � d S)Nc s s d Vd S)Nr r r r r �coro1J s z.TaskTests.test_wait_really_done.<locals>.coro1c s s d Vd Vd S)Nr r r r r �coro2N s z.TaskTests.test_wait_really_done.<locals>.coro2r& r� ) r r, r9 r& r� r� r/ r< r: r; r� r= )r r� r� r� r� r3 r; r� r r r �test_wait_really_doneF s z TaskTests.test_wait_really_donec s) � f d d � } � j | � } t j t j d d | �d | �} t j d d � � } t j | � d | �} t j t j | | g d t j d | �d | �} | j | � \ } } � j | h | � � j | h | � � j d | j
� � | j d � | j t j | | g d | �� d S)
Nc 3 s d V} � j d | � d Vd S)Ng $@r )r� )r� )r r r rq c s z0TaskTests.test_wait_first_exception.<locals>.geng $@r& c S s t d � � d S)N�err)�ZeroDivisionErrorr r r r �excm s z0TaskTests.test_wait_first_exception.<locals>.excr� r r� ) r% r r9 r r, r� �FIRST_EXCEPTIONr/ r< r� r� r� ) r rq r& r� r� r� r3 r; r� r )r r �test_wait_first_exceptiona s $
z#TaskTests.test_wait_first_exceptionc s � f d d � } � j | � � t j t j d d � �d � �} t j � f d d � � } t j | � d � �} t j | | g d t j d � �} � j | � \ } } � j | h | � � j | h | � � j d � j
� � � j d � � j t j | | g d � �� d S)
Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng $@r g{�G�z�?)r� )r� )r r r rq � s
z8TaskTests.test_wait_first_exception_in_wait.<locals>.geng $@r& c 3 s( t j d d � �Ed Ht d � � d S)Ng{�G�z�?r& r� )r r r� r )r& r r r� � s z8TaskTests.test_wait_first_exception_in_wait.<locals>.excr� g{�G�z�?r� ) r% r r9 r r, r� r� r/ r< r� r� r� )r rq r� r� r� r3 r; r� r )r& r r �!test_wait_first_exception_in_wait� s $
z+TaskTests.test_wait_first_exception_in_waitc s
� f d d � } � j | � � t j t j d d � �d � �� t j � f d d � � } t j | � d � �� t j � � � � f d d � � } � j t j | � d � �� � j d � j � � � j t j | � d � �� � j d � j � � d S)
Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?r g333333�?)r� )r� )r r r rq � s
z/TaskTests.test_wait_with_exception.<locals>.geng�������?r& c 3 s( t j d d � �Ed Ht d � � d S)Ng333333�?r& Zreally)r r r� r )r& r r �sleeper� s z3TaskTests.test_wait_with_exception.<locals>.sleeperc 3 s� t j � � g d � �Ed H\ } } � j t | � d � � j | t � � t d d � | D� � } � j t | � d � d S)Nr& r� c s s' | ] } | j � d k r | Vq d S)N)r� )r� rH r r r r� � s zBTaskTests.test_wait_with_exception.<locals>.foo.<locals>.<genexpr>r )r r� r< �lenr� )r; r� �errors)r� r� r& r r r rV � s
&z/TaskTests.test_wait_with_exception.<locals>.foog333333�?)r% r r9 r r, r/ r� r� )r rq r� rV r )r� r� r& r r �test_wait_with_exception� s $$ z"TaskTests.test_wait_with_exceptionc s� � f d d � } � j | � � t j t j d d � �d � �� t j t j d d � �d � �� t j � � � � f d d � � } � j t j | � d � �� � j d � j � � � j d � � j t j � � g d � �� d S)
Nc 3 sN d V} � j d | � d V} � j d | � d V} � j d | � d Vd S)Ng�������?r g333333�?g)\��(�?)r� )r� )r r r rq � s z-TaskTests.test_wait_with_timeout.<locals>.geng�������?r& g333333�?c 3 sb t j � � g d d d � �Ed H\ } } � j | t � g � � � j | t � g � � d S)Nr� g)\��(�?r& )r r� r< r� )r; r� )r� r� r& r r r rV � s z-TaskTests.test_wait_with_timeout.<locals>.foog)\��(�?r� )
r% r r9 r r, r/ r� r� r� r� )r rq rV r )r� r� r& r r �test_wait_with_timeout� s $$$
z TaskTests.test_wait_with_timeoutc s � f d d � } � j | � } t j t j d d | �d | �} t j t j d d | �d | �} | j t j | | g d d d | �� \ } } � j | t | g � � � j | t | g � � � j d | j � � | j
d � | j t j | | g d | �� d S)Nc 3 sN d V} � j d | � d V} � j d | � d V} � j d | � d Vd S)Ng�������?r g333333�?)r� )r� )r r r rq � s z4TaskTests.test_wait_concurrent_complete.<locals>.geng�������?r& g333333�?r� r� ) r% r r9 r r/ r� r< r� r� r� r� )r rq r& r� r� r; r� r )r r � test_wait_concurrent_complete� s $$*
z'TaskTests.test_wait_concurrent_completec se d d � } | j | � � d � _ t � � d � t j � � � f d d � � } | d d � � | d d � � | d
d � � t j � � � � f d d
� � } � j t j | � d � �� } | j d
� j � � | j d | d d � k � | j d | d d � k � | j
| d d � � j t j | � d � �� } | j d
� j � � d S)Nc s s d Vd Vd Vd Vd S)Nr g{�G�z�?r r r r r rq � s z(TaskTests.test_as_completed.<locals>.geng �?Fc 3 s[ t j | d � �Ed H� j | � � rW d � k rW d � k rW d � � j d � | S)Nr& r� r� Tg�Q� ��?)r r �addr� )�dtr� )� completedr& � time_shiftedr r r� s
z,TaskTests.test_as_completed.<locals>.sleeperg{�G�z�?r� r� g333333�?r� c 3 sE g } x8 t j � � � g d � �D] } | j | Ed H� q% W| S)Nr& )r � as_completedr� )�valuesrH )r� r� r� r& r r rV s %z(TaskTests.test_as_completed.<locals>.foor& r� ) r% Zslow_callback_durationr� r r, r/ r9 r� r� r: r< )r rq r� rV r� r )r� r� r� r� r& r� r �test_as_completed� s" !
$! !zTaskTests.test_as_completedc s6 d d � } | j | � � t j d d d � �� t j d d d � �� t j � � � f d d � � } � j t j | � d � �� } | j t | � d
| � | j | d d � | j | d d d
� | j | d d t j � | j
d
� j � � � j d � � j t j
� � g d � �� d S)Nc s s d Vd Vd Vd Vd S)Nr g�������?r r r r r rq * s z5TaskTests.test_as_completed_with_timeout.<locals>.geng�������?r� r& g333333�?r� c 3 s� g } x� t j � � g d d d � �D]y } | rA � j d � y" | Ed H} | j d | f � Wq( t j k
r� } z | j d | f � WYd d } ~ Xq( Xq( W| S)Nr� g� ��Q��?r& g{�G�z�?r r� )r r� r� r� r� )r� rH �vr� )r� r� r& r r rV 5 s (
*z5TaskTests.test_as_completed_with_timeout.<locals>.foor� r r g� ��Q��?r� )r r� )r% r r r, r/ r9 r< r� �assertIsInstancer� r� r� r� r� )r rq rV r� r )r� r� r& r � test_as_completed_with_timeout( s !
!
z(TaskTests.test_as_completed_with_timeoutc sw d d � } � j | � � t j d d d � �� t j � � � f d d � � } � j t j | � d � �� d S)Nc s s d Vd Vd Vd S)Nr g{�G�z�?r r r r r rq O s z<TaskTests.test_as_completed_with_unused_timeout.<locals>.geng{�G�z�?r� r& c 3 sH xA t j � g d d d � �D]! } | Ed H} � j | d � q Wd S)Nr� r r& r� )r r� r< )rH r� )r� r& r r r rV X s % z<TaskTests.test_as_completed_with_unused_timeout.<locals>.foo)r% r r r, r/ r9 )r rq rV r )r� r& r r �%test_as_completed_with_unused_timeoutM s
!z/TaskTests.test_as_completed_with_unused_timeoutc C s d d � } | j | � } t j d d d | �} t j d d d | �} | | h } t t j | d | �� } | j t | � d � | j | d � } | j | d � | j d | j � � | j
d � | j | d
� } | j | d � | j d | j � � d S) Nc s s d Vd Vd Vd S)Nr g�������?r r r r r rq b s z5TaskTests.test_as_completed_reverse_wait.<locals>.geng�������?r� r& g�������?r� r� r r ) r% r r �listr� r< r� r/ r� r� r� ) r rq r& r� r� �fs�futsr� r� r r r � test_as_completed_reverse_wait` s
z(TaskTests.test_as_completed_reverse_waitc
s� � f d d � } � j | � } t j d d d | �} t j d d d | �} | | h } t t j | d | �� } � j t | � d � t j | d | �} | j | � \ } } � j t d d � | D� � d d h � d S)
Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?r )r� )r� )r r r rq y s
z3TaskTests.test_as_completed_concurrent.<locals>.geng�������?r� r& r� r� c s s | ] } | j � Vq d S)N)r= )r� rH r r r r� � s z9TaskTests.test_as_completed_concurrent.<locals>.<genexpr>)
r% r r r� r� r< r� r� r/ r� )
r rq r& r� r� r� r� �waiterr; r� r )r r � test_as_completed_concurrentw s z&TaskTests.test_as_completed_concurrentc s� t j d d � � � t j � � f d d � � } t j | � d � j �} � j j | � | j � } � j t | � d d h � � j t | � d � d S) Nc S s | S)Nr )r� r r r rK � s z>TaskTests.test_as_completed_duplicate_coroutines.<locals>.coroc 3 sZ g } � d � } xA t j | | � d � g d � j �D] } | j | Ed H� q: W| S)N�hamr� r& )r r� r& r� )r= r� rH )rK r r r rc � s z@TaskTests.test_as_completed_duplicate_coroutines.<locals>.runnerr& r� r� r� ) r r, r9 r& r/ r= r< r� r� )r rc r( r= r )rK r r �&test_as_completed_duplicate_coroutines� s z0TaskTests.test_as_completed_duplicate_coroutinesc s� � f d d � } � j | � � t j � f d d � � } t j | d d � d � �} � j | � � j | j � � � j | j � d � � j d � j
� � d S)Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?g�������?)r� )r� )r r r rq � s
z!TaskTests.test_sleep.<locals>.genc 3 sA t j | d d � �Ed Ht j | d | d � �Ed H} | S)Nr� r& )r r )r� �argr� )r& r r r� � s !z%TaskTests.test_sleep.<locals>.sleeperg�������?�yeahr& ) r% r r, r9 r/ r: r; r< r= r� r� )r rq r� rA r )r& r r �
test_sleep� s
zTaskTests.test_sleepc s� � f d d � } � j | � } t j t j d d d | �d | �} d � | j � � � f d d � } | | _ t j | � � j � j � | j � t j | � � j
� j � d S)Nc 3 s d V} � j d | � d Vd S)Ng $@r )r� )r� )r r r rq � s z(TaskTests.test_sleep_cancel.<locals>.geng $@r� r& c s � | | | � � � S)Nr )�delay�callbackr )�handle�orig_call_laterr r r� � s z/TaskTests.test_sleep_cancel.<locals>.call_later) r% r r9 r r� r r{ r� Z
_cancelledrm r: )r rq r& rA r� r )r� r� r r �test_sleep_cancel� s
zTaskTests.test_sleep_cancelc s� � f d d � } � j | � � t j � f d d � � � t j � � f d d � � } | � } � j � j | � d � � j d � j � � d S) Nc 3 s7 d V} � j d | � d V} � j d | � d Vd S)Ng�������?r i� )r� )r� )r r r rq � s
z5TaskTests.test_task_cancel_sleeping_task.<locals>.genc 3 s t j | d � �Ed Hd S)Nr& )r r )r� )r& r r r � s z7TaskTests.test_task_cancel_sleeping_task.<locals>.sleepc 3 s_ t j � d � d � �} � j d | j � y
| Ed HWn t j k
rV d SYn Xd Sd S)Ni� r& g�������?r� zslept in)r r9 r� rm rn )r� )r& r r r �doit� s
z6TaskTests.test_task_cancel_sleeping_task.<locals>.doitr� g�������?)r% r r, r< r/ r� r� )r rq r� Zdoerr )r& r r r � test_task_cancel_sleeping_task� s z(TaskTests.test_task_cancel_sleeping_taskc s� t j d | j � � t j � f d d � � } t j | � d | j �} t j | j � | j | j � � | j � t j | j � | j
t j | j j | � | j
| j � | j � j � � d S)Nr& c 3 s
� Ed Hd S)Nr r )r( r r rK � s z6TaskTests.test_task_cancel_waiter_future.<locals>.coro)r r+ r& r, r9 r r{ r> r� rm rF rn r/ r� r: r� )r rK r3 r )r( r � test_task_cancel_waiter_future� s
z(TaskTests.test_task_cancel_waiter_futurec C sd t j d d � � } | � } t j | d | j �} | j d � | j t | j � | j � d S)Nc S s d S)NrD r r r r r r8 s z6TaskTests.test_step_in_completed_task.<locals>.notmuchr& r7 ) r r, r9 r& rE rF �AssertionError�_stepr0 )r r8 rq r3 r r r �test_step_in_completed_task s
z%TaskTests.test_step_in_completed_taskc C s5 t j d d � � } | j t | j j | � � d S)Nc s s d Vd Vd S)Nr rD r r r r r r8 s z+TaskTests.test_step_result.<locals>.notmuch)r r, rF r. r& r/ )r r8 r r r �test_step_result s zTaskTests.test_step_resultc s� Gd d � d t j � } | d | j � � d � t j � � f d d � � } t j | � d | j �} t j | j � | j � j � t � } � j
| � t j | j � | j | � � | j | j � � | j
| j � � d S)Nc s4 e Z d Z � f d d � Z � f d d � Z � S)z.TaskTests.test_step_result_future.<locals>.Futc s d | _ t � j | | � d S)NF)�cb_addedr# rL )r r �kwds)r' r r rL s z7TaskTests.test_step_result_future.<locals>.Fut.__init__c s d | _ t � j | � d S)NT)r� r# rj )r �fn)r' r r rj " s z@TaskTests.test_step_result_future.<locals>.Fut.add_done_callback)r r r! rL rj r r )r' r �Fut s r� r& c 3 s � Ed H� d S)Nr r )r( r= r r �wait_for_future) s z:TaskTests.test_step_result_future.<locals>.wait_for_future)r r+ r& r, r9 r r{ r: r� �objectrE r> r; r� r= )r r� r� rA r� r )r( r= r �test_step_result_future s
z!TaskTests.test_step_result_futurec C sp t j d d � � } t j | � d | j �} | j t | j � | j | j � � | j | j
� t � d S)Nc S s
t � � d S)N)�
BaseExceptionr r r r �notmutch: s z8TaskTests.test_step_with_baseexception.<locals>.notmutchr& ) r r, r9 r& rF r r� r: r; r� r� )r r r3 r r r � test_step_with_baseexception9 s
z&TaskTests.test_step_with_baseexceptionc s� � f d d � } � j | � � t j � f d d � � � t � � t j � � f d d � � } t j | � d � �} t j � � | j � � j | j � � � j
t t j � � � j | j � � � j | j � � � j
| j � � � d S)Nc 3 s d V} � j d | � d Vd S)Ng $@r )r� )r� )r r r rq F s z7TaskTests.test_baseexception_during_cancel.<locals>.genc 3 s t j d d � �Ed Hd S)Nr� r& )r r r )r& r r r� M s z;TaskTests.test_baseexception_during_cancel.<locals>.sleeperc 3 s2 y � � Ed HWn t j k
r- � � Yn Xd S)N)r rn r )�base_excr� r r r S s z<TaskTests.test_baseexception_during_cancel.<locals>.notmutchr& )r% r r, r r9 r r{ rm r� r; rF r: r� r> r� )r rq r r3 r )r r& r r� r � test_baseexception_during_cancelD s
z*TaskTests.test_baseexception_during_cancelc C s� d d � } | j t j | � � d d � } | j t j | � � t j d d � � } | j t j | � � | j t j t j � � � d S)Nc S s d S)Nr r r r r r� g s z.TaskTests.test_iscoroutinefunction.<locals>.fnc s s d Vd S)Nr r r r r �fn1l s z/TaskTests.test_iscoroutinefunction.<locals>.fn1c s s d Vd S)Nr r r r r �fn2p s z/TaskTests.test_iscoroutinefunction.<locals>.fn2)r� r �iscoroutinefunctionr, r: r �Mock)r r� r r r r r �test_iscoroutinefunctionf s z"TaskTests.test_iscoroutinefunctionc sw t j d | j � � t j � f d d � � } | � } | j t � � | j j | � Wd QRX| j � j � � d S)Nr& c 3 s � Vd S)Nr r )r( r r r� z s z;TaskTests.test_yield_vs_yield_from.<locals>.wait_for_future) r r+ r& r, rF r. r/ r� r; )r r� r3 r )r( r �test_yield_vs_yield_fromw s z"TaskTests.test_yield_vs_yield_fromc sV t j d d � � � t j � f d d � � } | � } | j t | j j | � d S)Nc s s d Vd S)Nr r r r r rK � s z:TaskTests.test_yield_vs_yield_from_generator.<locals>.coroc
3 s$ � � } z | VWd | j � Xd S)N)r0 )rq )rK r r r� � s zETaskTests.test_yield_vs_yield_from_generator.<locals>.wait_for_future)r r, rF r. r& r/ )r r� r3 r )rK r �"test_yield_vs_yield_from_generator� s z,TaskTests.test_yield_vs_yield_from_generatorc C sp t j d d � � } | j t j | � � | � } | j t j | � � | j j | � } | j | d � d S)Nc S s d S)Nr4 r r r r r r� � s z7TaskTests.test_coroutine_non_gen_function.<locals>.funcr4 )r r, r: r � iscoroutiner& r/ r< )r r� rK r� r r r � test_coroutine_non_gen_function� s z)TaskTests.test_coroutine_non_gen_functionc s� t j d | j � � t j � f d d � � } t j � f d d � � } t j | � d | j �} t j | � d | j �} | j j | � } | j | d � | j | j � � d S)Nr& c s � S)Nr r )r( r r r� � s zETaskTests.test_coroutine_non_gen_function_return_future.<locals>.funcc s � j d � d S)Nr4 )rE r )r( r r rK � s zETaskTests.test_coroutine_non_gen_function_return_future.<locals>.coror4 ) r r+ r& r, r9 r/ r< r� r= )r r� rK �t1�t2r� r )r( r �-test_coroutine_non_gen_function_return_future� s z7TaskTests.test_coroutine_non_gen_function_return_futurec s� � j t j j d � j � � t j � � f d d � � } t j | � j � d � j �� � j j � � � j t j j d � j � � d S)Nr& c s& � j t j j d | � � k � d S)Nr& )r: r r9 � current_task)r& )r r3 r r rK � s z)TaskTests.test_current_task.<locals>.coro)r� r r9 r r& r, r/ )r rK r )r r3 r �test_current_task� s
!zTaskTests.test_current_taskc s � j t j j d � j � � t j d � j � � t j d � j � � t j � � � � f d d � � } t j � � � � f d d � � } t j | � j � d � j �� t j | � j � d � j �� � j j t j � � f d � j �� � j t j j d � j � � d S)Nr& c 3 s^ � j t j j d | � � k � � Ed H� j t j j d | � � k � � j d � d S)Nr& T)r: r r9 r rE )r& )r� r� r �task1r r r� � s " "zBTaskTests.test_current_task_with_interleaving_tasks.<locals>.coro1c 3 s^ � j t j j d | � � k � � j d � � Ed H� j t j j d | � � k � d S)Nr& T)r: r r9 r rE )r& )r� r� r �task2r r r� � s "
zBTaskTests.test_current_task_with_interleaving_tasks.<locals>.coro2) r� r r9 r r& r+ r, r/ r� )r r� r� r )r� r� r r r r �)test_current_task_with_interleaving_tasks� s $$!!
z3TaskTests.test_current_task_with_interleaving_tasksc s� d � t j d � j � � t j � � � f d d � � � t j � � f d d � � } t j | � d � j �} t j � j � | j � � j j | � � j � d � � j
� j � � d S)Nr r& c 3 sC y
� Ed HWn" t j k
r1 � d 7� � Yn X� j d � d S)Nr zgot past sleep() in inner())r rn �failr )�proofr r� r r �inner� s
z8TaskTests.test_yield_future_passes_cancel.<locals>.innerc 3 s@ y � � Ed HWn t j k
r1 � d 7� Yn X� d 7� d S)Nr� r� )r rn r )r r r r r� � s
z8TaskTests.test_yield_future_passes_cancel.<locals>.outer�e ) r r+ r& r, r5 r r{ rm r/ r< r: r� )r r� rH r )r r r r� r � test_yield_future_passes_cancel� s !
z)TaskTests.test_yield_future_passes_cancelc s� d � t j d � j � � t j � � f d d � � � t j � � � f d d � � } t j | � d � j �} t j � j � | j � � j t j � j j
| � � j d � t j � j � � j � d � d S)Nr r& c 3 s � Ed H� d 7� d S)Nr r r )r r� r r r s z?TaskTests.test_yield_wait_does_not_shield_cancel.<locals>.innerc 3 s7 t j � � g d � j �Ed H\ } } � d 7� d S)Nr& r� )r r� r& )�d�p)r r r r r r�
s )z?TaskTests.test_yield_wait_does_not_shield_cancel.<locals>.outerr )
r r+ r& r, r5 r r{ rm rF rn r/ rE r< )r r� rH r )r r r r� r �&test_yield_wait_does_not_shield_cancel� s !
z0TaskTests.test_yield_wait_does_not_shield_cancelc C sW t j d | j � } t j | � } | j d � | j j | � } | j | d � d S)Nr& r� )r r+ r& �shieldrE r/ r< )r r r� r� r r r �test_shield_result s
z TaskTests.test_shield_resultc C sw t j d | j � } t j | � } t j | j � t d � } | j | � t j | j � | j | j � | � d S)Nr& r� )
r r+ r& r r r{ r. �
set_exceptionr> r� )r r r� r� r r r �test_shield_exception s
z TaskTests.test_shield_exceptionc C se t j d | j � } t j | � } t j | j � | j � t j | j � | j | j � � d S)Nr& ) r r+ r& r r r{ rm r: r� )r r r� r r r �test_shield_cancel) s
z TaskTests.test_shield_cancelc C sQ t j d | j � } | j d � | j j t j | � � } | j | d � d S)Nr& r� )r r+ r& rE r/ r r< )r r( r� r r r �test_shield_shortcut1 s
z TaskTests.test_shield_shortcutc
s� d � t j d � j � � t j � � f d d � � � t j � � � f d d � � } t j | � d � j �} t j � j � | j � � j t j � � � j j
| � Wd QRX� j d � t j � j � � j � d � d S)Nr r& c 3 s � Ed H� d 7� d S)Nr r r )r r� r r r < s z+TaskTests.test_shield_effect.<locals>.innerc 3 s, t j � � d � j �Ed H� d 7� d S)Nr& r� )r r r& r )r r r r r r� B s z+TaskTests.test_shield_effect.<locals>.outerr )
r r+ r& r, r5 r r{ rm rF rn r/ rE r< )r r� rH r )r r r r� r �test_shield_effect7 s !
z TaskTests.test_shield_effectc C s� t j d | j � } t j d | j � } t j | | d | j �} t j | d | j �} t j | j � | j � t j | j � | j | j � � | j
d � | j
d � t j | j � | j | j � d d g � d S)Nr& r r� )
r r+ r& �gatherr r r{ rm r: r� rE r< r= )r �child1�child2�parentr� r r r �test_shield_gatherQ s
z TaskTests.test_shield_gatherc C s t j d | j � } t j d | j � } t j | d | j �} t j | d | j �} t j | | d | j �} t j | j � | j � t j | j � | j | j � t j
� | j | j � � | j | j � � | j
d � | j
d � t j | j � d S)Nr& r r� )r r+ r& r r' r r{ rm r� r� rn r: r� rE )r r( r) r� r� r* r r r �test_gather_shield_ s
z TaskTests.test_gather_shieldc C s� t j d | j � } | j t | j j t j | d | j �� t � } | j t | j j t j | d | j �� | j � d S)Nr& ) r r+ r& rF rP r/ r� r
r0 )r r( rK r r r � test_as_completed_invalid_argsp s z(TaskTests.test_as_completed_invalid_argsc C s� t j d | j � } | j t | j j t j | d | j �� t � } | j t | j j t j | d | j �� | j � | j t | j j t j g d | j �� d S)Nr& )
r r+ r& rF rP r/ r� r
r0 rG )r r( rK r r r �test_wait_invalid_args{ s
z TaskTests.test_wait_invalid_argsc sL � f d d � } t d � � | � Wd QRXt d � � | � Wd QRXd S)Nc s� t j � � � f d d � � } t j d � j � � � j d � | � � � j t j � � � � j � j t j
� � j � j � � j � j
t j � � j j � � � j � j � d S)Nc 3 s � j � j � � Ed Hd S)N)r: �
gi_runningr )r( rq r r r rK � s zGTaskTests.test_corowrapper_mocks_generator.<locals>.check.<locals>.coror& )r r, r+ r& rE r: r r� �gi_frame�types� FrameTyper� r/ ru �CodeTyper/ r� )rK )r )r( rq r �check� s !
z9TaskTests.test_corowrapper_mocks_generator.<locals>.checkFT)r )r r4 r )r r � test_corowrapper_mocks_generator� s
z*TaskTests.test_corowrapper_mocks_generatorc s� t d � �� t j � f d d � � } t j � � f d d � � � t j d d � � � t j | � d � j �} � j j | � } � j | d � Wd QRXd S)
NTc 3 s � � Ed HS)Nr r )r r r r � s z1TaskTests.test_yield_from_corowrapper.<locals>.t1c 3 s: t j d � j � } t j � | � d � j �| Ed HS)Nr& )r r+ r& r9 )rH )r �t3r r r � s z1TaskTests.test_yield_from_corowrapper.<locals>.t2c S s | j d � d S)Nr r� r )r r� r )rE )rH r r r r6 � s z1TaskTests.test_yield_from_corowrapper.<locals>.t3r& r r� r )r r� r )r r r, r9 r& r/ r< )r r r3 �valr )r r r6 r �test_yield_from_corowrapper� s
z%TaskTests.test_yield_from_corowrapperc sN d d � � � f d d � } | j | d � d � | j | d � d � d S)
Nc s s d V} | S)Nr )r� r r r rV � s z7TaskTests.test_yield_from_corowrapper_send.<locals>.fooc sv t j j � � � } | j d � y | j | � Wn0 t k
re } z | j d SWYd d } ~ Xn
Xt d � � d S)Nr zStopIteration was expected)r r � CoroWrapper�send�
StopIterationr r� )r� �cw�ex)rV r r �call� s
z8TaskTests.test_yield_from_corowrapper_send.<locals>.callr r� r� )r r� )r r� )r< )r r> r )rV r � test_yield_from_corowrapper_send� s
z*TaskTests.test_yield_from_corowrapper_sendc C sD t j � } d d � } t j j | � � } | | d <d | _ d S)Nc s s
g Ed Hd S)Nr r r r r rV � s z/TaskTests.test_corowrapper_weakref.<locals>.foor< )�weakref�WeakValueDictionaryr r r9 rq )r ZwdrV r< r r r �test_corowrapper_weakref� s
z"TaskTests.test_corowrapper_weakrefc C sF d d � } t d � } t j j | � � } | j d � | j | | j | � � t j j | � � } | j d � | j | | j t | � � t j j | � � } | j d � | j t d � } | j | t � | j | j d � t j j | � � } | j d � | j t d d � } | j | t � | j | j d � d S)Nc s sJ d } x= y | V} Wq t k
rA } z | } WYd d } ~ Xq Xq Wd S)N)� Exception)�value�er r r rV � s z-TaskTests.test_corowrapper_throw.<locals>.foorV )rV )rV )
rC r r r9 r: r> �throwr� r< r )r rV r� r< r r r �test_corowrapper_throw� s$
z TaskTests.test_corowrapper_throwzneed python 3.4 or laterc C s= t j d d � � } t j � } | j j d � | j j | � | | j � } t j | d | j �} | j t j j
d | j � | h � | j j � | j t | j j
� d � | j j d =d } | j } d } t j � | j t j j
d | j � t � � | j | j d d d t j d
| i � | j � d S) Nc s s+ t j d | � } | Ed Ht d � � d S)Nr& zcode never reached)r r+ rC )r& �futurer r r �kill_me s z:TaskTests.test_log_destroyed_pending_task.<locals>.kill_meTr& r rH �messagez%Task was destroyed but it is pending!r3 r )r r, r r
r& rg �set_exception_handlerr5 r< r9 � all_tasksZ _run_oncer� �_readyr0 �f_localsrk r �
gc_collectr� Zassert_called_with�ANYZ
reset_mock)r rI Z mock_handlerrK r3 r r r r � test_log_destroyed_pending_task� s( %
%
z)TaskTests.test_log_destroyed_pending_taskzasyncio.base_events.loggerc ss t j � � | j � � t j d d � � � t j � � f d d � � } � j | � � | j | j j � d S)Nc S s
t � d S)N)rP r r r r rK , s z>TaskTests.test_tb_logger_not_called_after_cancel.<locals>.coroc 3 s> � j � � � } t j d d � �Ed H| j � d } d S)Ng�������?r& )r� r r rm )r3 )rK r& r r rc 0 s
z@TaskTests.test_tb_logger_not_called_after_cancel.<locals>.runner)r r* r@ r, r/ r� �error�called)r �m_logrc r )rK r& r �&test_tb_logger_not_called_after_cancel' s
z0TaskTests.test_tb_logger_not_called_after_cancelzasyncio.coroutines.loggerc
C s� t d � � t j d d � � } Wd QRXt } t j � j d } | � t j � | j | j
j � | j
j d d } t
j | � \ } } d t j | j � t j | � | t j | � | f } | j | t j | t j � � d S)NTc S s d S)Nr r r r r � coro_noop= s z9TaskTests.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\):\n.*\n File "%s", line %s, in test_coroutine_never_yielded\n coro_noop\(\)$)r r r, �__file__�sys� _getframe�f_linenor rO r: rR rS � call_argsr ri r| r} r! r` �compile�DOTALL) r rT rV Z tb_filename� tb_linenorJ Z
func_filenameZ func_linenoZregexr r r � test_coroutine_never_yielded: s
z&TaskTests.test_coroutine_never_yieldedc sL � f d d � } t d � � | � Wd QRXt d � � | � Wd QRXd S)Nc s> t j d d � � } � j j | � � } � j | d � d S)Nc S s t j d d � � } | � S)Nc S s d S)Nr r r r r r �
inner_coro^ s zeTaskTests.test_return_coroutine_from_coroutine.<locals>.check.<locals>.outer_coro.<locals>.inner_coro)r r, )r` r r r �
outer_coro\ s zQTaskTests.test_return_coroutine_from_coroutine.<locals>.check.<locals>.outer_coror )r r, r& r/ r< )ra r= )r r r r4 [ s z=TaskTests.test_return_coroutine_from_coroutine.<locals>.checkFT)r )r r4 r )r r �$test_return_coroutine_from_coroutineW s
z.TaskTests.test_return_coroutine_from_coroutinec C s� | j j d � t j t � d | j �} t j � j d } | j | j t
� | j | j d d d � t | d f � | j j
| � d S)NTr& r r �test_task_source_tracebackr )r& rg r r9 r
rX rY rZ r� rk r� r< rW r/ )r r3 rp r r r rc o s
z$TaskTests.test_task_source_tracebackc s� t j � � | j � j � t j � f d d � � } � j | � � } � j t j | | d � �� } � j | j � | j t j
� j | � | j | j
� � d S)Nc 3 s t j d � � } | Ed Hd S)Nr& )r r+ )r( )r& r r �blocking_coroutine s z;TaskTests._test_cancel_wait_for.<locals>.blocking_coroutiner& )r r* r� r0 r, r� rz r� rm rF rn r/ r: r� )r r� rd r3 r� r )r& r �_test_cancel_wait_for{ s !
z TaskTests._test_cancel_wait_forc C s | j d � d S)N)re )r r r r � test_cancel_blocking_wait_for� s z'TaskTests.test_cancel_blocking_wait_forc C s | j d � d S)Ng N@)re )r r r r �test_cancel_wait_for� s z TaskTests.test_cancel_wait_forc s� t j � } | j | j � t j d | � � � f d d � } t j | � d | �} t j | d | �� d � � � f d d � } � j | � � j d � | j � � | j
� d � | j � j � � | j
� j
� d g � d S)Nr& c 3 s � Ed HS)Nr r )r( r r �
child_coro� s z0TaskTests.test_cancel_gather.<locals>.child_coroc s � j � � d S)N)rm )�_)�
cancel_result� gather_taskr r �cancelling_callback� s z9TaskTests.test_cancel_gather.<locals>.cancelling_callbackr� F)r r* r� r0 r+ r' r5 rj rE r/ r< r� r� r= )r r& rh Z
gather_futurerl r )rj r( rk r �test_cancel_gather� s
z TaskTests.test_cancel_gather)`r r r! r$ r2 r6 rB rC rI rJ �unittestZ
skipUnlessrh rO rQ rT rd rs ry 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$ r% r& r+ r, r- r. r5 r8 r? rB rG �PY34rQ r ZpatchrU r_ rb rc re rf rg rm r r )r' r r" K s� ; D ) !
" - %
"
! ' ) r" c sv e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
� S)�GatherTestsBasec sE t � j � | j � | _ | j � | _ | j | j d d �d S)N�cleanupF)r# r$ r% �one_loopr1 r@ )r )r' r r r$ � s
zGatherTestsBase.setUpc C s! x | j r t j | � q Wd S)N)rM r r{ )r r& r r r � _run_loop� s zGatherTestsBase._run_loopc s� � f d d � t d � D� \ } } } t j � j | | | � | � } t j � } | j | � | j d � | j d � � j � j � � j
| j d � � j | j
� � | j d � � j � j � | j | � � j
| j � d d d g � d S)Nc s% g | ] } t j d � j � � q S)r& )r r+ rr )r� �i)r r r �
<listcomp>� s z2GatherTestsBase._check_success.<locals>.<listcomp>r r r� F)�ranger r' � wrap_futuresr � MockCallbackrj rE rs rr r< rS r� r; �assert_called_once_withr= )r �kwargsr� r� r� r( �cbr )r r �_check_success� s (!
z GatherTestsBase._check_successc C s | j � | j d d � d S)N�return_exceptionsF)r| )r r r r � test_success� s
z GatherTestsBase.test_successc C s | j d d � d S)Nr} T)r| )r r r r � test_result_exception_success� s z-GatherTestsBase.test_result_exception_successc s � f d d � t d � D� \ } } } } } t j � j | | | | | � � } t j � } | j | � t � } | j d � | j | � � j
� j � � j | j
� � | j | � � j | j � | � | j d � | j � | j t � � | j � d S)Nc s% g | ] } t j d � j � � q S)r& )r r+ rr )r� rt )r r r ru � s z6GatherTestsBase.test_one_exception.<locals>.<listcomp>r
r r )rv r r' rw r rx rj r� rE r" rs rr r: r; ry r> r� rm r. ) r r� r� r� r rE r( r{ r� r )r r �test_one_exception� s .$
z"GatherTestsBase.test_one_exceptionc s* � f d d � t d � D� \ } } } } t j � j | | | | � d d � } t j � } | j | � t � } t � } | j d � | j
| � | j d � � j � j � � j
| j � � | j
| � � j � j � � j | j � � | j | � � j | j � d d | | g � d S)Nc s% g | ] } t j d � j � � q S)r& )r r+ rr )r� rt )r r r ru � s z:GatherTestsBase.test_return_exceptions.<locals>.<listcomp>r r} Tr r )rv r r' rw r rx rj r� r. rE r" rs rr r� r; r: ry r< r= ) r r� r� r� r r( r{ r� Zexc2r )r r �test_return_exceptions� s" +
z&GatherTestsBase.test_return_exceptionsc C s t j j t j j t j � � } d j d � } t d d | d | �\ } } } | j | j � d � t d | d d d | �\ } } } | j | j � d � t d | d d
d | �\ } } } | j | j � d � t d d | d d
d | �\ } } } | j | j � d � d S)
N�
�import asyncio.coroutines� print(asyncio.coroutines._DEBUG)z-Ez-c�
PYTHONPATHs FalseZPYTHONASYNCIODEBUG� �1s True)r� r� ) �os�path�dirnamer rW �joinr r< �rstrip)r Zaio_pathrx �sts�stdout�stderrr r r �test_env_var_debug� s& ! z"GatherTestsBase.test_env_var_debug) r r r! r$ rs r| r~ r r� r� r� r r )r' r rp � s rp c @ sd e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d S)�FutureGatherTestsc G s | S)Nr )r �futuresr r r rw s z FutureGatherTests.wrap_futuresc C s� t j | j � | j t j d � t j | � } | j | t j � | j | j | j � | j | j � | j
| j � � | j | j
� g � t j | d | j � } | j | j | j � d S)Nr& )r r@ rr r� r' r� r+ r> r? rs r: r; r< r= r1 )r Z seq_or_iterr( r r r �_check_empty_sequence s z'FutureGatherTests._check_empty_sequencec C sA | j g � | j f � | j t � � | j t d � � d S)Nr� )r� r� �iter)r r r r � test_constructor_empty_sequence* s
z1FutureGatherTests.test_constructor_empty_sequencec C s� t j d | j � } t j d | j � } | j t � � t j | | � Wd QRX| j t � � t j | d | j �Wd QRXd S)Nr& )r r+ rr r1 rF rG r' )r r� r� r r r �%test_constructor_heterogenous_futures0 s z7FutureGatherTests.test_constructor_heterogenous_futuresc s� � f d d � t d � D� } t j | � } � j | j � j � � j � j � � j | j � � t j | d � j � } � j | j � j � � j � j � � j | j � � d S)Nc s% g | ] } t j d � j � � q S)r& )r r+ r1 )r� rt )r r r ru 9 s zIFutureGatherTests.test_constructor_homogenous_futures.<locals>.<listcomp>r r& ) rv r r' r> r? r1 rs r� r; )r Zchildrenr( r )r r �#test_constructor_homogenous_futures8 s z5FutureGatherTests.test_constructor_homogenous_futuresc s
� f d d � t d � D� \ } } } } } t j | | | | | � } t j � } | j | � | j d � | j � � j � j � � j
| j � � | j | � � j
| j � � � j | j � t j � | j d � | j � | j t � � | j � d S)Nc s% g | ] } t j d � j � � q S)r& )r r+ rr )r� rt )r r r ru D s z;FutureGatherTests.test_one_cancellation.<locals>.<listcomp>r
r r )rv r r' r rx rj rE rm rs rr r: r; ry r� r� r� r� rn r" r. )r r� r� r� r rE r( r{ r )r r �test_one_cancellationC s .
z'FutureGatherTests.test_one_cancellationc sp � f d d � t d � D� \ } } } } } } t j | | | | | | d d �} t j � } | j | � | j d � t � } | j | � | j � � j
� j � � j | j
� � | j d � | j � t � }
| j |
� � j j | � } � j | d t j � � j | d t j � d | d <| d <� j | d | d d d |
g � | j | � d S)
Nc s% g | ] } t j d � j � � q S)r& )r r+ rr )r� rt )r r r ru V s zLFutureGatherTests.test_result_exception_one_cancellation.<locals>.<listcomp>� r} Tr r r� r )rv r r' r rx rj rE r� r" rm rs rr r� r; r. r/ r� rn r< ry ) r r� r� r� r rE rH r( r{ ZzdeZrter� r )r r �&test_result_exception_one_cancellationU s* "$
"z8FutureGatherTests.test_result_exception_one_cancellationN)
r r r! rw r� r� r� r� r� r� r r r r r� s r� c s^ e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z � S)
�CoroutineGatherTestsc s! t � j � t j | j � d S)N)r# r$ r r@ rr )r )r' r r r$ o s
zCoroutineGatherTests.setUpc G sC g } x6 | D]. } t j | d d � � } | j | � � q
W| S)Nc s s | Ed HS)Nr )r( r r r rK v s z/CoroutineGatherTests.wrap_futures.<locals>.coro)r r, r� )r r� Zcorosr( rK r r r rw s s
z!CoroutineGatherTests.wrap_futuresc C s� t j d d � � } | � } | � } t j | | � } | j | j | j � | j j | � | j | j d d �| � } | � } t j | | d | j �} | j | j | j � | j j | � d S)Nc S s d S)Nre r r r r r rK } s zBCoroutineGatherTests.test_constructor_loop_selection.<locals>.cororq Fr& ) r r, r' r> r? rr r/ r@ r1 )r rK Zgen1Zgen2r( Zgen3Zgen4r� r r r � test_constructor_loop_selection| s z4CoroutineGatherTests.test_constructor_loop_selectionc C s~ t j d d � � } | d � } t j | | | d � | d | j �} | j | j � | j | j � d d d d g � d S)Nc S s | S)Nr )r� r r r rK � s z<CoroutineGatherTests.test_duplicate_coroutines.<locals>.corore �defr& )r r, r' rr rs r< r= )r rK r� r( r r r �test_duplicate_coroutines� s
'z.CoroutineGatherTests.test_duplicate_coroutinesc
st d � t j d � j � � t j � � f d d � � } t j | � d � j �� t j | � d � j �� d � t j � � � � � f d d � � } t j | � d � j �} t j � j � � j | j � � � j t j
� � � j j | � Wd QRX� j � j � � � j � j
� � � j � j
� � � j � j
� � t j � j � � j � d � d S)Nr r& c 3 s � Ed H� d 7� d S)Nr r r )r r� r r r � s z?CoroutineGatherTests.test_cancellation_broadcast.<locals>.innerc 3 s2 t j � � d � j �� � Ed H� d 7� d S)Nr& r� )r r' rr r )r( r) �gathererr r r r r� � s z?CoroutineGatherTests.test_cancellation_broadcast.<locals>.outer)r r+ rr r, r5 r r{ r: rm rF rn r/ r� r� r< )r r r� rH r )r( r) r� r r r� r �test_cancellation_broadcast� s$ 'z0CoroutineGatherTests.test_cancellation_broadcastc s� t j d d � � � t j d � j � � t j d � j � � t j � � � � f d d � � } t j | � d � j �} t j � j � � j d � t j � j � � j d � t j � j � � j | j � t
� d S)Nc s s | Ed Ht d � � d S)Nzshould not be ignored)r. )rH r r r r � s z:CoroutineGatherTests.test_exception_marking.<locals>.innerr& c 3 s. t j � � � � � � d � j �Ed Hd S)Nr& )r r' rr r )r� r� r r r r r� � s z:CoroutineGatherTests.test_exception_marking.<locals>.outer) r r, r+ rr r5 r r{ rE r� r� r. )r r� rH r )r� r� r r r �test_exception_marking� s $
z+CoroutineGatherTests.test_exception_marking) r r r! r$ rw r� r� r� r� r r )r' r r� m s "r� c s� e Z d Z � f d d � Z e j d d d d � � Z d d d d d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z � S)�RunCoroutineThreadsafeTestsc s0 t � j � t j � | _ | j | j � d S)N)r# r$ r r* r& r@ )r )r' r r r$ � s
z!RunCoroutineThreadsafeTests.setUpFc c s\ t j d d | j �Ed H| r- t d � � | rT t j j j | j � j � d V| | S)Ng�������?r& zFail!)r r r& r. Ztasksr9 r rm )r r� r� r rm r r r r� � s z RunCoroutineThreadsafeTests.addNc
C s~ | j d d d | d | �} t j | | j � } | rO | j j | j d � z | j | � SWd | j � px | j � Xd S)Nr r� r rm ) r� r Zrun_coroutine_threadsafer& Zcall_soon_threadsafer: r= r; rm )r r rm r� � advance_cororK rH r r r �target� s z"RunCoroutineThreadsafeTests.targetc C s> | j j d | j � } | j j | � } | j | d � d S)Nr )r& �run_in_executorr� r/ r< )r rH r= r r r � test_run_coroutine_threadsafe� s z9RunCoroutineThreadsafeTests.test_run_coroutine_threadsafec
C s^ | j j d | j d � } | j t � � } | j j | � Wd QRX| j d | j j � d S)NTzFail!) r& r� r� rF r. r/ �assertInr� r )r rH � exc_contextr r r �,test_run_coroutine_threadsafe_with_exception� s zHRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_exceptionc
s� � f d d � } � j j d | � } � j t j � � � j j | � Wd QRXt j � j � x0 t j j � j � D] } � j
| j � � qw Wd S)Nc s � j d d � S)Nr� r )r� r )r r r �<lambda> s zXRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeout.<locals>.<lambda>) r& r� rF r r� r/ r r{ r9 rL r: r; )r r� rH r3 r )r r �*test_run_coroutine_threadsafe_with_timeout s zFRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeoutc sU � f d d � } � j j d | � } � j t j � � � j j | � Wd QRXd S)Nc s � j d d � S)Nrm T)r� r )r r r r� s zZRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_cancelled.<locals>.<lambda>)r& r� rF r rn r/ )r r� rH r )r r �,test_run_coroutine_threadsafe_task_cancelled s zHRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_cancelledc s� � j j d � f d d � � } � j j d d � � t j � } � j j | � � j t � � } � j j | � Wd QRX� j d | j
j d � � j t
| j � d � | j \ \ } } } � j | d | j
� d S)Nc s � j d d � S)Nr� T)r� r )r r r r� s zbRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_factory_exception.<locals>.<lambda>c S s t S)N)�
wrong_name)r& rK r r r r� s r� r r r� )r& r� Zset_task_factoryr rx rK rF � NameErrorr/ r� r� r r< r� Zcall_args_listr[ )r rH r� r� r& �contextrz r )r r �4test_run_coroutine_threadsafe_task_factory_exception s zPRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_factory_exception)
r r r! r$ r r, r� r� r� r� r� r� r� r r )r' r r� � s
r� c s@ e Z d Z � f d d � Z � f d d � Z d d � Z � S)�
SleepTestsc s- t � j � t j � | _ t j d � d S)N)r# r$ r r* r& r@ )r )r' r r r$ . s
zSleepTests.setUpc s' | j j � d | _ t � j � d S)N)r& r0 r# �tearDown)r )r' r r r� 3 s
zSleepTests.tearDownc s` d � � f d d � � t j � � � f d d � � } � j j | � � � j � d � d S)Nr c s � | 7� d S)Nr )�num)r= r r �
inc_result; s z.SleepTests.test_sleep_zero.<locals>.inc_resultc 3 sd � j j � d � � j � d � t j d d � j d d �Ed H} � j � d � � | � d S)Nr r r& r= r� )r& r� r< r r )r� )r� r= r r r rK ? s
#z(SleepTests.test_sleep_zero.<locals>.coro� )r r, r& r/ r< )r rK r )r� r= r r �test_sleep_zero8 s
!zSleepTests.test_sleep_zero)r r r! r$ r� r� r r )r' r r� - s r� �__main__)(�
contextlibr� r[ r� r| rX r� r1 rn r@ r r r r r4 r � ImportErrorr Ztest.support.script_helperr Ztest.script_helperZasyncio.test_support� version_inforo rh r, r
�contextmanagerr r r ZTestCaser" rp r� r� r� r� r �mainr r r r �<module> sZ
� � � � � � � qd T c]