f
��_� c @ s0 d dl Z d dlZddlmZ G dd� de�ZdS )i Ni )�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�Z __class__� �,/usr/lib64/python3.10/unittest/async_case.pyr " s z IsolatedAsyncioTestCase.__init__c � � d S r r
�r r
r
r �
asyncSetUp' � z"IsolatedAsyncioTestCase.asyncSetUpc � r r r
r
r
r
r �
asyncTearDown* r z%IsolatedAsyncioTestCase.asyncTearDownc O � | 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 Zmethodr
r
r �_callTestMethod@ r z'IsolatedAsyncioTestCase._callTestMethodc C s | � | j� | �� d S r )r r ZtearDownr
r
r
r �
_callTearDownC s z%IsolatedAsyncioTestCase._callTearDownc O r r r )r Zfunctionr r r
r
r � _callCleanupG r z$IsolatedAsyncioTestCase._callCleanupc O s4 ||i |��}| j �� }| j�||f� | j �|�S r )r �
create_futurer �
put_nowait�run_until_complete�r r r r �ret�futr
r
r r J s
z"IsolatedAsyncioTestCase._callAsyncc O sB ||i |��}t �|�r>| j�� }| j�||f� | j�|�S |S r )�inspectZ isawaitabler r r 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 r ) �asyncioZQueuer Z
set_resultZgetZ task_done� cancelledZ
SystemExitZKeyboardInterruptZ
BaseExceptionZCancelledErrorZ
set_exception)r r" ZqueueZqueryZ awaitabler! Zexr
r
r �_asyncioLoopRunner\ s
z*IsolatedAsyncioTestCase._asyncioLoopRunnerc C sJ t �� }t �|� |�d� || _|�� }|�| �|��| _|� |� d S )NT)
r$ Znew_event_loop�set_event_loopZ set_debugr r Z create_taskr& Z_asyncioCallsTaskr )r �loopr" r
r
r �_setupAsyncioLoopo s
z)IsolatedAsyncioTestCase._setupAsyncioLoopc C s� | j }d | _ | j�d � |�| j�� � z�t�|�}|sPW t�d � |�� d S |D ] }|� � qT|�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)Zmessage� exception�task)r r r r Zjoinr$ Z all_tasksr' ZcloseZcancelZgatherr% r* Zcall_exception_handlerZshutdown_asyncgens)r r( Z to_cancelr+ r
r
r �_tearDownAsyncioLoopy s<
�
� �
�
z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s* | � � zt� �|�W | �� S | �� 0 r )r) r �runr, )r Zresultr r
r r- � s �zIsolatedAsyncioTestCase.run)r r )Z__name__Z
__module__Z __qualname__r r r r r r r r r r r& r) r, r- Z
__classcell__r
r
r r r s
"r )r$ r# Zcaser r r
r
r
r Z<module> s