0x1949 Team - FAZEMRX - MANAGER
Edit File: async_case.cpython-310.pyc
o ��+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� fd d �Zd!d"� Z� ZS )%�IsolatedAsyncioTestCase�runTestc s t � �|� d | _d | _d S �N)�super�__init__�_asyncioTestLoop�_asyncioCallsQueue)�self� methodName�� __class__� �*/usr/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 )� addCleanup)r �func�args�kwargsr r r �addAsyncCleanup, s z'IsolatedAsyncioTestCase.addAsyncCleanupc C s | � � | �| j� d S r )�setUp� _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 �tearDownr r r r � _callTearDownB s z%IsolatedAsyncioTestCase._callTearDownc O r r r )r �functionr r r r r � _callCleanupF s z$IsolatedAsyncioTestCase._callCleanupc O s^ | j d us J d��||i |��}t�|�s J |�d���| j �� }| j�||f� | j �|�S )N�$asyncio test loop is not initializedz returned non-awaitable�r �inspect� isawaitable� create_futurer � put_nowait�run_until_complete�r r r r �ret�futr r r r I s z"IsolatedAsyncioTestCase._callAsyncc O sT | j d us J d��||i |��}t�|�r(| j �� }| j�||f� | j �|�S |S )Nr' 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 r0 �queue�query� awaitabler/ �exr r r �_asyncioLoopRunner[ s, � � ���z*IsolatedAsyncioTestCase._asyncioLoopRunnerc C s\ | j d u s J d��t�� }t�|� |�d� || _ |�� }|�| �|��| _|� |� d S )Nz%asyncio test loop already initializedT) r r1 �new_event_loop�set_event_loop� set_debugr+ � create_taskr@ �_asyncioCallsTaskr- )r �loopr0 r r r �_setupAsyncioLoopn s z)IsolatedAsyncioTestCase._setupAsyncioLoopc C s$ | j d us J d��| j }d | _ | j�d � |�| j�� � zct�|�}|s8W |�|�� � t�d � |� � d S |D ]}|� � q:|�tj |ddi�� |D ]}|� � rUqN|� � d urf|�d|� � |d�� qN|�|�� � W |�|�� � t�d � |� � d S |�|�� � t�d � |� � w )Nr' �return_exceptionsTz(unhandled exception during test shutdown)�message� exception�task)r r r, r- �joinr1 � all_tasks�shutdown_default_executorrB �close�cancel�gatherr6 rJ �call_exception_handler�shutdown_asyncgens)r rF � to_cancelrK r r r �_tearDownAsyncioLoopx sD � � �� � z,IsolatedAsyncioTestCase._tearDownAsyncioLoopNc s* | � � z t� �|�W | �� S | �� w r )rG r �runrU )r �resultr r r rV � s zIsolatedAsyncioTestCase.runc s | � � t� �� | �� d S r )rG r �debugrU r r r r rX � s z IsolatedAsyncioTestCase.debugc C s | j d ur | �� d S d S r )r rU r r r r �__del__� s �z IsolatedAsyncioTestCase.__del__)r r )�__name__� __module__� __qualname__r r r r r r" r$ r& r r r@ rG rU rV rX rY � __classcell__r r r r r s" $ r )r1 r) �caser r r r r r �<module> s