Your IP : 216.73.216.213
�
f f�/ � @ s� d 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
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
rTy d d l m Z Wn" e k
rOd 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& Gd! d" � d" e j � Z' e( d# k r�e j) � n d S)$zTests for tasks.py.� N)�mock)�
coroutines)�
test_utils)�support)� test_support)�assert_python_ok� � � c C s d S)N� r r r �A/opt/alt/python34/lib64/python3.4/test/test_asyncio/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 )Zqualname�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 @ s e Z 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 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 e j
eQ d� � d� d� � � ZR eS jT d� � d� d� � � ZU d� d� � ZV d� d� � ZW d� d� � ZX d� d� � ZY d� S)�� TaskTestsc C s | j � | _ d S)N)�
new_test_loop�loop)r r r r �setUpM s zTaskTests.setUpc C sx t j � } t j d | � } t j d d � � } z3 | j t d � � | j j | | � � Wd QXWd | j � Xd S)Nr# c s s
| Dd Hd S)Nr )�futr r r �runT s z-TaskTests.test_other_loop_future.<locals>.runz Task .* got Future .* attached) r �new_event_loop�Future� coroutineZassertRaisesRegex� RuntimeErrorr# �run_until_complete�close)r �
other_loopr% r& r r r �test_other_loop_futureP s
z TaskTests.test_other_loop_futurec 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 �notmuch` s z*TaskTests.test_task_class.<locals>.notmuchr# r/ )r r) �Taskr# r+ �
assertTrue�done� assertEqual�result�assertIs�_loopr' �set_event_loopr, )r r0 �tr# r r r �test_task_class_ 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)Nr/ r r r r r r0 q s z7TaskTests.test_ensure_future_coroutine.<locals>.notmuchr# r/ )r r) �
ensure_futurer# r+ r2 r3 r4 r5 r6 r7 r' r8 r, )r r0 r9 r# r r r � test_ensure_future_coroutinep 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 QX| j � t j | d | j �} | j
| | � d S)Nr# �ko)r r( r# �
set_resultr; r+ r2 r3 r4 r5 r6 r' r8 � assertRaises�
ValueErrorr, )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 QX| j � t j | d | j �} | j
| | � d S)Nc S s d S)Nr/ r r r r r r0 � s z2TaskTests.test_ensure_future_task.<locals>.notmuchr# r/ )r r) r1 r# r; r+ r2 r3 r4 r5 r6 r' r8 r? r@ r, )r r0 Zt_origr9 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 rD r r r �__init__� s z;TaskTests.test_ensure_future_awaitable.<locals>.Aw.__init__c s s | j Dd HS)N)rD )r r r r � __await__� s z<TaskTests.test_ensure_future_awaitable.<locals>.Aw.__await__N)r r r rE rF r r r r �Aw� s rG c S s d S)Nr/ r r r r r rD � s z4TaskTests.test_ensure_future_awaitable.<locals>.coror# )r r) r' r8 r; r+ )r rG rD r# r% r r r � test_ensure_future_awaitable� s
z&TaskTests.test_ensure_future_awaitablec C s' | j t � � t j d � Wd QXd S)Nr/ )r? � TypeErrorr r; )r r r r �test_ensure_future_neither� s z$TaskTests.test_ensure_future_neitherc C sK t j d | j � } | j t d � � | j | t j | � � Wd QXd S)Nr# z#function is deprecated, use ensure_)r r( r# ZassertWarnsRegex�DeprecationWarningr6 Zasync)r rA 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 � � Dd 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 rN �filezfoo\(\) running)
Z get_stackr4 �f_code�co_name�io�StringIO� print_stack�seek�read� assertRegex)rA rP �tb)�Tr r r rM � s
z%TaskTests.test_get_stack.<locals>.barc 3 s( t j � � d � j �� � Dd Hd S)Nr# )r r; r# r )rZ rN r r r �runner� s z(TaskTests.test_get_stack.<locals>.runner)r r) r# r+ )r r[ r )rZ rM rN r r �test_get_stack� s
!zTaskTests.test_get_stackc C s | j j d � t j d d � � } | j | j d � t rT | j | j d � n | j | j t � t j
| � \ } } d | | f } | � } t j s� t r� d } n d } | j | j d � t r� | j | j | � n 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 Dd Hd S)N�abcr r r r r r0 � s z)TaskTests.test_task_repr.<locals>.notmuchr0 z%s:%sr# �runningr Tz <Task pending %s cb=[<Dummy>()]>z#<Task cancelling %s cb=[<Dummy>()]>r3 z<Task cancelled %s>z <Task finished %s result='abc'>)r# � set_debugr r) r4 r �PY35r r r �get_function_sourcer r r1 �add_done_callbackr r �_source_traceback�repr�cancelr? �CancelledErrorr+ ) r r0 �filename�linenor �gen�
coro_qualnamer9 rD r r r �test_task_repr� sL
zTaskTests.test_task_reprc C s | j j d � t j d d � � } | j | j d � t rT | j | j d � n | j | j t � | � } t j
s t r� d } d } n
d } } | j | j | � t r� | j | j | � n t j
r^t j
r t j | � \ } } | j
d } d | | | | d | d f } n" | j } d | | j | j f } | j t | � d
| � n 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 r0 % s z8TaskTests.test_task_repr_coro_decorator.<locals>.notmuchr0 rD r z0%s() running, defined at %s:%s, created at %s:%sr z%s() running at %s:%sz<CoroWrapper %s>r# z%s:%sr^ r z <Task pending %s cb=[<Dummy>()]>r )r# r_ r r) r4 r r` r r r r r ra rc �gi_code� co_filename�co_firstlinenord r1 rb r r r+ ) r r0 ri Z coro_namerj rg rh r rD �coder9 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 | Dd HS)Nr )r% r r r �wait_fori s z3TaskTests.test_task_repr_wait_for.<locals>.wait_forr# z<Task .* wait_for=%s>)r# r_ r r) r( r1 r � run_brieflyrX rd �re�escaper> r+ )r rr r% �taskr r r �test_task_repr_wait_forf 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 QXt | j
� } d } | j
| j | � | � d S)NTc s s t j d � Dd 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# r_ r r) � functools�partial� create_taskZ_log_destroy_pending�
addCleanupZ_coror, rd r2 �
startswith)r r{ Z partial_funcrv Z coro_repr�expectedr r r �"test_task_repr_partial_corowrapperv 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$ � � Dd H} � � Dd 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) r4 r# r+ )r r� r9 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 QX� 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 ri � s z"TaskTests.test_cancel.<locals>.genc 3 s t j d d � �Dd Hd S)Ng $@r# � )r rx r )r# r r rv � s z#TaskTests.test_cancel.<locals>.taskr# )
r" r r) r1 � call_soonre r? rf r+ r2 r3 � cancelled� assertFalse)r ri rv r9 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 rv � s z)TaskTests.test_cancel_yield.<locals>.taskr# )r r) r1 r# r rs re r? rf r+ r2 r3 r� r� )r rv r9 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 QX| j � j � � | j | j � � d S)Nr# c 3 s
� Dd Hd S)Nr� r r )rA r r rv � s z0TaskTests.test_cancel_inner_future.<locals>.task)
r r( r# r) r1 r rs re r? rf r+ r2 r� )r rv r9 r )rA 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 QX| j | j � � | j � j
� � | j | j
� � d S)Nr# c 3 s
� Dd Hd S)Nr� r r )rA r r rv � s z>TaskTests.test_cancel_both_task_and_inner_future.<locals>.task)r r( r# r) r1 r rs re r? rf r+ r2 r3 r� )r rv r9 r )rA 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 � Dd Hy
� Dd HWn t j k
r1 d SYn Xd S)Nr� )r rf r )�fut1�fut2r r rv � s
z1TaskTests.test_cancel_task_catching.<locals>.taskr� )r r( r# r) r1 r rs r6 � _fut_waiterr> re r2 r� r+ r4 r� )r rv r9 �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= � Dd Hy
� Dd HWn t j k
r- Yn X� Dd H} | S)N)r rf )r� )r� r� �fut3r r rv s
z1TaskTests.test_cancel_task_ignoring.<locals>.taskr� )r r( r# r) r1 r rs r6 r� r> re r2 r� r+ r4 r� )r rv r9 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 � �Dd Hd S)N�d r# r� )re r2 � _must_cancelr rx r )r# r r9 r r rv % s
z0TaskTests.test_cancel_current_task.<locals>.taskr# )
r r' r8 r) r1 r? rf r+ r2 r3 r� r� re )r rv r )r# r r9 r �test_cancel_current_task! s
!z"TaskTests.test_cancel_current_taskc
s � f d d � } � j | � � d � g � t j � � � f d d � � } t j | � d � �} � j t � � } � j | � Wd QX� 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 ri 6 s z6TaskTests.test_stop_while_run_in_complete.<locals>.genr c 3 sc x\ � d k r^ � j t j d d � �� � d Dd H� d 7� � d k r � j � q q Wd S)N�
g�������?r# r r| r )�appendr rx �stopr )r# �waitersry r r rv D s
z7TaskTests.test_stop_while_run_in_complete.<locals>.taskr# z+Event loop stopped before Future completed.r| g333333�?)r" r r) r1 r? r* r+ r4 �str� exceptionr� r3 r� �timer, re rf )r ri rv r9 �cm�wr )r# r r� ry r � test_stop_while_run_in_complete4 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 QX� 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 ri _ s
z$TaskTests.test_wait_for.<locals>.genc
3 s0 d � z t j d d � �Dd HWd d � Xd S)NTg�������?r# Fr3 )r rx r )� foo_runningr# r r rN j s
z$TaskTests.test_wait_for.<locals>.foor# g�������?F)r" r r) r1 r? � TimeoutErrorr+ rr r2 r3 r� r� r� r4 )r ri rN 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)Nr3 r r r r r rD � s z.TaskTests.test_wait_for_blocking.<locals>.coro�timeoutr# r3 )r" r r) r+ rr r4 )r r# rD 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 � � zN t j | � d � �} � j t j � � � j t j | d � � Wd QXWd 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 ri � s
z5TaskTests.test_wait_for_with_global_loop.<locals>.genc 3 s t j d d � �Dd Hd S)Ng�������?r# r3 )r rx r )r# r r rN � s z5TaskTests.test_wait_for_with_global_loop.<locals>.foor# g{�G�z�?)r" r r) r8 r1 r? r� r+ rr r� r� r2 r3 r� )r ri rN 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 ri � s z3TaskTests.test_wait_for_race_condition.<locals>.genr# r� g�������?g�������?r/ )r" r r( rr �
call_laterr> r+ r4 )r ri r# r% rv 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 ri � s
z TaskTests.test_wait.<locals>.geng�������?r# g333333�?c 3 sY t j � � g d � �Dd H\ } } � j | t � � g � � � j | t � � d S)Nr# r� )r �waitr4 �set)r3 �pending)r� r� r# r r r rN � s & z TaskTests.test_wait.<locals>.foor� ) r" r r1 rx r) r+ r4 r� r� )r ri rN 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 ri � s
z1TaskTests.test_wait_with_global_loop.<locals>.geng{�G�z�?r# g� ��Q��?c 3 sS t j � � g � Dd H\ } } � j | t � � g � � � j | t � � d S)Nr� )r r� r4 r� )r3 r� )r� r� r r r rN � s z1TaskTests.test_wait_with_global_loop.<locals>.foor� )r" r r1 rx r) r8 r+ r4 )r ri r# rN 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 rD � s z6TaskTests.test_wait_duplicate_coroutines.<locals>.coro�test�spamr# c s s | ] } | j � Vq d S)N)r5 )�.0rA r r r � <genexpr>� s z;TaskTests.test_wait_duplicate_coroutines.<locals>.<genexpr>) r r) r1 r� r# r+ r� r4 r� )r rD �crv r3 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 ) r? r@ r# r+ r r� r� rx r, )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 ri s
z0TaskTests.test_wait_first_completed.<locals>.geng $@r# g�������?r� r� )r" r r1 rx r� �FIRST_COMPLETEDr+ r4 r� r3 r2 � assertIsNoner5 r� r� � advance_time)r ri r# r� r� rv r3 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 �coro1, s z.TaskTests.test_wait_really_done.<locals>.coro1c s s d Vd Vd S)Nr r r r r �coro20 s z.TaskTests.test_wait_really_done.<locals>.coro2r# r� ) r r) r1 r# r� r� r+ r4 r2 r3 r� r5 )r r� r� r� r� rv r3 r� r r r �test_wait_really_done( 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 ri E s z0TaskTests.test_wait_first_exception.<locals>.geng $@r# c S s t d � � d S)N�err)�ZeroDivisionErrorr r r r �excO s z0TaskTests.test_wait_first_exception.<locals>.excr� r r� ) r" r r1 rx r) r� �FIRST_EXCEPTIONr+ r4 r� r� r� ) r ri r# r� r� r� rv r3 r� r )r r �test_wait_first_exceptionC 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 ri d s
z8TaskTests.test_wait_first_exception_in_wait.<locals>.geng $@r# c 3 s( t j d d � �Dd Ht d � � d S)Ng{�G�z�?r# r� )r rx r� r )r# r r r� p s z8TaskTests.test_wait_first_exception_in_wait.<locals>.excr� g{�G�z�?r� ) r" r r1 rx r) r� r� r+ r4 r� r� r� )r ri r� r� r� rv r3 r� r )r# r r �!test_wait_first_exception_in_waitb 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 ri � s
z/TaskTests.test_wait_with_exception.<locals>.geng�������?r# c 3 s( t j d d � �Dd Ht d � � d S)Ng333333�?r# Zreally)r rx r� r )r# r r �sleeper� s z3TaskTests.test_wait_with_exception.<locals>.sleeperc 3 s� t j � � g d � �Dd 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� rA r r r r� � s zBTaskTests.test_wait_with_exception.<locals>.foo.<locals>.<genexpr>r )r r� r4 �lenr� )r3 r� �errors)r� r� r# r r r rN � s
&z/TaskTests.test_wait_with_exception.<locals>.foog333333�?)r" r r1 rx r) r+ r� r� )r ri r� rN 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 ri � s z-TaskTests.test_wait_with_timeout.<locals>.geng�������?r# g333333�?c 3 sb t j � � g d d d � �Dd H\ } } � j | t � g � � � j | t � g � � d S)Nr� g)\��(�?r# )r r� r4 r� )r3 r� )r� r� r# r r r rN � s z-TaskTests.test_wait_with_timeout.<locals>.foog)\��(�?r� )
r" r r1 rx r) r+ r� r� r� r� )r ri rN 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 ri � s z4TaskTests.test_wait_concurrent_complete.<locals>.geng�������?r# g333333�?r� r� ) r" r r1 rx r+ r� r4 r� r� r� r� )r ri r# r� r� r3 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 ri � s z(TaskTests.test_as_completed.<locals>.geng �?Fc 3 s^ t j | d � �Dd H� j | � � rZ d � k rZ d � k rZ d � � j d � n | S)Nr# r� r� Tg�Q� ��?)r rx �addr� )�dtry )� 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 | Dd H� q% W| S)Nr# )r � as_completedr� )�valuesrA )r� r� r� r# r r rN � s %z(TaskTests.test_as_completed.<locals>.foor# r| ) r" Zslow_callback_durationr� r r) r+ r1 r� r� r2 r4 )r ri r� rN 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 ri 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]| } | rD � j d � n y" | Dd 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� rA �vr� )r� r� r# r r rN s ( *z5TaskTests.test_as_completed_with_timeout.<locals>.foor| r r g� ��Q��?r� )r r� )r" r rx r) r+ r1 r4 r� �assertIsInstancer� r� r� r� r� )r ri rN 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 ri 1 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]! } | Dd H} � j | d � q Wd S)Nr� r r# r� )r r� r4 )rA r� )r� r# r r r rN : s % z<TaskTests.test_as_completed_with_unused_timeout.<locals>.foo)r" r rx r) r+ r1 )r ri rN r )r� r# r r �%test_as_completed_with_unused_timeout/ 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 ri D s z5TaskTests.test_as_completed_reverse_wait.<locals>.geng�������?r� r# g�������?r� r| r r ) r" r rx �listr� r4 r� r+ r� r� r� ) r ri r# r� r� �fs�futsry rz r r r � test_as_completed_reverse_waitB 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 ri [ s
z3TaskTests.test_as_completed_concurrent.<locals>.geng�������?r� r# r� r| c s s | ] } | j � Vq d S)N)r5 )r� rA r r r r� k s z9TaskTests.test_as_completed_concurrent.<locals>.<genexpr>)
r" r rx r� r� r4 r� r� r+ r� )
r ri r# r� r� r� r� �waiterr3 r� r )r r � test_as_completed_concurrentY 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 rD o s z>TaskTests.test_as_completed_duplicate_coroutines.<locals>.coroc 3 sZ g } � d � } xA t j | | � d � g d � j �D] } | j | Dd H� q: W| S)N�hamr� r# )r r� r# r� )r5 r� rA )rD r r r r[ s s z@TaskTests.test_as_completed_duplicate_coroutines.<locals>.runnerr# r� r� r| ) r r) r1 r# r+ r5 r4 r� r� )r r[ r% r5 r )rD r r �&test_as_completed_duplicate_coroutinesm 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 ri � s
z!TaskTests.test_sleep.<locals>.genc 3 sA t j | d d � �Dd Ht j | d | d � �Dd H} | S)Nr| r# )r rx )r� �argr� )r# r r r� � s !z%TaskTests.test_sleep.<locals>.sleeperg�������?�yeahr# ) r" r r) r1 r+ r2 r3 r4 r5 r� r� )r ri r� r9 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 ri � s z(TaskTests.test_sleep_cancel.<locals>.geng $@r� r# c s � | | | � � � S)Nr )Zdelay�callbackr )�handle�orig_call_laterr r r� � s z/TaskTests.test_sleep_cancel.<locals>.call_later) r" r r1 rx r� r rs r� Z
_cancelledre r2 )r ri r# r9 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 ri � s
z5TaskTests.test_task_cancel_sleeping_task.<locals>.genc 3 s t j | d � �Dd Hd S)Nr# )r rx )r� )r# r r rx � s z7TaskTests.test_task_cancel_sleeping_task.<locals>.sleepc 3 s_ t j � d � d � �} � j d | j � y
| Dd HWn t j k
rV d SYn Xd Sd S)Ni� r# g�������?r� zslept in)r r1 r� re rf )r� )r# rx r r �doit� s
z6TaskTests.test_task_cancel_sleeping_task.<locals>.doitr� g�������?)r" r r) r4 r+ r� r� )r ri r� Zdoerr )r# r rx 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
� Dd Hd S)Nr r )r% r r rD � s z6TaskTests.test_task_cancel_waiter_future.<locals>.coro)r r( r# r) r1 r rs r6 r� re r? rf r+ r� r2 r� )r rD rv 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)Nr= r r r r r r0 � s z6TaskTests.test_step_in_completed_task.<locals>.notmuchr# r/ ) r r) r1 r# r> r? �AssertionError�_stepr, )r r0 ri rv 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 r= r r r r r r0 � s z+TaskTests.test_step_result.<locals>.notmuch)r r) r? r* r# r+ )r r0 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_added�superrE )r r �kwds)� __class__r r rE s z7TaskTests.test_step_result_future.<locals>.Fut.__init__c s d | _ t � j | � d S)NT)r� r� rb )r �fn)r� r r rb s z@TaskTests.test_step_result_future.<locals>.Fut.add_done_callback)r r r rE rb r r )r� r �Fut� s r� r# c 3 s � Dd H� d S)Nr r )r% r5 r r �wait_for_future s z:TaskTests.test_step_result_future.<locals>.wait_for_future)r r( r# r) r1 r rs r2 r� �objectr> r6 r3 r� r5 )r r� r� r9 r� r )r% r5 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) r1 r# r? r� r� r2 r3 r� r� )r r� rv r r r � test_step_with_baseexception 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 ri ( s z7TaskTests.test_baseexception_during_cancel.<locals>.genc 3 s t j d d � �Dd Hd S)Nr� r# )r rx r )r# r r r� / s z;TaskTests.test_baseexception_during_cancel.<locals>.sleeperc 3 s2 y � � Dd HWn t j k
r- � � Yn Xd S)N)r rf r )�base_excr� r r r� 5 s z<TaskTests.test_baseexception_during_cancel.<locals>.notmutchr# )r" r r) r� r1 r rs re r� r3 r? r2 r� r6 r� )r ri r� rv r )r� r# r r� r � test_baseexception_during_cancel& s
z*TaskTests.test_baseexception_during_cancelc C ss d d � } | j t j | � � d d � } | j t j | � � t j d d � � } | j t j | � � d S)Nc S s d S)Nr r r r r r� I s z.TaskTests.test_iscoroutinefunction.<locals>.fnc s s d Vd S)Nr r r r r �fn1N s z/TaskTests.test_iscoroutinefunction.<locals>.fn1c s s d Vd S)Nr r r r r �fn2R s z/TaskTests.test_iscoroutinefunction.<locals>.fn2)r� r �iscoroutinefunctionr) r2 )r r� r r r r r �test_iscoroutinefunctionH s z"TaskTests.test_iscoroutinefunctionc sv t j d | j � � t j � f d d � � } | � } | j t � � | j j | � Wd QX| 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) r? r* r+ r� r3 )r r� rv 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 rD e s z:TaskTests.test_yield_vs_yield_from_generator.<locals>.coroc
3 s$ � � } z | VWd | j � Xd S)N)r, )ri )rD r r r� i s zETaskTests.test_yield_vs_yield_from_generator.<locals>.wait_for_future)r r) r? r* r# r+ )r r� rv r )rD r �"test_yield_vs_yield_from_generatord 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)Nr� r r r r r r{ w s z7TaskTests.test_coroutine_non_gen_function.<locals>.funcr� )r r) r2 r � iscoroutiner# r+ r4 )r r{ rD r� r r r � test_coroutine_non_gen_functionv 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)Nr� )r> r )r% r r rD � s zETaskTests.test_coroutine_non_gen_function_return_future.<locals>.coror� ) r r( r# r) r1 r+ r4 r� r5 )r r{ rD �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# )r2 r r1 � current_task)r# )r rv r r rD � s z)TaskTests.test_current_task.<locals>.coro)r� r r1 r r# r) r+ )r rD r )r rv 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 � � Dd H� j t j j d | � � k � � j d � d S)Nr# T)r2 r r1 r r> )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 � � Dd H� j t j j d | � � k � d S)Nr# T)r2 r r1 r r> )r# )r� r� r �task2r r r� � s "
zBTaskTests.test_current_task_with_interleaving_tasks.<locals>.coro2) r� r r1 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
� Dd HWn" t j k
r1 � d 7� � Yn X� j d � d S)Nr zgot past sleep() in inner())r rf �failr )�proofr r� r r �inner� s
z8TaskTests.test_yield_future_passes_cancel.<locals>.innerc 3 s@ y � � Dd HWn t j k
r1 � d 7� Yn X� d 7� d S)Nr� r� )r rf r )r r r r r� � s
z8TaskTests.test_yield_future_passes_cancel.<locals>.outer�e ) r r( r# r) r; r rs re r+ r4 r2 r� )r r� rA 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 � Dd 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 �Dd 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) r; r rs re r? rf r+ r> r4 )r r� rA 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# �shieldr> r+ r4 )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 rs r* �
set_exceptionr6 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 rs re r2 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# r> r+ r r4 )r r% r� r r r �test_shield_shortcut 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 QX� j d � t j � j � � j � d � d S)Nr r# c 3 s � Dd 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 �Dd H� d 7� d S)Nr# r� )r r r# r )r r r r r r� " s z+TaskTests.test_shield_effect.<locals>.outerr )
r r( r# r) r; r rs re r? rf r+ r> r4 )r r� rA r )r r r r� r �test_shield_effect 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 rs re r2 r� r> r4 r5 )r �child1�child2�parentr� r r r �test_shield_gather1 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 rs re r� r� rf r2 r� r> )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# r? rI r+ r� r
r, )r r% rD 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# r? rI r+ r� r
r, r@ )r r% rD r r r �test_wait_invalid_args[ s
z TaskTests.test_wait_invalid_argsc sJ � f d d � } t d � � | � Wd QXt d � � | � Wd QXd 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 � � Dd Hd S)N)r2 �
gi_runningr )r% ri r r r rD p s zGTaskTests.test_corowrapper_mocks_generator.<locals>.check.<locals>.coror# )r r) r( r# r> r2 r r� �gi_frame�types� FrameTyper� r( rm �CodeTyper+ r� )rD )r )r% ri r �checkl s !
z9TaskTests.test_corowrapper_mocks_generator.<locals>.checkFT)r )r r- r )r r � test_corowrapper_mocks_generatorj 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 QXd S)
NTc 3 s � � Dd 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 �| Dd HS)Nr# )r r( r# r1 )rA )r �t3r r r
� s z1TaskTests.test_yield_from_corowrapper.<locals>.t2c S s | j d � d S)Nr r| r )r r| r )r> )rA r r r r/ � s z1TaskTests.test_yield_from_corowrapper.<locals>.t3r# r r| r )r r| r )r r r) r1 r# r+ r4 )r r rv �valr )r r
r/ 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 rN � 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� �cwZex)rN r r �call� s
z8TaskTests.test_yield_from_corowrapper_send.<locals>.callr r| r� )r r| )r r| )r4 )r r6 r )rN 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 Dd Hd S)Nr r r r r rN � s z/TaskTests.test_corowrapper_weakref.<locals>.foor5 )�weakref�WeakValueDictionaryr r r2 ri )r ZwdrN r5 r r r �test_corowrapper_weakref� s
z"TaskTests.test_corowrapper_weakrefzneed 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 i d d 6t j d 6| d
6� | j � d S) Nc s s+ t j d | � } | Dd Ht d � � d S)Nr# zcode never reached)r r( � Exception)r# �futurer r r �kill_me� s z:TaskTests.test_log_destroyed_pending_task.<locals>.kill_meTr# r r< z%Task was destroyed but it is pending!�messagerv r )r r) r ZMockr# r_ �set_exception_handlerr; r4 r1 � all_tasksZ _run_oncer� �_readyr) �f_localsrc r �
gc_collectr� Zassert_called_with�ANYZ
reset_mock)r r= Z mock_handlerrD rv r r r r � test_log_destroyed_pending_task� s( %
%
z)TaskTests.test_log_destroyed_pending_taskzasyncio.coroutines.loggerc C s� t d � � t j d d � � } Wd QXt } 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 rC r2 �error�called� call_argsr ra rt ru r rX �compile�DOTALL) r Zm_logrF Z tb_filename� tb_linenor> Z
func_filenameZ func_linenoZregexr r r � test_coroutine_never_yielded� s
z&TaskTests.test_coroutine_never_yieldedc 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# r_ r r1 r
rH rI rJ r� rc r� r4 rG r+ )r rv rh r r r rR 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 � � } | Dd Hd S)Nr# )r r( )r% )r# r r �blocking_coroutine s z;TaskTests._test_cancel_wait_for.<locals>.blocking_coroutiner# )r r' r� r, r) r rr r� re r? rf r+ r2 r� )r r� rS rv r� r )r# r �_test_cancel_wait_for s !
z TaskTests._test_cancel_wait_forc C s | j d � d S)N)rT )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@)rT )r r r r �test_cancel_wait_for) s z TaskTests.test_cancel_wait_forN)Zr r r r$ r. r: r< rB rC �unittestZ
skipUnlessr` rH rJ rL r\ rk rq rw 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. r1 r7 r: �PY34rE r ZpatchrQ rR rT rU rV r r r r r! K s� ; D ) !
" - %
"
! ' ) r! c @ sp 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 d � Z
d S)�GatherTestsBasec C s8 | j � | _ | j � | _ | j | j d d �d S)N�cleanupF)r" �one_loopr- r8 )r r r r r$ / s zGatherTestsBase.setUpc C s! x | j r t j | � q Wd S)N)rA r rs )r r# r r r � _run_loop4 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( r[ )r� �i)r r r �
<listcomp>9 s z2GatherTestsBase._check_success.<locals>.<listcomp>r r r| F)�ranger r � wrap_futuresr � MockCallbackrb r> r\ r[ r4 rL r� r3 �assert_called_once_withr5 )r �kwargsr� r� r� r% �cbr )r r �_check_success8 s (!
z GatherTestsBase._check_successc C s | j � | j d d � d S)N�return_exceptionsF)re )r r r r � test_successG s
z GatherTestsBase.test_successc C s | j d d � d S)Nrf T)re )r r r r � test_result_exception_successK 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( r[ )r� r] )r r r r^ O s z6GatherTestsBase.test_one_exception.<locals>.<listcomp>r
r r )r_ r r r` r ra rb r� r> r r\ r[ r2 r3 rb r6 r� re r* ) r r� r� r� r �er% rd r� r )r r �test_one_exceptionN s .$
z"GatherTestsBase.test_one_exceptionc s* � f d d � t d � D� \ } } } } t j d d � j | | | | � � } 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( r[ )r� r] )r r r r^ a s z:GatherTestsBase.test_return_exceptions.<locals>.<listcomp>r rf Tr r )r_ r r r` r ra rb r� r* r> r r\ r[ r� r3 r2 rb r4 r5 ) r r� r� r� r r% rd 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)rm rn ) �os�path�dirnamer rG �joinr r4 �rstrip)r Zaio_pathrp �sts�stdout�stderrr r r �test_env_var_debugs s& ! z"GatherTestsBase.test_env_var_debugN) r r r r$ r\ re rg rh rj rk rz r r r r rY - s rY 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 r` � 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 r8 r[ r� r r� r( r6 r7 r\ r2 r3 r4 r5 r- )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)Nrp )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 QX| j t � � t j | d | j �Wd QXd S)Nr# )r r( r[ r- r? r@ r )r r� r� r r r �%test_constructor_heterogenous_futures� 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( r- )r� r] )r r r r^ � s zIFutureGatherTests.test_constructor_homogenous_futures.<locals>.<listcomp>r r# ) r_ r r r6 r7 r- r\ r� r3 )r Zchildrenr% r )r r �#test_constructor_homogenous_futures� 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( r[ )r� r] )r r r r^ � s z;FutureGatherTests.test_one_cancellation.<locals>.<listcomp>r
r r )r_ r r r ra rb r> re r\ r[ r2 r3 rb r� r� r� r� rf r r* )r r� r� r� r ri r% rd r )r r �test_one_cancellation� 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( r[ )r� r] )r r r r^ � s zLFutureGatherTests.test_result_exception_one_cancellation.<locals>.<listcomp>� rf Tr r r| r )r_ r r r ra rb r> r� r re r\ r[ r� r3 r* r+ r� rf r4 rb ) r r� r� r� r ri rA r% rd 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{ � 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 r8 r[ )r )r� r r r$ � s
zCoroutineGatherTests.setUpc G sC g } x6 | D]. } t j | d d � � } | j | � � q
W| S)Nc s s | Dd HS)Nr )r% r r r rD � s z/CoroutineGatherTests.wrap_futures.<locals>.coro)r r) r� )r r| Zcorosr% rD r r r r` � 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)Nr] r r r r r rD � s zBCoroutineGatherTests.test_constructor_loop_selection.<locals>.cororZ Fr# ) r r) r r6 r7 r[ r+ r8 r- )r rD 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 rD s z<CoroutineGatherTests.test_duplicate_coroutines.<locals>.coror] �defr# )r r) r r[ r\ r4 r5 )r rD r� r% r r r �test_duplicate_coroutines s
'z.CoroutineGatherTests.test_duplicate_coroutinesc ss 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 QX� j � j � � � j � j
� � � j � j
� � � j � j
� � t j � j � � j � d � d S)Nr r# c 3 s � Dd 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 �� � Dd H� d 7� d S)Nr# r� )r r r[ r )r! r" �gathererr r r r r� s z?CoroutineGatherTests.test_cancellation_broadcast.<locals>.outer)r r( r[ r) r; r rs r2 re r? rf r+ r� r� r4 )r r r� rA 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 | Dd Ht d � � d S)Nzshould not be ignored)r* )rA r r r r 2 s z:CoroutineGatherTests.test_exception_marking.<locals>.innerr# c 3 s. t j � � � � � � d � j �Dd Hd S)Nr# )r r r[ r )r� r� r r r r r� : s z:CoroutineGatherTests.test_exception_marking.<locals>.outer) r r) r( r[ r; r rs r> r� r� r* )r r� rA 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� � s "r� c @ s� e Z d Z d Z 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
d S)�RunCoroutineThreadsafeTestsz/Test case for asyncio.run_coroutine_threadsafe.c C s# t j � | _ | j | j � d S)N)r r' r# r8 )r r r r r$ J s z!RunCoroutineThreadsafeTests.setUpFc c sb t j d d | j �Dd H| r0 t d � � n | rZ t j j j | j � j � d Vn | | S)z"Wait 0.05 second and return a + b.g�������?r# NzFail!)r rx r# r* Ztasksr1 r re )r r� r� r re r r r r� N s z RunCoroutineThreadsafeTests.addNc
C s� | j d d d | d | �} t j | | j � } | rR | j j | j d � n z | j | � SWd | j � p{ | j � Xd S)z$Run add coroutine in the event loop.r r| r re N) r� r Zrun_coroutine_threadsafer# Zcall_soon_threadsafer3 r5 r3 re )r r re r� � advance_cororD r< r r r �targetY s z"RunCoroutineThreadsafeTests.targetc C s> | j j d | j � } | j j | � } | j | d � d S)z9Test coroutine submission from a thread to an event loop.Nr )r# �run_in_executorr� r+ r4 )r r< r5 r r r � test_run_coroutine_threadsafel s z9RunCoroutineThreadsafeTests.test_run_coroutine_threadsafec C s] | j j d | j d � } | j t � � } | j j | � Wd QX| j d | j j � d S)z]Test coroutine submission from a thread to an event loop
when an exception is raised.NTzFail!) r# r� r� r? r* r+ �assertInr� r )r r< � exc_contextr r r �,test_run_coroutine_threadsafe_with_exceptionr s zHRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_exceptionc s� � f d d � } � j j d | � } � j t j � � � j j | � Wd QXt j � j � x0 t j j � j � D] } � j
| j � � qv Wd S)zZTest coroutine submission from a thread to an event loop
when a timeout is raised.c s � j d d � S)Nr� r )r� r )r r r �<lambda>} s zXRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeout.<locals>.<lambda>N) r# r� r? r r� r+ r rs r1 r@ r2 r3 )r r� r< rv r )r r �*test_run_coroutine_threadsafe_with_timeoutz s zFRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_with_timeoutc sT � f d d � } � j j d | � } � j t j � � � j j | � Wd QXd S)z[Test coroutine submission from a tread to an event loop
when the task is cancelled.c s � j d d � S)Nre T)r� r )r r r r� � s zZRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_cancelled.<locals>.<lambda>N)r# r� r? r rf r+ )r r� r< 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 QX� j d | j
j d � � j t
| j � d � | j \ \ } } } � j | d | j
� d S) ziTest coroutine submission from a tread to an event loop
when the task factory raise an exception.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# rD r r r r� � s r� r r r� )r# r� Zset_task_factoryr ra r? r? � NameErrorr+ r� r� r r4 r� Zcall_args_listrM )r r< r� r� r# �contextrc r )r r �4test_run_coroutine_threadsafe_task_factory_exception� s zPRunCoroutineThreadsafeTests.test_run_coroutine_threadsafe_task_factory_exception)r r r �__doc__r$ r r) r� r� r� r� r� r� r� r r r r r� G s
r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�
SleepTestsc C s t j � | _ t j d � d S)N)r r' r# r8 )r r r r r$ � s zSleepTests.setUpc C s | j j � d | _ d S)N)r# r, )r r r r �tearDown� 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)r5 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 �Dd H} � j � d � � | � d S)Nr r r# r5 r� )r# r� r4 r rx )r� )r� r5 r r r rD � s
#z(SleepTests.test_sleep_zero.<locals>.coro� )r r) r# r+ r4 )r rD r )r� r5 r r �test_sleep_zero� s
!zSleepTests.test_sleep_zeroN)r r r r$ r� r� r r r r r� � s r� c @ s� 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 d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d S) � TimeoutTestsc C s t j � | _ t j d � d S)N)r r' r# r8 )r r r r r$ � s zTimeoutTests.setUpc C s | j j � d | _ d S)N)r# r, )r r r r r� � s
zTimeoutTests.tearDownc sp d g � t j � � f d d � � � t j � � f d d � � } � j j | � � � j � d d � d S)NFc 3 sH y t j d d � j �Dd HWn" t j k
rC d � d <� Yn Xd S)Nr� r# Tr )r rx r# rf r )�canceled_raisedr r r �long_running_task� s
z4TimeoutTests.test_timeout.<locals>.long_running_taskc 3 s` � j t j � �H t j d d � j ��) } � � Dd H� j | j � j � Wd QXWd QXd S)Ng{�G�z�?r# )r? r r� r� r# r6 r7 )r9 )r� r r r �go� s z%TimeoutTests.test_timeout.<locals>.gor z CancelledError was not raised)r r) r# r+ r2 )r r� r )r� r� r r � test_timeout� s
zTimeoutTests.test_timeoutc sP t j � f d d � � � t j � � f d d � � } � j j | � � d S)Nc 3 s t j d d � j �Dd Hd S)Ng{�G�z�?r# r3 )r rx r# r )r r r r� � s zCTimeoutTests.test_timeout_finish_in_time.<locals>.long_running_taskc 3 sA t j d d � j �� � � Dd H} Wd QX� j | d � d S)Ng�������?r# r3 )r r� r# r4 )�resp)r� r r r r� � s z4TimeoutTests.test_timeout_finish_in_time.<locals>.go)r r) r# r+ )r r� r )r� r r �test_timeout_finish_in_time� s z(TimeoutTests.test_timeout_finish_in_timec sB t j � j � t j � f d d � � } � j j | � � d S)Nc
3 sD t j d � �/ } t j d � Dd H� j | j � j � Wd QXd S)Ng�������?g{�G�z�?)r r� rx r6 r7 r# )r9 )r r r r& � s z3TimeoutTests.test_timeout_gloabal_loop.<locals>.run)r r8 r# r) r+ )r r& r )r r �test_timeout_gloabal_loop� s z&TimeoutTests.test_timeout_gloabal_loopc s2 t j � f d d � � } � j j | � � d S)Nc 3 sZ t j d d � j �Dd H� j t � �* t j d d � j �� t � Wd QXWd QXd S)Nr r# g�������?)r rx r# r? �KeyErrorr� r )r r r r� � s z<TimeoutTests.test_timeout_not_relevant_exception.<locals>.go)r r) r# r+ )r r� r )r r �#test_timeout_not_relevant_exception� s z0TimeoutTests.test_timeout_not_relevant_exceptionc s2 t j � f d d � � } � j j | � � d S)Nc 3 s` t j d d � j �Dd H� j t j � �- t j d d � j �� t j � Wd QXWd QXd S)Nr r# g����MbP?)r rx r# r? rf r� r )r r r r� s zLTimeoutTests.test_timeout_canceled_error_is_converted_to_timeout.<locals>.go)r r) r# r+ )r r� r )r r �3test_timeout_canceled_error_is_converted_to_timeout s z@TimeoutTests.test_timeout_canceled_error_is_converted_to_timeoutc sJ t j d d � � � t j � � f d d � � } � j j | � � d S)Nc S s t j d � d S)Ng�������?r3 )r� rx r r r r r� s
zBTimeoutTests.test_timeout_blocking_loop.<locals>.long_running_taskc 3 sA t j d d � j �� � � Dd H} Wd QX� j | d � d S)Ng{�G�z�?r# r3 )r r� r# r4 )r5 )r� r r r r� s z3TimeoutTests.test_timeout_blocking_loop.<locals>.go)r r) r# r+ )r r� r )r� r r �test_timeout_blocking_loop
s z'TimeoutTests.test_timeout_blocking_loopc sf t j d � j � � � j j d � j d � � t j � � f d d � � } � j j | � � d S)Nr# g�������?r3 c 3 s> t j d d � j �� � Dd H} Wd QX� j | d � d S)Ng�������?r# r3 )r r� r# r4 )r� )r% r r r r� s z1TimeoutTests.test_for_race_conditions.<locals>.go)r r( r# r� r> r) r+ )r r� r )r% r r �test_for_race_conditions s z%TimeoutTests.test_for_race_conditionsc s2 t j � f d d � � } � j j | � � d S)Nc 3 s� d } � j j � } � j t j � �S t j d d � j ��4 d } z t j d d � j �Dd HWd d } XWd QXWd QX� j j � | } � j d | k o� d k n | � � j | � d S)Ng�������?r# Tg�������?Fg
ףp=
�?g� ��Q��?) r# r� r? r r� r� rx r2 r� )r� �startr� )r r r r� % s &z*TimeoutTests.test_timeout_time.<locals>.go)r r) r# r+ )r r� r )r r �test_timeout_time$ s z TimeoutTests.test_timeout_timec C s9 | j t � �$ t j d d | j �� Wd QXWd QXd S)Ng�������?r# )r? r* r r� r# )r r r r �"test_raise_runtimeerror_if_no_task9 s z/TimeoutTests.test_raise_runtimeerror_if_no_taskc s_ d g � t j � � f d d � � � t j � � � f d d � � } � j j | � � d S)NFc 3 sd y> t j d d � j �� t j d d � j �Dd HWd QXWn t j k
r_ d � d <Yn Xd S)Ng����MbP?r# r Tr )r r� r# rx r� r )� has_timeoutr r r r� B s
%z<TimeoutTests.test_outer_coro_is_not_cancelled.<locals>.outerc 3 s_ t j � � d � j �} | Dd H� j � d � � j | j � � � j | j � � d S)Nr# r )r r; r# r2 r� r� r3 )rv )r� r� r r r r� J s
z9TimeoutTests.test_outer_coro_is_not_cancelled.<locals>.go)r r) r# r+ )r r� r )r� r� r r � test_outer_coro_is_not_cancelled> s !z-TimeoutTests.test_outer_coro_is_not_cancelledc sk t j d � j � � t j � � f d d � � � t j � � � f d d � � } � j j | � � d S)Nr# c 3 s, � j d � t j d d � j �Dd Hd S)Nr r# )r> r rx r# r )r% r r r r� W s
z2TimeoutTests.test_cancel_outer_coro.<locals>.outerc 3 sz t j � � d � j �} � Dd H| j � � j t j � � | Dd HWd QX� j | j � � � j | j � � d S)Nr# ) r r; r# re r? rf r2 r� r3 )rv )r% r� r r r r� \ s
z/TimeoutTests.test_cancel_outer_coro.<locals>.go)r r( r# r) r+ )r r� r )r% r� r r �test_cancel_outer_coroT s !
z#TimeoutTests.test_cancel_outer_coroN)r r r r$ r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s
r� �__main__)*r� �
contextlibr} rS rr rt rH r� r* rW r8 r r r r r� r � ImportErrorr Ztest.support.script_helperr Ztest.script_helperZasyncio.test_support� version_inforX r` r) r
�contextmanagerr r r ZTestCaser! rY r{ r� r� r� r� r �mainr r r r �<module> s\
� � � � � � �c T c\ �