Your IP : 216.73.216.213
3
� f9 � @ s� d Z ddlZddlZddlZyddlmZ W n ek
rL ddlm Z Y nX ddlm
Z
ddlZddlm Z G dd� de j �Z
G d d
� d
e
�ZG d d � d e
�ZG d
d� de
�Zedkr�ej� dS )z3Tests support for new syntax introduced by PEP 492.� N)�support)� test_support)�mock)�
test_utilsc s e Zd Z� fdd�Z� ZS )�BaseTestc sH t � j� tj� | _tj� | j_tj� | j_f | jjj _
| j | j� d S )N) �super�setUp�asyncioZ
BaseEventLoop�loopr ZMockZ_process_eventsZ _selectorZselectZ return_valueZset_event_loop)�self)� __class__� �B/opt/alt/python36/lib64/python3.6/test/test_asyncio/test_pep492.pyr s
zBaseTest.setUp)�__name__�
__module__� __qualname__r �
__classcell__r
r
)r r r s r c @ s e Zd Zdd� Zdd� ZdS )� LockTestsc sp t j� jd�t j� jd�t j� jd�t j� jd�g}� fdd�}x*|D ]"}� jj||�� � j|j� � qFW d S )N)r
c � s� t jd� jd�I d H � j| j� � | 4 I d H �D}� j|d � � j| j� � t jd� jd�I d H � j| j� � W d Q I d H R X � 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 sp t j� jd�t j� jd�t j� jd�t j� jd�g}� fdd�}x*|D ]"}� jj||�� � j|j� � qFW d S )N)r
c � s� t jd� jd�I d H � j| j� � | I d H �D}� j|d � � j| j� � t jd� jd�I d H � j| j� � W d Q R X � 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 Zd Zdd� ZdS )�StreamReaderTestsc sT d}t j| jd�� � j|� � j� � fdd�}| jj|� �}| j|dddg� d S )Ns line1
line2
line3)r
c � sB g } x8� 2 d H y 3 d H }W n
t k
s2X | j|� q Y W | 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 @ sV e Zd Zdd� Zdd� Zdd� Zdd� Zejd �d
d � �Z d d
� Z
dd� Z dd� Z dS )�CoroutineTestsc C sR dd� }|� }z| j tj|�� W d |j� X G dd� d�}| j tj|� �� d S )Nc � s d S )Nr
r
r
r
r �fooa s z,CoroutineTests.test_iscoroutine.<locals>.fooc @ s. e Zd Zdd� Zd
dd�Zdd� Zdd � ZdS ) 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 V d S )Nr
)r r
r
r � __await__n s z;CoroutineTests.test_iscoroutine.<locals>.FakeCoro.__await__)NN)r r r r2 r6 r7 r8 r
r
r
r �FakeCoroj s
r9 )r r Z iscoroutiner7 )r r0 �fr9 r
r
r �test_iscoroutine` s
z CoroutineTests.test_iscoroutinec C s dd� }| j tj|�� d S )Nc � s d S )Nr
r
r
r
r r0 s s z4CoroutineTests.test_iscoroutinefunction.<locals>.foo)r r Ziscoroutinefunction)r r0 r
r
r �test_iscoroutinefunctionr s z'CoroutineTests.test_iscoroutinefunctionc sD G dd� d�� t j� fdd��}|� }| j|jd �d� |j� d S )Nc @ s e Zd Zdd� ZdS )zCCoroutineTests.test_function_returning_awaitable.<locals>.Awaitablec S s dS )N�spam)r= r
)r r
r
r r8 x s zMCoroutineTests.test_function_returning_awaitable.<locals>.Awaitable.__await__N)r r r r8 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- r2 r7 )r r? �coror
)r> r �!test_function_returning_awaitablev s
z0CoroutineTests.test_function_returning_awaitablec sX dd� � � fdd�}| j j|� �}| j|d� | j jd� | j j|� �}| j|d� d S )Nc � s dS )Nr= r
r
r
r
r �bar� s z5CoroutineTests.test_async_def_coroutines.<locals>.barc � s � � I d H S )Nr
r
)rC r
r r0 � s z5CoroutineTests.test_async_def_coroutines.<locals>.foor= T)r
r r- � set_debug)r r0 r( r
)rC r �test_async_def_coroutines� s z(CoroutineTests.test_async_def_coroutineszasyncio.coroutines.loggerc sR dd� � � ��fdd�}�j jd� �j j|� � � �fdd�}�j j|� � d S )Nc � s d S )Nr
r
r
r
r r0 � s z2CoroutineTests.test_async_def_wrapped.<locals>.fooc
� sj � � } �j t| �d� tjdtf��< d } tj� �j�jj� �jj d d }�j |d� W d Q R X d 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)r0 �m_logr r
r �start� s
z4CoroutineTests.test_async_def_wrapped.<locals>.startTc � s* � � } t j| �jd�}�jt|�d� d S )N)r
zTask.*foo.*running)r �
ensure_futurer
rF rG )rK Ztask)r0 r r
r rN � s )r
rD r )r rM rN r
)r0 rM r r �test_async_def_wrapped� s z%CoroutineTests.test_async_def_wrappedc sF dd� �t j�fdd��� � �fdd�}�jj|� �}�j|d� d S )Nc s s f E d H dS )Nr= r
r
r
r
r �gen� s
z0CoroutineTests.test_types_coroutine.<locals>.genc s � � S )Nr
r
)rQ r
r r? � s z1CoroutineTests.test_types_coroutine.<locals>.funcc � s � � } �j | tj� | I d H S )N)ZassertIsInstance�types�_GeneratorWrapper)�wrapper)r? r r
r rA � s z1CoroutineTests.test_types_coroutine.<locals>.coror= )rR r@ r
r r- )r rA r( r
)r? rQ r r �test_types_coroutine� s
z#CoroutineTests.test_types_coroutinec s4 d � � �fdd��� ��fdd�}�j j|� � d S )Nc
� s0 � j dd�} z�j| d jjd� W d d } X d S )N� )�limitr r0 )Z get_stackr- �f_code�co_name)r: )�Tr r
r r0 � s z1CoroutineTests.test_task_print_stack.<locals>.fooc � s t j�� �jd�� � I d H d S )N)r
)r rO r
r
)rZ r0 r r
r �runner� s z4CoroutineTests.test_task_print_stack.<locals>.runner)r
r )r r[ r
)rZ r0 r r �test_task_print_stack� s z$CoroutineTests.test_task_print_stackc sP �fdd�� � �fdd�}�j jd� �jtd�� �j j|� � W d Q R X d S )Nc � s t jd� jd�I d H d S )Ng�������?)r
)r r r
r
)r r
r �afunc� s z/CoroutineTests.test_double_await.<locals>.afuncc
� sJ � � } t j| �jd�}z$t jd�jd�I d H | I d H W d |j� X d S )N)r
r )r ZTaskr
r Zcancel)rA �t)r] r r
r r[ � s z0CoroutineTests.test_double_await.<locals>.runnerTz<Cannot await.*test_double_await.*\bafunc\b.*while.*\bsleep\b)r
rD ZassertRaisesRegex� RuntimeErrorr )r r[ r
)r] r r �test_double_await� s z CoroutineTests.test_double_awaitN)
r r r r; r< rB rE r ZpatchrP rU r\ r` r
r
r
r r/ ^ s
r/ �__main__)�__doc__�collections.abc� collectionsrR Zunittestr r � ImportErrorr r r r ZTestCaser r r% r/ r �mainr
r
r
r �<module> s" /