Your IP : 216.73.216.213
��Yf9 � @ s d Z d d l Z d d l Z d d l Z y d d l m Z Wn" e k
rb d d l m Z Yn Xd d l m
Z
d d l Z d d l m Z Gd d � d e j � Z
Gd d
� d
e
� Z Gd d � d e
� Z Gd
d � d e
� Z e d k r e j � d S)z3Tests support for new syntax introduced by PEP 492.� N)�support)� test_support)�mock)�
test_utilsc s" e Z d Z � f d d � Z � S)�BaseTestc sf t � j � t j � | _ t j � | j _ t j � | j _ f | j j j _
| j | j � d S)N) �super�setUp�asyncioZ
BaseEventLoop�loopr ZMockZ_process_eventsZ _selectorZselectZ return_valueZset_event_loop)�self)� __class__� �B/opt/alt/python35/lib64/python3.5/test/test_asyncio/test_pep492.pyr s
zBaseTest.setUp)�__name__�
__module__� __qualname__r r
r
)r r r s r c @ s( e Z d Z d d � Z d d � Z d S)� LockTestsc s� t j d � j � t j d � j � t j d � j � t j d � j � g } � f d d � } x7 | D]/ } � j j | | � � � j | j � � qg Wd S)Nr
c � s� t j d d � j �Id H� j | j � � | 4Id H�X } � j | d � � j | j � � t j d d � j �Id H� j | j � � Wd QId HRX� j | j � � d S)Ng{�G�z�?r
)r �sleepr
� assertFalse�locked�assertIs�
assertTrue)�lock�_lock)r r
r �test& s z7LockTests.test_context_manager_async_with.<locals>.test) r �Lockr
� Condition� Semaphore�BoundedSemaphore�run_until_completer r )r �
primitivesr � primitiver
)r r � test_context_manager_async_with s
z)LockTests.test_context_manager_async_withc s� t j d � j � t j d � j � t j d � j � t j d � j � g } � f d d � } x7 | D]/ } � j j | | � � � j | j � � qg Wd S)Nr
c � s� t j d d � j �Id H� j | j � � | Id H�X } � j | d � � j | j � � t j d d � j �Id H� j | j � � Wd QRX� j | j � � d S)Ng{�G�z�?r
)r r r
r r r r )r r )r r
r r < s z7LockTests.test_context_manager_with_await.<locals>.test) r r r
r r r r r r )r r r r! r
)r r � test_context_manager_with_await4 s
z)LockTests.test_context_manager_with_awaitN)r r r r" r# r
r
r
r r s r c @ s e Z d Z d d � Z d S)�StreamReaderTestsc sv d } t j d | j � � � j | � � j � � f d d � } | j j | � � } | j | d d d g � d S)Ns line1
line2
line3r
c � sK g } x> � 2d Hy 3d H} Wn t k
r2 YWqG X| j | � q WqG | S)N)�StopAsyncIteration�append)�data�line)�streamr
r �readerT s -z/StreamReaderTests.test_readline.<locals>.readers line1
s line2
s line3)r � StreamReaderr
Z feed_dataZfeed_eofr � assertEqual)r ZDATAr* r' r
)r) r �
test_readlineM s
z StreamReaderTests.test_readlineN)r r r r- r
r
r
r r$ K s r$ c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z e j d � d
d � � Z d d
� Z
d d � Z d d � Z d S)�CoroutineTestsc C sm d d � } | � } z | j t j | � � Wd | j � XGd d � d � } | j t j | � � � d S)Nc � s d S)Nr
r
r
r
r �fooa s z,CoroutineTests.test_iscoroutine.<locals>.fooc @ sF e Z d Z d d � Z d d d d � Z d d � Z d d � Z d S)
z1CoroutineTests.test_iscoroutine.<locals>.FakeCoroc S s d S)Nr
)r �valuer
r
r �sendk s z6CoroutineTests.test_iscoroutine.<locals>.FakeCoro.sendNc S s d S)Nr
)r �typ�val�tbr
r
r �throwl s z7CoroutineTests.test_iscoroutine.<locals>.FakeCoro.throwc S s d S)Nr
)r r
r
r �closem s z7CoroutineTests.test_iscoroutine.<locals>.FakeCoro.closec s s d Vd S)Nr
)r r
r
r � __await__n s z;CoroutineTests.test_iscoroutine.<locals>.FakeCoro.__await__)r r r r1 r5 r6 r7 r
r
r
r �FakeCoroj s r8 )r r Z iscoroutiner6 )r r/ �fr8 r
r
r �test_iscoroutine` s z CoroutineTests.test_iscoroutinec C s& d d � } | j t j | � � d S)Nc � s d S)Nr
r
r
r
r r/ s s z4CoroutineTests.test_iscoroutinefunction.<locals>.foo)r r Ziscoroutinefunction)r r/ r
r
r �test_iscoroutinefunctionr s z'CoroutineTests.test_iscoroutinefunctionc s^ Gd d � d � � t j � f d d � � } | � } | j | j d � d � | j � d S)Nc @ s e Z d Z d d � Z d S)zCCoroutineTests.test_function_returning_awaitable.<locals>.Awaitablec S s d S)N�spam)r< r
)r r
r
r r7 x s zMCoroutineTests.test_function_returning_awaitable.<locals>.Awaitable.__await__N)r r r r7 r
r
r
r � Awaitablew s r= c s � � S)Nr
r
)r= r
r �func{ s z>CoroutineTests.test_function_returning_awaitable.<locals>.funcr< )r � coroutiner, r1 r6 )r r> �coror
)r= r �!test_function_returning_awaitablev s
z0CoroutineTests.test_function_returning_awaitablec s| d d � � � f d d � } | j j | � � } | j | d � | j j d � | j j | � � } | j | d � d S)Nc � s d S)Nr< r
r
r
r
r �bar� s z5CoroutineTests.test_async_def_coroutines.<locals>.barc � s � � Id HS)Nr
r
)rB r
r r/ � s z5CoroutineTests.test_async_def_coroutines.<locals>.foor< T)r
r r, � set_debug)r r/ r' r
)rB r �test_async_def_coroutines� s z(CoroutineTests.test_async_def_coroutineszasyncio.coroutines.loggerc ss d d � � � � � f d d � } � j j d � � j j | � � � � f d d � } � j j | � � d S)Nc � s d S)Nr
r
r
r
r r/ � s z2CoroutineTests.test_async_def_wrapped.<locals>.fooc
� s� � � } � j t | � d � t j d t f � �L d } t j � � j � j j � � j j d d } � j | d � Wd QRXd S)Nz/<CoroWrapper .*\.foo\(\) running at .*pep492.*>z.*foo.*was neverr zCoroWrapper.*foo.*was never)
� assertRegex�reprr Zcheck_warnings�RuntimeWarningZ
gc_collectr �errorZcalled� call_args)�foo_coro�message)r/ �m_logr r
r �start� s
z4CoroutineTests.test_async_def_wrapped.<locals>.startTc � s; � � } t j | d � j �} � j t | � d � d S)Nr
zTask.*foo.*running)r �
ensure_futurer
rE rF )rJ Ztask)r/ r r
r rM � s )r
rC r )r rL rM r
)r/ rL r r �test_async_def_wrapped� s z%CoroutineTests.test_async_def_wrappedc se d d � � t j � f d d � � � � � f d d � } � j j | � � } � j | d � d S)Nc s s
f Ed Hd S)Nr< r
r
r
r
r �gen� s z0CoroutineTests.test_types_coroutine.<locals>.genc s � � S)Nr
r
)rP r
r r> � s z1CoroutineTests.test_types_coroutine.<locals>.funcc � s% � � } � j | t j � | Id HS)N)ZassertIsInstance�types�_GeneratorWrapper)�wrapper)r> r r
r r@ � s z1CoroutineTests.test_types_coroutine.<locals>.coror< )rQ r? r
r r, )r r@ r' r
)r> rP r r �test_types_coroutine� s
z#CoroutineTests.test_types_coroutinec sJ d � � � f d d � � � � � f d d � } � j j | � � d S)Nc
� s> � j d d � } z � j | d j j d � Wd d } Xd S)N�limit� r r/ )Z get_stackr, �f_code�co_name)r9 )�Tr r
r r/ � s z1CoroutineTests.test_task_print_stack.<locals>.fooc � s( t j � � d � j �� � Id Hd S)Nr
)r rN r
r
)rY r/ r r
r �runner� s z4CoroutineTests.test_task_print_stack.<locals>.runner)r
r )r rZ r
)rY r/ r r �test_task_print_stack� s z$CoroutineTests.test_task_print_stackc sh � f d d � � � � f d d � } � j j d � � j t d � � � j j | � � Wd QRXd S)Nc � s t j d d � j �Id Hd S)Ng�������?r
)r r r
r
)r r
r �afunc� s z/CoroutineTests.test_double_await.<locals>.afuncc
� s[ � � } t j | d � j �} z( t j d d � j �Id H| Id HWd | j � Xd S)Nr
r )r ZTaskr
r Zcancel)r@ �t)r\ r r
r rZ � s
z0CoroutineTests.test_double_await.<locals>.runnerTz<Cannot await.*test_double_await.*\bafunc\b.*while.*\bsleep\b)r
rC ZassertRaisesRegex� RuntimeErrorr )r rZ r
)r\ r r �test_double_await� s
z CoroutineTests.test_double_awaitN)
r r r r: r; rA rD r ZpatchrO rT r[ r_ r
r
r
r r. ^ s
r. �__main__)�__doc__�collections.abc� collectionsrQ Zunittestr r � ImportErrorr r r r ZTestCaser r r$ r. r �mainr
r
r
r �<module> s
/�