Your IP : 216.73.216.213
���]9 � @ s� d d l Z d d l Z d d l Z y d d l m Z Wn" e k
r\ 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)� 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_events� _selector�selectZ return_valueZset_event_loop)�self)� __class__� �0/opt/alt/python35/lib64/python3.5/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 r3 r7 r8 r9 r r r r �FakeCoroj s r: )r r Z iscoroutiner8 )r
r1 �fr: 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 r1 s s z4CoroutineTests.test_iscoroutinefunction.<locals>.foo)r r Ziscoroutinefunction)r
r1 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 r9 x s zMCoroutineTests.test_function_returning_awaitable.<locals>.Awaitable.__await__N)r r r r9 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. r3 r8 )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 )rD r r r1 � s z5CoroutineTests.test_async_def_coroutines.<locals>.foor> T)r
r! r. � set_debug)r
r1 r) r )rD 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 r1 � 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)r1 �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
rG rH )rL Ztask)r1 r
r r rO � s )r
rE r! )r
rN rO r )r1 rN 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 )rR 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 rB � s z1CoroutineTests.test_types_coroutine.<locals>.coror> )rS rA r
r! r. )r
rB r) r )r@ rR 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 r1 )Z get_stackr. �f_code�co_name)r; )�Tr
r r r1 � s z1CoroutineTests.test_task_print_stack.<locals>.fooc � s( t j � � d � j �� � Id Hd S)Nr
)r rP r
r )r[ r1 r
r r �runner� s z4CoroutineTests.test_task_print_stack.<locals>.runner)r
r! )r
r\ r )r[ r1 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 �cancel)rB �t)r^ r
r r r\ � s
z0CoroutineTests.test_double_await.<locals>.runnerTz<Cannot await.*test_double_await.*\bafunc\b.*while.*\bsleep\b)r
rE ZassertRaisesRegex� RuntimeErrorr! )r
r\ r )r^ r
r �test_double_await� s
z CoroutineTests.test_double_awaitN)
r r r r< r= rC rF r ZpatchrQ rV r] rb r r r r r0 ^ s
r0 �__main__)�collections.abc� collectionsrS Zunittestr r � ImportErrorr r r r ZTestCaser r r& r0 r �mainr r r r �<module> s
/�