File: //usr/lib/python3.9/unittest/__pycache__/async_case.cpython-39.pyc
a
kx�g� � @ s0 d dl Z d dlZddlmZ G dd� de�ZdS )� N� )�TestCasec s� e Zd Zd � fdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd � fd d � Z� ZS )!�IsolatedAsyncioTestCase�runTestc s t � �|� d | _d | _d S �N)�super�__init__�_asyncioTestLoop�_asyncioCallsQueue)�selfZ
methodName�� __class__� �)/usr/lib/python3.9/unittest/async_case.pyr " s z IsolatedAsyncioTestCase.__init__c � s d S r r �r r r r �
asyncSetUp' s z"IsolatedAsyncioTestCase.asyncSetUpc � s d S r r r r r r �
asyncTearDown* s z%IsolatedAsyncioTestCase.asyncTearDownc O s | j |g|�R i |�� d S r )Z
addCleanup)r �func�args�kwargsr r r �addAsyncCleanup- s
z'IsolatedAsyncioTestCase.addAsyncCleanupc C s | � � | �| j� d S r )ZsetUp�
_callAsyncr r r r r �
_callSetUp< s z"IsolatedAsyncioTestCase._callSetUpc C s | � |� d S r ��_callMaybeAsync)r �methodr r r �_callTestMethod@ s z'IsolatedAsyncioTestCase._callTestMethodc C s | � | j� | �� d S r )r r ZtearDownr r r r �
_callTearDownC s z%IsolatedAsyncioTestCase._callTearDownc O s | j |g|�R i |�� d S r r )r Zfunctionr r r r r � _callCleanupG s z$IsolatedAsyncioTestCase._callCleanupc O sP | j d usJ �||i |��}t�|�s*J �| j �� }| j�||f� | j �|�S r �r �inspectZ isawaitable�
create_futurer
�
put_nowait�run_until_complete�r r r r �ret�futr r r r J s
z"IsolatedAsyncioTestCase._callAsyncc O sT | j d usJ �||i |��}t�|�rL| j �� }| j�||f� | j �|�S |S d S r r r$ r r r r R s
z'IsolatedAsyncioTestCase._callMaybeAsyncc
� s� t �� | _}|�d � |�� I d H }|�� |d u r:d S |\}}z |I d H }|�� s`|�|� W q ttfyz � Y q t t j
fy� } z |�� s�|� |� W Y d }~qd }~0 0 qd S r ) �asyncioZQueuer
Z
set_result�getZ task_done� cancelled�
SystemExit�KeyboardInterrupt�
BaseExceptionZCancelledErrorZ
set_exception)r r& ZqueueZqueryZ awaitabler% Zexr r r �_asyncioLoopRunner\ s
z*IsolatedAsyncioTestCase._asyncioLoopRunnerc C sX | j d u sJ �t�� }t�|� |�d� || _ |�� }|�| �|��| _|� |� d S )NT)
r r' Znew_event_loop�set_event_loopZ set_debugr! Z create_taskr- Z_asyncioCallsTaskr# )r �loopr& r r r �_setupAsyncioLoopo s
z)IsolatedAsyncioTestCase._setupAsyncioLoopc C s� | j d usJ �| j }d | _ | j�d � |�| j�� � z�t�|�}|s^W t�d � |�� d S |D ] }|� � qb|�tj
||dd��� |D ]0}|� � r�q�|� � d ur�|�
d|� � |d�� q�|�|�� � W t�d � |�� nt�d � |�� 0 d S )NT)r/ Zreturn_exceptionsz(unhandled exception during test shutdown)�message� exception�task)r r
r" r# �joinr' Z all_tasksr. �closeZcancelZgatherr) r2 Zcall_exception_handlerZshutdown_asyncgens)r r/ Z to_cancelr3 r r r �_tearDownAsyncioLoopy s>
�
� �
�
z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s. | � � zt� �|�W | �� S | �� 0 d S r )r0 r �runr6 )r �resultr r r r7 � s �zIsolatedAsyncioTestCase.run)r )N)�__name__�
__module__� __qualname__r r r r r r r r r r r- r0 r6 r7 �
__classcell__r r r r r s
"r )r' r Zcaser r r r r r �<module> s