File: //opt/cppython/lib/python3.8/test/test_asyncio/__pycache__/test_asyncio_waitfor.cpython-38.pyc
U
/D�g� � @ sT d dl Z d dlZd dlZdd� ZG dd� d�ZG dd� dej�ZedkrPe�� dS ) � Nc C s t �d � d S �N)�asyncioZset_event_loop_policy� r r �E/opt/cppython/lib/python3.8/test/test_asyncio/test_asyncio_waitfor.py�tearDownModule s r c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�SlowTaskz? Task will run for this defined time, ignoring cancel requests g�������?c C s
d| _ d S )NF)�exited��selfr r r �__init__
s zSlowTask.__init__c � s\ t �� | j }|t �� }|dkr$qRzt�|�I d H W q tjk
rN Y qX 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 S )
�AsyncioWaitForTestc � s\ t � }t�t�|�� |jd ��}t�d�I d H |�� t�|h�I d H | � |j
� d S )N� r ) r r Z create_task�wait_forr r r �cancel�wait�
assertTruer )r
�tZ waitfortaskr r r � atest_asyncio_wait_for_cancelled! s z3AsyncioWaitForTest.atest_asyncio_wait_for_cancelledc C s t �| �� � d S r )r r r r r r r � test_asyncio_wait_for_cancelled+ s z2AsyncioWaitForTest.test_asyncio_wait_for_cancelledc � sN t � }z t�|�� |jd �I d H W n tjk
r< Y nX | �|j� d S )Nr )r r r r r � TimeoutErrorr r )r
r r r r � atest_asyncio_wait_for_timeout. s z1AsyncioWaitForTest.atest_asyncio_wait_for_timeoutc C s t �| �� � d S r )r r r r r r r � test_asyncio_wait_for_timeout8 s z0AsyncioWaitForTest.test_asyncio_wait_for_timeoutN)r r r r r r r! r r r r r s
r �__main__) r Zunittestr r r ZTestCaser r �mainr r r r �<module> s