File: //usr/local/lib/python3.10/test/test_asyncio/__pycache__/test_waitfor.cpython-310.pyc
o
�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 �;/usr/local/lib/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�t0�ret�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) r0 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_running�startedr r r0 e s �
�zFAsyncioWaitForTest.test_wait_for_timeout_less_then_0_or_0.<locals>.fooFr# )r r$ ZsubTestr% r r1 r r
r r r" � cancelledr' r( )r r) r5 r0 r* r+ r- r r8 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 r6 �r r r �r9 r r r0 � s ��z-AsyncioWaitForTest.test_wait_for.<locals>.foor# g �?F) r r$ r r1 r r
r r r" r; r( r' )r r) r0 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>.coror4 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 r4 r# �ok)r r$ r% r Z
call_laterr& r' )r r) r* �taskrB 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� )r1 r r r r r
r r �inner� s �
��zKAsyncioWaitForTest.test_wait_for_cancellation_race_condition.<locals>.innerg{�G�z�?r4 rG rA )r rH �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 �Nr7 T�r r r �_EPSILONr �� task_doner r rH � � ���zKAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation.<locals>.innerr4 ) r r r1 r r rN r � exception� __context__r' �typer )r rH �
inner_task�cm�chainedr rO 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 � rK rL rM r rO r r rH � rQ zdAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo.<locals>.innerr r4 )r r r rN r �rH rU rO r r r0 � s
�
zUAsyncioWaitForTest.test_wait_for_waits_for_task_cancellation_w_timeout_0.<locals>.foo) r1 r r r rR rS r' rT r )r r0 rV rW r rO 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 r[ c � s2 �� fdd�} t �| � �}t j|td�I d H d S )Nc � s �z t �d�I d H W � �� �)Nr r= r �r[ r r rH � s �zcAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo.<locals>.innerr4 )r r r rN rY r\ r r r0 � s � zTAsyncioWaitForTest.test_wait_for_reraises_exception_during_cancellation.<locals>.foo)� Exceptionr1 )r r0 r r\ 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 rH � s �����zAAsyncioWaitForTest.test_wait_for_self_cancellation.<locals>.innerr# r4 r r_ )r r r r r r1 r r' )r rH rU r rE 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 r1 r r r; )r r5 rb rE r r ra r �_test_cancel_wait_for s � �z(AsyncioWaitForTest._test_cancel_wait_forc � s �| � d �I d H d S r �rc r
r r r � test_cancel_blocking_wait_for � �z0AsyncioWaitForTest.test_cancel_blocking_wait_forc � s �| � d�I d H d S )Ng N@rd r
r r r �test_cancel_wait_for! rf z'AsyncioWaitForTest.test_cancel_wait_forN)r r r r r! r. r2 r< r? rC rF rJ rX rZ r^ r` rc re rg r r r r r ' s"
r �__main__)
r Zunittestr
r rN r ZIsolatedAsyncioTestCaser r �mainr r r r �<module> s �