File: //usr/local/lib/python3.10/unittest/__pycache__/async_case.cpython-310.opt-2.pyc
o
�i� � @ 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� fd d �Zd!d"� Z� ZS )%�IsolatedAsyncioTestCase�runTestc s t � �|� d | _d | _d S �N)�super�__init__�_asyncioTestLoop�_asyncioCallsQueue)�selfZ
methodName�� __class__� �0/usr/local/lib/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 �methodr r r �_callTestMethod? s z'IsolatedAsyncioTestCase._callTestMethodc C s | � | j� | �� d S r )r r ZtearDownr r r r �
_callTearDownB s z%IsolatedAsyncioTestCase._callTearDownc O r r r )r �functionr r r r r � _callCleanupF s 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 I 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 Q s
z'IsolatedAsyncioTestCase._callMaybeAsyncc
� s� �t �� | _}|�d � |�� I d H }|�� |d u r d S |\}}z|I d H }|�� s2|�|� W n' ttfy= � t t j
fyZ } z|�� sP|� |� W Y d }~nd }~ww qr ) �asyncio�Queuer
�
set_result�get� task_done� cancelled�
SystemExit�KeyboardInterrupt�
BaseException�CancelledError�
set_exception)r r( �queue�queryZ awaitabler' �exr 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_taskr8 Z_asyncioCallsTaskr% )r �loopr( r r r �_setupAsyncioLoopn s
z)IsolatedAsyncioTestCase._setupAsyncioLoopc
C s | j }d | _ | j�d � |�| j�� � zct�|�}|s/W |�|�� � t�d � |� � d S |D ]}|�
� q1|�tj |ddi�� |D ]}|� � rLqE|�
� d ur]|�d|�
� |d�� qE|�|�� � W |�|�� � t�d � |� � d S |�|�� � t�d � |� � w )NZreturn_exceptionsTz(unhandled exception during test shutdown)�message� exception�task)r r
r$ r% �joinr* Z all_tasksZshutdown_default_executorr9 �close�cancelZgatherr/ r= Zcall_exception_handlerZshutdown_asyncgens)r r: Z to_cancelr> r r r �_tearDownAsyncioLoopx sB
�
� ��
�
z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s* | � � z t� �|�W | �� S | �� w r )r; r �runrB )r �resultr r r rC � s zIsolatedAsyncioTestCase.runc s | � � t� �� | �� d S r )r; r �debugrB r r r r rE � s
z IsolatedAsyncioTestCase.debugc C s | j d ur | �� d S d S r )r rB r r r r �__del__� s
�z IsolatedAsyncioTestCase.__del__)r r )�__name__�
__module__� __qualname__r r r r r r r r" r r r8 r; rB rC rE rF �
__classcell__r r r r r s"
$ r )r* r) �caser r r r r r �<module> s