Your IP : 216.73.216.52
o
0�i�! � @ s\ d dl Z d dlZd dlZdd� ZdZG dd� d�ZG dd� dej�Zed kr,e� � dS dS )
� Nc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �E/opt/alt/python310/lib64/python3.10/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 � s^ �t �� | j } |t �� }|dkrnz
t�|�I d H W n
tjy( Y nw q d| _d S )NTr )�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 � sN �t � }zt�|�� |jd �I d H W n
tjy Y nw | �|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 � sl �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 s,w 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 � � zRAsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0_coroutine_do_not_startedc
� s� �t �� }dD ]s}| j|d��c d � |�� �� �fdd�}t �|� �}�I d H | �t j�� |�� }t �||�I d H W d � n1 sFw Y |�� }| � |�
� � | � |� � � | � � d� | �
|| d� W d � n1 suw Y qd S )N)r �����timeoutc � s2 �d� �� d � z
t�d�I d H W d� dS d� w �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* �
� ���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 s1w Y |�� }| �|�� � | �|� � � | �
|| d� | � � d� d S )Nc � s( �d� z
t �d�I d H W d� dS d� w 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 � �z AsyncioWaitForTest.test_wait_forc � s0 �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 � sJ �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 � s4 �� fdd�}t j|� dd�I d H }� �|d� d S )Nc � sD �� � tj�� t�d�I d H W d � dS 1 sw 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 s*w Y | �� � |jj}| � t
|�t j � d S )NFc � �L �z!z
t �d�I d H W n t jy t �t�I d H � w W d� d S d� w �Nr6 T�r r r �_EPSILONr �Z task_doner r rF � � ���zKAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation.<locals>.innerr3 ) r r r0 r r rL r � exception� __context__r' �typer )r rF �
inner_task�cm�chainedr rM r �)test_wait_for_waits_for_task_cancellation� s �
�
z<AsyncioWaitForTest.test_wait_for_waits_for_task_cancellationc � sr �d� � fdd�}| � tj��}|� I d H W d � n1 s w Y | �� � |jj}| �t|�tj� d S )NFc � sB �� fdd�} t �| � �}t �t�I d H t j|dd�I d H d S )Nc � rI rJ rK r rM r r rF � rN zdAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo.<locals>.innerr r3 )r r r rL r �rF rR rM r r r/ � s
�
zUAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo) r0 r r r rO rP r' rQ r )r r/ rS rT r rM r �5test_wait_for_waits_for_task_cancellation_w_timeout_0� s � �
zHAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0c � sZ �G dd� dt �� � fdd�}| �� �� |� I d H W d � d S 1 s&w 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 rX c � s2 �� fdd�} t �| � �}t j|td�I d H d S )Nc � s �z t �d�I d H W � �� �)Nr r; r �rX r r rF � s �zcAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo.<locals>.innerr3 )r r r rL rV rY r r r/ � s � zTAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo)� Exceptionr0 )r r/ r rY 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 s9w Y | �|I d H d� d S )Nc � sl �z t �d�I d H W dS t jy5 z t �d�I d H W Y dS t jy4 t �d�I d H Y Y dS w w )Ng333333�?�* )r r r r r r r rF � s �����zAAsyncioWaitForTest.test_wait_for_self_cancellation.<locals>.innerr# r3 r r\ )r r r r r r0 r r' )r rF rR 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 s6w 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 � �z0AsyncioWaitForTest.test_cancel_blocking_wait_forc � s �| � d�I d H d S )Ng N@ra r
r r r �test_cancel_wait_for! rc z'AsyncioWaitForTest.test_cancel_wait_forN)r r r r r! r- r1 r: r= rA rD rH rU rW r[ r] r` rb rd r r r r r ' s"
r �__main__)
r Zunittestr
r rL r ZIsolatedAsyncioTestCaser r �mainr r r r �<module> s �