Your IP : 216.73.216.213
a
XC?h�! � @ sX d dl Z d dlZd dlZdd� ZdZG dd� d�ZG dd� dej�Zed krTe� � dS )
� Nc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �C/opt/alt/python39/lib64/python3.9/test/test_asyncio/test_waitfor.py�tearDownModule s r g-C ��6?c @ s e Zd ZdZdd� Zdd� ZdS )�SlowTask皙�����?c C s
d| _ d S )NF)�exited��selfr r r �__init__ s zSlowTask.__init__c � sZ t �� | j }|t �� }|dkr$qPzt�|�I d H W q tjyL Y q0 qd| _d S )Nr T)�time� monotonic� TASK_TIMEOUTr �sleep�CancelledErrorr )r ZexitatZtosleepr r r �run s z SlowTask.runN)�__name__�
__module__� __qualname__r r r r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�AsyncioWaitForTestc � s\ t � }t�t�|�� |jd ��}t�d�I d H |�� t�|h�I d H | � |j
� d S )N� r ) r r � create_task�wait_forr r r �cancel�wait�
assertTruer )r �tZ waitfortaskr r r � test_asyncio_wait_for_cancelled) s �z2AsyncioWaitForTest.test_asyncio_wait_for_cancelledc � sL t � }z t�|�� |jd �I d H W n tjy: Y n0 | �|j� d S )Nr )r r r r r � TimeoutErrorr r )r r r r r � test_asyncio_wait_for_timeout4 s z0AsyncioWaitForTest.test_asyncio_wait_for_timeoutc � sj t �� }|�� }|�d� |�� }t �|d�I d H }|�� }| �|d� | �|�� � | � || d� d S )N�doner 皙�����?)
r �get_running_loop�
create_future�
set_resultr
r � assertEqualr r! �
assertLess)r �loop�fut�t0Zret�t1r r r �2test_wait_for_timeout_less_then_0_or_0_future_done> s
zEAsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0_future_donec � s� t �� }d� � fdd�}| �t j��, |�� }t �|� d�I d H W d � n1 sV0 Y |�� }| �� d� | �|| d� d S )NFc � s d� d S )NTr r �Z foo_startedr r �fooQ s z_AsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_started.<locals>.foor r" )r r# � assertRaisesr r
r r& r' )r r( r. r* r+ r r- r �?test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_startedL s 2 zRAsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_startedc
� s� t �� }dD ]�}| j|d��� d � |�� �� �fdd�}t �|� �}�I d H | �t j��* |�� }t �||�I d H W d � n1 s�0 Y |�� }| � |�
� � | � |� � � | � � d� | �
|| d� W d � q 1 s�0 Y q d S )N)r �����timeoutc � s2 d� �� d � zt�d�I d H W d� nd� 0 dS �NT�
Fr! )r% r r r �� foo_runningZstartedr r r. e s
zFAsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0.<locals>.fooFr" )r r# ZsubTestr$ r r/ r r
r r r! � cancelledr& r' )r r( r3 r. r) r* r+ r r6 r �&test_wait_for_timeout_less_then_0_or_0] s
0 z9AsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0c � s� t �� }d � � fdd�}t �|� �}| �t j��* |�� }t �|d�I d H W d � n1 s`0 Y |�� }| �|�� � | �|� � � | �
|| d� | � � d� d S )Nc � s( d� zt �d�I d H W d� nd� 0 dS r4 �r r r �r7 r r r. � s
z-AsyncioWaitForTest.test_wait_for.<locals>.foor" g �?F) r r# r r/ r r
r r r! r8 r' r& )r r( r. r) r* r+ r r; r �
test_wait_for} s 0z AsyncioWaitForTest.test_wait_forc � s. dd� }t j|� d d�I d H }| �|d� d S )Nc � s dS )Nr! r r r r r �coro� s z7AsyncioWaitForTest.test_wait_for_blocking.<locals>.coror2 r! �r r r& )r r= �resr r r �test_wait_for_blocking� s z)AsyncioWaitForTest.test_wait_for_blockingc � sH t �� }|�� }t j|dd�}|�d|jd� |I d H }| �|d� d S )Nr r2 r" �ok)r r# r$ r Z
call_laterr% r& )r r( r) �taskr? r r r � test_wait_for_race_condition� s
z/AsyncioWaitForTest.test_wait_for_race_conditionc � s2 � fdd�}t j|� dd�I d H }� �|d� d S )Nc � s@ � � tj�� t�d�I d H W d � n1 s20 Y dS )N� )r/ r r r r r
r r �inner� s .zKAsyncioWaitForTest.test_wait_for_cancellation_race_condition.<locals>.innerg{�G�z�?r2 rD r> )r rE �resultr r
r �)test_wait_for_cancellation_race_condition� s z<AsyncioWaitForTest.test_wait_for_cancellation_race_conditionc � s� d� � fdd�}t �|� �}| �t j��$}t j|td�I d H W d � n1 sR0 Y | �� � |jj}| � t
|�t j � d S )NFc � sP zDzt �d�I d H W n& t jy< t �t�I d H � Y n0 W d� nd� 0 d S �Nr5 T�r r r �_EPSILONr �Z task_doner r rE � s
zKAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation.<locals>.innerr2 ) r r r/ r r rJ r � exception� __context__r& �typer )r rE �
inner_task�cm�chainedr rK r �)test_wait_for_waits_for_task_cancellation� s
2
z<AsyncioWaitForTest.test_wait_for_waits_for_task_cancellationc � sp d� � fdd�}| � tj�� }|� I d H W d � n1 s>0 Y | �� � |jj}| �t|�tj� d S )NFc � s@ � fdd�} t �| � �}t �t�I d H t j|dd�I d H d S )Nc � sP zDzt �d�I d H W n& t jy< t �t�I d H � Y n0 W d� nd� 0 d S rH rI r rK r r rE � s
zdAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo.<locals>.innerr r2 )r r r rJ r �rE rO rK r r r. � s
zUAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo) r/ r r r rL rM r& rN r )r r. rP rQ r rK r �5test_wait_for_waits_for_task_cancellation_w_timeout_0� s *
zHAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0c � sV G dd� dt �� � fdd�}| �� �� |� I d H W d � n1 sH0 Y d S )Nc @ s e Zd ZdS )z]AsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.FooExceptionN)r r r r r r r � FooException� s rU c � s0 � fdd�} t �| � �}t j|td�I d H d S )Nc � s$ zt �d�I d H W � �n� �0 d S )Nr r: r �rU r r rE � s zcAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo.<locals>.innerr2 )r r r rJ rS rV r r r. � s zTAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo)� Exceptionr/ )r r. r rV r �4test_wait_for_reraises_exception_during_cancellation� s zGAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellationc � s� dd� }t �|� �}t j|dd�}t �|�}t �d�I d H |�� | �t j�� |I d H W d � n1 sp0 Y | �|I d H d� d S )Nc � sh zt �d�I d H W nN t jyb zt �d�I d H W n$ t jy\ t �d�I d H Y n0 Y n0 dS )Ng333333�?�* )r r r r r r r rE � s zAAsyncioWaitForTest.test_wait_for_self_cancellation.<locals>.innerr" r2 r rY )r r r r r r/ r r& )r rE rO r rB r r r � test_wait_for_self_cancellation� s
(z2AsyncioWaitForTest.test_wait_for_self_cancellationc � s� t �� � � fdd�}t �|� �}t �t �||��}� �|j� | �t j�� |I d H W d � n1 sj0 Y | �|� � � d S )Nc � s � � � } | I d H d S r )r$ )r) �r( r r �blocking_coroutine s zDAsyncioWaitForTest._test_cancel_wait_for.<locals>.blocking_coroutine)
r r# r r Z call_soonr r/ r r r8 )r r3 r\ rB r r r[ r �_test_cancel_wait_for s (z(AsyncioWaitForTest._test_cancel_wait_forc � s | � d �I d H d S r �r] r
r r r � test_cancel_blocking_wait_for s z0AsyncioWaitForTest.test_cancel_blocking_wait_forc � s | � d�I d H d S )Ng N@r^ r
r r r �test_cancel_wait_for! s z'AsyncioWaitForTest.test_cancel_wait_forN)r r r r r r, r0 r9 r< r@ rC rG rR rT rX rZ r] r_ r` r r r r r ' s
r �__main__)
r Zunittestr
r rJ r ZIsolatedAsyncioTestCaser r �mainr r r r �<module> s