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 ZdZdZdd� Zdd� ZdS )�SlowTaskz? Task will run for this defined time, ignoring cancel requests 皙�����?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__�__doc__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 r0 r r
r r r" � cancelledr' r( )r r) r4 r/ r* r+ r, r r7 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 r5 �r r r �r8 r r r/ � s
z-AsyncioWaitForTest.test_wait_for.<locals>.foor# g �?F) r r$ r r0 r r
r r r" r9 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>.coror3 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 r3 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� )r0 r r r r r
r r �inner� s .zKAsyncioWaitForTest.test_wait_for_cancellation_race_condition.<locals>.innerg{�G�z�?r3 rE r? )r rF �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 �Nr6 T�r r r �_EPSILONr �Z task_doner r rF � s
zKAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation.<locals>.innerr3 ) r r r0 r r rK r � exception� __context__r' �typer )r rF �
inner_task�cm�chainedr rL 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 rI rJ r rL r r rF � s
zdAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo.<locals>.innerr r3 )r r r rK r �rF rP rL r r r/ � s
zUAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo) r0 r r r rM rN r' rO r )r r/ rQ rR r rL 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 rV 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 �rV r r rF � s zcAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo.<locals>.innerr3 )r r r rK rT rW r r r/ � s zTAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo)� Exceptionr0 )r r/ r rW 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 rF � s zAAsyncioWaitForTest.test_wait_for_self_cancellation.<locals>.innerr# r3 r rZ )r r r r r r0 r r' )r rF rP r rC 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 r0 r r r9 )r r4 r] rC 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- r1 r: r= rA rD rH rS rU rY r[ r^ r` ra r r r r r ' s
r �__main__)
r Zunittestr
r rK r ZIsolatedAsyncioTestCaser r �mainr r r r �<module> s